Split initialization to 2 phases. initialize() and connectToServer()

1 files changed, 31 insertions, 3 deletions

diff --git a/tests/test-resource.cpp b/tests/test-resource.cpp
index 3dce276..eb2b59d 100644
--- a/tests/test-resource.cpp
+++ b/tests/test-resource.cpp
@@ -7,13 +7,14 @@ TestResource::~TestResource()
 {
 }
 
-void TestResource::initTestCase()
+void TestResource::init()
 {
     resource = new Resource(MediaClass, RP_FLAGS_AUDIO|RP_FLAGS_VIDEO, this);
-    resourceLibrary = new MockResourceLibrary(resource, false);
+    resourceLibrary = new MockResourceLibrary(resource, false, false);
     QVERIFY(resource != NULL);
     QVERIFY(resource->resourceClass == MediaClass);
     QVERIFY(resource->flags == (RP_FLAGS_AUDIO|RP_FLAGS_VIDEO));
+    qDebug("INIT!");
 }
 
 void TestResource::testInitializeSucceeds()
@@ -34,7 +35,34 @@ void TestResource::testInitializeFails()
 
     QEXPECT_FAIL("", "Expecting resourceLibrary->initialize() to fail", Continue);
     QVERIFY(initializeSucceeded);
-    delete resource;
+}
+
+void TestResource::testConnectToServerSucceeds()
+{
+    bool initializeSucceeded = resource->initialize(resourceLibrary);
+
+    QVERIFY(resource->resourceLibrary == resourceLibrary);
+    QVERIFY(initializeSucceeded);
+
+    bool connectToServerSucceeded = resource->connectToServer();
+    QVERIFY(connectToServerSucceeded);
+}
+
+void TestResource::testConnectToServerFails()
+{
+    MockResourceLibrary *mockResourceLibrary =
+	static_cast<MockResourceLibrary *>(resourceLibrary);
+
+    bool initializeSucceeded = resource->initialize(resourceLibrary);
+
+    QVERIFY(resource->resourceLibrary == resourceLibrary);
+    QVERIFY(initializeSucceeded);
+
+    mockResourceLibrary->makeServerConnectFail();
+    bool connectToServerSucceeded = resource->connectToServer();
+
+    QEXPECT_FAIL("", "Expecting resourceLibrary->connectToServer() to fail", Continue);
+    QVERIFY(connectToServerSucceeded);
 }
 
 QTEST_MAIN(TestResource)