on bionic help sign in
130 lines 4.4 kB history raw
1/*
2 * Copyright (C) 2015-2016 Canonical, Ltd.
3 *
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; version 3.
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
12 *
13 * You should have received a copy of the GNU General Public License
14 * along with this program. If not, see <http://www.gnu.org/licenses/>.
15 */
17#include "SessionsModel.h"
19#include "MockController.h"
21#include <QLightDM/SessionsModel>
22#include <QtCore/QModelIndex>
23#include <QtTest>
24#include <QString>
26class GreeterSessionsModelTest : public QObject
28 Q_OBJECT
30private Q_SLOTS:
32 void init()
33 {
34 model = new SessionsModel();
35 QVERIFY(model);
36 mock = QLightDM::MockController::instance();
37 QVERIFY(mock);
38 }
40 void cleanup()
41 {
42 delete model;
43 // mock will be a singleton, doesn't need to be cleaned
44 }
46 static QModelIndex findByKey(QAbstractItemModel *model, const QString& key)
47 {
48 for (int i = 0; i < model->rowCount(QModelIndex()); i++) {
49 if (model->data(model->index(i, 0), QLightDM::SessionsModel::KeyRole).toString() == key)
50 return model->index(i, 0);
51 }
53 return QModelIndex();
54 }
56 void testIconDirectoriesAreValid()
57 {
58 Q_FOREACH(const QUrl& searchDirectory, model->iconSearchDirectories())
59 {
60 QVERIFY(searchDirectory.isValid());
61 }
62 }
64 void testIconLookupLogic()
65 {
66 const QString data[][2] = {
67 //{SESSION KEY, ICON NAME}
68 {"gnome-classic", "gnome_badge.png"},
69 {"gnome-flashback-compiz", "gnome_badge.png"},
70 {"gnome-flashback-metacity", "gnome_badge.png"},
71 {"gnome-shell", "gnome_badge.png"},
72 {"gnome-wayland", "gnome_badge.png"},
73 {"gnome", "gnome_badge.png"},
74 {"kde", "kde_badge.png"},
75 {"plasma", "kde_badge.png"},
76 {"recovery_console", "recovery_console_badge.png"},
77 {"ubuntu","ubuntu_badge.png"},
78 {"ubuntu-2d", "ubuntu_badge.png"},
79 {"xterm", "recovery_console_badge.png"},
80 {"made up session", "unknown_badge.png"},
81 {"", "unknown_badge.png"}
82 };
84 const short length = sizeof(data) / sizeof(data[0]);
85 for (int i = 0; i < length; i++) {
86 std::string key = data[i][0].toStdString();
87 std::string icon = data[i][1].toStdString();
88 std::string url = model->iconUrl(data[i][0]).toString().toStdString();
89 std::string errorMessage = "Session icon url for " + key +
90 " should contain " + icon + " but " + url + " was returned";
91 QVERIFY2(model->iconUrl(data[i][0]).toString().contains(data[i][1]),
92 errorMessage.c_str());
93 }
94 }
96 void testMultipleSessionsCountIsCorrect()
97 {
98 mock->setProperty("sessionMode", "full");
99 QVERIFY(model->rowCount(QModelIndex()) > 1);
100 }
102 void testNoSessionsCountIsCorrect()
103 {
104 mock->setProperty("sessionMode", "none");
105 QVERIFY(model->rowCount(QModelIndex()) == 0);
106 }
108 void testSingleSessionCountIsCorrect()
109 {
110 mock->setProperty("sessionMode", "single");
111 QVERIFY(model->rowCount(QModelIndex()) == 1);
112 }
114 void testSessionNameIsCorrect()
115 {
116 // This is testing the lookup, not the correctness of the strings,
117 // so one test should be sufficient
118 mock->setProperty("sessionMode", "full");
119 QVERIFY(model->data(findByKey(model, "ubuntu"),
120 Qt::DisplayRole).toString() == "Ubuntu");
121 }
123private:
124 SessionsModel *model;
125 QObject *mock;
126};
128QTEST_MAIN(GreeterSessionsModelTest)
130#include "sessionsmodel.moc"