From 4318b7e1c4c6c296dca763a18e124881df4ed5e2 Mon Sep 17 00:00:00 2001 From: Xuan Wang Date: Mon, 29 Apr 2024 12:23:07 -0700 Subject: [PATCH] [Python tests] Allow passing kwargs to grpc.server in test_server (#36455) Allow passing `kwargs` to `grpc.server` in test server. Closes #36455 COPYBARA_INTEGRATE_REVIEW=https://github.com/grpc/grpc/pull/36455 from XuanWang-Amos:disable_reflection_in_tests d9a7e3d7fa26d34fda512eaf63aa8c28250bda19 PiperOrigin-RevId: 629149234 --- src/python/grpcio_tests/tests/unit/test_common.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/python/grpcio_tests/tests/unit/test_common.py b/src/python/grpcio_tests/tests/unit/test_common.py index 97dcbf59101..ff3ade1c284 100644 --- a/src/python/grpcio_tests/tests/unit/test_common.py +++ b/src/python/grpcio_tests/tests/unit/test_common.py @@ -13,8 +13,10 @@ # limitations under the License. """Common code used throughout tests of gRPC.""" +import ast import collections from concurrent import futures +import os import threading import grpc @@ -110,9 +112,12 @@ def test_server(max_workers=10, reuse_port=False): These servers have SO_REUSEPORT disabled to prevent cross-talk. """ + server_kwargs = os.environ.get("GRPC_ADDITIONAL_SERVER_KWARGS", "{}") + server_kwargs = ast.literal_eval(server_kwargs) return grpc.server( futures.ThreadPoolExecutor(max_workers=max_workers), options=(("grpc.so_reuseport", int(reuse_port)),), + **server_kwargs, )