From deebcfea01a5308b47833711c86a7d0e99becf33 Mon Sep 17 00:00:00 2001 From: kpayson64 Date: Tue, 29 May 2018 14:21:21 -0700 Subject: Use the correct proto for benchmark_serivce --- src/python/grpcio_tests/tests/qps/worker_server.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'src/python/grpcio_tests/tests') diff --git a/src/python/grpcio_tests/tests/qps/worker_server.py b/src/python/grpcio_tests/tests/qps/worker_server.py index db145fbf64..740bdcf1eb 100644 --- a/src/python/grpcio_tests/tests/qps/worker_server.py +++ b/src/python/grpcio_tests/tests/qps/worker_server.py @@ -20,6 +20,7 @@ import time from concurrent import futures import grpc from src.proto.grpc.testing import control_pb2 +from src.proto.grpc.testing import benchmark_service_pb2_grpc from src.proto.grpc.testing import worker_service_pb2_grpc from src.proto.grpc.testing import stats_pb2 @@ -72,7 +73,7 @@ class WorkerServer(worker_service_pb2_grpc.WorkerServiceServicer): server = test_common.test_server(max_workers=server_threads) if config.server_type == control_pb2.ASYNC_SERVER: servicer = benchmark_server.BenchmarkServer() - worker_service_pb2_grpc.add_BenchmarkServiceServicer_to_server( + benchmark_service_pb2_grpc.add_BenchmarkServiceServicer_to_server( servicer, server) elif config.server_type == control_pb2.ASYNC_GENERIC_SERVER: resp_size = config.payload_config.bytebuf_params.resp_size -- cgit v1.2.3