Fix grpc aio handler typing

Author: XiaXia009

stubs/grpcio/@tests/test_cases/check_server_interceptor.py

@@ -1,8 +1,8 @@
 from __future__ import annotations
 
-from collections.abc import Callable
+from collections.abc import AsyncIterator, Callable
 from concurrent.futures.thread import ThreadPoolExecutor
-from typing import Awaitable, TypeVar
+from typing import Awaitable, TypeVar, cast
 
 import grpc
 import grpc.aio
@@ -26,10 +26,28 @@ def intercept_service(
 class NoopAioInterceptor(grpc.aio.ServerInterceptor):
     async def intercept_service(
         self,
-        continuation: Callable[[grpc.HandlerCallDetails], Awaitable[grpc.RpcMethodHandler[RequestT, ResponseT]]],
+        continuation: Callable[[grpc.HandlerCallDetails], Awaitable[grpc.aio.RpcMethodHandler[RequestT, ResponseT]]],
         handler_call_details: grpc.HandlerCallDetails,
-    ) -> grpc.RpcMethodHandler[RequestT, ResponseT]:
+    ) -> grpc.aio.RpcMethodHandler[RequestT, ResponseT]:
         return await continuation(handler_call_details)
 
 
 grpc.aio.server(interceptors=[NoopAioInterceptor()])
+
+aio_handler = grpc.aio.RpcMethodHandler[RequestT, ResponseT]()
+aio_handler.unary_unary = cast(
+    Callable[[RequestT, grpc.aio.ServicerContext[RequestT, ResponseT]], Awaitable[ResponseT]],
+    None,
+)
+aio_handler.unary_stream = cast(
+    Callable[[RequestT, grpc.aio.ServicerContext[RequestT, ResponseT]], AsyncIterator[ResponseT]],
+    None,
+)
+aio_handler.stream_unary = cast(
+    Callable[[AsyncIterator[RequestT], grpc.aio.ServicerContext[RequestT, ResponseT]], Awaitable[ResponseT]],
+    None,
+)
+aio_handler.stream_stream = cast(
+    Callable[[AsyncIterator[RequestT], grpc.aio.ServicerContext[RequestT, ResponseT]], AsyncIterator[ResponseT]],
+    None,
+)

stubs/grpcio/grpc/aio/__init__.pyi

@@ -15,7 +15,6 @@ from grpc import (
     GenericRpcHandler,
     HandlerCallDetails,
     RpcError,
-    RpcMethodHandler,
     ServerCredentials,
     StatusCode,
     _Options,
@@ -369,6 +368,22 @@ class StreamStreamClientInterceptor(ClientInterceptor, metaclass=abc.ABCMeta):
         request_iterator: AsyncIterable[_TRequest] | Iterable[_TRequest],
     ) -> AsyncIterator[_TResponse] | StreamStreamCall[_TRequest, _TResponse]: ...
 
+# Service-Side Handler:
+
+class RpcMethodHandler(Generic[_TRequest, _TResponse], metaclass=abc.ABCMeta):
+    request_streaming: bool
+    response_streaming: bool
+
+    request_deserializer: _Deserializer[_TRequest] | None
+    response_serializer: _Serializer[_TResponse] | None
+
+    unary_unary: Callable[[_TRequest, ServicerContext[_TRequest, _TResponse]], Awaitable[_TResponse]] | None
+    unary_stream: Callable[[_TRequest, ServicerContext[_TRequest, _TResponse]], AsyncIterator[_TResponse]] | None
+    stream_unary: Callable[[AsyncIterator[_TRequest], ServicerContext[_TRequest, _TResponse]], Awaitable[_TResponse]] | None
+    stream_stream: Callable[
+        [AsyncIterator[_TRequest], ServicerContext[_TRequest, _TResponse]], AsyncIterator[_TResponse]
+    ] | None
+
 # Server-Side Interceptor:
 
 class ServerInterceptor(metaclass=abc.ABCMeta):