Add generic persistence wrapper

tests/test_item.py

@@ -74,6 +74,5 @@ class TestStateItems(unittest.TestCase):
             self.states.state(item)
 
 
-
 if __name__ == '__main__':
     unittest.main()

tests/test_store.py

@@ -0,0 +1,84 @@
+# standard imports
+import unittest
+
+# local imports
+from shep.persist import PersistedState
+from shep.error import (
+        StateExists,
+        StateItemExists,
+        StateInvalid,
+        StateItemNotFound,
+        )
+
+class MockStore:
+
+    def __init__(self):
+        self.v = {}
+        self.for_state = 0
+
+
+    def add(self, k):
+        self.v[k] = True
+
+
+    def remove(self, k):
+        del self.v[k]
+
+
+class TestStateItems(unittest.TestCase):
+        
+    def setUp(self):
+        self.mockstore = MockStore()
+
+        def mockstore_factory(v):
+            self.mockstore.for_state = v
+            return self.mockstore
+
+        self.states = PersistedState(mockstore_factory, 4)
+        self.states.add('foo') 
+        self.states.add('bar') 
+        self.states.add('baz') 
+        self.states.alias('xyzzy', self.states.BAZ | self.states.BAR) 
+        self.states.alias('plugh', self.states.FOO | self.states.BAR) 
+
+
+    def test_persist_new(self):
+        item = b'foo'
+        self.states.put(item)
+        self.assertTrue(self.mockstore.v.get(item))
+
+
+    def test_persist_move(self):
+        item = b'foo'
+        self.states.put(item, self.states.FOO)
+        self.states.move(item, self.states.XYZZY) 
+        self.assertEqual(self.mockstore.for_state, self.states.name(self.states.XYZZY))
+
+
+    def test_persist_move(self):
+        item = b'foo'
+        self.states.put(item, self.states.FOO)
+        self.states.move(item, self.states.XYZZY) 
+        self.assertEqual(self.mockstore.for_state, self.states.name(self.states.XYZZY))
+        # TODO: cant check the add because remove happens after remove, need better mock
+        self.assertIsNone(self.mockstore.v.get(item)) 
+
+
+    def test_persist_purge(self):
+        item = b'foo'
+        self.states.put(item, self.states.FOO)
+        self.states.purge(item)
+        self.assertEqual(self.mockstore.for_state, self.states.name(self.states.FOO))
+        self.assertIsNone(self.mockstore.v.get(item))
+
+
+    def test_persist_move_new(self):
+        item = b'foo'
+        self.states.put(item)
+        self.states.move(item, self.states.BAZ)
+        self.assertEqual(self.mockstore.for_state, self.states.name(self.states.BAZ))
+        self.assertIsNone(self.mockstore.v.get(item))
+
+
+if __name__ == '__main__':
+    unittest.main()