From cc8d6424e26cfe33de15a8ce712969be5ed87d25 Mon Sep 17 00:00:00 2001 From: Florian Geyer Date: Sat, 21 Apr 2012 23:17:26 +0200 Subject: [PATCH] Add test for deleted objects. --- tests/CMakeLists.txt | 2 + tests/TestDeletedObjects.cpp | 107 +++++++++++++++++++++++++++++++++++ tests/TestDeletedObjects.h | 39 +++++++++++++ 3 files changed, 148 insertions(+) create mode 100644 tests/TestDeletedObjects.cpp create mode 100644 tests/TestDeletedObjects.h diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt index d64ac0249..df87c3d89 100644 --- a/tests/CMakeLists.txt +++ b/tests/CMakeLists.txt @@ -125,6 +125,8 @@ add_unit_test(NAME testkeepass2randomstream SOURCES TestKeePass2RandomStream.cpp add_unit_test(NAME testmodified SOURCES TestModified.cpp MOCS TestModified.h LIBS ${TEST_LIBRARIES}) +add_unit_test(NAME testdeletedobjects SOURCES TestDeletedObjects.cpp MOCS TestDeletedObjects.h LIBS ${TEST_LIBRARIES}) + if(WITH_GUI_TESTS) add_subdirectory(gui) diff --git a/tests/TestDeletedObjects.cpp b/tests/TestDeletedObjects.cpp new file mode 100644 index 000000000..c247eb2a6 --- /dev/null +++ b/tests/TestDeletedObjects.cpp @@ -0,0 +1,107 @@ +/* + * Copyright (C) 2012 Felix Geyer + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 2 or (at your option) + * version 3 of the License. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ + +#include "TestDeletedObjects.h" + +#include + +#include "tests.h" +#include "core/Database.h" +#include "core/Group.h" +#include "crypto/Crypto.h" +#include "format/KeePass2XmlReader.h" +#include "config-keepassx-tests.h" + +void TestDeletedObjects::initTestCase() +{ + Crypto::init(); +} + +void TestDeletedObjects::createAndDelete(Database* db, int delObjectsSize) +{ + + QVERIFY(db->deletedObjects().size() == delObjectsSize); + Group* root = db->rootGroup(); + int rootChildrenCount = root->children().size(); + + Group* g = new Group(); + g->setParent(root); + Uuid gUuid = Uuid::random(); + g->setUuid(gUuid); + delete g; + QVERIFY(db->deletedObjects().size() == ++delObjectsSize); + QVERIFY(db->deletedObjects().at(delObjectsSize-1).uuid == gUuid); + QVERIFY(rootChildrenCount == root->children().size()); + + Group* g1 = new Group(); + g1->setParent(root); + Uuid g1Uuid = Uuid::random(); + g1->setUuid(g1Uuid); + Entry* e1 = new Entry(); + e1->setGroup(g1); + Uuid e1Uuid = Uuid::random(); + e1->setUuid(e1Uuid); + Group* g2 = new Group(); + g2->setParent(g1); + Uuid g2Uuid = Uuid::random(); + g2->setUuid(g2Uuid); + Entry* e2 = new Entry(); + e2->setGroup(g2); + Uuid e2Uuid = Uuid::random(); + e2->setUuid(e2Uuid); + + delete g1; + delObjectsSize += 4; + + QVERIFY(db->deletedObjects().size() == delObjectsSize); + QVERIFY(db->deletedObjects().at(delObjectsSize-4).uuid == e1Uuid); + QVERIFY(db->deletedObjects().at(delObjectsSize-3).uuid == e2Uuid); + QVERIFY(db->deletedObjects().at(delObjectsSize-2).uuid == g2Uuid); + QVERIFY(db->deletedObjects().at(delObjectsSize-1).uuid == g1Uuid); + QVERIFY(rootChildrenCount == root->children().size()); + + Entry* e3 = new Entry(); + e3->setGroup(root); + Uuid e3Uuid = Uuid::random(); + e3->setUuid(e3Uuid); + + delete e3; + + QVERIFY(db->deletedObjects().size() == ++delObjectsSize); + QVERIFY(db->deletedObjects().at(delObjectsSize-1).uuid == e3Uuid); + QVERIFY(rootChildrenCount == root->children().size()); + + +} + +void TestDeletedObjects::testDeletedObjectsFromFile() +{ + KeePass2XmlReader* reader = new KeePass2XmlReader(); + QString xmlFile = QString(KEEPASSX_TEST_DATA_DIR).append("/NewDatabase.xml"); + Database* db = reader->readDatabase(xmlFile); + + createAndDelete(db, 2); +} + +void TestDeletedObjects::testDeletedObjectsFromNewDb() +{ + Database* db = new Database(); + + createAndDelete(db, 0); +} + +KEEPASSX_QTEST_CORE_MAIN(TestDeletedObjects) diff --git a/tests/TestDeletedObjects.h b/tests/TestDeletedObjects.h new file mode 100644 index 000000000..4d2a0b372 --- /dev/null +++ b/tests/TestDeletedObjects.h @@ -0,0 +1,39 @@ +/* + * Copyright (C) 2012 Felix Geyer + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 2 or (at your option) + * version 3 of the License. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ + +#ifndef KEEPASSX_TESTDELETEDOBJECTS_H +#define KEEPASSX_TESTDELETEDOBJECTS_H + +#include + +class Database; + +class TestDeletedObjects : public QObject +{ + Q_OBJECT + +private: + void createAndDelete(Database* db, int delObjectsSize); + +private Q_SLOTS: + void initTestCase(); + void testDeletedObjectsFromFile(); + void testDeletedObjectsFromNewDb(); + +}; + +#endif // KEEPASSX_TESTDELETEDOBJECTS_H