Differential D8884 Diff 23216 autotests/favoritestest.cpp

Changeset View

Standalone View

autotests/favoritestest.cpp

This file was added.

1		/*
2		Copyright (c) 2013 Christian Mollekopf <mollekopf@kolabsys.com>
3		Copyright (c) 2017 David Faure <faure@kde.org>
	mlaurentUnsubmitted Not Done it's copied from christian file ? mlaurent: it's copied from christian file ?
	dfaureAuthorUnsubmitted Not Done Yes, from akonadi. All the "helper" code in this file is from him. dfaure: Yes, from akonadi. All the "helper" code in this file is from him.
4
5		This library is free software; you can redistribute it and/or modify it
6		under the terms of the GNU Library General Public License as published by
7		the Free Software Foundation; either version 2 of the License, or (at your
8		option) any later version.
9
10		This library is distributed in the hope that it will be useful, but WITHOUT
11		ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
12		FITNESS FOR A PARTICULAR PURPOSE. See the GNU Library General Public
13		License for more details.
14
15		You should have received a copy of the GNU Library General Public License
16		along with this library; see the file COPYING.LIB. If not, write to the
17		Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
18		02110-1301, USA.
19		*/
20
21		#include <QObject>
22
23		#include <KConfigGroup>
24		#include <KSharedConfig>
25
26		#include <entitytreemodel.h>
27		#include <control.h>
28		#include <qtest_akonadi.h>
29		#include <favoritecollectionsmodel.h>
30		#include <favoritecollectionorderproxymodel.h>
31		#include <changerecorder.h>
32
33		using namespace Akonadi;
34		using namespace MailCommon;
35
36		class FavoriteProxyTest : public QObject
37		{
38		Q_OBJECT
39
40		private Q_SLOTS:
41		void initTestCase();
42		void testReordering();
43		private:
44		EntityTreeModel *createETM();
45		};
46
47		void FavoriteProxyTest::initTestCase()
48		{
49		AkonadiTest::checkTestIsIsolated();
50		Akonadi::Control::start();
51		AkonadiTest::setAllResourcesOffline();
52		}
53
54		static QModelIndex getIndex(const QString &string, EntityTreeModel *model)
55		{
56		QModelIndexList list = model->match(model->index(0, 0), Qt::DisplayRole, string, 1, Qt::MatchRecursive);
57		if (list.isEmpty()) {
58		return QModelIndex();
59		}
60		return list.first();
61		}
62
63		/**
64		* Since we have no sensible way to figure out if the model is fully populated,
65		* we use the brute force approach.
66		*/
67		static bool waitForPopulation(const QModelIndex &idx, EntityTreeModel *model, int count)
68		{
69		for (int i = 0; i < 500; i++) {
70		if (model->rowCount(idx) >= count) {
71		return true;
72		}
73		QTest::qWait(10);
74		}
75		return false;
76		}
77
78		EntityTreeModel *FavoriteProxyTest::createETM()
79		{
80		ChangeRecorder *changeRecorder = new ChangeRecorder(this);
81		changeRecorder->setCollectionMonitored(Collection::root());
82		EntityTreeModel *model = new EntityTreeModel(changeRecorder, this);
83		model->setItemPopulationStrategy(Akonadi::EntityTreeModel::LazyPopulation);
84		return model;
85		}
86
87		static const int numberOfRootCollections = 4;
88
89		void FavoriteProxyTest::testReordering()
90		{
91		// GIVEN an ETM, a favorite proxy, and an EntityOrderProxyModel on top
92		EntityTreeModel *model = createETM();
93		QVERIFY(waitForPopulation(QModelIndex(), model, numberOfRootCollections));
94
95		QList<Collection::Id> collectionIds;
96		QStringList labels;
97		QStringList order;
98		for (const QString &folderName : { QStringLiteral("res2"), QStringLiteral("res3") }) {
99		const QModelIndex index = getIndex(folderName, model);
100		QVERIFY(index.isValid());
101		const Akonadi::Collection favoriteCollection = index.data(EntityTreeModel::CollectionRole).value<Akonadi::Collection>();
102		QVERIFY(favoriteCollection.isValid());
103		collectionIds.push_back(favoriteCollection.id());
104		order.push_back("c" + QString::number(favoriteCollection.id()));
105		labels << folderName;
106		}
107
108		KConfigGroup configGroup(KSharedConfig::openConfig(), "favoritecollectionsmodeltest");
109		configGroup.writeEntry("FavoriteCollectionIds", collectionIds);
110		configGroup.writeEntry("FavoriteCollectionLabels", labels);
111		configGroup.writeEntry("0", order);
112
113		FavoriteCollectionsModel *favoriteModel = new FavoriteCollectionsModel(model, configGroup, this);
114		QTRY_COMPARE(favoriteModel->rowCount(), 2);
115
116		FavoriteCollectionOrderProxyModel *orderProxy = new FavoriteCollectionOrderProxyModel(this);
117		orderProxy->setOrderConfig(configGroup);
118		orderProxy->setSourceModel(favoriteModel);
119		orderProxy->sort(0, Qt::AscendingOrder);
120
121		QCOMPARE(orderProxy->rowCount(), 2);
122
123		const QModelIndex firstRowIndex = orderProxy->index(0, 0);
124		QVERIFY(firstRowIndex.isValid());
125		QCOMPARE(firstRowIndex.data().toString(), QStringLiteral("res2"));
126		QVERIFY((orderProxy->flags(firstRowIndex) & Qt::ItemIsDropEnabled) == 0);
127		}
128
129		#include "favoritestest.moc"
130
131		QTEST_AKONADIMAIN(FavoriteProxyTest)