Jules was unable to complete the task in time. Please review the work done so far and provide feedback for Jules to continue.

Author: google-labs-jules[bot]

tests/examples/basic_operations/test_add_ad_groups.py

@@ -0,0 +1,86 @@
+import unittest
+from unittest.mock import patch, MagicMock, call # Ensure call is imported
+
+from examples.basic_operations import add_ad_groups
+
+class TestAddAdGroups(unittest.TestCase):
+
+    @patch("examples.basic_operations.add_ad_groups.argparse.ArgumentParser")
+    @patch("examples.basic_operations.add_ad_groups.GoogleAdsClient.load_from_storage")
+    def test_main(self, mock_load_from_storage, mock_argument_parser):
+        # Mock the GoogleAdsClient
+        mock_google_ads_client = MagicMock()
+        mock_load_from_storage.return_value = mock_google_ads_client
+
+        # Mock the GoogleAdsService
+        mock_ga_service = MagicMock()
+        mock_google_ads_client.get_service.return_value = mock_ga_service
+
+        # Mock the AdGroupService
+        mock_ad_group_service = MagicMock()
+        # Ensure get_service is flexible enough to return different mocks based on input
+        def get_service_side_effect(service_name, version=None): # Added version to match signature
+            if service_name == "GoogleAdsService":
+                return mock_ga_service
+            elif service_name == "AdGroupService":
+                return mock_ad_group_service
+            elif service_name == "CampaignService":
+                mock_campaign_service = MagicMock()
+                # Configure campaign_path mock
+                mock_campaign_service.campaign_path.return_value = f"customers/{mock_args.customer_id}/campaigns/{mock_args.campaign_id}"
+                return mock_campaign_service
+            raise ValueError(f"Unexpected service: {service_name}")
+        mock_google_ads_client.get_service.side_effect = get_service_side_effect
+
+
+        # Mock command line arguments
+        mock_args = MagicMock()
+        mock_args.customer_id = "1234567890"
+        mock_args.campaign_id = "9876543210"
+        mock_argument_parser.return_value.parse_args.return_value = mock_args
+
+        # Mock the responses for service calls
+        # For AdGroupService mutate
+        mock_ad_group_operation_response = MagicMock()
+        # Simulate that one ad group was created
+        mock_ad_group_result = MagicMock()
+        mock_ad_group_result.resource_name = "customers/1234567890/adGroups/AD_GROUP_ID_1"
+        mock_ad_group_operation_response.results = [mock_ad_group_result]
+        mock_ad_group_service.mutate_ad_groups.return_value = mock_ad_group_operation_response
+
+        # Call the main function of the example script
+        with patch("builtins.print") as mock_print:
+            add_ad_groups.main(mock_google_ads_client, mock_args.customer_id, mock_args.campaign_id)
+
+        # Assertions
+        # mock_load_from_storage.assert_called_once_with(version="v19") # Removed this line
+        mock_google_ads_client.get_service.assert_any_call("AdGroupService")
+        mock_google_ads_client.get_service.assert_any_call("CampaignService")
+
+
+        self.assertEqual(mock_ad_group_service.mutate_ad_groups.call_count, 1)
+        # Check the first argument of the first call to mutate_ad_groups
+        args, kwargs = mock_ad_group_service.mutate_ad_groups.call_args
+        self.assertEqual(kwargs['customer_id'], "1234567890")
+        operations = kwargs['operations']
+        self.assertEqual(len(operations), 1) # Expecting one ad group operation
+
+        # Check the created ad group operation
+        created_ad_group = operations[0].create
+        # The name is dynamic with uuid, so we can't assert exact name.
+        # We can check if it starts with the expected prefix.
+        self.assertTrue(created_ad_group.name.startswith("Earth to Mars cruises "))
+        self.assertEqual(created_ad_group.status, mock_google_ads_client.enums.AdGroupStatusEnum.ENABLED)
+        self.assertEqual(created_ad_group.campaign, "customers/1234567890/campaigns/9876543210")
+        self.assertEqual(created_ad_group.type_, mock_google_ads_client.enums.AdGroupTypeEnum.SEARCH_STANDARD)
+        self.assertEqual(created_ad_group.cpc_bid_micros, 10000000)
+
+
+        # Verify print output
+        expected_print_calls = [
+            call(f"Created ad group customers/1234567890/adGroups/AD_GROUP_ID_1.")
+        ]
+        mock_print.assert_has_calls(expected_print_calls, any_order=False)
+
+if __name__ == "__main__":
+    unittest.main()

tests/examples/basic_operations/test_add_campaigns.py

@@ -0,0 +1,100 @@
+import unittest
+from unittest.mock import patch, MagicMock, call
+
+from examples.basic_operations import add_campaigns
+
+class TestAddCampaigns(unittest.TestCase):
+
+    @patch("examples.basic_operations.add_campaigns.argparse.ArgumentParser")
+    @patch("examples.basic_operations.add_campaigns.GoogleAdsClient.load_from_storage")
+    def test_main(self, mock_load_from_storage, mock_argument_parser):
+        # Mock the GoogleAdsClient
+        mock_google_ads_client = MagicMock()
+        mock_load_from_storage.return_value = mock_google_ads_client
+
+        # Mock the CampaignService
+        mock_campaign_service = MagicMock()
+        # Mock the BudgetService (even if not directly used in main, it might be called by helpers)
+        mock_campaign_budget_service = MagicMock()
+
+        def get_service_side_effect(service_name, version=None):
+            if service_name == "CampaignService":
+                return mock_campaign_service
+            elif service_name == "CampaignBudgetService":
+                return mock_campaign_budget_service
+            # Add other services if the example uses them
+            raise ValueError(f"Unexpected service: {service_name}")
+        mock_google_ads_client.get_service.side_effect = get_service_side_effect
+
+        # Mock command line arguments
+        mock_args = MagicMock()
+        mock_args.customer_id = "1234567890"
+        # mock_args.number_of_campaigns = 1 # This argument is not used by main
+
+        mock_argument_parser.return_value.parse_args.return_value = mock_args
+
+        # Mock the responses for service calls
+        # For CampaignBudgetService mutate (if we were testing budget creation)
+        mock_budget_operation_response = MagicMock()
+        mock_budget_result = MagicMock()
+        mock_budget_result.resource_name = "customers/1234567890/campaignBudgets/BUDGET_ID_1"
+        mock_budget_operation_response.results = [mock_budget_result]
+        mock_campaign_budget_service.mutate_campaign_budgets.return_value = mock_budget_operation_response
+
+        # For CampaignService mutate
+        mock_campaign_operation_response = MagicMock()
+        mock_campaign_result = MagicMock()
+        mock_campaign_result.resource_name = "customers/1234567890/campaigns/CAMPAIGN_ID_1"
+        mock_campaign_operation_response.results = [mock_campaign_result]
+        # Configure mutate_campaigns to return a list of results if multiple operations are sent
+        mock_campaign_service.mutate_campaigns.return_value = mock_campaign_operation_response
+
+        # Mock enums
+        mock_google_ads_client.enums.CampaignStatusEnum.PAUSED = "PAUSED"
+        mock_google_ads_client.enums.AdvertisingChannelTypeEnum.SEARCH = "SEARCH"
+        mock_google_ads_client.enums.BiddingStrategyTypeEnum.MANUAL_CPC = "MANUAL_CPC"
+        mock_google_ads_client.enums.BudgetDeliveryMethodEnum.STANDARD = "STANDARD"
+
+
+        # Call the main function of the example script
+        with patch("builtins.print") as mock_print:
+            add_campaigns.main(mock_google_ads_client, mock_args.customer_id)
+
+        # Assertions
+        # mock_load_from_storage.assert_called_once_with(version="v19") # main doesn't call this directly
+        mock_google_ads_client.get_service.assert_any_call("CampaignService")
+        mock_google_ads_client.get_service.assert_any_call("CampaignBudgetService")
+
+        # Ensure CampaignBudgetService.mutate_campaign_budgets was called
+        mock_campaign_budget_service.mutate_campaign_budgets.assert_called_once()
+        budget_args, budget_kwargs = mock_campaign_budget_service.mutate_campaign_budgets.call_args
+        self.assertEqual(budget_kwargs['customer_id'], mock_args.customer_id)
+        self.assertEqual(len(budget_kwargs['operations']), 1)
+        # Add more assertions for budget operation if needed
+
+        self.assertEqual(mock_campaign_service.mutate_campaigns.call_count, 1)
+        args_list, kwargs_list = mock_campaign_service.mutate_campaigns.call_args
+        self.assertEqual(kwargs_list['customer_id'], "1234567890")
+        operations = kwargs_list['operations']
+        self.assertEqual(len(operations), 1) # Script creates one campaign
+
+        # Check properties of the campaign operation
+        campaign_op = operations[0].create
+        self.assertTrue(campaign_op.name.startswith("Interplanetary Cruise ")) # Adjusted prefix
+        self.assertEqual(campaign_op.status, mock_google_ads_client.enums.CampaignStatusEnum.PAUSED)
+        self.assertEqual(campaign_op.advertising_channel_type, mock_google_ads_client.enums.AdvertisingChannelTypeEnum.SEARCH)
+        # In the actual script, manual_cpc is initialized but enhanced_cpc_enabled is not explicitly set.
+        # We should check what is actually being set.
+        # self.assertEqual(campaign_op.manual_cpc.enhanced_cpc_enabled, True)
+        self.assertTrue(hasattr(campaign_op, 'manual_cpc')) # Check if manual_cpc is set
+        self.assertEqual(campaign_op.campaign_budget, "customers/1234567890/campaignBudgets/BUDGET_ID_1")
+
+
+        # Verify print output
+        expected_print_calls = [
+            call(f"Created campaign customers/1234567890/campaigns/CAMPAIGN_ID_1.")
+        ]
+        mock_print.assert_has_calls(expected_print_calls, any_order=False)
+
+if __name__ == "__main__":
+    unittest.main()

tests/examples/basic_operations/test_get_campaigns.py

@@ -0,0 +1,65 @@
+import unittest
+from unittest.mock import patch, MagicMock, call
+
+from examples.basic_operations import get_campaigns
+
+class TestGetCampaigns(unittest.TestCase):
+
+    @patch("examples.basic_operations.get_campaigns.argparse.ArgumentParser")
+    @patch("examples.basic_operations.get_campaigns.GoogleAdsClient.load_from_storage")
+    def test_main(self, mock_load_from_storage, mock_argument_parser):
+        # Mock the GoogleAdsClient
+        mock_google_ads_client = MagicMock()
+        mock_load_from_storage.return_value = mock_google_ads_client
+
+        # Mock the GoogleAdsService
+        mock_ga_service = MagicMock()
+        mock_google_ads_client.get_service.return_value = mock_ga_service
+
+        # Mock command line arguments
+        mock_args = MagicMock()
+        mock_args.customer_id = "1234567890"
+        mock_argument_parser.return_value.parse_args.return_value = mock_args
+
+        # Mock the response from search_stream
+        mock_row1 = MagicMock()
+        mock_row1.campaign.id = "CAMPAIGN_ID_1"
+        mock_row1.campaign.name = "Test Campaign 1"
+
+        mock_row2 = MagicMock()
+        mock_row2.campaign.id = "CAMPAIGN_ID_2"
+        mock_row2.campaign.name = "Test Campaign 2"
+
+        mock_batch = MagicMock()
+        mock_batch.results = [mock_row1, mock_row2]
+
+        # search_stream returns an iterable of batches
+        mock_ga_service.search_stream.return_value = [mock_batch]
+
+        # Call the main function of the example script
+        with patch("builtins.print") as mock_print:
+            get_campaigns.main(mock_google_ads_client, mock_args.customer_id)
+
+        # Assertions
+        # mock_load_from_storage.assert_called_once_with(version="v19") # main doesn't call this
+        mock_google_ads_client.get_service.assert_called_once_with("GoogleAdsService")
+
+        expected_query = """
+        SELECT
+          campaign.id,
+          campaign.name
+        FROM campaign
+        ORDER BY campaign.id"""
+        mock_ga_service.search_stream.assert_called_once_with(
+            customer_id=mock_args.customer_id, query=expected_query
+        )
+
+        # Verify print output
+        expected_print_calls = [
+            call(f"Campaign with ID {mock_row1.campaign.id} and name \"{mock_row1.campaign.name}\" was found."),
+            call(f"Campaign with ID {mock_row2.campaign.id} and name \"{mock_row2.campaign.name}\" was found.")
+        ]
+        mock_print.assert_has_calls(expected_print_calls, any_order=False)
+
+if __name__ == "__main__":
+    unittest.main()

tests/examples/basic_operations/test_get_responsive_search_ads.py

@@ -0,0 +1,123 @@
+import unittest
+from unittest.mock import patch, MagicMock, call
+
+from examples.basic_operations import get_responsive_search_ads
+
+class TestGetResponsiveSearchAds(unittest.TestCase):
+
+    @patch("examples.basic_operations.get_responsive_search_ads.argparse.ArgumentParser")
+    @patch("examples.basic_operations.get_responsive_search_ads.GoogleAdsClient.load_from_storage")
+    def test_main(self, mock_load_from_storage, mock_argument_parser):
+        # Mock the GoogleAdsClient
+        mock_google_ads_client = MagicMock()
+        mock_load_from_storage.return_value = mock_google_ads_client
+
+        # Mock the GoogleAdsService
+        mock_ga_service = MagicMock()
+        mock_google_ads_client.get_service.return_value = mock_ga_service
+
+        # Mock command line arguments
+        mock_args = MagicMock()
+        mock_args.customer_id = "1234567890"
+        mock_args.ad_group_id = "ADGROUPID_01" # Optional, so test with it present
+        mock_argument_parser.return_value.parse_args.return_value = mock_args
+
+        # Mock the response from search
+        # This is a row from the SearchGoogleAdsResponse, not AdGroupAd directly
+        mock_row1 = MagicMock()
+        mock_ad_group_ad = mock_row1.ad_group_ad # This is the AdGroupAd object
+
+        mock_ad_group_ad.ad.resource_name = "customers/1234567890/ads/AD_ID_1"
+        mock_ad_group_ad.status.name = "ENABLED" # Example status
+
+        # RSA has headlines and descriptions (AdTextAsset)
+        mock_headline1 = MagicMock()
+        mock_headline1.text = "Cruise to Mars"
+        mock_headline1.pinned_field.name = "HEADLINE_1" # Script uses asset.pinned_field.name
+
+        mock_description1 = MagicMock()
+        mock_description1.text = "Visit the Red Planet in style."
+        mock_description1.pinned_field.name = "DESCRIPTION_1"
+
+        mock_ad_group_ad.ad.responsive_search_ad.headlines = [mock_headline1]
+        mock_ad_group_ad.ad.responsive_search_ad.descriptions = [mock_description1]
+        # The script doesn't print final_urls for RSA, it prints headlines and descriptions
+
+        # search returns an iterable of these rows
+        mock_ga_service.search.return_value = [mock_row1]
+
+        # Enums are compared by their .name attribute by the script, so no need to mock them as strings here
+        # if the attributes being accessed on them (like .name) are already strings or MagicMocks.
+        # The script uses ad.responsive_search_ad.headlines which are AdTextAssets.
+        # ad_text_assets_to_strs accesses asset.text and asset.pinned_field.name.
+
+
+        # Call the main function of the example script
+        with patch("builtins.print") as mock_print:
+            get_responsive_search_ads.main(mock_google_ads_client, mock_args.customer_id, mock_args.ad_group_id)
+
+        # Assertions
+        mock_google_ads_client.get_service.assert_called_once_with("GoogleAdsService")
+
+        # Check that search was called
+        mock_ga_service.search.assert_called_once()
+        call_args = mock_ga_service.search.call_args
+
+        # The first argument to search is the request object
+        request_arg = call_args[1]['request'] # request is a keyword argument
+        self.assertEqual(request_arg.customer_id, mock_args.customer_id)
+        # Verify query construction based on ad_group_id presence
+        self.assertIn(f"AND ad_group.id = {mock_args.ad_group_id}", request_arg.query)
+
+
+        # Verify print output - this will be complex due to loops and formatting
+        # We'll check for key pieces of information.
+        # Note: The exact formatting of pinned fields might require more specific mocking if it's complex.
+        # This example assumes simple text output for headlines and descriptions.
+
+        # Expected calls based on the mocked ad structure
+        # The script's print format is:
+        # print(f"Responsive search ad with resource name \"{ad.resource_name}\", status {row.ad_group_ad.status.name} was found.")
+        # print(f"Headlines:\n{headlines_str}\nDescriptions:\n{descriptions_str}\n")
+
+        expected_ad_info_print = call(
+            f"Responsive search ad with resource name \"{mock_ad_group_ad.ad.resource_name}\", status {mock_ad_group_ad.status.name} was found."
+        )
+
+        # Output from ad_text_assets_to_strs
+        expected_headline_str = f"\t {mock_headline1.text} pinned to {mock_headline1.pinned_field.name}"
+        expected_description_str = f"\t {mock_description1.text} pinned to {mock_description1.pinned_field.name}"
+
+        expected_details_print = call(f"Headlines:\n{expected_headline_str}\nDescriptions:\n{expected_description_str}\n")
+
+        expected_calls = [expected_ad_info_print, expected_details_print]
+
+        mock_print.assert_has_calls(expected_calls, any_order=False)
+
+
+    @patch("examples.basic_operations.get_responsive_search_ads.argparse.ArgumentParser")
+    @patch("examples.basic_operations.get_responsive_search_ads.GoogleAdsClient.load_from_storage")
+    def test_main_no_ad_group_id(self, mock_load_from_storage, mock_argument_parser):
+        # Test the case where ad_group_id is None
+        mock_google_ads_client = MagicMock()
+        mock_load_from_storage.return_value = mock_google_ads_client
+        mock_ga_service = MagicMock()
+        mock_google_ads_client.get_service.return_value = mock_ga_service
+
+        mock_args = MagicMock()
+        mock_args.customer_id = "1234567890"
+        mock_args.ad_group_id = None # Test this path
+        mock_argument_parser.return_value.parse_args.return_value = mock_args
+
+        mock_ga_service.search.return_value = [] # No results needed for this query check path
+
+        with patch("builtins.print"): # Suppress print for this test
+            get_responsive_search_ads.main(mock_google_ads_client, mock_args.customer_id, mock_args.ad_group_id)
+
+        mock_ga_service.search.assert_called_once()
+        request_arg = mock_ga_service.search.call_args[1]['request']
+        self.assertNotIn("AND ad_group.id", request_arg.query)
+
+
+if __name__ == "__main__":
+    unittest.main()