82 lines
2.3 KiB
C++
82 lines
2.3 KiB
C++
#include "CppUnitTest.h"
|
|
#include "../GhettoCrypt/GhettoCryptWrapper.h"
|
|
#include "../GhettoCrypt/Flexblock.h"
|
|
#include "../GhettoCrypt/Util.h"
|
|
|
|
using namespace Microsoft::VisualStudio::CppUnitTestFramework;
|
|
using namespace GhettoCipher;
|
|
|
|
namespace SimpleTests
|
|
{
|
|
TEST_CLASS(GCWrapper)
|
|
{
|
|
public:
|
|
|
|
// Tests that encrypting and decrypting strings using the wrapper works.
|
|
// This test will start from scratch after encryption, to ensure EVERYTHING has to be re-calculated.
|
|
TEST_METHOD(String)
|
|
{
|
|
// Setup
|
|
const std::string plaintext = "Hello, World!";
|
|
const std::string password = "Der Affe will Zucker";
|
|
|
|
std::string ciphertext;
|
|
std::string decrypted;
|
|
|
|
// Encryption
|
|
{
|
|
ciphertext = GhettoCryptWrapper::EncryptString(plaintext, password);
|
|
}
|
|
|
|
// Decryption
|
|
{
|
|
decrypted = GhettoCryptWrapper::DecryptString(ciphertext, password);
|
|
}
|
|
|
|
// Assertion
|
|
Assert::AreEqual(
|
|
plaintext,
|
|
decrypted
|
|
);
|
|
}
|
|
|
|
// Tests that encrypting and decrypting files using the wrapper works.
|
|
// This test will start from scratch after encryption, to ensure EVERYTHING has to be re-calculated.
|
|
TEST_METHOD(File)
|
|
{
|
|
// Setup
|
|
#if defined _WIN64
|
|
const std::string testfile_dir = "../../SimpleTests/";
|
|
#elif defined _WIN32
|
|
const std::string testfile_dir = "../SimpleTests/";
|
|
#endif
|
|
|
|
const std::string filename_plain = testfile_dir + "testfile.png";
|
|
const std::string filename_encrypted = testfile_dir + "testfile.png.crypt";
|
|
const std::string filename_decrypted = testfile_dir + "testfile.png.clear.png";
|
|
const std::string password = "Der Affe will Zucker";
|
|
|
|
|
|
// Encryption
|
|
{
|
|
GhettoCryptWrapper::EncryptFile(filename_plain, filename_encrypted, password);
|
|
}
|
|
|
|
// Decryption
|
|
{
|
|
GhettoCryptWrapper::DecryptFile(filename_encrypted, filename_decrypted, password);
|
|
}
|
|
|
|
// Read in both the base, and the decrypted file
|
|
const Flexblock plainfile = ReadFileToBits(filename_plain);
|
|
const Flexblock decryptfile = ReadFileToBits(filename_decrypted);
|
|
|
|
// Assertion (If this fails, maybe check if the image is even readable by an image viewer)
|
|
Assert::AreEqual(
|
|
PadStringToLength(plainfile, decryptfile.length(), '0', false),
|
|
decryptfile
|
|
);
|
|
}
|
|
};
|
|
}
|