test: fix tests by including search_query where needed

Signed-off-by: Zack Koppert <[email protected]>

Author: zkoppert

test_env.py

@@ -39,6 +39,7 @@ def setUp(self):
             "TYPE",
             "UPDATE_EXISTING",
             "REPO_SPECIFIC_EXEMPTIONS",
+            "REPOSITORY_SEARCH_QUERY",
             "SCHEDULE",
             "SCHEDULE_DAY",
             "LABELS",
@@ -68,6 +69,7 @@ def test_get_env_vars_with_org(self):
         expected_result = (
             "my_organization",
             [],
+            "",  # search_query
             None,
             None,
             b"",
@@ -120,6 +122,7 @@ def test_get_env_vars_with_org_and_repo_specific_exemptions(self):
         expected_result = (
             "my_organization",
             [],
+            "",  # search_query
             None,
             None,
             b"",
@@ -232,6 +235,7 @@ def test_get_env_vars_with_repos(self):
         expected_result = (
             None,
             ["org/repo1", "org2/repo2"],
+            "",  # search_query
             None,
             None,
             b"",
@@ -289,6 +293,7 @@ def test_get_env_vars_with_team(self):
         expected_result = (
             "my_organization",
             [],
+            "",  # search_query
             None,
             None,
             b"",
@@ -364,6 +369,7 @@ def test_get_env_vars_optional_values(self):
         expected_result = (
             "my_organization",
             [],
+            "",  # search_query
             None,
             None,
             b"",
@@ -410,6 +416,7 @@ def test_get_env_vars_with_update_existing(self):
         expected_result = (
             "my_organization",
             [],
+            "",  # search_query
             None,
             None,
             b"",
@@ -450,7 +457,7 @@ def test_get_env_vars_missing_org_or_repo(self):
         the_exception = cm.exception
         self.assertEqual(
             str(the_exception),
-            "ORGANIZATION and REPOSITORY environment variables were not set. Please set one",
+            "ORGANIZATION, REPOSITORY, and REPOSITORY_SEARCH_QUERY environment variables were not set. Please set one",
         )
 
     @patch.dict(
@@ -470,6 +477,7 @@ def test_get_env_vars_auth_with_github_app_installation(self):
         expected_result = (
             "my_organization",
             [],
+            "",  # search_query
             12345,
             678910,
             b"hello",
@@ -559,6 +567,7 @@ def test_get_env_vars_with_repos_no_dry_run(self):
         expected_result = (
             "my_organization",
             [],
+            "",  # search_query
             None,
             None,
             b"",
@@ -605,6 +614,7 @@ def test_get_env_vars_with_repos_disabled_security_updates(self):
         expected_result = (
             "my_organization",
             [],
+            "",  # search_query
             None,
             None,
             b"",
@@ -652,6 +662,7 @@ def test_get_env_vars_with_repos_filter_visibility_multiple_values(self):
         expected_result = (
             "my_organization",
             [],
+            "",  # search_query
             None,
             None,
             b"",
@@ -699,6 +710,7 @@ def test_get_env_vars_with_repos_filter_visibility_single_value(self):
         expected_result = (
             "my_organization",
             [],
+            "",  # search_query
             None,
             None,
             b"",
@@ -776,6 +788,7 @@ def test_get_env_vars_with_repos_filter_visibility_no_duplicates(self):
         expected_result = (
             "my_organization",
             [],
+            "",  # search_query
             None,
             None,
             b"",
@@ -824,6 +837,7 @@ def test_get_env_vars_with_repos_exempt_ecosystems(self):
         expected_result = (
             "my_organization",
             [],
+            "",  # search_query
             None,
             None,
             b"",
@@ -871,6 +885,7 @@ def test_get_env_vars_with_no_batch_size(self):
         expected_result = (
             "my_organization",
             [],
+            "",  # search_query
             None,
             None,
             b"",
@@ -919,6 +934,7 @@ def test_get_env_vars_with_batch_size(self):
         expected_result = (
             "my_organization",
             [],
+            "",  # search_query
             None,
             None,
             b"",
@@ -1056,6 +1072,7 @@ def test_get_env_vars_with_valid_schedule_and_schedule_day(self):
         expected_result = (
             "my_organization",
             [],
+            "",  # search_query
             None,
             None,
             b"",
@@ -1141,6 +1158,7 @@ def test_get_env_vars_with_a_valid_label(self):
         expected_result = (
             "my_organization",
             [],
+            "",  # search_query
             None,
             None,
             b"",
@@ -1187,6 +1205,7 @@ def test_get_env_vars_with_valid_labels_containing_spaces(self):
         expected_result = (
             "my_organization",
             [],
+            "",  # search_query
             None,
             None,
             b"",

test_evergreen.py

@@ -320,6 +320,7 @@ def test_get_repos_iterator_with_organization(self, mock_github):
         """Test the get_repos_iterator function with an organization"""
         organization = "my_organization"
         repository_list = []
+        search_query = ""
         github_connection = mock_github.return_value
 
         mock_organization = MagicMock()
@@ -328,7 +329,7 @@ def test_get_repos_iterator_with_organization(self, mock_github):
         github_connection.organization.return_value = mock_organization
 
         result = get_repos_iterator(
-            organization, None, repository_list, github_connection
+            organization, None, repository_list, search_query, github_connection
         )
 
         # Assert that the organization method was called with the correct argument
@@ -345,14 +346,15 @@ def test_get_repos_iterator_with_repository_list(self, mock_github):
         """Test the get_repos_iterator function with a repository list"""
         organization = None
         repository_list = ["org/repo1", "org/repo2"]
+        search_query = ""
         github_connection = mock_github.return_value
 
         mock_repository = MagicMock()
         mock_repository_list = [mock_repository, mock_repository]
         github_connection.repository.side_effect = mock_repository_list
 
         result = get_repos_iterator(
-            organization, None, repository_list, github_connection
+            organization, None, repository_list, search_query, github_connection
         )
 
         # Assert that the repository method was called with the correct arguments for each repository in the list
@@ -371,6 +373,7 @@ def test_get_repos_iterator_with_team(self, mock_github):
         organization = "my_organization"
         repository_list = []
         team_name = "my_team"
+        search_query = ""
         github_connection = mock_github.return_value
 
         mock_team_repositories = MagicMock()
@@ -382,6 +385,7 @@ def test_get_repos_iterator_with_team(self, mock_github):
             organization,
             team_name,
             repository_list,
+            search_query,
             github_connection,
         )