mirror of
https://github.com/keepassxreboot/keepassxc.git
synced 2024-09-20 08:05:40 +00:00
79 lines
2.3 KiB
C++
79 lines
2.3 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 "core/Database.h"
|
||
|
|
#include "core/Metadata.h"
|
||
|
|
#include "crypto/Crypto.h"
|
||
|
|
#include "format/KeePass2Reader.h"
|
||
|
|
#include "format/KeePass2Writer.h"
|
||
|
|
#include "keys/PasswordKey.h"
|
||
|
|
|
||
|
|
#include "format/KeePass2XmlWriter.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 = new Group();
|
||
|
|
group->setUuid(Uuid::random());
|
||
|
|
group->setParent(m_dbOrg);
|
||
|
|
m_dbOrg->setRootGroup(group);
|
||
|
|
Entry* entry = new Entry();
|
||
|
|
entry->setUuid(Uuid::random());
|
||
|
|
entry->addAttribute("test", "protectedTest", true);
|
||
|
|
QVERIFY(entry->isAttributeProtected("test"));
|
||
|
|
group->addEntry(entry);
|
||
|
|
|
||
|
|
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.error());
|
||
|
|
QVERIFY(m_dbTest);
|
||
|
|
}
|
||
|
|
|
||
|
|
void TestKeePass2Writer::testBasic()
|
||
|
|
{
|
||
|
|
QCOMPARE(m_dbTest->metadata()->name(), m_dbOrg->metadata()->name());
|
||
|
|
QVERIFY(m_dbTest->rootGroup());
|
||
|
|
}
|
||
|
|
|
||
|
|
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);
|
||
|
|
}
|
||
|
|
|
||
|
|
QTEST_MAIN(TestKeePass2Writer);
|