1 files changed, 22 insertions, 4 deletions
diff --git a/tests/mock-resource-library.cpp b/tests/mock-resource-library.cpp
index 49c29e4..565e745 100644
--- a/tests/mock-resource-library.cpp
+++ b/tests/mock-resource-library.cpp
@@ -1,11 +1,14 @@
 #include "mock-resource-library.h"
 
-MockResourceLibrary::MockResourceLibrary(QObject *parent,
+MockResourceLibrary::MockResourceLibrary(Resource *resource,
 					 bool makeInitializationFail,
-					 bool makeConnectingToServerFail)
-    : QObject(parent), initializeFails(makeInitializationFail),
-      connectToServerFails(makeConnectingToServerFail)
+					 bool makeConnectingToServerFail,
+					 bool makeReserveFail)
+    : QObject(resource), initializeFails(makeInitializationFail),
+      connectToServerFails(makeConnectingToServerFail),
+      reserveFails(makeReserveFail)
 {
+    this->resource = resource;
 }
 
 MockResourceLibrary::~MockResourceLibrary()
@@ -37,3 +40,18 @@ bool MockResourceLibrary::connectToServer()
     else
 	return false;
 }
+
+bool MockResourceLibrary::reserve()
+{
+    if(!reserveFails) {
+	resource->handleStateChange(OwnedState);
+    }
+    else {
+	resource->handleStateChange(NotOwnedState);
+    }
+}
+
+void MockResourceLibrary::makeReserveFail()
+{
+    reserveFails = true;
+}