1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
#include <QtTest>
#include <iostream>
#include <QDebug>
#include <QString>
#include <QtConcurrent/QtConcurrentRun>
#include "common/storage.h"
class StorageTest : public QObject
{
Q_OBJECT
private:
//This should point to a directory on disk and not a ramdisk (since we're measuring performance)
QString testDataPath;
QString dbName;
const char *keyPrefix = "key";
void populate(int count)
{
Storage storage(testDataPath, dbName);
for (int i = 0; i < count; i++) {
//This should perhaps become an implementation detail of the db?
if (i % 10000 == 0) {
if (i > 0) {
storage.commitTransaction();
}
storage.startTransaction();
}
storage.write(keyPrefix + std::to_string(i), keyPrefix + std::to_string(i));
}
storage.commitTransaction();
}
bool verify(Storage &storage, int i)
{
bool success = true;
bool keyMatch = true;
const auto reference = keyPrefix + std::to_string(i);
success = storage.read(keyPrefix + std::to_string(i), [&error, &reference](const std::string &value) {
if (value != reference) {
qDebug() << "Mismatch while reading";
keyMatch = false;
}
});
return succes && keyMatch;
}
private Q_SLOTS:
void initTestCase()
{
testDataPath = "./testdb";
dbName = "test";
}
void cleanupTestCase()
{
Database db(testDataPath, dbName);
db.removeFromDisk();
}
void testRead()
{
const int count = 100;
populate(count);
//ensure we can read everything back correctly
{
Database db(testDataPath, dbName);
for (int i = 0; i < count; i++) {
QVERIFY(verify(db, i));
}
}
Database db(testDataPath, dbName);
db.removeFromDisk();
}
void testConcurrentRead()
{
const int count = 10000;
populate(count);
//Try to concurrently read
QList<QFuture<void> > futures;
const int concurrencyLevel = 4;
for (int num = 0; num < concurrencyLevel; num++) {
futures << QtConcurrent::run([this, count](){
Database db(testDataPath, dbName);
for (int i = 0; i < count; i++) {
if (!verify(db, i)) {
qWarning() << "invalid value";
break;
}
}
});
}
for(auto future : futures) {
future.waitForFinished();
}
Database db(testDataPath, dbName);
db.removeFromDisk();
}
};
QTEST_MAIN(StorageTest)
#include "storagetest.moc"
|
