From 01398d3b672628f4e7aa3bc26b8d4e5d44c9b1a0 Mon Sep 17 00:00:00 2001 From: Menghan Li Date: Mon, 12 Feb 2018 13:42:22 -0800 Subject: [PATCH] interop test: Expect io.EOF from stream.Send() --- interop/test_utils.go | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/interop/test_utils.go b/interop/test_utils.go index 7f288c1b7cc9..c3cd3fc8ebb2 100644 --- a/interop/test_utils.go +++ b/interop/test_utils.go @@ -241,10 +241,8 @@ func DoTimeoutOnSleepingServer(tc testpb.TestServiceClient, args ...grpc.CallOpt ResponseType: testpb.PayloadType_COMPRESSABLE, Payload: pl, } - if err := stream.Send(req); err != nil { - if status.Code(err) != codes.DeadlineExceeded { - grpclog.Fatalf("%v.Send(_) = %v", stream, err) - } + if err := stream.Send(req); err != nil && err != io.EOF { + grpclog.Fatalf("%v.Send(_) = %v", stream, err) } if _, err := stream.Recv(); status.Code(err) != codes.DeadlineExceeded { grpclog.Fatalf("%v.Recv() = _, %v, want error code %d", stream, err, codes.DeadlineExceeded)