commit 653803e3109c38fa54a21ee7e93cdf4b42482bd1
author vjpai <vpai@google.com> Mon Mar 23 10:19:12 2015 -0700
committer vjpai <vpai@google.com> Mon Mar 23 10:19:12 2015 -0700
tree db66b5f02bdabf67d52ed2bee5892f1bfe458a4b
parent 46f65239cf5bb55c106558196b9f293188746ef7

Make RPC type configurable

test/cpp/qps/qps_driver.cc

diff --git a/test/cpp/qps/qps_driver.cc b/test/cpp/qps/qps_driver.cc
index bf51e74..764fe13 100644
--- a/test/cpp/qps/qps_driver.cc
+++ b/test/cpp/qps/qps_driver.cc
@@ -42,6 +42,8 @@
 
 // Common config
 DEFINE_bool(enable_ssl, false, "Use SSL");
+DEFINE_string(rpc_type, "UNARY_TEST",
+	      "Type of RPC: UNARY_TEST or STREAMING_TEST");
 
 // Server config
 DEFINE_int32(server_threads, 1, "Number of server threads");
@@ -59,6 +61,7 @@
 using grpc::testing::ServerConfig;
 using grpc::testing::ClientType;
 using grpc::testing::ServerType;
+using grpc::testing::RpcType;
 using grpc::testing::ResourceUsage;
 using grpc::testing::sum;
 
@@ -73,6 +76,9 @@
   grpc_init();
   ParseCommandLineFlags(&argc, &argv, true);
 
+  RpcType rpc_type;
+  RpcType_Parse(FLAGS_rpc_type, &rpc_type);
+
   ClientType client_type;
   ServerType server_type;
   GPR_ASSERT(ClientType_Parse(FLAGS_client_type, &client_type));
@@ -86,6 +92,7 @@
   client_config.set_client_channels(FLAGS_client_channels);
   client_config.set_payload_size(FLAGS_payload_size);
   client_config.set_async_client_threads(FLAGS_async_client_threads);
+  client_config.set_rpc_type(rpc_type);
 
   ServerConfig server_config;
   server_config.set_server_type(server_type);