Add session update unit test for sqlalchemy

Change-Id: I8e664fd4487d7dfe15627a513e300006182a50fd
This commit is contained in:
gengchc2 2018-11-14 00:56:34 -08:00 committed by gecong
parent d9007cf473
commit 08e22adfbb
2 changed files with 72 additions and 0 deletions

View File

@ -498,6 +498,39 @@ fake_session_1 = {
"project_id": "tecs"
}
fake_session_2 = {
"session_id": 'turistidellademocrazia',
"session_tag": 5,
"description": 'This is a session about test',
"hold_off": 100,
"schedule": {
"time_created": 1234,
"time_started": 1234,
"time_ended": 0,
"status": "stop",
"schedule_date": "2018-11-14T16:20:00"
},
"jobs": {
'venerescollataincorpodalolita': {
"client_id": 'bruco',
"status": 'running',
"start_time": 12344321,
},
'job_id_2': {
"client_id": "cocktail",
"status": 'completed',
"result": 'success',
"start_time": 123321,
"end_time": 123325,
}
},
"time_start": 123412344,
"time_end": 432234432,
"status": "running",
"user_id": "califfo",
"project_id": "tecs"
}
def get_fake_session_0():
return copy.deepcopy(fake_session_0)
@ -507,6 +540,10 @@ def get_fake_session_1():
return copy.deepcopy(fake_session_1)
def get_fake_session_2():
return copy.deepcopy(fake_session_2)
class FakeReqResp(object):
def __init__(self, method='GET', body=''):
self.method = method

View File

@ -29,6 +29,8 @@ class DbSessionTestCase(base.DbTestCase):
super(DbSessionTestCase, self).setUp()
self.fake_session_0 = common.get_fake_session_0()
self.fake_session_0.pop('session_id')
self.fake_session_2 = common.get_fake_session_2()
self.fake_session_2.pop('session_id')
def test_add_and_get_session(self):
session_doc = copy.deepcopy(self.fake_session_0)
@ -84,3 +86,36 @@ class DbSessionTestCase(base.DbTestCase):
get('user_id'),
session_id=session_id)
self.assertEqual(len(result), 0)
def test_add_and_update_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.update_session(user_id=self.fake_session_0.
get('user_id'),
session_id=session_id,
patch_doc=patch_doc,
project_id=self.fake_session_0.
get('project_id'))
self.assertIsNotNone(result)
self.assertEqual(result, 0)
result = self.dbapi.get_session(project_id=self.fake_session_0.
get('project_id'),
user_id=self.fake_session_0.
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'))