mirror of
				https://github.com/keepassxreboot/keepassxc.git
				synced 2025-10-31 14:39:17 -04:00 
			
		
		
		
	
		
			
				
	
	
		
			402 lines
		
	
	
	
		
			12 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			402 lines
		
	
	
	
		
			12 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| /*
 | |
|  *  Copyright (C) 2010 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 "TestGroup.h"
 | |
| 
 | |
| #include <QtCore/QPointer>
 | |
| #include <QtTest/QSignalSpy>
 | |
| #include <QtTest/QTest>
 | |
| 
 | |
| #include "tests.h"
 | |
| #include "core/Database.h"
 | |
| #include "core/Group.h"
 | |
| #include "core/Metadata.h"
 | |
| #include "crypto/Crypto.h"
 | |
| 
 | |
| void TestGroup::initTestCase()
 | |
| {
 | |
|     qRegisterMetaType<Entry*>("Entry*");
 | |
|     qRegisterMetaType<Group*>("Group*");
 | |
|     Crypto::init();
 | |
| }
 | |
| 
 | |
| void TestGroup::testParenting()
 | |
| {
 | |
|     Database* db = new Database();
 | |
|     QPointer<Group> rootGroup = db->rootGroup();
 | |
|     Group* tmpRoot = new Group();
 | |
| 
 | |
|     QPointer<Group> g1 = new Group();
 | |
|     QPointer<Group> g2 = new Group();
 | |
|     QPointer<Group> g3 = new Group();
 | |
|     QPointer<Group> g4 = new Group();
 | |
| 
 | |
|     g1->setParent(tmpRoot);
 | |
|     g2->setParent(tmpRoot);
 | |
|     g3->setParent(tmpRoot);
 | |
|     g4->setParent(tmpRoot);
 | |
| 
 | |
|     g2->setParent(g1);
 | |
|     g4->setParent(g3);
 | |
|     g3->setParent(g1);
 | |
|     g1->setParent(db->rootGroup());
 | |
| 
 | |
|     QVERIFY(g1->parent() == rootGroup);
 | |
|     QVERIFY(g2->parent() == g1);
 | |
|     QVERIFY(g3->parent() == g1);
 | |
|     QVERIFY(g4->parent() == g3);
 | |
| 
 | |
|     QVERIFY(g1->database() == db);
 | |
|     QVERIFY(g2->database() == db);
 | |
|     QVERIFY(g3->database() == db);
 | |
|     QVERIFY(g4->database() == db);
 | |
| 
 | |
|     QCOMPARE(tmpRoot->children().size(), 0);
 | |
|     QCOMPARE(rootGroup->children().size(), 1);
 | |
|     QCOMPARE(g1->children().size(), 2);
 | |
|     QCOMPARE(g2->children().size(), 0);
 | |
|     QCOMPARE(g3->children().size(), 1);
 | |
|     QCOMPARE(g4->children().size(), 0);
 | |
| 
 | |
|     QVERIFY(rootGroup->children().at(0) == g1);
 | |
|     QVERIFY(g1->children().at(0) == g2);
 | |
|     QVERIFY(g1->children().at(1) == g3);
 | |
|     QVERIFY(g3->children().contains(g4));
 | |
| 
 | |
|     Group* g5 = new Group();
 | |
|     Group* g6 = new Group();
 | |
|     g5->setParent(db->rootGroup());
 | |
|     g6->setParent(db->rootGroup());
 | |
|     QVERIFY(db->rootGroup()->children().at(1) == g5);
 | |
|     QVERIFY(db->rootGroup()->children().at(2) == g6);
 | |
| 
 | |
|     g5->setParent(db->rootGroup());
 | |
|     QVERIFY(db->rootGroup()->children().at(1) == g6);
 | |
|     QVERIFY(db->rootGroup()->children().at(2) == g5);
 | |
| 
 | |
|     QSignalSpy spy(db, SIGNAL(groupDataChanged(Group*)));
 | |
|     g2->setName("test");
 | |
|     g4->setName("test");
 | |
|     g3->setName("test");
 | |
|     g1->setName("test");
 | |
|     g3->setIcon(Uuid::random());
 | |
|     g1->setIcon(2);
 | |
|     QCOMPARE(spy.count(), 6);
 | |
| 
 | |
|     delete db;
 | |
| 
 | |
|     QVERIFY(rootGroup.isNull());
 | |
|     QVERIFY(g1.isNull());
 | |
|     QVERIFY(g2.isNull());
 | |
|     QVERIFY(g3.isNull());
 | |
|     QVERIFY(g4.isNull());
 | |
| 
 | |
|     delete tmpRoot;
 | |
| }
 | |
| 
 | |
| void TestGroup::testSignals()
 | |
| {
 | |
|     Database* db = new Database();
 | |
|     Database* db2 = new Database();
 | |
|     QPointer<Group> root = db->rootGroup();
 | |
| 
 | |
|     QSignalSpy spyAboutToAdd(db, SIGNAL(groupAboutToAdd(Group*,int)));
 | |
|     QSignalSpy spyAdded(db, SIGNAL(groupAdded()));
 | |
|     QSignalSpy spyAboutToRemove(db, SIGNAL(groupAboutToRemove(Group*)));
 | |
|     QSignalSpy spyRemoved(db, SIGNAL(groupRemoved()));
 | |
|     QSignalSpy spyAboutToMove(db, SIGNAL(groupAboutToMove(Group*,Group*,int)));
 | |
|     QSignalSpy spyMoved(db, SIGNAL(groupMoved()));
 | |
| 
 | |
|     QSignalSpy spyAboutToAdd2(db2, SIGNAL(groupAboutToAdd(Group*,int)));
 | |
|     QSignalSpy spyAdded2(db2, SIGNAL(groupAdded()));
 | |
|     QSignalSpy spyAboutToRemove2(db2, SIGNAL(groupAboutToRemove(Group*)));
 | |
|     QSignalSpy spyRemoved2(db2, SIGNAL(groupRemoved()));
 | |
|     QSignalSpy spyAboutToMove2(db2, SIGNAL(groupAboutToMove(Group*,Group*,int)));
 | |
|     QSignalSpy spyMoved2(db2, SIGNAL(groupMoved()));
 | |
| 
 | |
|     Group* g1 = new Group();
 | |
|     Group* g2 = new Group();
 | |
| 
 | |
|     g1->setParent(root);
 | |
|     QCOMPARE(spyAboutToAdd.count(), 1);
 | |
|     QCOMPARE(spyAdded.count(), 1);
 | |
|     QCOMPARE(spyAboutToRemove.count(), 0);
 | |
|     QCOMPARE(spyRemoved.count(), 0);
 | |
|     QCOMPARE(spyAboutToMove.count(), 0);
 | |
|     QCOMPARE(spyMoved.count(), 0);
 | |
| 
 | |
|     g2->setParent(root);
 | |
|     QCOMPARE(spyAboutToAdd.count(), 2);
 | |
|     QCOMPARE(spyAdded.count(), 2);
 | |
|     QCOMPARE(spyAboutToRemove.count(), 0);
 | |
|     QCOMPARE(spyRemoved.count(), 0);
 | |
|     QCOMPARE(spyAboutToMove.count(), 0);
 | |
|     QCOMPARE(spyMoved.count(), 0);
 | |
| 
 | |
|     g2->setParent(root);
 | |
|     QCOMPARE(spyAboutToAdd.count(), 2);
 | |
|     QCOMPARE(spyAdded.count(), 2);
 | |
|     QCOMPARE(spyAboutToRemove.count(), 0);
 | |
|     QCOMPARE(spyRemoved.count(), 0);
 | |
|     QCOMPARE(spyAboutToMove.count(), 0);
 | |
|     QCOMPARE(spyMoved.count(), 0);
 | |
| 
 | |
|     g2->setParent(root, 0);
 | |
|     QCOMPARE(spyAboutToAdd.count(), 2);
 | |
|     QCOMPARE(spyAdded.count(), 2);
 | |
|     QCOMPARE(spyAboutToRemove.count(), 0);
 | |
|     QCOMPARE(spyRemoved.count(), 0);
 | |
|     QCOMPARE(spyAboutToMove.count(), 1);
 | |
|     QCOMPARE(spyMoved.count(), 1);
 | |
| 
 | |
|     g1->setParent(g2);
 | |
|     QCOMPARE(spyAboutToAdd.count(), 2);
 | |
|     QCOMPARE(spyAdded.count(), 2);
 | |
|     QCOMPARE(spyAboutToRemove.count(), 0);
 | |
|     QCOMPARE(spyRemoved.count(), 0);
 | |
|     QCOMPARE(spyAboutToMove.count(), 2);
 | |
|     QCOMPARE(spyMoved.count(), 2);
 | |
| 
 | |
|     delete g1;
 | |
|     QCOMPARE(spyAboutToAdd.count(), 2);
 | |
|     QCOMPARE(spyAdded.count(), 2);
 | |
|     QCOMPARE(spyAboutToRemove.count(), 1);
 | |
|     QCOMPARE(spyRemoved.count(), 1);
 | |
|     QCOMPARE(spyAboutToMove.count(), 2);
 | |
|     QCOMPARE(spyMoved.count(), 2);
 | |
| 
 | |
|     g2->setParent(db2->rootGroup());
 | |
|     QCOMPARE(spyAboutToAdd.count(), 2);
 | |
|     QCOMPARE(spyAdded.count(), 2);
 | |
|     QCOMPARE(spyAboutToRemove.count(), 2);
 | |
|     QCOMPARE(spyRemoved.count(), 2);
 | |
|     QCOMPARE(spyAboutToMove.count(), 2);
 | |
|     QCOMPARE(spyMoved.count(), 2);
 | |
|     QCOMPARE(spyAboutToAdd2.count(), 1);
 | |
|     QCOMPARE(spyAdded2.count(), 1);
 | |
|     QCOMPARE(spyAboutToRemove2.count(), 0);
 | |
|     QCOMPARE(spyRemoved2.count(), 0);
 | |
|     QCOMPARE(spyAboutToMove2.count(), 0);
 | |
|     QCOMPARE(spyMoved2.count(), 0);
 | |
| 
 | |
|     Group* g3 = new Group();
 | |
|     Group* g4 = new Group();
 | |
| 
 | |
|     g3->setParent(root);
 | |
|     QCOMPARE(spyAboutToAdd.count(), 3);
 | |
|     QCOMPARE(spyAdded.count(), 3);
 | |
|     QCOMPARE(spyAboutToRemove.count(), 2);
 | |
|     QCOMPARE(spyRemoved.count(), 2);
 | |
|     QCOMPARE(spyAboutToMove.count(), 2);
 | |
|     QCOMPARE(spyMoved.count(), 2);
 | |
| 
 | |
|     g4->setParent(root);
 | |
|     QCOMPARE(spyAboutToAdd.count(), 4);
 | |
|     QCOMPARE(spyAdded.count(), 4);
 | |
|     QCOMPARE(spyAboutToRemove.count(), 2);
 | |
|     QCOMPARE(spyRemoved.count(), 2);
 | |
|     QCOMPARE(spyAboutToMove.count(), 2);
 | |
|     QCOMPARE(spyMoved.count(), 2);
 | |
| 
 | |
|     g3->setParent(root);
 | |
|     QCOMPARE(spyAboutToAdd.count(), 4);
 | |
|     QCOMPARE(spyAdded.count(), 4);
 | |
|     QCOMPARE(spyAboutToRemove.count(), 2);
 | |
|     QCOMPARE(spyRemoved.count(), 2);
 | |
|     QCOMPARE(spyAboutToMove.count(), 3);
 | |
|     QCOMPARE(spyMoved.count(), 3);
 | |
| 
 | |
|     delete db;
 | |
|     delete db2;
 | |
| 
 | |
|     QVERIFY(root.isNull());
 | |
| }
 | |
| 
 | |
| void TestGroup::testEntries()
 | |
| {
 | |
|     Group* group = new Group();
 | |
| 
 | |
|     QPointer<Entry> entry1 = new Entry();
 | |
|     entry1->setGroup(group);
 | |
| 
 | |
|     QPointer<Entry> entry2 = new Entry();
 | |
|     entry2->setGroup(group);
 | |
| 
 | |
|     QCOMPARE(group->entries().size(), 2);
 | |
|     QVERIFY(group->entries().at(0) == entry1);
 | |
|     QVERIFY(group->entries().at(1) == entry2);
 | |
| 
 | |
|     delete group;
 | |
| 
 | |
|     QVERIFY(entry1.isNull());
 | |
|     QVERIFY(entry2.isNull());
 | |
| }
 | |
| 
 | |
| void TestGroup::testDeleteSignals()
 | |
| {
 | |
|     Database* db = new Database();
 | |
|     Group* groupRoot = db->rootGroup();
 | |
|     Group* groupChild = new Group();
 | |
|     Group* groupChildChild = new Group();
 | |
|     groupRoot->setObjectName("groupRoot");
 | |
|     groupChild->setObjectName("groupChild");
 | |
|     groupChildChild->setObjectName("groupChildChild");
 | |
|     groupChild->setParent(groupRoot);
 | |
|     groupChildChild->setParent(groupChild);
 | |
|     QSignalSpy spyAboutToRemove(db, SIGNAL(groupAboutToRemove(Group*)));
 | |
|     QSignalSpy spyRemoved(db, SIGNAL(groupRemoved()));
 | |
| 
 | |
|     delete groupChild;
 | |
|     QVERIFY(groupRoot->children().isEmpty());
 | |
|     QCOMPARE(spyAboutToRemove.count(), 1);
 | |
|     QCOMPARE(spyRemoved.count(), 1);
 | |
|     delete db;
 | |
| 
 | |
| 
 | |
|     Group* group = new Group();
 | |
|     Entry* entry = new Entry();
 | |
|     entry->setGroup(group);
 | |
|     QSignalSpy spyEntryAboutToRemove(group, SIGNAL(entryAboutToRemove(Entry*)));
 | |
|     QSignalSpy spyEntryRemoved(group, SIGNAL(entryRemoved(Entry*)));
 | |
| 
 | |
|     delete entry;
 | |
|     QVERIFY(group->entries().isEmpty());
 | |
|     QCOMPARE(spyEntryAboutToRemove.count(), 1);
 | |
|     QCOMPARE(spyEntryRemoved.count(), 1);
 | |
|     delete group;
 | |
| 
 | |
|     Database* db2 = new Database();
 | |
|     Group* groupRoot2 = db2->rootGroup();
 | |
|     Group* group2 = new Group();
 | |
|     group2->setParent(groupRoot2);
 | |
|     Entry* entry2 = new Entry();
 | |
|     entry2->setGroup(group2);
 | |
|     QSignalSpy spyEntryAboutToRemove2(group2, SIGNAL(entryAboutToRemove(Entry*)));
 | |
|     QSignalSpy spyEntryRemoved2(group2, SIGNAL(entryRemoved(Entry*)));
 | |
| 
 | |
|     delete group2;
 | |
|     QCOMPARE(spyEntryAboutToRemove2.count(), 0);
 | |
|     QCOMPARE(spyEntryRemoved2.count(), 0);
 | |
|     delete db2;
 | |
| }
 | |
| 
 | |
| void TestGroup::testCopyCustomIcon()
 | |
| {
 | |
|     Database* dbSource = new Database();
 | |
| 
 | |
|     Uuid groupIconUuid = Uuid::random();
 | |
|     QImage groupIcon(16, 16, QImage::Format_RGB32);
 | |
|     groupIcon.setPixel(0, 0, qRgb(255, 0, 0));
 | |
|     dbSource->metadata()->addCustomIcon(groupIconUuid, groupIcon);
 | |
| 
 | |
|     Uuid entryIconUuid = Uuid::random();
 | |
|     QImage entryIcon(16, 16, QImage::Format_RGB32);
 | |
|     entryIcon.setPixel(0, 0, qRgb(255, 0, 0));
 | |
|     dbSource->metadata()->addCustomIcon(entryIconUuid, entryIcon);
 | |
| 
 | |
|     Group* group = new Group();
 | |
|     group->setParent(dbSource->rootGroup());
 | |
|     group->setIcon(groupIconUuid);
 | |
|     QCOMPARE(group->icon(), groupIcon);
 | |
| 
 | |
|     Entry* entry = new Entry();
 | |
|     entry->setGroup(dbSource->rootGroup());
 | |
|     entry->setIcon(entryIconUuid);
 | |
|     QCOMPARE(entry->icon(), entryIcon);
 | |
| 
 | |
|     Database* dbTarget = new Database();
 | |
| 
 | |
|     group->setParent(dbTarget->rootGroup());
 | |
|     QVERIFY(dbTarget->metadata()->containsCustomIcon(groupIconUuid));
 | |
|     QCOMPARE(dbTarget->metadata()->customIcon(groupIconUuid), groupIcon);
 | |
|     QCOMPARE(group->icon(), groupIcon);
 | |
| 
 | |
|     entry->setGroup(dbTarget->rootGroup());
 | |
|     QVERIFY(dbTarget->metadata()->containsCustomIcon(entryIconUuid));
 | |
|     QCOMPARE(dbTarget->metadata()->customIcon(entryIconUuid), entryIcon);
 | |
|     QCOMPARE(entry->icon(), entryIcon);
 | |
| 
 | |
|     delete dbSource;
 | |
|     delete dbTarget;
 | |
| }
 | |
| 
 | |
| void TestGroup::testSearch()
 | |
| {
 | |
|     Group* groupRoot = new Group();
 | |
|     Group* group1 = new Group();
 | |
|     Group* group2 = new Group();
 | |
|     Group* group3 = new Group();
 | |
| 
 | |
|     group1->setParent(groupRoot);
 | |
|     group2->setParent(groupRoot);
 | |
|     group3->setParent(groupRoot);
 | |
| 
 | |
|     Group* group11 = new Group();
 | |
| 
 | |
|     group11->setParent(group1);
 | |
| 
 | |
|     Group* group21 = new Group();
 | |
|     Group* group211 = new Group();
 | |
|     Group* group2111 = new Group();
 | |
| 
 | |
|     group21->setParent(group2);
 | |
|     group211->setParent(group21);
 | |
|     group2111->setParent(group211);
 | |
| 
 | |
|     group1->setSearchingEnabled(Group::Disable);
 | |
|     group11->setSearchingEnabled(Group::Enable);
 | |
| 
 | |
|     Entry* eRoot = new Entry();
 | |
|     eRoot->setNotes("test search term test");
 | |
|     eRoot->setGroup(groupRoot);
 | |
| 
 | |
|     Entry* eRoot2 = new Entry();
 | |
|     eRoot2->setNotes("test term test");
 | |
|     eRoot2->setGroup(groupRoot);
 | |
| 
 | |
|     Entry* e1 = new Entry();
 | |
|     e1->setNotes("test search term test");
 | |
|     e1->setGroup(group1);
 | |
| 
 | |
|     Entry* e2111 = new Entry();
 | |
|     e2111->setNotes("test search term test");
 | |
|     e2111->setGroup(group2111);
 | |
| 
 | |
|     Entry* e2111b = new Entry();
 | |
|     e2111b->setNotes("test search test");
 | |
|     e2111b->setGroup(group2111);
 | |
| 
 | |
|     Entry* e3 = new Entry();
 | |
|     e3->setNotes("test search term test");
 | |
|     e3->setGroup(group3);
 | |
| 
 | |
|     Entry* e3b = new Entry();
 | |
|     e3b->setNotes("test search test");
 | |
|     e3b->setGroup(group3);
 | |
| 
 | |
|     QList<Entry*> searchResult;
 | |
| 
 | |
|     searchResult = groupRoot->search("search term", Qt::CaseInsensitive);
 | |
|     QCOMPARE(searchResult.count(), 3);
 | |
| 
 | |
|     searchResult = group211->search("search term", Qt::CaseInsensitive);
 | |
|     QCOMPARE(searchResult.count(), 1);
 | |
| 
 | |
|     delete groupRoot;
 | |
| }
 | |
| 
 | |
| KEEPASSX_QTEST_CORE_MAIN(TestGroup)
 | 
