blob: 7de76582c5479bd4cbbcdc89ea3200b8480076ec (
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
28
29
|
import unittest
from pyapd import objects
from pyapd.stores import sqlite, exceptions
class TestSqliteStore(unittest.TestCase):
def setUp(self):
self.store = sqlite.Store(file=':memory:')
self.store.createdb()
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('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)
|