keepassxc

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

19
#include "TestCsvExporter.h"
20

21
#include <QBuffer>
22
#include <QTest>
23

24
#include "core/Group.h"
25
#include "core/Totp.h"
26
#include "crypto/Crypto.h"
27
#include "format/CsvExporter.h"
28

29
QTEST_GUILESS_MAIN(TestCsvExporter)
30

31
const QString TestCsvExporter::ExpectedHeaderLine =
32
    QString("\"Group\",\"Title\",\"Username\",\"Password\",\"URL\",\"Notes\",\"TOTP\",\"Icon\",\"Last "
33
            "Modified\",\"Created\"\n");
34

35
void TestCsvExporter::init()
36
{
37
    m_db = QSharedPointer<Database>::create();
38
    m_csvExporter = QSharedPointer<CsvExporter>::create();
39
}
40

41
void TestCsvExporter::initTestCase()
42
{
43
    Crypto::init();
44
}
45

46
void TestCsvExporter::cleanup()
47
{
48
}
49

50
void TestCsvExporter::testExport()
51
{
52
    Group* groupRoot = m_db->rootGroup();
53
    auto* group = new Group();
54
    group->setName("Test Group Name");
55
    group->setParent(groupRoot);
56
    auto* entry = new Entry();
57
    entry->setGroup(group);
58
    entry->setTitle("Test Entry Title");
59
    entry->setUsername("Test Username");
60
    entry->setPassword("Test Password");
61
    entry->setUrl("http://test.url");
62
    entry->setNotes("Test Notes");
63
    entry->setTotp(Totp::createSettings("DFDF", Totp::DEFAULT_DIGITS, Totp::DEFAULT_STEP));
64
    entry->setIcon(5);
65

66
    QBuffer buffer;
67
    QVERIFY(buffer.open(QIODevice::ReadWrite));
68
    m_csvExporter->exportDatabase(&buffer, m_db);
69
    auto exported = QString::fromUtf8(buffer.buffer());
70

71
    QString expectedResult = QString()
72
                                 .append(ExpectedHeaderLine)
73
                                 .append("\"Passwords/Test Group Name\",\"Test Entry Title\",\"Test Username\",\"Test "
74
                                         "Password\",\"http://test.url\",\"Test Notes\"");
75

76
    QVERIFY(exported.startsWith(expectedResult));
77
    exported.remove(expectedResult);
78
    QVERIFY(exported.contains("otpauth://"));
79
    QVERIFY(exported.contains(",\"5\","));
80
}
81

82
void TestCsvExporter::testEmptyDatabase()
83
{
84
    QBuffer buffer;
85
    QVERIFY(buffer.open(QIODevice::ReadWrite));
86
    m_csvExporter->exportDatabase(&buffer, m_db);
87

88
    QCOMPARE(QString::fromUtf8(buffer.buffer().constData()), ExpectedHeaderLine);
89
}
90

91
void TestCsvExporter::testNestedGroups()
92
{
93
    Group* groupRoot = m_db->rootGroup();
94
    auto* group = new Group();
95
    group->setName("Test Group Name");
96
    group->setParent(groupRoot);
97
    auto* childGroup = new Group();
98
    childGroup->setName("Test Sub Group Name");
99
    childGroup->setParent(group);
100
    auto* entry = new Entry();
101
    entry->setGroup(childGroup);
102
    entry->setTitle("Test Entry Title");
103

104
    QBuffer buffer;
105
    QVERIFY(buffer.open(QIODevice::ReadWrite));
106
    m_csvExporter->exportDatabase(&buffer, m_db);
107
    auto exported = QString::fromUtf8(buffer.buffer());
108
    QVERIFY(exported.startsWith(
109
        QString()
110
            .append(ExpectedHeaderLine)
111
            .append("\"Passwords/Test Group Name/Test Sub Group Name\",\"Test Entry Title\",\"\",\"\",\"\",\"\"")));
112
}
113