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/__init__.py

@@ -0,0 +1,77 @@
+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
+            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")
+        mock_google_ads_client.get_service.assert_any_call("AdGroupService")
+        # Allow GoogleAdsService to be called or not, as it's used for campaign existence check which might be mocked differently or not reached in a minimal test
+        # mock_google_ads_client.get_service.assert_any_call("GoogleAdsService")
+
+
+        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), 2) # Expecting two ad group operations
+
+        # Example of checking one operation (can be more detailed)
+        self.assertEqual(operations[0].create.name, f"Earth to Mars Cruises #{(0 + 1)}")
+        self.assertEqual(operations[0].create.status, mock_google_ads_client.enums.AdGroupStatusEnum.PAUSED)
+        self.assertEqual(operations[0].create.campaign, f"customers/1234567890/campaigns/9876543210")
+
+
+        # 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/__init__.py

@@ -0,0 +1,67 @@
+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_google_ads_client.get_service.return_value = mock_campaign_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 responses for service calls
+        mock_campaign_operation_response = MagicMock()
+        # Simulate that one campaign was created
+        mock_campaign_result = MagicMock()
+        mock_campaign_result.resource_name = "customers/1234567890/campaigns/CAMPAIGN_ID_1"
+        mock_campaign_operation_response.results = [mock_campaign_result]
+        mock_campaign_service.mutate_campaigns.return_value = mock_campaign_operation_response
+
+        # 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")
+        mock_google_ads_client.get_service.assert_called_once_with("CampaignService")
+
+        self.assertEqual(mock_campaign_service.mutate_campaigns.call_count, 1)
+        args, kwargs = mock_campaign_service.mutate_campaigns.call_args
+        self.assertEqual(kwargs['customer_id'], "1234567890")
+        operations = kwargs['operations']
+        # The example creates 1 campaign, but the script has a loop for _NUMBER_OF_CAMPAIGNS (default 1)
+        # For simplicity here, assuming _NUMBER_OF_CAMPAIGNS = 1 or that the test setup matches one operation.
+        # If _NUMBER_OF_CAMPAIGNS was different, this would need adjustment or the constant mocked.
+        self.assertEqual(len(operations), 1)
+
+        # Check properties of the created campaign operation
+        campaign_op = operations[0].create
+        self.assertTrue(campaign_op.name.startswith("Interplanetary Cruise #"))
+        self.assertEqual(campaign_op.advertising_channel_type, mock_google_ads_client.enums.AdvertisingChannelTypeEnum.SEARCH)
+        self.assertEqual(campaign_op.status, mock_google_ads_client.enums.CampaignStatusEnum.PAUSED)
+        self.assertEqual(campaign_op.manual_cpc.enhanced_cpc_enabled, True)
+        self.assertEqual(campaign_op.campaign_budget, mock_campaign_service.client.get_type("CampaignBudgetOperation").create.name) # This needs careful mocking if budget is created separately
+        self.assertEqual(campaign_op.network_settings.target_google_search, True)
+
+
+        # Verify print output
+        # Assuming one campaign result as mocked
+        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_add_ad_groups.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 = "111"
+        mock_row1.campaign.name = "Test Campaign 1"
+
+        mock_row2 = MagicMock()
+        mock_row2.campaign.id = "222"
+        mock_row2.campaign.name = "Test Campaign 2"
+
+        mock_batch1 = MagicMock()
+        mock_batch1.results = [mock_row1, mock_row2]
+
+        mock_stream_response = [mock_batch1]
+        mock_ga_service.search_stream.return_value = mock_stream_response
+
+        # 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")
+        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("Campaign with ID 111 and name "Test Campaign 1" was found."),
+            call("Campaign with ID 222 and name "Test Campaign 2" was found.")
+        ]
+        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,103 @@
+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"  # Optional, so test with and without
+        mock_argument_parser.return_value.parse_args.return_value = mock_args
+
+        # Mock the response from search_stream
+        mock_row1_ad = MagicMock()
+        mock_row1_ad.resource_name = "customers/123/ads/RSA1"
+        # Mock headlines
+        mock_headline1 = MagicMock()
+        mock_headline1.text = "Headline 1 RSA1"
+        mock_headline1.asset_performance_label = mock_google_ads_client.enums.AssetPerformanceLabelEnum.PENDING
+        mock_headline1.policy_summary.review_status = mock_google_ads_client.enums.PolicyReviewStatusEnum.REVIEWED
+        mock_headline1.policy_summary.approval_status = mock_google_ads_client.enums.PolicyApprovalStatusEnum.APPROVED
+        # Mock descriptions
+        mock_desc1 = MagicMock()
+        mock_desc1.text = "Description 1 RSA1"
+        mock_desc1.asset_performance_label = mock_google_ads_client.enums.AssetPerformanceLabelEnum.LEARNING
+
+        mock_row1_ad.responsive_search_ad.headlines = [mock_headline1]
+        mock_row1_ad.responsive_search_ad.descriptions = [mock_desc1]
+
+
+        mock_row1 = MagicMock()
+        mock_row1.ad_group_ad.ad = mock_row1_ad
+
+        mock_batch1 = MagicMock()
+        mock_batch1.results = [mock_row1]
+
+        mock_stream_response = [mock_batch1]
+        mock_ga_service.search_stream.return_value = mock_stream_response
+
+        # Call the main function of the example script
+        with patch("builtins.print") as mock_print:
+            # Test with ad_group_id
+            get_responsive_search_ads.main(mock_google_ads_client, mock_args.customer_id, mock_args.ad_group_id)
+
+            # Test without ad_group_id (reset mock_args and mock_ga_service calls for a clean second run)
+            mock_args.ad_group_id = None
+            mock_ga_service.reset_mock() # Reset call counts etc.
+            mock_ga_service.search_stream.return_value = mock_stream_response # Re-assign return value
+            get_responsive_search_ads.main(mock_google_ads_client, mock_args.customer_id, mock_args.ad_group_id)
+
+
+        # Assertions
+        self.assertEqual(mock_load_from_storage.call_count, 2) # Called twice
+        mock_load_from_storage.assert_called_with(version="v19") # Check one of the calls
+
+        self.assertEqual(mock_google_ads_client.get_service.call_count, 2)
+        mock_google_ads_client.get_service.assert_called_with("GoogleAdsService")
+
+        # Check query for call WITH ad_group_id
+        args_with, kwargs_with = mock_ga_service.search_stream.call_args_list[0]
+        self.assertEqual(kwargs_with['customer_id'], "1234567890")
+        self.assertIn("AND ad_group.id = 'ADGROUPID'", kwargs_with['query'])
+
+        # Check query for call WITHOUT ad_group_id
+        args_without, kwargs_without = mock_ga_service.search_stream.call_args_list[1]
+        self.assertEqual(kwargs_without['customer_id'], "1234567890")
+        self.assertNotIn("AND ad_group.id", kwargs_without['query'])
+
+        # Verify print output (simplified for one call, can be expanded)
+        # This will check the print calls from the first main() call (with ad_group_id)
+        expected_print_calls = [
+            call("Responsive search ad with resource name "
+                 ""customers/123/ads/RSA1" was found."),
+            call("	Headlines:"),
+            call("		Headline text: Headline 1 RSA1"),
+            call("		Performance: PENDING"),
+            call("		Review status: REVIEWED"),
+            call("		Approval status: APPROVED"),
+            call("	Descriptions:"),
+            call("		Description text: Description 1 RSA1"),
+            call("		Performance: LEARNING"),
+        ]
+        # mock_print.assert_has_calls(expected_print_calls, any_order=False)
+        # Due to two calls to main, the above assert_has_calls will be tricky without more complex print mocking.
+        # We can check if a specific print happened:
+        mock_print.assert_any_call("Responsive search ad with resource name "
+                                   ""customers/123/ads/RSA1" was found.")
+
+
+if __name__ == "__main__":
+    unittest.main()