Mock a stream without implying an `io.EOF` return

[seanlafferty-ibm]

I'm mocking a server with MockServerWithBufConn, and I'm trying to mock a grpc client stream like so

server.ExpectClientStream(myMethodName).Once().Return(myResponse)

however when my app code calls Send() on the stream, it gets back an io.EOF error. Looking through the source code I think this is happening because the underlying bufconn only has one message in the queue, and once there's none left it sends an io.EOF. Im looking for a way to return a nil error instead, as I want to replicate what happens with a real server with a long-lived connection.

Said simply, I looking to to mock a stream response, without also implying the server is terminating the connection.

I tried using Run() instead like

server.ExpectClientStream(myMethodName).Once().Run(func(ctx context.Context, s grpc.ServerStream) (any, error) {
  sends := 1
  for sends <= 2 {
    sends++
    // s.Recv()
    // s.Send()
  }
})

which fixes the app seeing an io.EOF, but ultimately causes other issues since Im only calling Send() once in the application code, not twice.

Does this make sense, or do I have a misunderstanding in how I should be stringing these mocks together?

[nhatthm]

Hey I'm away until next Thursday 13th Nov so I can't look into this atm. Could you please tell me what version of grpcmock and grpc you are using? And can you write a small logic for reproducing the issue?

[seanlafferty-ibm]

Hey sure, thing. The deps I'm using

	go.nhat.io/grpcmock v0.33.0
	google.golang.org/grpc v1.76.0
	google.golang.org/protobuf v1.36.10
	gotest.tools/v3 v3.5.2

Repro

import (
	"testing"

	"go.nhat.io/grpcmock"
	"go.nhat.io/grpcmock/planner"
	testpb "go.nhat.io/grpcmock/test/grpctest"
	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials/insecure"
	"gotest.tools/v3/assert"
)

func TestEOF(t *testing.T) {
	mockServer, mockDialer := grpcmock.MockServerWithBufConn(
		grpcmock.WithPlanner(planner.FirstMatch()),
		grpcmock.RegisterService(testpb.RegisterItemServiceServer),
	)(t)
	conn, err := grpc.NewClient("passthrough:mockURL", grpc.WithContextDialer(mockDialer), grpc.WithTransportCredentials(insecure.NewCredentials()))
	t.Cleanup(func() {
		err := conn.Close()
		if err != nil {
			t.Fatal(err)
		}
	})
	assert.NilError(t, err)
	client := testpb.NewItemServiceClient(conn)

	stream, err := client.CreateItems(t.Context())
	assert.NilError(t, err)

	mockServer.ExpectClientStream(testpb.ItemService_CreateItems_FullMethodName).Once().Return(&testpb.CreateItemsResponse{})
	err = stream.Send(&testpb.Item{Id: 52})
	assert.NilError(t, err) // <--- Fails, gets EOF
}

Again, Im not entirely confident I understand how the stream mock is supposed to behave, but my thinking is the mock server shouldn't return an io.EOF (end of stream) until I ask it to 🙂

(In my app, Im trying to write a test that covers a few different scenarios- server closing, server not closing, server erroring, Send()s kicking off at certain intervals, etc)