Storage: Move benchwrapper directory into 'tests/perf/'. (#9246)

storage/tests/perf/README.md

@@ -0,0 +1,21 @@
+# storage benchwrapp
+
+main.py is a gRPC wrapper around the storage library for benchmarking purposes.
+
+## Running
+
+```bash
+$ export STORAGE_EMULATOR_HOST=http://localhost:8080
+$ pip install grpcio
+$ cd storage
+$ pip install -e . # install google.cloud.storage locally
+$ cd tests/perf
+$ python3 benchwrapper.py --port 8081
+```
+
+## Re-generating protos
+
+```bash
+$ pip install grpcio-tools
+$ python -m grpc_tools.protoc -I. --python_out=. --grpc_python_out=. *.proto
+```

storage/test_utils/benchwrapper.py → storage/tests/perf/benchwrapper.py

@@ -12,18 +12,21 @@
 
 parser = argparse.ArgumentParser()
 
-if os.environ.get('STORAGE_EMULATOR_HOST') == None:
-    sys.exit('This benchmarking server only works when connected to an emulator. Please set STORAGE_EMULATOR_HOST.')
+if os.environ.get("STORAGE_EMULATOR_HOST") is None:
+    sys.exit(
+        "This benchmarking server only works when connected to an emulator. Please set STORAGE_EMULATOR_HOST."
+    )
 
-parser.add_argument('--port', help='The port to run on.')
+parser.add_argument("--port", help="The port to run on.")
 
 args = parser.parse_args()
 
-if args.port == None:
-    sys.exit('Usage: python3 main.py --port 8081')
+if args.port is None:
+    sys.exit("Usage: python3 main.py --port 8081")
 
 client = storage.Client()
 
+
 class StorageBenchWrapperServicer(storage_pb2_grpc.StorageBenchWrapperServicer):
     def Write(self, request, context):
         # TODO(deklerk): implement this
@@ -35,14 +38,17 @@ def Read(self, request, context):
         blob.download_as_string()
         return storage_pb2.EmptyResponse()
 
+
 server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
-storage_pb2_grpc.add_StorageBenchWrapperServicer_to_server(StorageBenchWrapperServicer(), server)
+storage_pb2_grpc.add_StorageBenchWrapperServicer_to_server(
+    StorageBenchWrapperServicer(), server
+)
 
-print('listening on localhost:'+args.port)
-server.add_insecure_port('[::]:'+args.port)
+print("listening on localhost:" + args.port)
+server.add_insecure_port("[::]:" + args.port)
 server.start()
 try:
     while True:
         time.sleep(_ONE_DAY_IN_SECONDS)
 except KeyboardInterrupt:
-    server.stop(0)
+    server.stop(0)