From 52c21188d8880c5e947027efb682e2d1b63848ad Mon Sep 17 00:00:00 2001
From: Ehsan Afzali <eafzali@gmail.com>
Date: Tue, 31 Mar 2020 20:24:32 +0300
Subject: [PATCH] add e2e test for preloader in server send stream

---
 test/end2end_test.go | 68 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 68 insertions(+)

diff --git a/test/end2end_test.go b/test/end2end_test.go
index eb91d09afdf..aba13a54ae5 100644
--- a/test/end2end_test.go
+++ b/test/end2end_test.go
@@ -2248,6 +2248,74 @@ func testPreloaderClientSend(t *testing.T, e env) {
 	}
 }
 
+// preparedMsgSendServer is a TestServiceServer whose
+// StreamingOutputCall makes a SendMsg calls using PreparedMsg,
+// sending prepared messaged with payload "0".
+// TestPreloaderSenderSend verifies it is being sent correctly.
+//
+// All other TestServiceServer methods crash if called.
+type preparedMsgSendServer struct {
+	testpb.TestServiceServer
+}
+
+func (s preparedMsgSendServer) StreamingOutputCall(args *testpb.StreamingOutputCallRequest, stream testpb.TestService_StreamingOutputCallServer) error {
+	preparedMsg := &grpc.PreparedMsg{}
+	err := preparedMsg.Encode(stream, &testpb.StreamingOutputCallResponse{
+		Payload: &testpb.Payload{
+			Body: []byte{'0'},
+		},
+	})
+	if err != nil {
+		return err
+	}
+	stream.SendMsg(preparedMsg)
+	return nil
+}
+
+func (s) TestPreloaderSenderSend(t *testing.T) {
+	for _, e := range listTestEnv() {
+		testPreloaderSenderSend(t, e)
+	}
+}
+
+func testPreloaderSenderSend(t *testing.T, e env) {
+	te := newTest(t, e)
+	te.startServer(preparedMsgSendServer{})
+	defer te.tearDown()
+
+	cc := te.clientConn()
+	tc := testpb.NewTestServiceClient(cc)
+
+	req := &testpb.StreamingOutputCallRequest{}
+	stream, err := tc.StreamingOutputCall(context.Background(), req)
+	if err != nil {
+		t.Errorf("%v.StreamingOutputCall(_) = _, %v, want <nil>", tc, err)
+		return
+	}
+	var ngot int
+	var buf bytes.Buffer
+	for {
+		reply, err := stream.Recv()
+		if err == io.EOF {
+			break
+		}
+		if err != nil {
+			t.Fatal(err)
+		}
+		ngot++
+		if buf.Len() > 0 {
+			buf.WriteByte(',')
+		}
+		buf.Write(reply.GetPayload().GetBody())
+	}
+	if want := 1; ngot != want {
+		t.Errorf("Got %d replies, want %d", ngot, want)
+	}
+	if got, want := buf.String(), "0"; got != want {
+		t.Errorf("Got replies %q; want %q", got, want)
+	}
+}
+
 func (s) TestMaxMsgSizeClientDefault(t *testing.T) {
 	for _, e := range listTestEnv() {
 		testMaxMsgSizeClientDefault(t, e)