Add session replace unit test for sqlalchemy
The patch add replace test case for sqlalchemy Change-Id: I311d620c76e21f4099f0778adc734fc909071b09
This commit is contained in:
parent
df2993f850
commit
15ef8247f8
|
@ -566,6 +566,10 @@ def get_fake_session_2():
|
|||
return copy.deepcopy(fake_session_2)
|
||||
|
||||
|
||||
def get_fake_session_id():
|
||||
return uuid.uuid4().hex
|
||||
|
||||
|
||||
class FakeReqResp(object):
|
||||
def __init__(self, method='GET', body=''):
|
||||
self.method = method
|
||||
|
|
|
@ -31,6 +31,7 @@ class DbSessionTestCase(base.DbTestCase):
|
|||
self.fake_session_0.pop('session_id')
|
||||
self.fake_session_2 = common.get_fake_session_2()
|
||||
self.fake_session_2.pop('session_id')
|
||||
self.fake_session_id = common.get_fake_session_id()
|
||||
|
||||
def test_add_and_get_session(self):
|
||||
session_doc = copy.deepcopy(self.fake_session_0)
|
||||
|
@ -119,3 +120,49 @@ class DbSessionTestCase(base.DbTestCase):
|
|||
self.assertEqual(result.get('schedule').get('schedule_date'),
|
||||
self.fake_session_2.get('schedule').
|
||||
get('schedule_date'))
|
||||
|
||||
def test_add_and_replace_session(self):
|
||||
session_doc = copy.deepcopy(self.fake_session_0)
|
||||
session_id = self.dbapi.add_session(project_id=self.fake_session_0.
|
||||
get('project_id'),
|
||||
user_id=self.fake_session_0.
|
||||
get('user_id'),
|
||||
doc=session_doc)
|
||||
self.assertIsNotNone(session_id)
|
||||
|
||||
patch_doc = copy.deepcopy(self.fake_session_2)
|
||||
result = self.dbapi.replace_session(user_id=self.
|
||||
fake_session_2.get('user_id'),
|
||||
session_id=session_id,
|
||||
doc=patch_doc,
|
||||
project_id=self.fake_session_2.
|
||||
get('project_id'))
|
||||
|
||||
self.assertIsNotNone(result)
|
||||
self.assertEqual(result, session_id)
|
||||
|
||||
result = self.dbapi.get_session(project_id=self.fake_session_2.
|
||||
get('project_id'),
|
||||
user_id=self.fake_session_2.
|
||||
get('user_id'),
|
||||
session_id=session_id)
|
||||
|
||||
self.assertIsNotNone(result)
|
||||
self.assertEqual(result.get('description'),
|
||||
self.fake_session_2.get('description'))
|
||||
self.assertEqual(result.get('hold_off'),
|
||||
self.fake_session_2.get('hold_off'))
|
||||
self.assertEqual(result.get('schedule').get('schedule_date'),
|
||||
self.fake_session_2.get('schedule').
|
||||
get('schedule_date'))
|
||||
|
||||
patch_doc1 = copy.deepcopy(self.fake_session_0)
|
||||
result = self.dbapi.replace_session(user_id=self.
|
||||
fake_session_0.get('user_id'),
|
||||
session_id=self.fake_session_id,
|
||||
doc=patch_doc1,
|
||||
project_id=self.fake_session_2.
|
||||
get('project_id'))
|
||||
self.assertIsNotNone(result)
|
||||
|
||||
self.assertEqual(result, self.fake_session_id)
|
||||
|
|
Loading…
Reference in New Issue