def test_init_job_update(self, mock_new_build_task): test_job = contract_job.ContractJob({"task_type": "update"}) mock_new_build_task.assert_called_once() self.assertIsNotNone(test_job.update_model)
def test_init_job_delete(self, mock_login, mock_new_build_task): test_job = contract_job.ContractJob({"task_type": "delete"}) mock_new_build_task.assert_called_once() self.assertIsNone(test_job.update_model) self.assertEqual(test_job.model.task_type, "delete")
def setUp(self, mock_login, mock_new_build_task): self.test_job = contract_job.ContractJob({"task_type": "create"}) importlib.reload(contract_job) contract_job.EVENT = '{"task_type": "create"}'