Don't run benchmark in test runner

5 years ago · 4ad50b8d2b
parent f09f59f1fd
commit 4ad50b8d2b
1 changed files with 6 additions and 2 deletions
--- a/src/python/grpcio_tests/tests/stress/unary_stream_benchmark.py
+++ b/src/python/grpcio_tests/tests/stress/unary_stream_benchmark.py
@ -6,8 +6,12 @@ import sys
 import time
 import contextlib

-from src.python.grpcio_tests.tests.stress import unary_stream_benchmark_pb2
-from src.python.grpcio_tests.tests.stress import unary_stream_benchmark_pb2_grpc
+try:
+    from src.python.grpcio_tests.tests.stress import unary_stream_benchmark_pb2
+    from src.python.grpcio_tests.tests.stress import unary_stream_benchmark_pb2_grpc
+except ImportError:
+    sys.stderr.write("This test must be run from Bazel.\n")
+    sys.exit(1)

 _PORT = 5741
 _MESSAGE_SIZE = 4