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)