[Python AIO] Change default code() return to StatusCode.OK instead of…

… None
XuanWang-Amos · Jan 4, 2024 · 7e86c31 · 7e86c31
1 parent b84d5ef
commit 7e86c31
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/python/grpcio/grpc/_cython/_cygrpc/aio/server.pyx.pxi b/src/python/grpcio/grpc/_cython/_cygrpc/aio/server.pyx.pxi
@@ -17,6 +17,7 @@ import inspect
 import traceback
 import functools
 
+import grpc
 
 cdef int _EMPTY_FLAG = 0
 cdef str _RPC_FINISHED_DETAILS = 'RPC already finished.'
@@ -55,7 +56,7 @@ cdef class RPCState:
         self.metadata_sent = False
         self.status_sent = False
         self.status_code = StatusCode.ok
-        self.py_status_code = None
+        self.py_status_code = grpc.StatusCode.OK
         self.status_details = ''
         self.trailing_metadata = _IMMUTABLE_EMPTY_METADATA
         self.compression_algorithm = None

diff --git a/src/python/grpcio_tests/tests_aio/unit/server_test.py b/src/python/grpcio_tests/tests_aio/unit/server_test.py
@@ -108,7 +108,8 @@ def __init__(self):
         }
 
     @staticmethod
-    async def _unary_unary(unused_request, unused_context):
+    async def _unary_unary(unused_request, context):
+        assert context.code() == grpc.StatusCode.OK
         return _RESPONSE
 
     async def _block_forever(self, unused_request, unused_context):