Adding upstream version 1.2+20240521.

Signed-off-by: Daniel Baumann <daniel@debian.org>

tests/inputs/example_service/example_service.proto

@@ -0,0 +1,20 @@
+syntax = "proto3";
+
+package example_service;
+
+service Test {
+  rpc ExampleUnaryUnary(ExampleRequest) returns (ExampleResponse);
+  rpc ExampleUnaryStream(ExampleRequest) returns (stream ExampleResponse);
+  rpc ExampleStreamUnary(stream ExampleRequest) returns (ExampleResponse);
+  rpc ExampleStreamStream(stream ExampleRequest) returns (stream ExampleResponse);
+}
+
+message ExampleRequest {
+  string example_string = 1;
+  int64  example_integer = 2;
+}
+
+message ExampleResponse {
+  string example_string = 1;
+  int64  example_integer = 2;
+}

tests/inputs/example_service/test_example_service.py

@@ -0,0 +1,86 @@
+from typing import (
+    AsyncIterable,
+    AsyncIterator,
+)
+
+import pytest
+from grpclib.testing import ChannelFor
+
+from tests.output_aristaproto.example_service import (
+    ExampleRequest,
+    ExampleResponse,
+    TestBase,
+    TestStub,
+)
+
+
+class ExampleService(TestBase):
+    async def example_unary_unary(
+        self, example_request: ExampleRequest
+    ) -> "ExampleResponse":
+        return ExampleResponse(
+            example_string=example_request.example_string,
+            example_integer=example_request.example_integer,
+        )
+
+    async def example_unary_stream(
+        self, example_request: ExampleRequest
+    ) -> AsyncIterator["ExampleResponse"]:
+        response = ExampleResponse(
+            example_string=example_request.example_string,
+            example_integer=example_request.example_integer,
+        )
+        yield response
+        yield response
+        yield response
+
+    async def example_stream_unary(
+        self, example_request_iterator: AsyncIterator["ExampleRequest"]
+    ) -> "ExampleResponse":
+        async for example_request in example_request_iterator:
+            return ExampleResponse(
+                example_string=example_request.example_string,
+                example_integer=example_request.example_integer,
+            )
+
+    async def example_stream_stream(
+        self, example_request_iterator: AsyncIterator["ExampleRequest"]
+    ) -> AsyncIterator["ExampleResponse"]:
+        async for example_request in example_request_iterator:
+            yield ExampleResponse(
+                example_string=example_request.example_string,
+                example_integer=example_request.example_integer,
+            )
+
+
+@pytest.mark.asyncio
+async def test_calls_with_different_cardinalities():
+    example_request = ExampleRequest("test string", 42)
+
+    async with ChannelFor([ExampleService()]) as channel:
+        stub = TestStub(channel)
+
+        # unary unary
+        response = await stub.example_unary_unary(example_request)
+        assert response.example_string == example_request.example_string
+        assert response.example_integer == example_request.example_integer
+
+        # unary stream
+        async for response in stub.example_unary_stream(example_request):
+            assert response.example_string == example_request.example_string
+            assert response.example_integer == example_request.example_integer
+
+        # stream unary
+        async def request_iterator():
+            yield example_request
+            yield example_request
+            yield example_request
+
+        response = await stub.example_stream_unary(request_iterator())
+        assert response.example_string == example_request.example_string
+        assert response.example_integer == example_request.example_integer
+
+        # stream stream
+        async for response in stub.example_stream_stream(request_iterator()):
+            assert response.example_string == example_request.example_string
+            assert response.example_integer == example_request.example_integer