aboutsummaryrefslogtreecommitdiff
path: root/tests/test_memory_store.py
blob: 899a77605a035203f987df5d6d10014757847531 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
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(oid, self.store.objects)

    def test_get_object(self):
        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(oid)