forked from mantzas/patron
-
Notifications
You must be signed in to change notification settings - Fork 67
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
11 changed files
with
177 additions
and
219 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,145 @@ | ||
| //go:build integration | ||
|
|
||
| package grpc | ||
|
|
||
| import ( | ||
| "context" | ||
| "fmt" | ||
| "log" | ||
| "net" | ||
| "testing" | ||
|
|
||
| "github.com/beatlabs/patron/examples" | ||
| "github.com/beatlabs/patron/internal/test" | ||
| "github.com/beatlabs/patron/observability/trace" | ||
| "github.com/stretchr/testify/assert" | ||
| "github.com/stretchr/testify/require" | ||
| "go.opentelemetry.io/otel/attribute" | ||
| "go.opentelemetry.io/otel/sdk/trace/tracetest" | ||
| "google.golang.org/grpc" | ||
| "google.golang.org/grpc/codes" | ||
| "google.golang.org/grpc/credentials/insecure" | ||
| "google.golang.org/grpc/status" | ||
| "google.golang.org/grpc/test/bufconn" | ||
| ) | ||
|
|
||
| type server struct { | ||
| examples.UnimplementedGreeterServer | ||
| } | ||
|
|
||
| func (s *server) SayHelloStream(_ *examples.HelloRequest, _ examples.Greeter_SayHelloStreamServer) error { | ||
| return status.Error(codes.Unavailable, "streaming not supported") | ||
| } | ||
|
|
||
| func (s *server) SayHello(_ context.Context, req *examples.HelloRequest) (*examples.HelloReply, error) { | ||
| if req.GetFirstname() == "" { | ||
| return nil, status.Error(codes.InvalidArgument, "first name cannot be empty") | ||
| } | ||
| return &examples.HelloReply{Message: fmt.Sprintf("Hello %s!", req.GetFirstname())}, nil | ||
| } | ||
|
|
||
| func TestSayHello(t *testing.T) { | ||
| ctx := context.Background() | ||
|
|
||
| client, closer := testServer() | ||
| defer closer() | ||
|
|
||
| // Tracing setup | ||
| exp := tracetest.NewInMemoryExporter() | ||
| tracePublisher := trace.Setup("test", nil, exp) | ||
|
|
||
| // Metrics monitoring set up | ||
| shutdownProvider, assertCollectMetrics := test.SetupMetrics(ctx, t) | ||
| defer shutdownProvider() | ||
|
|
||
| tt := map[string]struct { | ||
| req *examples.HelloRequest | ||
| wantErr bool | ||
| wantCode codes.Code | ||
| wantMsg string | ||
| wantCounter int | ||
| }{ | ||
| "ok": { | ||
| req: &examples.HelloRequest{Firstname: "John"}, | ||
| wantErr: false, | ||
| wantCode: codes.OK, | ||
| wantMsg: "Hello John!", | ||
| wantCounter: 1, | ||
| }, | ||
| "invalid": { | ||
| req: &examples.HelloRequest{}, | ||
| wantErr: true, | ||
| wantCode: codes.InvalidArgument, | ||
| wantMsg: "first name cannot be empty", | ||
| wantCounter: 1, | ||
| }, | ||
| } | ||
|
|
||
| for n, tt := range tt { | ||
| t.Run(n, func(t *testing.T) { | ||
| t.Cleanup(func() { exp.Reset() }) | ||
|
|
||
| res, err := client.SayHello(ctx, tt.req) | ||
| if tt.wantErr { | ||
| require.Nil(t, res) | ||
| require.Error(t, err) | ||
|
|
||
| rpcStatus, ok := status.FromError(err) | ||
| require.True(t, ok) | ||
| require.Equal(t, tt.wantCode, rpcStatus.Code()) | ||
| require.Equal(t, tt.wantMsg, rpcStatus.Message()) | ||
| } else { | ||
| require.NoError(t, err) | ||
| require.NotNil(t, res) | ||
| require.Equal(t, tt.wantMsg, res.GetMessage()) | ||
| } | ||
|
|
||
| require.NoError(t, tracePublisher.ForceFlush(context.Background())) | ||
|
|
||
| spans := exp.GetSpans().Snapshots() | ||
|
|
||
| assert.Len(t, spans, 1) | ||
| assert.Equal(t, "examples.Greeter/SayHello", spans[0].Name()) | ||
| assert.Equal(t, attribute.String("rpc.service", "examples.Greeter"), spans[0].Attributes()[0]) | ||
| assert.Equal(t, attribute.String("rpc.method", "SayHello"), spans[0].Attributes()[1]) | ||
| assert.Equal(t, attribute.String("rpc.system", "grpc"), spans[0].Attributes()[2]) | ||
| assert.Equal(t, attribute.Int64("rpc.grpc.status_code", int64(tt.wantCode)), spans[0].Attributes()[3]) | ||
|
|
||
| // Metrics | ||
| _ = assertCollectMetrics(4) | ||
| }) | ||
| } | ||
| } | ||
|
|
||
| func testServer() (examples.GreeterClient, func()) { | ||
| buffer := 101024 * 1024 | ||
| lis := bufconn.Listen(buffer) | ||
|
|
||
| baseServer := grpc.NewServer() | ||
| examples.RegisterGreeterServer(baseServer, &server{}) | ||
| go func() { | ||
| if err := baseServer.Serve(lis); err != nil { | ||
| log.Printf("error serving server: %v", err) | ||
| } | ||
| }() | ||
|
|
||
| conn, err := NewClient("123", | ||
| grpc.WithContextDialer(func(context.Context, string) (net.Conn, error) { | ||
| return lis.Dial() | ||
| }), grpc.WithTransportCredentials(insecure.NewCredentials())) | ||
| if err != nil { | ||
| log.Printf("error connecting to server: %v", err) | ||
| } | ||
|
|
||
| closer := func() { | ||
| err := lis.Close() | ||
| if err != nil { | ||
| log.Printf("error closing listener: %v", err) | ||
| } | ||
| baseServer.Stop() | ||
| } | ||
|
|
||
| client := examples.NewGreeterClient(conn) | ||
|
|
||
| return client, closer | ||
| } |
Oops, something went wrong.