def test_remove_files_when_cancelling_multiple_jobs(self, mock_rq_job, mock_rmtree):
_, mock_job = mock_rq_job
mock_job.is_queued.return_value = True
files_path = "something"
mock_job.kwargs = {"files_path": files_path}
cli.cancel_test("something", [1, 2])
assert mock_rmtree.call_count == 2
def test_remove_files_when_cancelling(self, mock_rq_job, mock_rmtree):
_, mock_job = mock_rq_job
mock_job.is_queued.return_value = True
files_path = "something"
mock_job.kwargs = {"files_path": files_path}
cli.cancel_test("something", [1])
mock_rmtree.assert_called_once_with(files_path, onerror=ANY)
def test_cancel_multiple_jobs(self, mock_rq_job):
_, mock_job = mock_rq_job
mock_job.is_queued.return_value = True
mock_job.kwargs = {"files_path": None}
cli.cancel_test("something", [1, 2])
assert mock_job.cancel.call_count == 2
def test_cancel_job(self, mock_rq_job):
_, mock_job = mock_rq_job
mock_job.is_queued.return_value = True
mock_job.kwargs = {"files_path": None}
cli.cancel_test("something", [1])
mock_job.cancel.assert_called_once()
def test_do_nothing_if_job_not_enqueued(self, mock_rq_job):
_, mock_job = mock_rq_job
mock_job.is_queued.return_value = False
cli.cancel_test("something", [1])
mock_job.cancel.assert_not_called()
def test_do_nothing_if_job_does_not_exist(self, mock_rq_job):
job_class, mock_job = mock_rq_job
job_class.fetch.side_effect = NoSuchJobError
cli.cancel_test("something", [1])
mock_job.cancel.assert_not_called()