keepassxc/tests/TestDeletedObjects.cpp
angelsl 3461cbfb06
Rename KeePass2{,Xml}{R,W} to Kdbx3{,Xml}{R,W}, and add a redirection class
This class will in future select Kdbx4{R,W} as appropriate.
2018-01-13 14:23:26 -05:00

163 lines
4.6 KiB
C++

/*
* Copyright (C) 2012 Felix Geyer <debfx@fobos.de>
*
* 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 <http://www.gnu.org/licenses/>.
*/
#include "TestDeletedObjects.h"
#include <QTest>
#include "core/Database.h"
#include "core/Group.h"
#include "crypto/Crypto.h"
#include "format/Kdbx3XmlReader.h"
#include "config-keepassx-tests.h"
QTEST_GUILESS_MAIN(TestDeletedObjects)
void TestDeletedObjects::initTestCase()
{
QVERIFY(Crypto::init());
}
void TestDeletedObjects::createAndDelete(Database* db, int delObjectsSize)
{
QCOMPARE(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;
QCOMPARE(db->deletedObjects().size(), ++delObjectsSize);
QCOMPARE(db->deletedObjects().at(delObjectsSize-1).uuid, gUuid);
QCOMPARE(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;
QCOMPARE(db->deletedObjects().size(), delObjectsSize);
QCOMPARE(db->deletedObjects().at(delObjectsSize-4).uuid, e1Uuid);
QCOMPARE(db->deletedObjects().at(delObjectsSize-3).uuid, e2Uuid);
QCOMPARE(db->deletedObjects().at(delObjectsSize-2).uuid, g2Uuid);
QCOMPARE(db->deletedObjects().at(delObjectsSize-1).uuid, g1Uuid);
QCOMPARE(rootChildrenCount, root->children().size());
Entry* e3 = new Entry();
e3->setGroup(root);
Uuid e3Uuid = Uuid::random();
e3->setUuid(e3Uuid);
delete e3;
QCOMPARE(db->deletedObjects().size(), ++delObjectsSize);
QCOMPARE(db->deletedObjects().at(delObjectsSize-1).uuid, e3Uuid);
QCOMPARE(rootChildrenCount, root->children().size());
}
void TestDeletedObjects::testDeletedObjectsFromFile()
{
Kdbx3XmlReader reader;
reader.setStrictMode(true);
QString xmlFile = QString(KEEPASSX_TEST_DATA_DIR).append("/NewDatabase.xml");
Database* db = reader.readDatabase(xmlFile);
createAndDelete(db, 2);
delete db;
}
void TestDeletedObjects::testDeletedObjectsFromNewDb()
{
Database* db = new Database();
createAndDelete(db, 0);
delete db;
}
void TestDeletedObjects::testDatabaseChange()
{
Database* db = new Database();
Group* root = db->rootGroup();
int delObjectsSize = 0;
Database* db2 = new Database();
Group* root2 = db2->rootGroup();
int delObjectsSize2 = 0;
Entry* e = new Entry();
e->setGroup(root);
QCOMPARE(db->deletedObjects().size(), delObjectsSize);
QCOMPARE(db2->deletedObjects().size(), delObjectsSize2);
e->setGroup(root2);
QCOMPARE(db->deletedObjects().size(), ++delObjectsSize);
QCOMPARE(db2->deletedObjects().size(), delObjectsSize2);
delete e;
QCOMPARE(db->deletedObjects().size(), delObjectsSize);
QCOMPARE(db2->deletedObjects().size(), ++delObjectsSize2);
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);
g1->setParent(root2);
delObjectsSize += 2;
QCOMPARE(db->deletedObjects().size(), delObjectsSize);
QCOMPARE(db2->deletedObjects().size(), delObjectsSize2);
QCOMPARE(db->deletedObjects().at(delObjectsSize-2).uuid, e1Uuid);
QCOMPARE(db->deletedObjects().at(delObjectsSize-1).uuid, g1Uuid);
Group* group = new Group();
Entry* entry = new Entry();
entry->setGroup(group);
entry->setGroup(root);
QCOMPARE(db->deletedObjects().size(), delObjectsSize);
QCOMPARE(db2->deletedObjects().size(), delObjectsSize2);
delete group;
delete db;
delete db2;
}