1 files changed, 185 insertions, 0 deletions
diff --git a/tests/client/modeltest.cpp b/tests/client/modeltest.cpp
new file mode 100644
index 0000000..1963327
--- /dev/null
+++ b/tests/client/modeltest.cpp
@@ -0,0 +1,185 @@
+#include "qdbusmenumodel.h"
+#include "dbusmenuscript.h"
+
+#include <QObject>
+#include <QSignalSpy>
+#include <QtTestGui>
+#include <QDebug>
+
+class ModelTest : public QObject
+{
+    Q_OBJECT
+private:
+    DBusMenuScript m_script;
+    QDBusMenuModel m_model;
+
+private Q_SLOTS:
+    void initTestCase()
+    {
+        Q_ASSERT(m_script.connect());
+    }
+
+    void cleanupTestCase()
+    {
+        m_script.quit();
+    }
+
+    void init()
+    {
+        m_model.stop();
+        m_model.setBusType(QDBusObject::SessionBus);
+        m_model.setBusName(MENU_SERVICE_NAME);
+        m_model.setObjectPath(MENU_OBJECT_PATH);
+    }
+
+    void cleanup()
+    {
+        m_script.unpublishMenu();
+    }
+
+    void testColumnCount()
+    {
+        QCOMPARE(m_model.columnCount(), 1);
+    }
+
+    void testParent()
+    {
+        QCOMPARE(m_model.parent(QModelIndex()), QModelIndex());
+    }
+
+    void testBusTypeProperty()
+    {
+        m_model.setProperty("busType", 1);
+        QCOMPARE(m_model.busType(), QDBusObject::SessionBus);
+
+        m_model.setProperty("busType", 2);
+        QCOMPARE(m_model.busType(), QDBusObject::SystemBus);
+
+        m_model.setProperty("busType", 0);
+        QCOMPARE(m_model.busType(), QDBusObject::SystemBus);
+
+        m_model.setProperty("busType", 10);
+        QCOMPARE(m_model.busType(), QDBusObject::SystemBus);
+
+    }
+
+    void testData()
+    {
+        // Make menu available
+        m_script.publishMenu();
+        m_script.run();
+
+        // start model
+        m_model.start();
+
+        // Wait for dbus sync
+        QTest::qWait(500);
+
+        QCOMPARE(m_model.status(), QDBusObject::Connected);
+        QCOMPARE(m_model.rowCount(), 4);
+
+        // Label (String)
+        QVariant label = m_model.data(m_model.index(0, 0), QMenuModel::Label);
+        QVERIFY(label.isValid());
+        QCOMPARE(label.type(), QVariant::String);
+        QCOMPARE(label.toString(), QString("Menu0"));
+
+        // Action (String)
+        QVariant action = m_model.data(m_model.index(1, 0), QMenuModel::Action);
+        QVERIFY(action.isValid());
+        QCOMPARE(action.type(), QVariant::String);
+        QCOMPARE(action.toString(), QString("Menu1Act"));
+
+        // Section (QObject)
+        QVariant vSection = m_model.data(m_model.index(2, 0), QMenuModel::LinkSection);
+        QVERIFY(vSection.isValid());
+        QMenuModel *section = qobject_cast<QMenuModel*>(vSection.value<QObject*>());
+        QVERIFY(section);
+        QCOMPARE(section->rowCount(), 2);
+
+        // SubMenu (QObject)
+        QVariant vSubMenu = m_model.data(m_model.index(3, 0), QMenuModel::LinkSubMenu);
+        QVERIFY(vSubMenu.isValid());
+        QMenuModel *submenu = qobject_cast<QMenuModel*>(vSubMenu.value<QObject*>());
+        QVERIFY(submenu);
+
+        // Wait for menu load (submenus are loaded async)
+        QTest::qWait(500);
+        QCOMPARE(submenu->rowCount(), 2);
+    }
+
+    void testExtraData()
+    {
+        // Make menu available
+        m_script.publishMenu();
+        m_script.run();
+
+        // start model
+        m_model.start();
+
+        // Wait for dbus sync
+        QTest::qWait(500);
+
+        QCOMPARE(m_model.status(), QDBusObject::Connected);
+        QCOMPARE(m_model.rowCount(), 4);
+
+        QVariant e = m_model.data(m_model.index(0, 0), QMenuModel::Extra);
+        QVERIFY(e.isValid());
+        QObject *extra = e.value<QObject*>();
+
+        // Boolean
+        QVariant v = extra->property("x-boolean");
+        QCOMPARE(v.type(), QVariant::Bool);
+        QCOMPARE(v.toBool(), true);
+
+        // Byte
+        v = extra->property("x-byte");
+        QCOMPARE(v.typeName(), "uchar");
+        QCOMPARE(v.value<uchar>(), (uchar)42);
+
+        // Int16
+        v = extra->property("x-int16");
+        QCOMPARE(v.typeName(), "short");
+        QCOMPARE(v.value<short>(), (short)-42);
+
+        // UInt16
+        v = extra->property("x-uint16");
+        QCOMPARE(v.typeName(), "ushort");
+        QCOMPARE(v.value<ushort>(), (ushort)42);
+
+        // Int32
+        v = extra->property("x-int32");
+        QCOMPARE(v.type(), QVariant::Int);
+        QCOMPARE(v.toInt(), -42);
+
+        // UInt32
+        v = extra->property("x-uint32");
+        QCOMPARE(v.type(), QVariant::UInt);
+        QCOMPARE(v.toUInt(), (uint) 42);
+
+        // Int64
+        v = extra->property("x-int64");
+        QCOMPARE(v.typeName(), "long");
+        QCOMPARE(v.value<long>(), (long) -42);
+
+        // UInt64
+        v = extra->property("x-uint64");
+        QCOMPARE(v.typeName(), "ulong");
+        QCOMPARE(v.value<ulong>(), (ulong) 42);
+
+        // Double
+        v = extra->property("x-double");
+        QCOMPARE(v.type(), QVariant::Double);
+        QCOMPARE(v.toDouble(), 42.42);
+
+        // String
+        v = extra->property("x-string");
+        QCOMPARE(v.type(), QVariant::String);
+        QCOMPARE(v.toString(), QString("42"));
+    }
+
+};
+
+QTEST_MAIN(ModelTest)
+
+#include "modeltest.moc"