fix: adjusting tests to new run tests args

Signed-off-by: Cortland Goffena <30168413+cmgoffena13@users.noreply.github.com>

Author: cmgoffena13

tests/core/test_context.py

@@ -4,7 +4,7 @@
 import re
 from datetime import date, timedelta, datetime
 from tempfile import TemporaryDirectory
-from unittest.mock import PropertyMock, call, patch
+from unittest.mock import ANY, PropertyMock, call, patch
 
 import time_machine
 import pytest
@@ -3607,7 +3607,7 @@ def test_plan_explain_skips_tests(sushi_context: Context, mocker: MockerFixture)
     sushi_context.console = TerminalConsole()
     spy = mocker.spy(sushi_context, "_run_plan_tests")
     sushi_context.plan(environment="dev", explain=True, no_prompts=True, include_unmodified=True)
-    spy.assert_called_once_with(skip_tests=True)
+    spy.assert_called_once_with(skip_tests=True, all_tests=False, model_names=ANY)
 
 
 def test_dev_environment_virtual_update_with_environment_statements(tmp_path: Path) -> None:

tests/integrations/github/cicd/test_github_controller.py

@@ -3,7 +3,7 @@
 import os
 import pathlib
 from unittest import mock
-from unittest.mock import PropertyMock, call
+from unittest.mock import ANY, PropertyMock, call
 
 import pytest
 import time_machine
@@ -256,7 +256,9 @@ def test_pr_plan(github_client, make_controller):
     assert controller.pr_plan.skip_backfill
     assert not controller.pr_plan.no_gaps
     assert not controller._context.apply.called
-    assert controller._context._run_plan_tests.call_args == call(skip_tests=True)
+    assert controller._context._run_plan_tests.call_args == call(
+        skip_tests=True, all_tests=False, model_names=ANY
+    )
     assert (
         controller._pr_plan_builder._categorizer_config
         == controller._context.auto_categorize_changes
@@ -278,7 +280,9 @@ def test_pr_plan_auto_categorization(github_client, make_controller):
     assert controller.pr_plan.skip_backfill
     assert not controller.pr_plan.no_gaps
     assert not controller._context.apply.called
-    assert controller._context._run_plan_tests.call_args == call(skip_tests=True)
+    assert controller._context._run_plan_tests.call_args == call(
+        skip_tests=True, all_tests=False, model_names=ANY
+    )
     assert controller._pr_plan_builder._categorizer_config == custom_categorizer_config
     assert controller.pr_plan.start == default_start_absolute
     assert not controller.pr_plan.start_override_per_model
@@ -365,7 +369,9 @@ def test_prod_plan(github_client, make_controller):
     assert not controller.prod_plan.skip_backfill
     assert controller.prod_plan.no_gaps
     assert not controller._context.apply.called
-    assert controller._context._run_plan_tests.call_args == call(skip_tests=True)
+    assert controller._context._run_plan_tests.call_args == call(
+        skip_tests=True, all_tests=False, model_names=ANY
+    )
     assert (
         controller._prod_plan_builder._categorizer_config
         == controller._context.auto_categorize_changes
@@ -387,7 +393,9 @@ def test_prod_plan_auto_categorization(github_client, make_controller):
     assert not controller.prod_plan.skip_backfill
     assert controller.prod_plan.no_gaps
     assert not controller._context.apply.called
-    assert controller._context._run_plan_tests.call_args == call(skip_tests=True)
+    assert controller._context._run_plan_tests.call_args == call(
+        skip_tests=True, all_tests=False, model_names=ANY
+    )
     assert controller._prod_plan_builder._categorizer_config == custom_categorizer_config
     # default PR start should be ignored for prod plans
     assert controller.prod_plan.start != default_pr_start
@@ -404,7 +412,9 @@ def test_prod_plan_with_gaps(github_client, make_controller):
     assert controller._prod_plan_with_gaps_builder._auto_categorization_enabled
     assert not controller.prod_plan_with_gaps.no_gaps
     assert not controller._context.apply.called
-    assert controller._context._run_plan_tests.call_args == call(skip_tests=True)
+    assert controller._context._run_plan_tests.call_args == call(
+        skip_tests=True, all_tests=False, model_names=ANY
+    )
 
 
 def test_run_tests(github_client, make_controller):