def test_run_without_task(self, fetch_task): """Ensure that we don't run a task if fetch returns None. """ fetch_task.return_value = None queue_service = Mock() result = _run_random_task_from_queue(queue_service, self.test_queue) self.assertFalse(result) fetch_task.assert_called_once_with(queue_service, self.test_queue)
def test_run_without_task(self, fetch_task): """Ensure that we don't run a task if fetch returns None. """ fetch_task.return_value = None queue_service = Mock() result = _run_random_task_from_queue(queue_service, self.test_queue) self.assertFalse(result) fetch_task.assert_called_once_with(queue_service, self.test_queue)
def test_run_with_task(self, fetch_task, execute_task): """Ensure that we handle a task run properly. """ task = {'name': self.test_task} fetch_task.return_value = task queue_service = Mock() queue_service.DeleteTask = Mock() result = _run_random_task_from_queue(queue_service, self.test_queue) self.assertTrue(result) fetch_task.assert_called_once_with(queue_service, self.test_queue) execute_task.assert_called_once_with(task) queue_service.DeleteTask.assert_called_once_with( self.test_queue, self.test_task)
def test_run_with_task(self, fetch_task, execute_task): """Ensure that we handle a task run properly. """ task = {'name': self.test_task} fetch_task.return_value = task queue_service = Mock() queue_service.DeleteTask = Mock() result = _run_random_task_from_queue(queue_service, self.test_queue) self.assertTrue(result) fetch_task.assert_called_once_with(queue_service, self.test_queue) execute_task.assert_called_once_with(task) queue_service.DeleteTask.assert_called_once_with( self.test_queue, self.test_task)