Method TestUnsupportedEncodingResponse

test/compressor_test.go:46–117

TestUnsupportedEncodingResponse validates gRPC status codes for different client-server compression setups ensuring the correct behavior when compression is enabled or disabled on either side.

(t *testing.T)

// for different client-server compression setups
// ensuring the correct behavior when compression is enabled or disabled on either side.
func (s) TestUnsupportedEncodingResponse(t *testing.T) {
	tests := []struct {
		name           string
		clientCompress bool
		serverCompress bool
		wantStatus     codes.Code
	}{
		{
			name:           "client_server_compression",
			clientCompress: true,
			serverCompress: true,
			wantStatus:     codes.OK,
		},
		{
			name:           "client_compression",
			clientCompress: true,
			serverCompress: false,
			wantStatus:     codes.Unimplemented,
		},
		{
			name:           "server_compression",
			clientCompress: false,
			serverCompress: true,
			wantStatus:     codes.Internal,
		},
	}

	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {
			ss := &stubserver.StubServer{
				UnaryCallF: func(_ context.Context, in *testpb.SimpleRequest) (*testpb.SimpleResponse, error) {
					return &testpb.SimpleResponse{Payload: in.Payload}, nil
				},
			}
			sopts := []grpc.ServerOption{}
			if test.serverCompress {
				// Using deprecated methods to selectively apply compression
				// only on the server side. With encoding.registerCompressor(),
				// the compressor is applied globally, affecting client and server
				sopts = append(sopts, grpc.RPCCompressor(newNopCompressor()), grpc.RPCDecompressor(newNopDecompressor()))
			}
			if err := ss.StartServer(sopts...); err != nil {
				t.Fatalf("Error starting server: %v", err)
			}
			defer ss.Stop()

			dopts := []grpc.DialOption{grpc.WithTransportCredentials(insecure.NewCredentials())}
			if test.clientCompress {
				// UseCompressor() requires the compressor to be registered
				// using encoding.RegisterCompressor() which applies compressor globally,
				// Hence, using deprecated WithCompressor() and WithDecompressor()
				// to apply compression only on client.
				dopts = append(dopts, grpc.WithCompressor(newNopCompressor()), grpc.WithDecompressor(newNopDecompressor()))
			}
			if err := ss.StartClient(dopts...); err != nil {
				t.Fatalf("Error starting client: %v", err)
			}