1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
package grpctool
import (
"context"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/tool/grpctool/test"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials/insecure"
)
func TestValidator(t *testing.T) {
lis := NewDialListener()
defer lis.Close()
server := grpc.NewServer()
defer server.GracefulStop()
test.RegisterTestingServer(server, &test.GrpcTestingServer{
UnaryFunc: func(ctx context.Context, request *test.Request) (*test.Response, error) {
return &test.Response{
// invalid response because Message is not set
}, nil
},
StreamingFunc: func(server test.Testing_StreamingRequestResponseServer) error {
return server.Send(&test.Response{
// invalid response because Message is not set
})
},
})
go func() {
assert.NoError(t, server.Serve(lis))
}()
conn, err := grpc.DialContext(context.Background(), "bufnet",
grpc.WithTransportCredentials(insecure.NewCredentials()),
grpc.WithChainStreamInterceptor(StreamClientValidatingInterceptor),
grpc.WithChainUnaryInterceptor(UnaryClientValidatingInterceptor),
grpc.WithContextDialer(lis.DialContext),
)
require.NoError(t, err)
defer conn.Close()
client := test.NewTestingClient(conn)
t.Run("invalid unary response", func(t *testing.T) {
_, err = client.RequestResponse(context.Background(), &test.Request{})
assert.EqualError(t, err, "rpc error: code = InvalidArgument desc = invalid server response: invalid Response.Message: value is required")
})
t.Run("invalid streaming response", func(t *testing.T) {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
stream, err := client.StreamingRequestResponse(ctx)
require.NoError(t, err)
_, err = stream.Recv()
assert.EqualError(t, err, "rpc error: code = InvalidArgument desc = invalid server response: invalid Response.Message: value is required")
})
}
|
