openstack
/
freezer-api
Code Issues Proposed changes

Add session replace unit test for sqlalchemy 

The patch add replace test case for sqlalchemy

Change-Id: I311d620c76e21f4099f0778adc734fc909071b09
This commit is contained in:
gengchc2 2018-11-18 17:30:59 -08:00
parent df2993f850
commit 15ef8247f8
2 changed files with 51 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
freezer_api/tests/unit/common.py
View File

@ -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

47
freezer_api/tests/unit/sqlalchemy/test_session.py
View File

@ -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)