Add utility functions to test the value of DBus properties

1 files changed, 25 insertions, 0 deletions

diff --git a/qt4/tests/lib/test.h b/qt4/tests/lib/test.h
index 736213e72..77b3d34e8 100644
--- a/qt4/tests/lib/test.h
+++ b/qt4/tests/lib/test.h
@@ -3,6 +3,7 @@
 #include <QtTest>
 
 #include <TelepathyQt4/PendingOperation>
+#include <TelepathyQt4/PendingVariant>
 #include <TelepathyQt4/Constants>
 
 namespace Tp
@@ -23,10 +24,14 @@ public:
     QEventLoop *mLoop;
     void processDBusQueue(Tp::DBusProxy *proxy);
 
+protected:
+    template<typename T> bool waitForProperty(Tp::PendingVariant *pv, T *value);
+
 protected Q_SLOTS:
     void expectSuccessfulCall(QDBusPendingCallWatcher*);
     void expectSuccessfulCall(Tp::PendingOperation*);
     void expectFailure(Tp::PendingOperation*);
+    void expectSuccessfulProperty(Tp::PendingOperation *op);
     void onWatchdog();
 
     virtual void initTestCaseImpl();
@@ -34,4 +39,24 @@ protected Q_SLOTS:
 
     virtual void cleanupImpl();
     virtual void cleanupTestCaseImpl();
+
+private:
+    // The property retrieved by expectSuccessfulProperty()
+    QVariant mPropertyValue;
 };
+
+template<typename T>
+bool Test::waitForProperty(Tp::PendingVariant *pv, T *value)
+{
+    connect(pv,
+            SIGNAL(finished(Tp::PendingOperation*)),
+            SLOT(expectSuccessfulProperty(Tp::PendingOperation*)));
+    if (mLoop->exec() == 0) {
+        *value = qdbus_cast<T>(mPropertyValue);
+        return true;
+    }
+    else {
+        *value = T();
+        return false;
+    }
+}