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)
