Skip to content

Commit

Permalink
Address linting issues
Browse files Browse the repository at this point in the history
  • Loading branch information
cookiefission committed Nov 21, 2023
1 parent dda47a1 commit cce9836
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 11 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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())
Expand Down Expand Up @@ -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
Expand All @@ -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())
Expand Down

0 comments on commit cce9836

Please sign in to comment.