grpclib_client: handle trailer-only responses (#127)

Resolves: #123
2020-07-25 19:57:46 +02:00
parent c513853301
commit 3d8c0cb713
2 changed files with 52 additions and 4 deletions
--- a/src/betterproto/grpc/grpclib_client.py
+++ b/src/betterproto/grpc/grpclib_client.py
@@ -76,8 +76,8 @@ class ServiceStub(ABC):
        ) as stream:
            await stream.send_message(request, end=True)
            response = await stream.recv_message()
-            assert response is not None
-            return response
+        assert response is not None
+        return response

    async def _unary_stream(
        self,
@@ -122,8 +122,8 @@ class ServiceStub(ABC):
        ) as stream:
            await self._send_messages(stream, request_iterator)
            response = await stream.recv_message()
-            assert response is not None
-            return response
+        assert response is not None
+        return response

    async def _stream_stream(
        self,