Skip to content

Commit 8668c63

Browse files
committed
mocked out exporter in veneer tests
1 parent f025e3e commit 8668c63

2 files changed

Lines changed: 8 additions & 2 deletions

File tree

packages/google-cloud-bigtable/tests/unit/data/_async/test_client.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -225,7 +225,9 @@ async def test_metrics_exporter_init_implicit_project(self):
225225
assert client._gcp_metrics_exporter.project_id == client.project
226226

227227
@CrossSync.pytest
228-
async def test_veneer_grpc_headers(self):
228+
@mock.patch("google.cloud.bigtable.data._async.client.BigtableMetricsExporter")
229+
@mock.patch("google.cloud.bigtable.data._sync_autogen.client.BigtableMetricsExporter")
230+
async def test_veneer_grpc_headers(self, exporter_mock, exporter_mock_sync):
229231
client_component = "data-async" if CrossSync.is_async else "data"
230232
VENEER_HEADER_REGEX = re.compile(
231233
r"gapic\/[0-9]+\.[\w.-]+ gax\/[0-9]+\.[\w.-]+ gccl\/[0-9]+\.[\w.-]+-"

packages/google-cloud-bigtable/tests/unit/data/_sync_autogen/test_client.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -189,7 +189,11 @@ def test_metrics_exporter_init_implicit_project(self):
189189
with self._make_client() as client:
190190
assert client._gcp_metrics_exporter.project_id == client.project
191191

192-
def test_veneer_grpc_headers(self):
192+
@mock.patch("google.cloud.bigtable.data._async.client.BigtableMetricsExporter")
193+
@mock.patch(
194+
"google.cloud.bigtable.data._sync_autogen.client.BigtableMetricsExporter"
195+
)
196+
def test_veneer_grpc_headers(self, exporter_mock, exporter_mock_sync):
193197
client_component = "data-async" if CrossSync._Sync_Impl.is_async else "data"
194198
VENEER_HEADER_REGEX = re.compile(
195199
"gapic\\/[0-9]+\\.[\\w.-]+ gax\\/[0-9]+\\.[\\w.-]+ gccl\\/[0-9]+\\.[\\w.-]+-"

0 commit comments

Comments
 (0)