sped up test

Author: daniel-sanche

google/cloud/bigtable/data/_async/metrics_interceptor.py

@@ -173,7 +173,8 @@ async def response_wrapper(call):
                 raise
             finally:
                 if call is not None:
-                    _end_attempt(operation, encountered_exc, await _get_metadata(call))
+                    metadata = await _get_metadata(encountered_exc or call)
+                    _end_attempt(operation, encountered_exc, metadata)
 
         try:
             return response_wrapper(await continuation(client_call_details, request))

tests/system/data/test_metrics_async.py

@@ -831,7 +831,7 @@ async def test_sample_row_keys_failure_timeout(
 
     @CrossSync.pytest
     async def test_sample_row_keys_failure_mid_stream(
-        self, table, temp_rows, handler, error_injector, cluster_config
+        self, table, temp_rows, handler, error_injector
     ):
         """
         Test failure in grpc stream
@@ -851,11 +851,6 @@ async def test_sample_row_keys_failure_mid_stream(
         assert operation.op_type.value == "SampleRowKeys"
         assert operation.is_streaming is False
         assert len(operation.completed_attempts) == 2
-        assert operation.cluster_id == next(iter(cluster_config.keys()))
-        assert (
-            operation.zone
-            == cluster_config[operation.cluster_id].location.split("/")[-1]
-        )
         # validate retried attempt
         attempt = handler.completed_attempts[0]
         assert attempt.end_status.name == "ABORTED"