Method TestClientStreaming_ClientCallRecvMsgTwice

test/end2end_test.go:3911–3947

Tests the behavior for client-streaming RPC when client calls RecvMsg() twice. Second call to RecvMsg should fail with io.EOF.

(t *testing.T)

// Tests the behavior for client-streaming RPC when client calls RecvMsg() twice.
// Second call to RecvMsg should fail with io.EOF.
func (s) TestClientStreaming_ClientCallRecvMsgTwice(t *testing.T) {
	ss := stubserver.StubServer{
		StreamingInputCallF: func(stream testgrpc.TestService_StreamingInputCallServer) error {
			if err := stream.RecvMsg(&testpb.StreamingInputCallRequest{}); err != nil {
				t.Errorf("stream.RecvMsg() = %v, want <nil>", err)
			}
			if err := stream.SendAndClose(&testpb.StreamingInputCallResponse{}); err != nil {
				t.Errorf("stream.SendAndClose(_) = %v, want <nil>", err)
			}
			return nil
		},
	}
	if err := ss.Start(nil); err != nil {
		t.Fatal("Error starting server:", err)
	}
	defer ss.Stop()

	ctx, cancel := context.WithTimeout(context.Background(), defaultTestTimeout)
	defer cancel()
	stream, err := ss.Client.StreamingInputCall(ctx)
	if err != nil {
		t.Fatalf(".StreamingInputCall(_) = _, %v, want <nil>", err)
	}
	if err := stream.Send(&testpb.StreamingInputCallRequest{}); err != nil {
		t.Fatalf("stream.Send(_) = %v, want <nil>", err)
	}
	if err := stream.CloseSend(); err != nil {
		t.Fatalf("stream.CloseSend() = %v, want <nil>", err)
	}
	resp := new(testpb.StreamingInputCallResponse)
	if err := stream.RecvMsg(resp); err != nil {
		t.Fatalf("stream.RecvMsg() = %v , want <nil>", err)
	}
	if err = stream.RecvMsg(resp); err != io.EOF {
		t.Errorf("stream.RecvMsg() = %v, want error %v", err, io.EOF)
	}
}

// Tests the behavior for server-side streaming when client calls SendMsg twice.
// Second call to SendMsg should fail with Internal error and result in closing