95 lines
3 KiB
Python
95 lines
3 KiB
Python
#! /usr/bin/python
|
|
|
|
"""Tests for the 'scopes.storage' package."""
|
|
|
|
import config
|
|
|
|
from datetime import datetime
|
|
import transaction
|
|
import unittest
|
|
|
|
import scopes.storage.common
|
|
from scopes.storage.common import commit, Storage, getEngine, sessionFactory
|
|
from scopes.storage import proxy
|
|
from scopes.storage import folder, tracking
|
|
|
|
engine = getEngine(config.dbengine, config.dbname, config.dbuser, config.dbpassword)
|
|
scopes.storage.common.engine = engine
|
|
scopes.storage.common.Session = sessionFactory(engine)
|
|
|
|
storage = Storage(schema=config.dbschema)
|
|
|
|
|
|
class Test(unittest.TestCase):
|
|
"Basic tests for the cco.storage package."
|
|
|
|
def test_001_tracking(self):
|
|
storage.dropTable('tracks')
|
|
tracks = storage.create(tracking.Container)
|
|
|
|
tr01 = tracking.Track('t01', 'john')
|
|
tr01.update(dict(activity='testing'))
|
|
self.assertEqual(tr01.head, {'taskId': 't01', 'userName': 'john'})
|
|
self.assertEqual(tr01.taskId, 't01')
|
|
self.assertEqual(tr01.userName, 'john')
|
|
|
|
self.assertTrue(tracks.getTable() is not None)
|
|
|
|
trid01 = tracks.save(tr01)
|
|
self.assertTrue(trid01 > 0)
|
|
|
|
tr01a = tracks.get(trid01)
|
|
self.assertEqual(tr01a.head, tr01.head)
|
|
self.assertEqual(tr01a.trackId, trid01)
|
|
self.assertEqual(tr01a.data.get('activity'), 'testing')
|
|
|
|
tr01a.update(dict(text='Set up unit tests.'))
|
|
tr01a.timeStamp = None
|
|
self.assertTrue(tracks.save(tr01a) > 0)
|
|
|
|
tr01b = tracks.queryLast(taskId='t01')
|
|
self.assertEqual(tr01b.head, tr01.head)
|
|
self.assertNotEqual(tr01b.trackId, trid01)
|
|
self.assertEqual(tr01b.data.get('activity'), 'testing')
|
|
|
|
tr02 = tracking.Track('t02', 'jim', trackId=31, timeStamp=datetime(2023, 11, 30),
|
|
data=dict(activity='concept'))
|
|
trid02 = tracks.upsert(tr02)
|
|
self.assertEqual(trid02, 31)
|
|
self.assertEqual(tr02.uid, 'rec-31')
|
|
tr02.trackId = trid01
|
|
trid021 = tracks.upsert(tr02)
|
|
self.assertEqual(trid021, trid01)
|
|
self.assertEqual(tr02.uid, 'rec-' + str(trid01))
|
|
|
|
tr03 = storage.getItem('rec-31')
|
|
self.assertEqual(tr03.trackId, 31)
|
|
|
|
n = tracks.remove(31)
|
|
self.assertEqual(n, 1)
|
|
self.assertEqual(tracks.get(31), None)
|
|
|
|
commit(storage.session)
|
|
|
|
def test_002_folder(self):
|
|
storage.dropTable('folders')
|
|
root = folder.Root(storage)
|
|
self.assertEqual(list(root.keys()), [])
|
|
root['top'] = folder.Folder()
|
|
self.assertEqual(list(root.keys()), ['top'])
|
|
top = root['top']
|
|
top['child1'] = folder.Folder(data=dict(title='First Child'))
|
|
self.assertEqual(list(top.keys()), ['child1'])
|
|
ch1 = top['child1']
|
|
self.assertEqual(ch1.parent, top.rid)
|
|
assert list(top.keys()) == ['child1']
|
|
|
|
commit(storage.session)
|
|
|
|
def suite():
|
|
return unittest.TestSuite((
|
|
unittest.TestLoader().loadTestsFromTestCase(Test),
|
|
))
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(defaultTest='suite')
|