diff --git a/instrumentation/opentelemetry-instrumentation-grpc/src/opentelemetry/instrumentation/grpc/_aio_server.py b/instrumentation/opentelemetry-instrumentation-grpc/src/opentelemetry/instrumentation/grpc/_aio_server.py index fe4355d19e..4d43049c78 100644 --- a/instrumentation/opentelemetry-instrumentation-grpc/src/opentelemetry/instrumentation/grpc/_aio_server.py +++ b/instrumentation/opentelemetry-instrumentation-grpc/src/opentelemetry/instrumentation/grpc/_aio_server.py @@ -16,14 +16,13 @@ import grpc.aio import wrapt -from ._server import ( - OpenTelemetryServerInterceptor, - _wrap_rpc_behavior, -) - from opentelemetry.semconv.trace import SpanAttributes from opentelemetry.trace.status import Status, StatusCode +from ._server import OpenTelemetryServerInterceptor, _wrap_rpc_behavior + + +# pylint:disable=abstract-method class _OpenTelemetryAioServicerContext(wrapt.ObjectProxy): def __init__(self, servicer_context, active_span): super().__init__(servicer_context) diff --git a/instrumentation/opentelemetry-instrumentation-grpc/tests/test_aio_server_interceptor.py b/instrumentation/opentelemetry-instrumentation-grpc/tests/test_aio_server_interceptor.py index 09bab13e3b..242295c08c 100644 --- a/instrumentation/opentelemetry-instrumentation-grpc/tests/test_aio_server_interceptor.py +++ b/instrumentation/opentelemetry-instrumentation-grpc/tests/test_aio_server_interceptor.py @@ -520,7 +520,9 @@ async def request(channel): with testcase.assertRaises(grpc.RpcError) as cm: await channel.unary_unary(rpc_call)(msg) - self.assertEqual(cm.exception.code(), grpc.StatusCode.FAILED_PRECONDITION) + self.assertEqual( + cm.exception.code(), grpc.StatusCode.FAILED_PRECONDITION + ) self.assertEqual(cm.exception.details(), failure_message) await run_with_test_server(request, servicer=AbortServicer()) @@ -567,11 +569,11 @@ async def test_abort_with_trailing_metadata(self): class AbortServicer(GRPCTestServerServicer): # pylint:disable=C0103 async def SimpleMethod(self, request, context): - metadata = ( - ("meta", "data"), - ) + metadata = (("meta", "data"),) await context.abort( - grpc.StatusCode.FAILED_PRECONDITION, failure_message, trailing_metadata=metadata + grpc.StatusCode.FAILED_PRECONDITION, + failure_message, + trailing_metadata=metadata, ) testcase = self @@ -583,7 +585,9 @@ async def request(channel): with testcase.assertRaises(grpc.RpcError) as cm: await channel.unary_unary(rpc_call)(msg) - self.assertEqual(cm.exception.code(), grpc.StatusCode.FAILED_PRECONDITION) + self.assertEqual( + cm.exception.code(), grpc.StatusCode.FAILED_PRECONDITION + ) self.assertEqual(cm.exception.details(), failure_message) await run_with_test_server(request, servicer=AbortServicer())