Merge pull request #15566 from kpayson64/fix_qps_python

Use the correct proto for benchmark_service
pull/15569/head
kpayson64 7 years ago committed by GitHub
commit e4004eba96
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 3
      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

Loading…
Cancel
Save