mirror of
https://github.com/keepassxreboot/keepassxc.git
synced 2024-12-28 00:39:43 -05:00
83 lines
2.5 KiB
C++
83 lines
2.5 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 "TestKeePass2Writer.h"
|
|
|
|
#include <QtCore/QBuffer>
|
|
#include <QtTest/QTest>
|
|
|
|
#include "tests.h"
|
|
#include "core/Database.h"
|
|
#include "core/Group.h"
|
|
#include "core/Metadata.h"
|
|
#include "crypto/Crypto.h"
|
|
#include "format/KeePass2Reader.h"
|
|
#include "format/KeePass2Writer.h"
|
|
#include "keys/PasswordKey.h"
|
|
|
|
void TestKeePass2Writer::initTestCase()
|
|
{
|
|
Crypto::init();
|
|
|
|
CompositeKey key;
|
|
key.addKey(PasswordKey("test"));
|
|
|
|
m_dbOrg = new Database();
|
|
m_dbOrg->setKey(key);
|
|
m_dbOrg->metadata()->setName("TESTDB");
|
|
Group* group = m_dbOrg->rootGroup();
|
|
group->setUuid(Uuid::random());
|
|
Entry* entry = new Entry();
|
|
entry->setUuid(Uuid::random());
|
|
entry->addAttribute("test", "protectedTest", true);
|
|
QVERIFY(entry->isAttributeProtected("test"));
|
|
entry->setGroup(group);
|
|
Group* groupNew = new Group();
|
|
groupNew->setUuid(Uuid::random());
|
|
groupNew->setName("TESTGROUP");
|
|
groupNew->setParent(group);
|
|
|
|
QBuffer buffer;
|
|
buffer.open(QBuffer::ReadWrite);
|
|
|
|
KeePass2Writer writer;
|
|
writer.writeDatabase(&buffer, m_dbOrg);
|
|
QVERIFY(!writer.error());
|
|
buffer.seek(0);
|
|
KeePass2Reader reader;
|
|
m_dbTest = reader.readDatabase(&buffer, key);
|
|
QVERIFY(!reader.hasError());
|
|
QVERIFY(m_dbTest);
|
|
}
|
|
|
|
void TestKeePass2Writer::testBasic()
|
|
{
|
|
QCOMPARE(m_dbTest->metadata()->name(), m_dbOrg->metadata()->name());
|
|
QVERIFY(m_dbTest->rootGroup());
|
|
QCOMPARE(m_dbTest->rootGroup()->children()[0]->name(), m_dbOrg->rootGroup()->children()[0]->name());
|
|
}
|
|
|
|
void TestKeePass2Writer::testProtectedAttributes()
|
|
{
|
|
QCOMPARE(m_dbTest->rootGroup()->entries().size(), 1);
|
|
Entry* entry = m_dbTest->rootGroup()->entries().at(0);
|
|
QCOMPARE(entry->attributes().value("test"), QString("protectedTest"));
|
|
QCOMPARE(entry->isAttributeProtected("test"), true);
|
|
}
|
|
|
|
KEEPASSX_QTEST_CORE_MAIN(TestKeePass2Writer)
|