diff options
Diffstat (limited to 'tests/test_memory_store.py')
| -rw-r--r-- | tests/test_memory_store.py | 11 |
1 files changed, 3 insertions, 8 deletions
diff --git a/tests/test_memory_store.py b/tests/test_memory_store.py index 8c19954..899a776 100644 --- a/tests/test_memory_store.py +++ b/tests/test_memory_store.py @@ -15,18 +15,13 @@ class TestMemoryStore(unittest.TestCase): 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']) + self.assertIn(oid, self.store.objects) def test_get_object(self): - res = self.store.get('object', self.oid) + res = self.store.get(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) + self.store.get(oid) |