aboutsummaryrefslogtreecommitdiff
path: root/tests/test_memory_store.py
blob: e6b7f1a7f62c5f1d5d322f2b34be8fb243f0cbf4 (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.APObject(ap_id=self.oid)
        self.store.add(self.obj)

    def test_add_object(self):
        oid = 'https://test/object/12345'
        obj = objects.APObject(ap_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)