Fix formatting

This commit is contained in:
Danny Weinberg 2020-06-04 11:34:20 -07:00
parent 061bf86a9c
commit 67422db6b9

View File

@ -13,19 +13,14 @@ from betterproto.tests.output_betterproto.repeated.repeated import (
class ExampleService: class ExampleService:
async def DoThing( async def DoThing(self, stream: "grpclib.server.Stream[Test, Test]"):
self, stream: "grpclib.server.Stream[Test, Test]"
):
request = await stream.recv_message() request = await stream.recv_message()
await stream.send_message(request) await stream.send_message(request)
def __mapping__(self) -> Dict[str, grpclib.const.Handler]: def __mapping__(self) -> Dict[str, grpclib.const.Handler]:
return { return {
"/repeated.ExampleService/DoThing": grpclib.const.Handler( "/repeated.ExampleService/DoThing": grpclib.const.Handler(
self.DoThing, self.DoThing, grpclib.const.Cardinality.UNARY_UNARY, Test, Test,
grpclib.const.Cardinality.UNARY_UNARY,
Test,
Test,
), ),
} }
@ -34,6 +29,6 @@ class ExampleService:
async def test_sets_serialized_on_wire() -> None: async def test_sets_serialized_on_wire() -> None:
async with ChannelFor([ExampleService()]) as channel: async with ChannelFor([ExampleService()]) as channel:
stub = ExampleServiceStub(channel) stub = ExampleServiceStub(channel)
response = await stub.do_thing(names=['a', 'b', 'c']) response = await stub.do_thing(names=["a", "b", "c"])
assert betterproto.serialized_on_wire(response) assert betterproto.serialized_on_wire(response)
assert response.names == ['a', 'b', 'c'] assert response.names == ["a", "b", "c"]