Method TestDecompressionExceedsMaxMessageSize

encoding/compressor_test.go:215–239

TestDecompressionExceedsMaxMessageSize uses a fake compressor that produces messages of size 100 bytes on decompression. A server is started with the max receive message size restricted to 99 bytes. The test verifies that the client receives a ResourceExhausted response from the server.

(t *testing.T)

// max receive message size restricted to 99 bytes. The test verifies that the
// client receives a ResourceExhausted response from the server.
func (s) TestDecompressionExceedsMaxMessageSize(t *testing.T) {
	const messageLen = 100
	regFn := internal.RegisterCompressorForTesting.(func(encoding.Compressor) func())
	compressor := &fakeCompressor{decompressedMessageSize: messageLen}
	unreg := regFn(compressor)
	defer unreg()
	ss := &stubserver.StubServer{
		UnaryCallF: func(context.Context, *testpb.SimpleRequest) (*testpb.SimpleResponse, error) {
			return &testpb.SimpleResponse{}, nil
		},
	}
	if err := ss.Start([]grpc.ServerOption{grpc.MaxRecvMsgSize(messageLen - 1)}); err != nil {
		t.Fatalf("Error starting endpoint server: %v", err)
	}
	defer ss.Stop()

	ctx, cancel := context.WithTimeout(context.Background(), defaultTestTimeout)
	defer cancel()

	req := &testpb.SimpleRequest{Payload: &testpb.Payload{}}
	_, err := ss.Client.UnaryCall(ctx, req, grpc.UseCompressor(compressor.Name()))
	if got, want := status.Code(err), codes.ResourceExhausted; got != want {
		t.Errorf("Client.UnaryCall(%+v) returned status %v, want %v", req, got, want)
	}
}