2 files changed, 84 insertions, 0 deletions

diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
index 2706d5b..3bbbe37 100644
--- a/tests/CMakeLists.txt
+++ b/tests/CMakeLists.txt
@@ -55,6 +55,7 @@ auto_tests (
     notificationtest
     entitystoretest
     upgradetest
+    resourcecontroltest
 )
 
 if (WIN32)
diff --git a/tests/resourcecontroltest.cpp b/tests/resourcecontroltest.cpp
new file mode 100644
index 0000000..e66f444
--- /dev/null
+++ b/tests/resourcecontroltest.cpp
@@ -0,0 +1,83 @@
+#include <QtTest>
+
+#include "dummyresource/resourcefactory.h"
+#include "resourcecontrol.h"
+#include "store.h"
+#include "testutils.h"
+#include "test.h"
+#include "resourceconfig.h"
+
+/**
+ * Test starting and stopping of resources.
+ */
+class ResourceControlTest : public QObject
+{
+    Q_OBJECT
+
+    KAsync::Job<bool> socketIsAvailable(const QByteArray &identifier)
+    {
+        return Sink::ResourceAccess::connectToServer(identifier)
+            .then<void, QSharedPointer<QLocalSocket>>(
+                [&](const KAsync::Error &error, QSharedPointer<QLocalSocket> socket) {
+                    if (error) {
+                        return KAsync::value(false);
+                    }
+                    socket->close();
+                    return KAsync::value(true);
+                });
+
+    }
+
+    bool blockingSocketIsAvailable(const QByteArray &identifier)
+    {
+        auto job = socketIsAvailable(identifier);
+        auto future = job.exec();
+        future.waitForFinished();
+        return future.value();
+    }
+
+private slots:
+
+    void initTestCase()
+    {
+        Sink::Test::initTest();
+        auto factory = Sink::ResourceFactory::load("sink.dummy");
+        QVERIFY(factory);
+        ::DummyResource::removeFromDisk("sink.dummy.instance1");
+        ResourceConfig::addResource("sink.dummy.instance1", "sink.dummy");
+        ::DummyResource::removeFromDisk("sink.dummy.instance2");
+        ResourceConfig::addResource("sink.dummy.instance2", "sink.dummy");
+    }
+
+    void testResourceStart()
+    {
+        VERIFYEXEC(Sink::ResourceControl::start("sink.dummy.instance1"));
+        QVERIFY(blockingSocketIsAvailable("sink.dummy.instance1"));
+    }
+
+    void testResourceShutdown()
+    {
+        QVERIFY(!blockingSocketIsAvailable("sink.dummy.instance2"));
+        VERIFYEXEC(Sink::ResourceControl::start("sink.dummy.instance2"));
+        QVERIFY(blockingSocketIsAvailable("sink.dummy.instance2"));
+        VERIFYEXEC(Sink::ResourceControl::shutdown("sink.dummy.instance2"));
+        QVERIFY(!blockingSocketIsAvailable("sink.dummy.instance2"));
+    }
+
+    //This will produce a race where the synchronize command starts the resource,
+    //the shutdown command doesn't shutdown because it doesn't realize that the resource is up,
+    //and the resource ends up getting started, but doing nothing.
+    void testResourceShutdownAfterStartByCommand()
+    {
+        QVERIFY(!blockingSocketIsAvailable("sink.dummy.instance2"));
+        auto future = Sink::Store::synchronize(Sink::SyncScope{}.resourceFilter("sink.dummy.instance2")).exec();
+
+        VERIFYEXEC(Sink::ResourceControl::shutdown("sink.dummy.instance2"));
+
+        QVERIFY(!blockingSocketIsAvailable("sink.dummy.instance2"));
+    }
+
+};
+
+QTEST_MAIN(ResourceControlTest)
+#include "resourcecontroltest.moc"