keepassxc

1
/*
2
 *  Copyright (C) 2012 Felix Geyer <debfx@fobos.de>
3
 *
4
 *  This program is free software: you can redistribute it and/or modify
5
 *  it under the terms of the GNU General Public License as published by
6
 *  the Free Software Foundation, either version 2 or (at your option)
7
 *  version 3 of the License.
8
 *
9
 *  This program is distributed in the hope that it will be useful,
10
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
11
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
 *  GNU General Public License for more details.
13
 *
14
 *  You should have received a copy of the GNU General Public License
15
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
16
 */
17

18
#include "TestDeletedObjects.h"
19

20
#include "config-keepassx-tests.h"
21
#include "core/Group.h"
22
#include "crypto/Crypto.h"
23
#include "format/KdbxXmlReader.h"
24
#include "format/KeePass2.h"
25
#include <QTest>
26

27
QTEST_GUILESS_MAIN(TestDeletedObjects)
28

29
void TestDeletedObjects::initTestCase()
30
{
31
    QVERIFY(Crypto::init());
32
}
33

34
void TestDeletedObjects::createAndDelete(QSharedPointer<Database> db, int delObjectsSize)
35
{
36
    QCOMPARE(db->deletedObjects().size(), delObjectsSize);
37
    Group* root = db->rootGroup();
38
    int rootChildrenCount = root->children().size();
39

40
    auto g = new Group();
41
    g->setParent(root);
42
    QUuid gUuid = QUuid::createUuid();
43
    g->setUuid(gUuid);
44
    delete g;
45
    QCOMPARE(db->deletedObjects().size(), ++delObjectsSize);
46
    QCOMPARE(db->deletedObjects().at(delObjectsSize - 1).uuid, gUuid);
47
    QCOMPARE(rootChildrenCount, root->children().size());
48

49
    auto g1 = new Group();
50
    g1->setParent(root);
51
    QUuid g1Uuid = QUuid::createUuid();
52
    g1->setUuid(g1Uuid);
53
    auto e1 = new Entry();
54
    e1->setGroup(g1);
55
    QUuid e1Uuid = QUuid::createUuid();
56
    e1->setUuid(e1Uuid);
57
    auto g2 = new Group();
58
    g2->setParent(g1);
59
    QUuid g2Uuid = QUuid::createUuid();
60
    g2->setUuid(g2Uuid);
61
    auto e2 = new Entry();
62
    e2->setGroup(g2);
63
    QUuid e2Uuid = QUuid::createUuid();
64
    e2->setUuid(e2Uuid);
65

66
    delete g1;
67
    delObjectsSize += 4;
68

69
    QCOMPARE(db->deletedObjects().size(), delObjectsSize);
70
    QCOMPARE(db->deletedObjects().at(delObjectsSize - 4).uuid, e1Uuid);
71
    QCOMPARE(db->deletedObjects().at(delObjectsSize - 3).uuid, e2Uuid);
72
    QCOMPARE(db->deletedObjects().at(delObjectsSize - 2).uuid, g2Uuid);
73
    QCOMPARE(db->deletedObjects().at(delObjectsSize - 1).uuid, g1Uuid);
74
    QCOMPARE(rootChildrenCount, root->children().size());
75

76
    auto e3 = new Entry();
77
    e3->setGroup(root);
78
    QUuid e3Uuid = QUuid::createUuid();
79
    e3->setUuid(e3Uuid);
80

81
    delete e3;
82

83
    QCOMPARE(db->deletedObjects().size(), ++delObjectsSize);
84
    QCOMPARE(db->deletedObjects().at(delObjectsSize - 1).uuid, e3Uuid);
85
    QCOMPARE(rootChildrenCount, root->children().size());
86
}
87

88
void TestDeletedObjects::testDeletedObjectsFromFile()
89
{
90
    KdbxXmlReader reader(KeePass2::FILE_VERSION_3_1);
91
    reader.setStrictMode(true);
92
    QString xmlFile = QString(KEEPASSX_TEST_DATA_DIR).append("/NewDatabase.xml");
93
    auto db = reader.readDatabase(xmlFile);
94

95
    createAndDelete(db, 2);
96
}
97

98
void TestDeletedObjects::testDeletedObjectsFromNewDb()
99
{
100
    auto db = QSharedPointer<Database>::create();
101
    createAndDelete(db, 0);
102
}
103

104
void TestDeletedObjects::testDatabaseChange()
105
{
106
    auto db = QSharedPointer<Database>::create();
107
    Group* root = db->rootGroup();
108
    int delObjectsSize = 0;
109
    auto db2 = QSharedPointer<Database>::create();
110
    Group* root2 = db2->rootGroup();
111
    int delObjectsSize2 = 0;
112

113
    auto* e = new Entry();
114
    e->setGroup(root);
115

116
    QCOMPARE(db->deletedObjects().size(), delObjectsSize);
117
    QCOMPARE(db2->deletedObjects().size(), delObjectsSize2);
118

119
    e->setGroup(root2);
120

121
    QCOMPARE(db->deletedObjects().size(), ++delObjectsSize);
122
    QCOMPARE(db2->deletedObjects().size(), delObjectsSize2);
123

124
    delete e;
125

126
    QCOMPARE(db->deletedObjects().size(), delObjectsSize);
127
    QCOMPARE(db2->deletedObjects().size(), ++delObjectsSize2);
128

129
    auto* g1 = new Group();
130
    g1->setParent(root);
131
    QUuid g1Uuid = QUuid::createUuid();
132
    g1->setUuid(g1Uuid);
133
    auto* e1 = new Entry();
134
    e1->setGroup(g1);
135
    QUuid e1Uuid = QUuid::createUuid();
136
    e1->setUuid(e1Uuid);
137
    g1->setParent(root2);
138

139
    delObjectsSize += 2;
140
    QCOMPARE(db->deletedObjects().size(), delObjectsSize);
141
    QCOMPARE(db2->deletedObjects().size(), delObjectsSize2);
142
    QCOMPARE(db->deletedObjects().at(delObjectsSize - 2).uuid, e1Uuid);
143
    QCOMPARE(db->deletedObjects().at(delObjectsSize - 1).uuid, g1Uuid);
144

145
    auto* group = new Group();
146
    auto* entry = new Entry();
147
    entry->setGroup(group);
148
    entry->setGroup(root);
149

150
    QCOMPARE(db->deletedObjects().size(), delObjectsSize);
151
    QCOMPARE(db2->deletedObjects().size(), delObjectsSize2);
152

153
    delete group;
154
}
155

156
void TestDeletedObjects::testCustomIconDeletion()
157
{
158
    Database db;
159
    QCOMPARE(db.deletedObjects().size(), 0);
160

161
    QUuid uuid = QUuid::createUuid();
162
    db.metadata()->addCustomIcon(uuid, QByteArray());
163
    db.metadata()->removeCustomIcon(uuid);
164
    QCOMPARE(db.deletedObjects().size(), 1);
165
    QCOMPARE(db.deletedObjects().at(0).uuid, uuid);
166
}
167