def test_cancel(self, get_implementation): test_job = models.TestJob.objects.create(target=self.project, target_build=self.build, environment='myenv', backend=self.backend, submitted=True, job_id=123) impl = MagicMock() impl.cancel = MagicMock(return_value=True) get_implementation.return_value = impl test_job.cancel() impl.cancel.assert_called()
def test_cancel_not_submitted(self, get_implementation): test_job = models.TestJob.objects.create(target=self.project, target_build=self.build, environment='myenv', backend=self.backend, submitted=False) impl = MagicMock() impl.cancel = MagicMock(return_value=True) get_implementation.return_value = impl test_job.cancel() impl.cancel.assert_not_called() test_job.refresh_from_db() self.assertTrue(test_job.fetched) self.assertTrue(test_job.submitted) self.assertIsNotNone(test_job.failure)