diff --git a/tests/TestKeePass1Reader.cpp b/tests/TestKeePass1Reader.cpp index def8c6e6..b15e8a03 100644 --- a/tests/TestKeePass1Reader.cpp +++ b/tests/TestKeePass1Reader.cpp @@ -102,6 +102,8 @@ void TestKeePass1Reader::testBasic() QCOMPARE(group2->name(), QString("eMail")); QCOMPARE(group2->entries().size(), 1); QCOMPARE(group2->iconNumber(), 19); + + reopenDatabase(m_db, "masterpw", QString()); } void TestKeePass1Reader::testMasterKey() @@ -154,6 +156,8 @@ void TestKeePass1Reader::testFileKey() QCOMPARE(db->rootGroup()->children().size(), 1); QCOMPARE(db->rootGroup()->children().at(0)->name(), name); + reopenDatabase(db, QString(), keyFilename); + delete db; } @@ -180,6 +184,8 @@ void TestKeePass1Reader::testCompositeKey() QCOMPARE(db->rootGroup()->children().size(), 1); QCOMPARE(db->rootGroup()->children().at(0)->name(), name); + reopenDatabase(db, "mypassword", keyFilename); + delete db; } @@ -230,4 +236,31 @@ QDateTime TestKeePass1Reader::genDT(int year, int month, int day, int hour, int return QDateTime(date, time, Qt::UTC); } +void TestKeePass1Reader::reopenDatabase(Database* db, const QString& password, const QString& keyfileName) +{ + QBuffer buffer; + buffer.open(QIODevice::ReadWrite); + + KeePass2Writer writer; + writer.writeDatabase(&buffer, db); + QVERIFY(!writer.error()); + QVERIFY(buffer.seek(0)); + + CompositeKey key; + if (!password.isNull()) { + key.addKey(PasswordKey(password)); + } + if (!keyfileName.isEmpty()) { + FileKey fileKey; + QVERIFY(fileKey.load(keyfileName)); + key.addKey(fileKey); + } + + KeePass2Reader reader; + Database* newDb = reader.readDatabase(&buffer, key); + QVERIFY(newDb); + QVERIFY(!reader.hasError()); + delete newDb; +} + KEEPASSX_QTEST_CORE_MAIN(TestKeePass1Reader) diff --git a/tests/TestKeePass1Reader.h b/tests/TestKeePass1Reader.h index cc2d1c5d..a4fc8a0e 100644 --- a/tests/TestKeePass1Reader.h +++ b/tests/TestKeePass1Reader.h @@ -22,7 +22,6 @@ #include class Database; -class QIODevice; class TestKeePass1Reader : public QObject { @@ -43,6 +42,7 @@ private Q_SLOTS: private: static QDateTime genDT(int year, int month, int day, int hour, int min); + static void reopenDatabase(Database* db, const QString& password, const QString& keyfileName); Database* m_db; };