Курнат Андрей
108 lines
3.5 KiB
C++
108 lines
3.5 KiB
C++
|
|
/*
|
|
* This software is in the public domain, furnished "as is", without technical
|
|
* support, and with no warranty, express or implied, as to its usefulness for
|
|
* any purpose.
|
|
*
|
|
*/
|
|
|
|
#include "gui/models/protocolitemmodel.h"
|
|
#include "gui/accountmanager.h"
|
|
#include "gui/accountstate.h"
|
|
#include "gui/folderman.h"
|
|
|
|
#include "testutils/testutils.h"
|
|
|
|
#include <QTest>
|
|
#include <QAbstractItemModelTester>
|
|
#include <folder.h>
|
|
|
|
namespace OCC {
|
|
|
|
class TestProtocolModel : public QObject
|
|
{
|
|
Q_OBJECT
|
|
|
|
private Q_SLOTS:
|
|
void testInsertAndRemove()
|
|
{
|
|
auto model = new ProtocolItemModel(300, false, this);
|
|
|
|
new QAbstractItemModelTester(model, this);
|
|
|
|
auto dir = TestUtils::createTempDir();
|
|
|
|
auto newAccountState = TestUtils::createDummyAccount();
|
|
|
|
const QDir d(dir.path());
|
|
QVERIFY(d.mkdir(QStringLiteral("foo")));
|
|
QVERIFY(d.mkdir(QStringLiteral("bar")));
|
|
auto foo = TestUtils::folderMan()->addFolder(
|
|
newAccountState.get(), TestUtils::createDummyFolderDefinition(newAccountState->account(), dir.path() + QStringLiteral("/foo")));
|
|
auto bar = TestUtils::folderMan()->addFolder(
|
|
newAccountState.get(), TestUtils::createDummyFolderDefinition(newAccountState->account(), dir.path() + QStringLiteral("/bar")));
|
|
|
|
QVERIFY(foo);
|
|
QVERIFY(bar);
|
|
|
|
// populate with dummy data
|
|
// -1 to test the ring buffer window roll over
|
|
const auto size = model->rawData().capacity() - 1;
|
|
auto item = SyncFileItemPtr::create();
|
|
std::vector<ProtocolItem> tmp;
|
|
tmp.reserve(size);
|
|
for (size_t i = 0; i < size; ++i) {
|
|
item->setLocalName(QString::number(i));
|
|
tmp.emplace_back(foo, item);
|
|
}
|
|
model->reset(std::move(tmp));
|
|
|
|
// test some inserts
|
|
for (int i = 0; i < 5; ++i) {
|
|
item->setLocalName(QString::number(i));
|
|
model->addProtocolItem(ProtocolItem { bar, item });
|
|
}
|
|
|
|
const auto oldSize = model->rowCount();
|
|
QCOMPARE(oldSize, model->rawData().capacity());
|
|
// pick one from the middle
|
|
const auto toBeRemoved = {
|
|
model->protocolItem(model->index(0, 0)),
|
|
model->protocolItem(model->index(static_cast<int>(model->rawData().capacity()) / 2, 0)),
|
|
model->protocolItem(model->index(static_cast<int>(model->rawData().capacity()) / 3, 0))
|
|
};
|
|
|
|
std::vector<ProtocolItem> copy;
|
|
copy.reserve(model->rowCount());
|
|
for (int i = 0; i < model->rowCount(); ++i) {
|
|
copy.push_back(model->protocolItem(model->index(i, 0)));
|
|
}
|
|
|
|
int matches = 0;
|
|
const auto filter = [&toBeRemoved, &matches](const ProtocolItem &pi) {
|
|
for (const auto &tb : toBeRemoved) {
|
|
if (pi.folder() == tb.folder() && pi.path() == tb.path()) {
|
|
matches++;
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
};
|
|
copy.erase(std::remove_if(copy.begin(), copy.end(), filter), copy.cend());
|
|
QCOMPARE(matches, toBeRemoved.size());
|
|
matches = 0;
|
|
model->remove_if(filter);
|
|
QCOMPARE(matches, toBeRemoved.size());
|
|
QCOMPARE(oldSize - 3, model->rowCount());
|
|
|
|
// ensure we kept the original order
|
|
for (int i = 0; i < model->rowCount(); ++i) {
|
|
QCOMPARE(model->protocolItem(model->index(i, 0)).folder(), copy[i].folder());
|
|
}
|
|
}
|
|
};
|
|
}
|
|
|
|
QTEST_MAIN(OCC::TestProtocolModel)
|
|
#include "testprotocolmodel.moc"
|