Merge pull request #1040 from menghanl/interop_change

call cancel on context in DoTimeoutOnSleepingServer
grpc · Jan 5, 2017 · 85497e2 · 85497e2
2 parents c810507 + ec1de5b
commit 85497e2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/interop/test_utils.go b/interop/test_utils.go
@@ -240,7 +240,8 @@ func DoEmptyStream(tc testpb.TestServiceClient, args ...grpc.CallOption) {
 
 // DoTimeoutOnSleepingServer performs an RPC on a sleep server which causes RPC timeout.
 func DoTimeoutOnSleepingServer(tc testpb.TestServiceClient, args ...grpc.CallOption) {
-	ctx, _ := context.WithTimeout(context.Background(), 1*time.Millisecond)
+	ctx, cancel := context.WithTimeout(context.Background(), 1*time.Millisecond)
+	defer cancel()
 	stream, err := tc.FullDuplexCall(ctx, args...)
 	if err != nil {
 		if grpc.Code(err) == codes.DeadlineExceeded {