Added test cases for CryConfigCreator

2015-10-22 18:48:14 +02:00 · 2015-10-22 18:48:14 +02:00 · c936fca16a
commit c936fca16a
parent 9bf83a6fe7
1 changed files with 103 additions and 0 deletions
--- a/test/config/CryConfigCreatorTest.cpp
+++ b/test/config/CryConfigCreatorTest.cpp
@ -0,0 +1,103 @@
+#include <google/gtest/gtest.h>
+#include <google/gmock/gmock.h>
+#include "../../src/config/CryConfigCreator.h"
+#include "../../src/config/CryCipher.h"
+#include <messmer/blockstore/implementations/encrypted/ciphers/ciphers.h>
+#include "../testutils/MockConsole.h"
+
+using namespace cryfs;
+
+using boost::optional;
+using boost::none;
+using cpputils::Console;
+using cpputils::unique_ref;
+using cpputils::make_unique_ref;
+using std::string;
+using std::vector;
+using ::testing::_;
+using ::testing::Return;
+using ::testing::Invoke;
+using ::testing::ValuesIn;
+using ::testing::HasSubstr;
+using ::testing::UnorderedElementsAreArray;
+using ::testing::WithParamInterface;
+
+class CryConfigCreatorTest: public ::testing::Test {
+public:
+    CryConfigCreatorTest()
+            : _console(make_unique_ref<MockConsole>()),
+              console(_console.get()),
+              creator(std::move(_console), cpputils::Random::PseudoRandom()) {
+    }
+    unique_ref<MockConsole> _console;
+    MockConsole *console;
+    CryConfigCreator creator;
+};
+
+#define EXPECT_ASK_FOR_CIPHER() EXPECT_CALL(*console, ask(HasSubstr("block cipher"), UnorderedElementsAreArray(CryCiphers::supportedCipherNames())))
+
+ACTION(ChooseAnyCipher) {
+    return 0;
+}
+
+ACTION_P(ChooseCipher, cipherName) {
+    return std::find(arg1.begin(), arg1.end(), cipherName) - arg1.begin();
+}
+
+TEST_F(CryConfigCreatorTest, DoesNotCrash) {
+    EXPECT_ASK_FOR_CIPHER().WillOnce(ChooseAnyCipher());
+    CryConfig config = creator.create();
+}
+
+TEST_F(CryConfigCreatorTest, ChoosesEmptyRootBlobId) {
+    EXPECT_ASK_FOR_CIPHER().WillOnce(ChooseAnyCipher());
+    CryConfig config = creator.create();
+    EXPECT_EQ("", config.RootBlob()); // This tells CryFS to create a new root blob
+}
+
+TEST_F(CryConfigCreatorTest, ChoosesValidEncryptionKey_448) {
+    EXPECT_ASK_FOR_CIPHER().WillOnce(ChooseCipher("mars-448-gcm"));
+    CryConfig config = creator.create();
+    blockstore::encrypted::Mars448_GCM::EncryptionKey::FromString(config.EncryptionKey()); // This crashes if invalid
+}
+
+TEST_F(CryConfigCreatorTest, ChoosesValidEncryptionKey_256) {
+    EXPECT_ASK_FOR_CIPHER().WillOnce(ChooseCipher("aes-256-gcm"));
+    CryConfig config = creator.create();
+    blockstore::encrypted::AES256_GCM::EncryptionKey::FromString(config.EncryptionKey()); // This crashes if invalid
+}
+
+TEST_F(CryConfigCreatorTest, ChoosesValidEncryptionKey_128) {
+    EXPECT_ASK_FOR_CIPHER().WillOnce(ChooseCipher("aes-128-gcm"));
+    CryConfig config = creator.create();
+    blockstore::encrypted::AES128_GCM::EncryptionKey::FromString(config.EncryptionKey()); // This crashes if invalid
+}
+
+class CryConfigCreatorTest_ChooseCipher: public CryConfigCreatorTest, public ::testing::WithParamInterface<string> {
+public:
+    string cipherName = GetParam();
+    optional<string> cipherWarning = CryCiphers::find(cipherName).warning();
+
+    void EXPECT_DONT_SHOW_WARNING() {
+        EXPECT_CALL(*console, askYesNo(_)).Times(0);
+    }
+
+    void EXPECT_SHOW_WARNING(const string &warning) {
+        EXPECT_CALL(*console, askYesNo(HasSubstr(warning))).WillOnce(Return(true));
+    }
+};
+
+TEST_P(CryConfigCreatorTest_ChooseCipher, ChoosesCipherCorrectly) {
+    if (cipherWarning == none) {
+        EXPECT_DONT_SHOW_WARNING();
+    } else {
+        EXPECT_SHOW_WARNING(*cipherWarning);
+    }
+
+    EXPECT_ASK_FOR_CIPHER().WillOnce(ChooseCipher(cipherName));
+
+    CryConfig config = creator.create();
+    EXPECT_EQ(cipherName, config.Cipher());
+}
+
+INSTANTIATE_TEST_CASE_P(CryConfigCreatorTest_ChooseCipher, CryConfigCreatorTest_ChooseCipher, ValuesIn(CryCiphers::supportedCipherNames()));