1 files changed, 8 insertions, 1 deletions
diff --git a/ick2/workapi_tests.py b/ick2/workapi_tests.py
index 72bcaf1..8652110 100644
--- a/ick2/workapi_tests.py
+++ b/ick2/workapi_tests.py
@@ -15,6 +15,8 @@
 
 import copy
 import os
+import shutil
+import tempfile
 import unittest
 
 
@@ -24,9 +26,14 @@ import ick2
 class WorkAPITests(unittest.TestCase):
 
     def setUp(self):
-        self.state = ick2.MemoryStore()
+        self.tempdir = tempfile.mkdtemp()
+        self.statedir = os.path.join(self.tempdir, 'state/dir')
+        self.state = ick2.MemoryPersistentState()
         self.claims = None
 
+    def tearDown(self):
+        shutil.rmtree(self.tempdir)
+
     def create_project_api(self):
         pipeline = {
             'pipeline': 'build',