openstack/deb-sahara
Code Issues Proposed changes

Merge "Add unit test to cover cancel job operation in oozie engine"

Browse Source
This commit is contained in:
Jenkins 2015-11-10 12:00:24 +00:00 committed by Gerrit Code Review
commit 48ab3224f6
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
sahara/tests/unit/service/edp/oozie/test_oozie.py
View File

@ -119,6 +119,23 @@ class TestOozieEngine(base.SaharaTestCase):
self.assertEqual(expected_external_hdfs_urls, external_hdfs_urls)
@mock.patch('sahara.service.edp.oozie.oozie.OozieClient.get_job_status')
@mock.patch('sahara.service.edp.oozie.oozie.OozieClient.kill_job')
def test_cancel_job(self, kill_get, status_get):
status_get.return_value = {}
oje = FakeOozieJobEngine(u.create_cluster())
_, job_exec = u.create_job_exec(edp.JOB_TYPE_PIG)
# test cancel job without engine_job_id
job_exec.engine_job_id = None
oje.cancel_job(job_exec)
self.assertEqual(0, kill_get.call_count)
# test cancel job with engine_job_id
job_exec.engine_job_id = 123
oje.cancel_job(job_exec)
self.assertEqual(1, kill_get.call_count)
class FakeOozieJobEngine(oe.OozieJobEngine):
def get_hdfs_user(self):