156 lines
5.3 KiB
C++
156 lines
5.3 KiB
C++
#include "../io/passwordfile.h"
|
|
#include "../io/cryptoexception.h"
|
|
#include "../io/entry.h"
|
|
|
|
#include <c++utilities/tests/testutils.h>
|
|
#include <c++utilities/application/global.h>
|
|
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
#include <cppunit/TestFixture.h>
|
|
|
|
using namespace std;
|
|
using namespace Io;
|
|
|
|
using namespace CPPUNIT_NS;
|
|
|
|
class PasswordFileTests : public TestFixture
|
|
{
|
|
CPPUNIT_TEST_SUITE(PasswordFileTests);
|
|
CPPUNIT_TEST(testReading);
|
|
#ifdef PLATFORM_UNIX
|
|
CPPUNIT_TEST(testWriting);
|
|
#endif
|
|
CPPUNIT_TEST_SUITE_END();
|
|
|
|
public:
|
|
void setUp();
|
|
void tearDown();
|
|
|
|
void testReading();
|
|
void testReading(const string &testfile1path, const string &testfile1password, const string &testfile2, const string &testfile2password, bool testfile2Mod);
|
|
#ifdef PLATFORM_UNIX
|
|
void testWriting();
|
|
#endif
|
|
};
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION(PasswordFileTests);
|
|
|
|
void PasswordFileTests::setUp()
|
|
{}
|
|
|
|
void PasswordFileTests::tearDown()
|
|
{}
|
|
|
|
/*!
|
|
* \brief Tests reading the testfiles testfile{1,2}.pwmgr.
|
|
*/
|
|
void PasswordFileTests::testReading()
|
|
{
|
|
testReading(TestUtilities::testFilePath("testfile1.pwmgr"), "123456",
|
|
TestUtilities::testFilePath("testfile2.pwmgr"), string(), false);
|
|
}
|
|
|
|
void PasswordFileTests::testReading(const string &testfile1path, const string &testfile1password, const string &testfile2, const string &testfile2password, bool testfile2Mod)
|
|
{
|
|
PasswordFile file;
|
|
|
|
// open testfile 1 ...
|
|
file.setPath(testfile1path);
|
|
file.open(true);
|
|
|
|
CPPUNIT_ASSERT(file.isEncryptionUsed() == !testfile1password.empty());
|
|
// attempt to decrypt using a wrong password
|
|
file.setPassword(testfile1password + "asdf");
|
|
if(!testfile1password.empty()) {
|
|
CPPUNIT_ASSERT_THROW(file.load(), CryptoException);
|
|
}
|
|
// attempt to decrypt using the correct password
|
|
file.setPassword(testfile1password);
|
|
file.load();
|
|
// test root entry
|
|
const NodeEntry *rootEntry = file.rootEntry();
|
|
CPPUNIT_ASSERT(rootEntry->label() == "testfile1");
|
|
CPPUNIT_ASSERT(rootEntry->children().size() == 4);
|
|
|
|
// test testaccount1
|
|
CPPUNIT_ASSERT(rootEntry->children()[0]->label() == "testaccount1");
|
|
CPPUNIT_ASSERT(rootEntry->children()[0]->type() == EntryType::Account);
|
|
CPPUNIT_ASSERT(static_cast<AccountEntry *>(rootEntry->children()[0])->fields().at(0).name() == "pin");
|
|
CPPUNIT_ASSERT(static_cast<AccountEntry *>(rootEntry->children()[0])->fields().at(0).value() == "123456");
|
|
CPPUNIT_ASSERT(static_cast<AccountEntry *>(rootEntry->children()[0])->fields().at(0).type() == FieldType::Password);
|
|
CPPUNIT_ASSERT(static_cast<AccountEntry *>(rootEntry->children()[0])->fields().at(0).tiedAccount() == static_cast<AccountEntry *>(rootEntry->children()[0]));
|
|
CPPUNIT_ASSERT(static_cast<AccountEntry *>(rootEntry->children()[0])->fields().at(1).type() == FieldType::Normal);
|
|
CPPUNIT_ASSERT_THROW(static_cast<AccountEntry *>(rootEntry->children()[0])->fields().at(2), out_of_range);
|
|
|
|
// test testaccount2
|
|
CPPUNIT_ASSERT(rootEntry->children()[1]->label() == "testaccount2");
|
|
CPPUNIT_ASSERT(rootEntry->children()[1]->type() == EntryType::Account);
|
|
CPPUNIT_ASSERT(static_cast<AccountEntry *>(rootEntry->children()[1])->fields().empty());
|
|
|
|
// test testcategory1
|
|
CPPUNIT_ASSERT(rootEntry->children()[2]->label() == "testcategory1");
|
|
CPPUNIT_ASSERT(rootEntry->children()[2]->type() == EntryType::Node);
|
|
const NodeEntry *category = static_cast<NodeEntry *>(rootEntry->children()[2]);
|
|
CPPUNIT_ASSERT(category->children().size() == 3);
|
|
CPPUNIT_ASSERT(category->children()[2]->type() == EntryType::Node);
|
|
CPPUNIT_ASSERT(static_cast<NodeEntry *>(category->children()[2])->children().size() == 2);
|
|
|
|
// test testaccount3
|
|
CPPUNIT_ASSERT(rootEntry->children()[3]->label() == "testaccount3");
|
|
|
|
// open testfile 2
|
|
file.setPath(testfile2);
|
|
file.open(true);
|
|
|
|
CPPUNIT_ASSERT(file.isEncryptionUsed() == !testfile2password.empty());
|
|
file.setPassword(testfile2password);
|
|
file.load();
|
|
rootEntry = file.rootEntry();
|
|
if(testfile2Mod) {
|
|
CPPUNIT_ASSERT(rootEntry->label() == "testfile2 - modified");
|
|
CPPUNIT_ASSERT(rootEntry->children().size() == 2);
|
|
CPPUNIT_ASSERT(rootEntry->children()[1]->label() == "newAccount");
|
|
} else {
|
|
CPPUNIT_ASSERT(rootEntry->label() == "testfile2");
|
|
CPPUNIT_ASSERT(rootEntry->children().size() == 1);
|
|
}
|
|
}
|
|
|
|
#ifdef PLATFORM_UNIX
|
|
/*!
|
|
* \brief Tests writing and reading.
|
|
*/
|
|
void PasswordFileTests::testWriting()
|
|
{
|
|
string testfile1 = TestUtilities::workingCopyPath("testfile1.pwmgr");
|
|
string testfile2 = TestUtilities::workingCopyPath("testfile2.pwmgr");
|
|
PasswordFile file;
|
|
|
|
// resave testfile 1
|
|
file.setPath(testfile1);
|
|
file.open(false);
|
|
file.setPassword("123456");
|
|
file.load();
|
|
file.doBackup();
|
|
file.save(false, true);
|
|
|
|
// resave testfile 2
|
|
file.setPath(testfile2);
|
|
file.open(false);
|
|
file.load();
|
|
file.rootEntry()->setLabel("testfile2 - modified");
|
|
new AccountEntry("newAccount", file.rootEntry());
|
|
file.setPassword("654321");
|
|
file.doBackup();
|
|
file.save(true, false);
|
|
|
|
// check results using the reading test
|
|
testReading(testfile1, string(),
|
|
testfile2, "654321", true);
|
|
|
|
// check backup files
|
|
testReading(testfile1 + ".backup", "123456",
|
|
testfile2 + ".backup", string(), false);
|
|
}
|
|
#endif
|