1 files changed, 74 insertions, 0 deletions
diff --git a/tests/glibext/storage.py b/tests/glibext/storage.py
new file mode 100644
index 0000000..b60377a
--- /dev/null
+++ b/tests/glibext/storage.py
@@ -0,0 +1,74 @@
+
+from chrysacase import ChrysalideTestCase
+from pychrysalide.glibext import ObjectStorage, SerializableObject
+import gi
+import os
+import tempfile
+
+
+class TestObjectStorage(ChrysalideTestCase):
+    """TestCase for analysis.storage."""
+
+    @classmethod
+    def setUpClass(cls):
+
+        super(TestObjectStorage, cls).setUpClass()
+
+        _, cls._tmp_filename = tempfile.mkstemp()
+
+        cls.log('Using temporary filename "%s"' % cls._tmp_filename)
+
+
+    @classmethod
+    def tearDownClass(cls):
+
+        super(TestObjectStorage, cls).tearDownClass()
+
+        cls.log('Delete filename "%s"' % cls._tmp_filename)
+
+        os.unlink(cls._tmp_filename)
+
+
+    def testGenericStorage(self):
+        """Store and load basic objects."""
+
+
+        class SimpleObject(gi._gi.GObject, SerializableObject):
+
+            def __init__(self, b=None):
+                super().__init__()
+                self._b = b
+
+            def _load(self, storage, fd):
+                assert(self._b is None)
+                self._b = os.read(fd, 1)[0]
+                return True
+
+            def _store(self, storage, fd):
+                os.write(fd, bytes([ self._b ]))
+                return True
+
+            def __eq__(self, other):
+                return self._b == other._b
+
+
+        # Store
+
+        storage = ObjectStorage('TestStorage', 0, 'my-storage-hash')
+        self.assertIsNotNone(storage)
+
+        so = SimpleObject(0x23)
+
+        pos = storage.store_object('simple', so)
+        self.assertIsNotNone(pos)
+
+        status = storage.store(self._tmp_filename)
+        self.assertTrue(status)
+
+        # Reload
+
+        storage2 = ObjectStorage.load(self._tmp_filename)
+
+        so2 = storage2.load_object('simple', pos)
+
+        self.assertEqual(so, so2)