Update misc/ example tests

Author: BenRKarl

examples/misc/campaign_report_to_csv.py

@@ -36,6 +36,19 @@
 
 
 _DEFAULT_FILE_NAME = "campaign_report_to_csv_results.csv"
+_QUERY: str = """
+  SELECT
+      customer.descriptive_name,
+      segments.date,
+      campaign.name,
+      metrics.impressions,
+      metrics.clicks,
+      metrics.cost_micros
+  FROM campaign
+  WHERE
+      segments.date DURING LAST_7_DAYS
+  ORDER BY metrics.impressions DESC
+  LIMIT 25"""
 
 
 def main(
@@ -55,24 +68,10 @@ def main(
     file_path: str = os.path.join(file_dir, output_file)
     ga_service = client.get_service("GoogleAdsService")
 
-    query: str = """
-        SELECT
-          customer.descriptive_name,
-          segments.date,
-          campaign.name,
-          metrics.impressions,
-          metrics.clicks,
-          metrics.cost_micros
-        FROM campaign
-        WHERE
-          segments.date DURING LAST_7_DAYS
-        ORDER BY metrics.impressions DESC
-        LIMIT 25"""
-
     # Issues a search request using streaming.
     search_request = client.get_type("SearchGoogleAdsStreamRequest")
     search_request.customer_id = customer_id
-    search_request.query = query
+    search_request.query = _QUERY
     stream = ga_service.search_stream(search_request)
 
     with open(file_path, "w", newline="") as f:

examples/misc/tests/__init__.py tests/examples/misc/__init__.pyexamples/misc/tests/__init__.py renamed to tests/examples/misc/__init__.py

@@ -1,6 +1,18 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 import unittest
 from unittest import mock
-import sys
 import argparse # Ensure argparse is imported
 
 from google.ads.googleads.errors import GoogleAdsException
@@ -203,7 +215,4 @@ def test_argument_parsing(
             expected_cust_id,
             expected_ag_id,
             expected_asset_id_val
-        )
-
-if __name__ == "__main__":
-    unittest.main()
+        )

…c/tests/test_add_ad_group_image_asset.py …es/misc/add_ad_group_image_asset_test.pyexamples/misc/tests/test_add_ad_group_image_asset.py renamed to tests/examples/misc/add_ad_group_image_asset_test.py examples/misc/tests/test_add_ad_group_image_asset.py renamed to tests/examples/misc/add_ad_group_image_asset_test.py

@@ -1,3 +1,16 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 import unittest
 from unittest import mock
 import csv
@@ -11,14 +24,6 @@
 from examples.misc import campaign_report_to_csv
 from google.ads.googleads.errors import GoogleAdsException
 from .test_utils import create_mock_google_ads_exception
-from google.ads.googleads.v19.services.services.google_ads_service import GoogleAdsServiceClient
-from google.ads.googleads.v19.services.types.google_ads_service import SearchGoogleAdsStreamResponse
-# Campaign, AdGroup, Customer, GoogleAdsRow types are not strictly needed for these tests
-# if we use the custom mock row/batch classes, but keeping them doesn't hurt.
-from google.ads.googleads.v19.resources.types.campaign import Campaign
-from google.ads.googleads.v19.resources.types.ad_group import AdGroup
-from google.ads.googleads.v19.resources.types.customer import Customer
-from google.ads.googleads.v19.services.types.google_ads_service import GoogleAdsRow
 
 
 # Helper classes for mocking Google Ads API CSV Row/Batch structures
@@ -332,7 +337,7 @@ def _simulate_script_main_block(
         # Script's GoogleAdsClient.load_from_storage call
         mock_client_instance = mock.Mock(name="GoogleAdsClientInstance")
         mock_gads_client_class_from_decorator.load_from_storage.return_value = mock_client_instance
-        googleads_client = mock_gads_client_class_from_decorator.load_from_storage(version="v19")
+        googleads_client = mock_gads_client_class_from_decorator.load_from_storage(version="v20")
 
         # Script's main function call
         mock_main_func_from_decorator(
@@ -384,7 +389,7 @@ def test_argument_parsing(self, mock_script_main, mock_gads_client_class,
 
         mock_parser_inst_for_assert.parse_args.assert_called_once_with()
 
-        mock_gads_client_class.load_from_storage.assert_called_once_with(version="v19")
+        mock_gads_client_class.load_from_storage.assert_called_once_with(version="v20")
 
         client_inst_for_assert = mock_gads_client_class.load_from_storage.return_value
         mock_script_main.assert_called_once_with(
@@ -425,15 +430,12 @@ def test_argument_parsing_no_headers_flag(self, mock_script_main, mock_gads_clie
         # For brevity, only checking parse_args and main call here, assuming add_argument calls are covered.
         mock_parser_inst_for_assert.parse_args.assert_called_once_with()
 
-        mock_gads_client_class.load_from_storage.assert_called_once_with(version="v19")
+        mock_gads_client_class.load_from_storage.assert_called_once_with(version="v20")
 
         client_inst_for_assert = mock_gads_client_class.load_from_storage.return_value
         mock_script_main.assert_called_once_with(
             client_inst_for_assert,
             expected_cust_id,
             expected_filepath,
             expected_headers_bool
-        )
-
-if __name__ == "__main__":
-    unittest.main()
+        )

…isc/tests/test_campaign_report_to_csv.py …ples/misc/campaign_report_to_csv_test.pyexamples/misc/tests/test_campaign_report_to_csv.py renamed to tests/examples/misc/campaign_report_to_csv_test.py examples/misc/tests/test_campaign_report_to_csv.py renamed to tests/examples/misc/campaign_report_to_csv_test.py

@@ -1,12 +1,23 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 import unittest
 from unittest import mock
 import argparse
-import sys
 
 # Assuming the script to be tested is in the parent directory.
 # Adjust the import path as necessary.
 from examples.misc import set_custom_client_timeouts
-from google.ads.googleads.errors import GoogleAdsException
 from google.api_core.exceptions import DeadlineExceeded
 from google.api_core.retry import Retry # Added import for Retry
 from .test_utils import create_mock_google_ads_exception
@@ -306,7 +317,4 @@ def test_argument_parsing_and_script_execution(
         mock_script_main_function.assert_called_once_with(
             client_instance_for_assert,
             expected_customer_id
-        )
-
-if __name__ == "__main__":
-    unittest.main()
+        )

…tests/test_set_custom_client_timeouts.py …/misc/set_custom_client_timeouts_test.pyexamples/misc/tests/test_set_custom_client_timeouts.py renamed to tests/examples/misc/set_custom_client_timeouts_test.py examples/misc/tests/test_set_custom_client_timeouts.py renamed to tests/examples/misc/set_custom_client_timeouts_test.py

@@ -1,3 +1,16 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 from unittest import mock
 from google.ads.googleads.errors import GoogleAdsException
 

examples/misc/tests/test_utils.py tests/examples/misc/test_utils.pyexamples/misc/tests/test_utils.py renamed to tests/examples/misc/test_utils.py

@@ -1,15 +1,25 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 import unittest
 from unittest import mock
 import argparse
-import sys
 
 # Assuming the script to be tested is in the parent directory.
 # Adjust the import path as necessary.
 from examples.misc import upload_image_asset
 from google.ads.googleads.errors import GoogleAdsException
 from .test_utils import create_mock_google_ads_exception
-from google.ads.googleads.v19.enums.types.asset_type import AssetTypeEnum
-from google.ads.googleads.v19.enums.types.mime_type import MimeTypeEnum
 
 
 class TestUploadImageAsset(unittest.TestCase):
@@ -181,7 +191,4 @@ def test_argument_parsing_and_script_execution(
         mock_script_main.assert_called_once_with(
             client_instance_for_assert,
             expected_cust_id
-        )
-
-if __name__ == "__main__":
-    unittest.main()
+        )