commit 6bb76488de15d050b16b0227a9ed28e3b9e07a97
author ZhouyihaiDing <ddyihai@google.com> Wed May 31 14:15:37 2017 -0700
committer Menghan Li <menghanl@google.com> Wed May 31 14:15:37 2017 -0700
tree 5157252545beae72e3db5c9f7d05e2d7ef4f890c
parent bb3728664d677cfa986024a99dca408c7daa67fc

add MaxConcurrentStreams to benchmark_test when start the server (#1271)

benchmark/benchmark_test.go

diff --git a/benchmark/benchmark_test.go b/benchmark/benchmark_test.go
index b0c3030..61c9fba 100644
--- a/benchmark/benchmark_test.go
+++ b/benchmark/benchmark_test.go
@@ -16,7 +16,7 @@
 func runUnary(b *testing.B, maxConcurrentCalls int) {
 	s := stats.AddStats(b, 38)
 	b.StopTimer()
-	target, stopper := StartServer(ServerInfo{Addr: "localhost:0", Type: "protobuf"})
+	target, stopper := StartServer(ServerInfo{Addr: "localhost:0", Type: "protobuf"}, grpc.MaxConcurrentStreams(uint32(maxConcurrentCalls+1)))
 	defer stopper()
 	conn := NewClientConn(target, grpc.WithInsecure())
 	tc := testpb.NewBenchmarkServiceClient(conn)
@@ -59,7 +59,7 @@
 func runStream(b *testing.B, maxConcurrentCalls int) {
 	s := stats.AddStats(b, 38)
 	b.StopTimer()
-	target, stopper := StartServer(ServerInfo{Addr: "localhost:0", Type: "protobuf"})
+	target, stopper := StartServer(ServerInfo{Addr: "localhost:0", Type: "protobuf"}, grpc.MaxConcurrentStreams(uint32(maxConcurrentCalls+1)))
 	defer stopper()
 	conn := NewClientConn(target, grpc.WithInsecure())
 	tc := testpb.NewBenchmarkServiceClient(conn)