Test sync stream unary and stream stream

Author: AdrienVannson

tests/test_sync_client.py

@@ -3,7 +3,6 @@
 from collections.abc import AsyncIterator
 
 import grpc
-import grpclib
 import pytest
 from grpclib.server import Server
 
@@ -24,10 +23,14 @@ async def get_unary_stream(self, message: "Request") -> "AsyncIterator[Response]
             yield Response(message=f"Hello {message.value} {i}")
 
     async def get_stream_unary(self, messages: "AsyncIterator[Request]") -> "Response":
-        raise grpclib.GRPCError(grpclib.const.Status.UNIMPLEMENTED)
+        s = 0
+        async for m in messages:
+            s += m.value
+        return Response(message=f"Hello {s}")
 
     async def get_stream_stream(self, messages: "AsyncIterator[Request]") -> "AsyncIterator[Response]":
-        raise grpclib.GRPCError(grpclib.const.Status.UNIMPLEMENTED)
+        async for message in messages:
+            yield Response(message=f"Hello {message.value}")
 
 
 @pytest.mark.asyncio
@@ -57,6 +60,12 @@ async def run_server():
         response = client.get_unary_stream(Request(value=42))
         assert [r.message for r in response] == [f"Hello 42 {i}" for i in range(5)]
 
+        response = client.get_stream_unary([Request(value=i) for i in range(5)])
+        assert response.message == "Hello 10"
+
+        response = client.get_stream_stream([Request(value=i) for i in range(5)])
+        assert [r.message for r in response] == [f"Hello {i}" for i in range(5)]
+
     # Create an async client
     # client = SimpleServiceStub(Channel(host="127.0.0.1", port=1234))
     # response = await client.get_unary_unary(Request(value=42))