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)