aboutsummaryrefslogtreecommitdiffhomepage
path: root/src/python/grpcio_tests/tests/qps/worker_server.py
diff options
context:
space:
mode:
authorGravatar Craig Tiller <ctiller@google.com>2017-03-21 15:16:14 -0700
committerGravatar Craig Tiller <ctiller@google.com>2017-03-21 15:16:14 -0700
commit31041c9e1247e26d5b9acfa9231bbb4689071bda (patch)
treec02880f9c1db48ab3fa74c8c788a0475bd51a4cc /src/python/grpcio_tests/tests/qps/worker_server.py
parentb7f35a658b00ae57d0341261d0944c26066dda04 (diff)
parent9f615de5ed5c622bd7abe0baaed418fcc560ba9b (diff)
Merge github.com:grpc/grpc into call_cases
Diffstat (limited to 'src/python/grpcio_tests/tests/qps/worker_server.py')
-rw-r--r--src/python/grpcio_tests/tests/qps/worker_server.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/src/python/grpcio_tests/tests/qps/worker_server.py b/src/python/grpcio_tests/tests/qps/worker_server.py
index ca1a777611..de9535f46e 100644
--- a/src/python/grpcio_tests/tests/qps/worker_server.py
+++ b/src/python/grpcio_tests/tests/qps/worker_server.py
@@ -35,7 +35,7 @@ import time
from concurrent import futures
import grpc
from src.proto.grpc.testing import control_pb2
-from src.proto.grpc.testing import services_pb2
+from src.proto.grpc.testing import services_pb2_grpc
from src.proto.grpc.testing import stats_pb2
from tests.qps import benchmark_client
@@ -45,7 +45,7 @@ from tests.qps import histogram
from tests.unit import resources
-class WorkerServer(services_pb2.WorkerServiceServicer):
+class WorkerServer(services_pb2_grpc.WorkerServiceServicer):
"""Python Worker Server implementation."""
def __init__(self):
@@ -87,8 +87,8 @@ class WorkerServer(services_pb2.WorkerServiceServicer):
futures.ThreadPoolExecutor(max_workers=server_threads))
if config.server_type == control_pb2.ASYNC_SERVER:
servicer = benchmark_server.BenchmarkServer()
- services_pb2.add_BenchmarkServiceServicer_to_server(servicer,
- server)
+ services_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
servicer = benchmark_server.GenericBenchmarkServer(resp_size)