blob: 7c282b7120f8bc57769e032bce5f9479abdc51ec (
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
30
31
32
33
|
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.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)
|
