patch gapic tests

googleapis · Jan 16, 2025 · 4a08f14 · 4a08f14
1 parent 86e69d3
commit 4a08f14
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/unit/gapic/bigtable_v2/test_bigtable.py b/tests/unit/gapic/bigtable_v2/test_bigtable.py
@@ -6868,7 +6868,8 @@ def test_execute_query_routing_parameters_request_1_grpc():
 
         assert args[0] == request_msg
 
-        expected_headers = {"name": "projects/sample1/instances/sample2"}
+        # expect app_profile_id while temporary patch is in place: https://github.com/googleapis/python-bigtable/pull/1072
+        expected_headers = {"name": "projects/sample1/instances/sample2", "app_profile_id": ""}
         assert (
             gapic_v1.routing_header.to_grpc_metadata(expected_headers) in kw["metadata"]
         )
@@ -7894,7 +7895,8 @@ async def test_execute_query_routing_parameters_request_1_grpc_asyncio():
 
         assert args[0] == request_msg
 
-        expected_headers = {"name": "projects/sample1/instances/sample2"}
+        # expect app_profile_id while temporary patch is in place: https://github.com/googleapis/python-bigtable/pull/1072
+        expected_headers = {"name": "projects/sample1/instances/sample2", "app_profile_id": ""}
         assert (
             gapic_v1.routing_header.to_grpc_metadata(expected_headers) in kw["metadata"]
         )
@@ -9915,7 +9917,8 @@ def test_execute_query_routing_parameters_request_1_rest():
 
         assert args[0] == request_msg
 
-        expected_headers = {"name": "projects/sample1/instances/sample2"}
+        # expect app_profile_id while temporary patch is in place: https://github.com/googleapis/python-bigtable/pull/1072
+        expected_headers = {"name": "projects/sample1/instances/sample2", "app_profile_id": ""}
         assert (
             gapic_v1.routing_header.to_grpc_metadata(expected_headers) in kw["metadata"]
         )