def test_get_tests(self, sample_task, execution, expected): mock_api = MagicMock() task = Task(sample_task, mock_api) if execution is None: tests = task.get_tests() else: tests = task.get_tests(execution=execution) mock_api.tests_by_task.assert_called_once() kwargs = mock_api.tests_by_task.call_args[1] assert "execution" in kwargs and kwargs["execution"] == expected assert tests == mock_api.tests_by_task.return_value
def test_get_tests(self, sample_task): mock_api = MagicMock() task = Task(sample_task, mock_api) tests = task.get_tests() mock_api.tests_by_task.assert_called_once() assert tests == mock_api.tests_by_task.return_value