def test_revoke_pipeline__process_is_none(self):
pipeline_model = MockPipelineModel(process=None)
with patch(PIPELINE_PIPELINE_MODEL_GET, MagicMock(return_value=pipeline_model)):
act_result = api.revoke_pipeline(self.pipeline_id)
self.assertFalse(act_result.result)
def test_revoke_pipeline__transit_success(self):
pipeline_model = MockPipelineModel()
with patch(PIPELINE_PIPELINE_MODEL_GET,
MagicMock(return_value=pipeline_model)):
act_result = api.revoke_pipeline(self.pipeline_id)
self.assertTrue(act_result.result)
pipeline_model.process.revoke_subprocess.assert_called_once()
pipeline_model.process.destroy_all.assert_called_once()
def test_revoke_pipeline__transit_success(self):
pipeline_model = MockPipelineModel()
with patch(PIPELINE_PIPELINE_MODEL_GET, MagicMock(return_value=pipeline_model)):
with mock.patch(PIPELINE_PROCESS_SELECT_FOR_UPDATE,
mock.MagicMock(return_value=MockQuerySet(get_return=pipeline_model.process))):
act_result = api.revoke_pipeline(self.pipeline_id)
self.assertTrue(act_result.result)
pipeline_model.process.revoke_subprocess.assert_called_once()
pipeline_model.process.destroy_all.assert_called_once()
def revoke_pipeline(pipeline_id):
return api.revoke_pipeline(pipeline_id)
def test_revoke_pipeline__transit_fail(self):
act_result = api.revoke_pipeline(self.pipeline_id)
self.assertFalse(act_result.result)
