Exemplo n.º 1
0
    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)
Exemplo n.º 2
0
    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)
Exemplo n.º 3
0
    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)
Exemplo n.º 4
0
    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)