fixed lint

Author: daniel-sanche

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

@@ -54,7 +54,10 @@ def wrapper(self, continuation, client_call_details, request):
         operation: "ActiveOperationMetric" = self.operation_map.get(key)
         if operation:
             # start a new attempt if not started
-            if operation.state == OperationState.CREATED or operation.state == OperationState.BETWEEN_ATTEMPTS:
+            if (
+                operation.state == OperationState.CREATED
+                or operation.state == OperationState.BETWEEN_ATTEMPTS
+            ):
                 operation.start_attempt()
             # wrap continuation in logic to process the operation
             return func(self, operation, continuation, client_call_details, request)
@@ -85,6 +88,7 @@ async def _get_metadata(source):
         # ignore errors while fetching metadata
         return None
 
+
 @CrossSync.convert_class(sync_name="BigtableMetricsInterceptor")
 class AsyncBigtableMetricsInterceptor(
     UnaryUnaryClientInterceptor, UnaryStreamClientInterceptor, MetricsHandler
@@ -167,4 +171,4 @@ async def response_wrapper(call):
         except Exception as rpc_error:
             # handle errors while intializing stream
             _end_attempt(operation, rpc_error, await _get_metadata(rpc_error))
-            raise rpc_error
+            raise rpc_error

tests/unit/data/_async/test_client.py

@@ -1160,7 +1160,9 @@ def _make_one(
     @CrossSync.pytest
     async def test_ctor(self):
         from google.cloud.bigtable.data._helpers import _WarmedInstanceKey
-        from google.cloud.bigtable.data._metrics import BigtableClientSideMetricsController
+        from google.cloud.bigtable.data._metrics import (
+            BigtableClientSideMetricsController,
+        )
 
         expected_table_id = "table-id"
         expected_instance_id = "instance-id"
@@ -1493,7 +1495,9 @@ def _make_one(
     @CrossSync.pytest
     async def test_ctor(self):
         from google.cloud.bigtable.data._helpers import _WarmedInstanceKey
-        from google.cloud.bigtable.data._metrics import BigtableClientSideMetricsController
+        from google.cloud.bigtable.data._metrics import (
+            BigtableClientSideMetricsController,
+        )
 
         expected_table_id = "table-id"
         expected_instance_id = "instance-id"

tests/unit/data/_async/test_metrics_interceptor.py

@@ -13,7 +13,6 @@
 # limitations under the License.
 
 import pytest
-import asyncio
 from grpc import RpcError
 
 from google.cloud.bigtable.data._metrics.data_model import OperationState
@@ -26,32 +25,43 @@
     import mock  # type: ignore
 
 if CrossSync.is_async:
-    from google.cloud.bigtable.data._async.metrics_interceptor import AsyncBigtableMetricsInterceptor
+    from google.cloud.bigtable.data._async.metrics_interceptor import (
+        AsyncBigtableMetricsInterceptor,
+    )
 else:
-    from google.cloud.bigtable.data._sync_autogen.metrics_interceptor import BigtableMetricsInterceptor
+    from google.cloud.bigtable.data._sync_autogen.metrics_interceptor import (  # noqa: F401
+        BigtableMetricsInterceptor,
+    )
 
 
 __CROSS_SYNC_OUTPUT__ = "tests.unit.data._sync_autogen.test_metrics_interceptor"
 
+
 @CrossSync.convert(replace_symbols={"__aiter__": "__iter__"})
 def _make_mock_stream_call(values, exc=None):
     """
     Create a mock call object that can be used for streaming calls
     """
     call = CrossSync.Mock()
+
     async def gen():
         for val in values:
             yield val
         if exc:
             raise exc
+
     call.__aiter__ = mock.Mock(return_value=gen())
     return call
 
 
 @CrossSync.convert_class(sync_name="TestMetricsInterceptor")
 class TestMetricsInterceptorAsync:
     @staticmethod
-    @CrossSync.convert(replace_symbols={"AsyncBigtableMetricsInterceptor": "BigtableMetricsInterceptor"})
+    @CrossSync.convert(
+        replace_symbols={
+            "AsyncBigtableMetricsInterceptor": "BigtableMetricsInterceptor"
+        }
+    )
     def _get_target_class():
         return AsyncBigtableMetricsInterceptor
 
@@ -170,7 +180,6 @@ async def test_unary_unary_interceptor_failure(self):
         exc.trailing_metadata = CrossSync.Mock(return_value=[("a", "b")])
         exc.initial_metadata = CrossSync.Mock(return_value=[("c", "d")])
         continuation = CrossSync.Mock(side_effect=exc)
-        call = continuation.return_value
         details = mock.Mock()
         details.metadata = [(OPERATION_INTERCEPTOR_METADATA_KEY, op.uuid)]
         request = mock.Mock()
@@ -235,7 +244,6 @@ async def test_unary_unary_interceptor_failure_generic(self):
         op.add_response_metadata.assert_not_called()
         op.end_attempt_with_status.assert_called_once_with(exc)
 
-
     @CrossSync.pytest
     async def test_unary_stream_interceptor_op_not_found(self):
         """Test that interceptor calls continuation if op is not found"""
@@ -446,4 +454,4 @@ async def test_unary_stream_interceptor_start_operation(self, initial_state):
         details.metadata = [(OPERATION_INTERCEPTOR_METADATA_KEY, op.uuid)]
         request = mock.Mock()
         await instance.intercept_unary_stream(continuation, details, request)
-        op.start_attempt.assert_called_once()
+        op.start_attempt.assert_called_once()

tests/unit/data/_sync_autogen/test_metrics_interceptor.py

@@ -152,7 +152,6 @@ def test_unary_unary_interceptor_failure(self):
         exc.trailing_metadata = CrossSync._Sync_Impl.Mock(return_value=[("a", "b")])
         exc.initial_metadata = CrossSync._Sync_Impl.Mock(return_value=[("c", "d")])
         continuation = CrossSync._Sync_Impl.Mock(side_effect=exc)
-        call = continuation.return_value
         details = mock.Mock()
         details.metadata = [(OPERATION_INTERCEPTOR_METADATA_KEY, op.uuid)]
         request = mock.Mock()