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)
