def test_fetch_with_no_tasks(self): """Ensure None is returned when GetTasks returns None. """ queue_service = Mock() queue_service.GetTasks.return_value = None result = _fetch_random_task_from_queue(queue_service, self.test_queue) self.assertIsNone(result) queue_service.GetTasks.assert_called_once_with(self.test_queue)
def test_fetch_with_no_tasks(self): """Ensure None is returned when GetTasks returns None. """ queue_service = Mock() queue_service.GetTasks.return_value = None result = _fetch_random_task_from_queue(queue_service, self.test_queue) self.assertIsNone(result) queue_service.GetTasks.assert_called_once_with(self.test_queue)
def test_fetch_with_tasks(self, choice): """Ensure None is returned when GetTasks returns None. """ queue_service = Mock() queue_service.GetTasks.return_value = ['a', 'b', 'c'] choice.return_value = 'b' result = _fetch_random_task_from_queue(queue_service, self.test_queue) self.assertEqual('b', result) queue_service.GetTasks.assert_called_once_with(self.test_queue)
def test_fetch_with_tasks(self, choice): """Ensure None is returned when GetTasks returns None. """ queue_service = Mock() queue_service.GetTasks.return_value = ['a', 'b', 'c'] choice.return_value = 'b' result = _fetch_random_task_from_queue(queue_service, self.test_queue) self.assertEqual('b', result) queue_service.GetTasks.assert_called_once_with(self.test_queue)