Add job replace unit test for sqlalchemy

The patch add replace test case for sqlalchemy

Change-Id: Ic0e0b6466e1dddc5be3b1e339572467e71df9a91
This commit is contained in:
gengchc2 2018-11-18 17:18:01 -08:00
parent 15ef8247f8
commit 8f4a306f3b
2 changed files with 49 additions and 0 deletions

View File

@ -297,6 +297,10 @@ def get_fake_job_2():
return copy.deepcopy(fake_job_2)
def get_fake_job_id():
return uuid.uuid4().hex
fake_job_0_elasticsearch_found = {
"_id": "e7181e5e-2c75-43f8-92c0-c037ae5f11e4",
"_index": "freezer",

View File

@ -32,6 +32,7 @@ class DbJobTestCase(base.DbTestCase):
self.fake_job_2 = common.get_fake_job_2()
self.fake_job_2.pop('job_id')
self.fake_project_id = self.fake_job_0.get('project_id')
self.fake_job_id = common.get_fake_job_id()
def test_add_and_get_job(self):
job_doc = copy.deepcopy(self.fake_job_0)
@ -111,3 +112,47 @@ class DbJobTestCase(base.DbTestCase):
get('schedule_interval'))
self.assertEqual(result.get('job_actions'),
self.fake_job_2.get('job_actions'))
def test_add_and_replace_job(self):
job_doc = copy.deepcopy(self.fake_job_0)
job_id = self.dbapi.add_job(user_id=self.fake_job_0.get('user_id'),
doc=job_doc,
project_id=self.fake_project_id)
self.assertIsNotNone(job_id)
patch_doc = copy.deepcopy(self.fake_job_2)
result = self.dbapi.replace_job(user_id=self.
fake_job_2.get('user_id'),
job_id=job_id,
doc=patch_doc,
project_id=self.fake_project_id)
self.assertIsNotNone(result)
self.assertEqual(result, job_id)
result = self.dbapi.get_job(project_id=self.fake_project_id,
user_id=self.fake_job_2.get('user_id'),
job_id=job_id)
self.assertIsNotNone(result)
self.assertEqual(result.get('client_id'),
self.fake_job_2.get('client_id'))
self.assertEqual(result.get('description'),
self.fake_job_2.get('description'))
self.assertEqual(result.get('job_schedule').
get('schedule_interval'),
self.fake_job_2.get('job_schedule').
get('schedule_interval'))
self.assertEqual(result.get('job_actions'),
self.fake_job_2.get('job_actions'))
patch_doc1 = copy.deepcopy(self.fake_job_0)
result = self.dbapi.replace_job(user_id=self.
fake_job_2.get('user_id'),
job_id=self.fake_job_id,
doc=patch_doc1,
project_id=self.fake_project_id)
self.assertIsNotNone(result)
self.assertEqual(result, self.fake_job_id)