From 615c439ad45eee9eb709200e3727d1d6eacaecfe Mon Sep 17 00:00:00 2001 From: Elena ``of Valhalla'' Grandi Date: Sun, 14 Apr 2019 17:46:11 +0200 Subject: Start working on storage backends --- tests/test_memory_store.py | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 tests/test_memory_store.py (limited to 'tests/test_memory_store.py') diff --git a/tests/test_memory_store.py b/tests/test_memory_store.py new file mode 100644 index 0000000..8c19954 --- /dev/null +++ b/tests/test_memory_store.py @@ -0,0 +1,32 @@ +import unittest + +from pyapd import objects +from pyapd.stores import memory, exceptions + + +class TestMemoryStore(unittest.TestCase): + def setUp(self): + self.store = memory.Store() + self.oid = 'https://test/object/12345' + self.obj = objects.Object(id=self.oid) + self.store.add(self.obj) + + def test_add_object(self): + oid = 'https://test/object/12345' + obj = objects.Object(id=oid) + self.store.add(obj) + self.assertIn('object', self.store.objects) + self.assertIn(oid, self.store.objects['object']) + + def test_get_object(self): + res = self.store.get('object', self.oid) + self.assertEqual(res, self.obj) + + def test_get_object_not_existing(self): + oid = 'https://test/object/does_not_exist' + with self.assertRaises(exceptions.DoesNotExist): + self.store.get('object', oid) + + def test_get_object_wrong_type(self): + with self.assertRaises(exceptions.UnknownObjectType): + self.store.get('no_such_type', self.oid) -- cgit v1.2.3