Run ruff

Author: sentrivana

instrumentation/opentelemetry-instrumentation-grpc/tests/test_aio_server_interceptor.py

@@ -153,7 +153,9 @@ async def request(channel):
             msg = request.SerializeToString()
             return await channel.unary_unary(rpc_call)(msg)
 
-        await run_with_test_server(request, interceptors=[aio_server_interceptor()])
+        await run_with_test_server(
+            request, interceptors=[aio_server_interceptor()]
+        )
 
         spans_list = self.memory_exporter.get_finished_spans()
         self.assertEqual(len(spans_list), 1)
@@ -205,7 +207,11 @@ async def request(channel):
             msg = request.SerializeToString()
             return await channel.unary_unary(rpc_call)(msg)
 
-        await run_with_test_server(request, servicer=TwoSpanServicer(), interceptors=[aio_server_interceptor()])
+        await run_with_test_server(
+            request,
+            servicer=TwoSpanServicer(),
+            interceptors=[aio_server_interceptor()],
+        )
 
         spans_list = self.memory_exporter.get_finished_spans()
         self.assertEqual(len(spans_list), 2)
@@ -252,7 +258,9 @@ async def request(channel):
             async for response in channel.unary_stream(rpc_call)(msg):
                 print(response)
 
-        await run_with_test_server(request, interceptors=[aio_server_interceptor()])
+        await run_with_test_server(
+            request, interceptors=[aio_server_interceptor()]
+        )
 
         spans_list = self.memory_exporter.get_finished_spans()
         self.assertEqual(len(spans_list), 1)
@@ -306,7 +314,11 @@ async def request(channel):
             async for response in channel.unary_stream(rpc_call)(msg):
                 print(response)
 
-        await run_with_test_server(request, servicer=TwoSpanServicer(), interceptors=[aio_server_interceptor()])
+        await run_with_test_server(
+            request,
+            servicer=TwoSpanServicer(),
+            interceptors=[aio_server_interceptor()],
+        )
 
         spans_list = self.memory_exporter.get_finished_spans()
         self.assertEqual(len(spans_list), 2)
@@ -366,7 +378,11 @@ async def request(channel):
         lifetime_servicer = SpanLifetimeServicer()
         active_span_before_call = trace.get_current_span()
 
-        await run_with_test_server(request, servicer=lifetime_servicer, interceptors=[aio_server_interceptor()])
+        await run_with_test_server(
+            request,
+            servicer=lifetime_servicer,
+            interceptors=[aio_server_interceptor()],
+        )
 
         active_span_in_handler = lifetime_servicer.span
         active_span_after_call = trace.get_current_span()
@@ -389,7 +405,9 @@ async def sequential_requests(channel):
             await request(channel)
             await request(channel)
 
-        await run_with_test_server(sequential_requests, interceptors=[aio_server_interceptor()])
+        await run_with_test_server(
+            sequential_requests, interceptors=[aio_server_interceptor()]
+        )
 
         spans_list = self.memory_exporter.get_finished_spans()
         self.assertEqual(len(spans_list), 2)
@@ -449,7 +467,9 @@ async def concurrent_requests(channel):
             await asyncio.gather(request(channel), request(channel))
 
         await run_with_test_server(
-            concurrent_requests, servicer=LatchedServicer(), interceptors=[aio_server_interceptor()]
+            concurrent_requests,
+            servicer=LatchedServicer(),
+            interceptors=[aio_server_interceptor()],
         )
 
         spans_list = self.memory_exporter.get_finished_spans()
@@ -503,7 +523,11 @@ async def request(channel):
             self.assertEqual(cm.exception.code(), grpc.StatusCode.INTERNAL)
             self.assertEqual(cm.exception.details(), failure_message)
 
-        await run_with_test_server(request, servicer=AbortServicer(), interceptors=[aio_server_interceptor()])
+        await run_with_test_server(
+            request,
+            servicer=AbortServicer(),
+            interceptors=[aio_server_interceptor()],
+        )
 
         spans_list = self.memory_exporter.get_finished_spans()
         self.assertEqual(len(spans_list), 1)
@@ -568,7 +592,11 @@ async def request(channel):
             )
             self.assertEqual(cm.exception.details(), failure_message)
 
-        await run_with_test_server(request, servicer=AbortServicer(), interceptors=[aio_server_interceptor()])
+        await run_with_test_server(
+            request,
+            servicer=AbortServicer(),
+            interceptors=[aio_server_interceptor()],
+        )
 
         spans_list = self.memory_exporter.get_finished_spans()
         self.assertEqual(len(spans_list), 1)
@@ -616,10 +644,14 @@ async def request(channel):
                 return await channel.unary_unary(rpc_call)(msg)
 
             class MockInterceptor(grpc.aio.ServerInterceptor):
-                async def intercept_service(self, continuation, handler_call_details):
+                async def intercept_service(
+                    self, continuation, handler_call_details
+                ):
                     return await continuation(handler_call_details)
 
-            await run_with_test_server(request, interceptors=(MockInterceptor(),))
+            await run_with_test_server(
+                request, interceptors=(MockInterceptor(),)
+            )
 
         finally:
             grpc_server_instrumentor.uninstrument()
@@ -651,6 +683,7 @@ async def intercept_service(self, continuation, handler_call_details):
             },
         )
 
+
 def get_latch(num):
     """Get a countdown latch function for use in n threads."""
     cv = asyncio.Condition()

instrumentation/opentelemetry-instrumentation-grpc/tests/test_server_interceptor.py

@@ -108,7 +108,9 @@ def handler(request, context):
 
         try:
             with self.server(max_workers=1) as (server, channel):
-                server.add_generic_rpc_handlers((UnaryUnaryRpcHandler(handler),))
+                server.add_generic_rpc_handlers(
+                    (UnaryUnaryRpcHandler(handler),)
+                )
                 rpc_call = "TestServicer/handler"
                 try:
                     server.start()