Implement context storage handles.

[shibboleth/cpp-xmltooling.git] / xmltoolingtest / MemoryStorageServiceTest.h

diff --git a/xmltoolingtest/MemoryStorageServiceTest.h b/xmltoolingtest/MemoryStorageServiceTest.h
index 1d91950..99e2e0e 100644 (file)
--- a/xmltoolingtest/MemoryStorageServiceTest.h
+++ b/xmltoolingtest/MemoryStorageServiceTest.h
@@ -30,18 +30,19 @@ public:
         auto_ptr<StorageService> storage(\r
             XMLToolingConfig::getConfig().StorageServiceManager.newPlugin(MEMORY_STORAGE_SERVICE,NULL)\r
             );\r
+        auto_ptr<StorageService::StorageHandle> handle(storage->createHandle());\r
 \r
         string data;\r
-        TSM_ASSERT("Record found in storage.", !storage->readString("foo1", data));\r
-        storage->createString("foo1", "bar1", time(NULL) - 300);\r
-        storage->createString("foo2", "bar2", time(NULL));\r
-        TSM_ASSERT("Record not found in storage.", storage->readString("foo1", data));\r
+        TSM_ASSERT("Record found in storage.", !storage->readString(handle.get(), "foo1", data));\r
+        storage->createString(handle.get(), "foo1", "bar1", time(NULL) - 300);\r
+        storage->createString(handle.get(), "foo2", "bar2", time(NULL));\r
+        TSM_ASSERT("Record not found in storage.", storage->readString(handle.get(), "foo1", data));\r
         TSM_ASSERT_EQUALS("Record value doesn't match.", data, "bar1");\r
-        TSM_ASSERT("Update failed.", storage->updateString("foo2", "bar1"));\r
-        TSM_ASSERT("Record not found in storage.", storage->readString("foo2", data));\r
+        TSM_ASSERT("Update failed.", storage->updateString(handle.get(), "foo2", "bar1"));\r
+        TSM_ASSERT("Record not found in storage.", storage->readString(handle.get(), "foo2", data));\r
         TSM_ASSERT_EQUALS("Record value doesn't match.", data, "bar1");\r
-        TSM_ASSERT("Delete failed.", storage->deleteString("foo2"));\r
-        storage->reap();\r
+        TSM_ASSERT("Delete failed.", storage->deleteString(handle.get(), "foo2"));\r
+        storage->reap(handle.get());\r
         Thread::sleep(1);\r
     }\r
 };\r