def test_run_queue(self, _execute_task): """When run() is called, ensure tasks are run, and the queue is flushed to remove run tasks. Also, ensure True is returned since messages were processed. """ from furious.test_stubs.appengine.queues import run_queue queue_service = Mock() queue_service.GetTasks.return_value = ['task1', 'task2', 'task3'] num_processed = run_queue(queue_service, 'default') # Expect _execute_task() to be called for each task expected_call_args_list = [call('task1', None, None), call('task2', None, None), call('task3', None, None)] self.assertEquals(_execute_task.call_args_list, expected_call_args_list) # Make sure FlushQueue was called once to clear the queue after # tasks were processed self.assertEqual(1, queue_service.FlushQueue.call_count) # We should have processed tasks, so verify the num processed. self.assertEqual(3, num_processed)
def test_run_queue(self, _execute_task): """When run() is called, ensure tasks are run, and the queue is flushed to remove run tasks. Also, ensure True is returned since messages were processed. """ from furious.test_stubs.appengine.queues import run_queue queue_service = Mock() queue_service.GetTasks.return_value = ['task1', 'task2', 'task3'] num_processed = run_queue(queue_service, 'default') # Expect _execute_task() to be called for each task expected_call_args_list = [call('task1'), call('task2'), call('task3')] self.assertEquals(_execute_task.call_args_list, expected_call_args_list) # Make sure FlushQueue was called once to clear the queue after # tasks were processed self.assertEqual(1, queue_service.FlushQueue.call_count) # We should have processed tasks, so verify the num processed. self.assertEqual(3, num_processed)
def test_run_queue_no_tasks(self, _execute_task): """When run() is called and there are no tasks in the queue, ensure _execute_task is not called. Ensure False is returned since no messages were processed. """ from furious.test_stubs.appengine.queues import run_queue queue_service = Mock() queue_service.GetTasks.return_value = [] num_processed = run_queue(queue_service, 'default') # Expect _execute_task() to not be called since there are no tasks self.assertFalse(_execute_task.called) # We should not have processed any tasks, so verify 0 processed. self.assertEqual(0, num_processed)