Skip to content

Commit

Permalink
cleanup
Browse files Browse the repository at this point in the history
Signed-off-by: Yuri Shkuro <github@ysh.us>
  • Loading branch information
yurishkuro committed Jan 26, 2025
1 parent bf468c7 commit 8786f23
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 38 deletions.
16 changes: 0 additions & 16 deletions cmd/query/app/grpc_handler.go
Original file line number Diff line number Diff line change
Expand Up @@ -29,8 +29,6 @@ const (
var (
errNilRequest = status.Error(codes.InvalidArgument, "a nil argument is not allowed")
errUninitializedTraceID = status.Error(codes.InvalidArgument, "uninitialized TraceID is not allowed")
errMissingServiceNames = status.Error(codes.InvalidArgument, "please provide at least one service name")
errMissingQuantile = status.Error(codes.InvalidArgument, "please provide a quantile between (0, 1]")
)

// GRPCHandler implements the gRPC endpoint of the query service.
Expand Down Expand Up @@ -241,17 +239,3 @@ func (g *GRPCHandler) GetDependencies(ctx context.Context, r *api_v2.GetDependen

return &api_v2.GetDependenciesResponse{Dependencies: dependencies}, nil
}

func (g *GRPCHandler) handleErr(msg string, err error) error {
if err == nil {
return nil
}
g.logger.Error(msg, zap.Error(err))

if _, ok := status.FromError(err); ok {
return err
}

// Received an "unexpected" error.
return status.Errorf(codes.Internal, "%s: %v", msg, err)
}
33 changes: 11 additions & 22 deletions cmd/query/app/grpc_handler_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -116,13 +116,12 @@ var (
)

type grpcServer struct {
server *grpc.Server
lisAddr net.Addr
spanReader *spanstoremocks.Reader
depReader *depsmocks.Reader
metricsQueryService querysvc.MetricsQueryService
archiveSpanReader *spanstoremocks.Reader
archiveSpanWriter *spanstoremocks.Writer
server *grpc.Server
lisAddr net.Addr
spanReader *spanstoremocks.Reader
depReader *depsmocks.Reader
archiveSpanReader *spanstoremocks.Reader
archiveSpanWriter *spanstoremocks.Writer
}

type grpcClient struct {
Expand Down Expand Up @@ -167,13 +166,8 @@ func newGRPCClient(t *testing.T, addr string) *grpcClient {
}
}

type testOption func(*testQueryService)

type testQueryService struct {
}

func withServerAndClient(t *testing.T, actualTest func(server *grpcServer, client *grpcClient), options ...testOption) {
server := initializeTenantedTestServerGRPCWithOptions(t, &tenancy.Manager{}, options...)
func withServerAndClient(t *testing.T, actualTest func(server *grpcServer, client *grpcClient)) {
server := initializeTenantedTestServerGRPC(t, &tenancy.Manager{})
client := newGRPCClient(t, server.lisAddr.String())
defer server.server.Stop()
defer client.conn.Close()
Expand Down Expand Up @@ -620,7 +614,7 @@ func TestSendSpanChunksError(t *testing.T) {
require.EqualError(t, err, expectedErr.Error())
}

func initializeTenantedTestServerGRPCWithOptions(t *testing.T, tm *tenancy.Manager, options ...testOption) *grpcServer {
func initializeTenantedTestServerGRPC(t *testing.T, tm *tenancy.Manager) *grpcServer {
archiveSpanReader := &spanstoremocks.Reader{}
archiveSpanWriter := &spanstoremocks.Writer{}

Expand All @@ -635,11 +629,6 @@ func initializeTenantedTestServerGRPCWithOptions(t *testing.T, tm *tenancy.Manag
ArchiveSpanWriter: archiveSpanWriter,
})

tqs := &testQueryService{}
for _, opt := range options {
opt(tqs)
}

logger := zap.NewNop()
tracer := jtracer.NoOp()

Expand All @@ -655,8 +644,8 @@ func initializeTenantedTestServerGRPCWithOptions(t *testing.T, tm *tenancy.Manag
}
}

func withTenantedServerAndClient(t *testing.T, tm *tenancy.Manager, actualTest func(server *grpcServer, client *grpcClient), options ...testOption) {
server := initializeTenantedTestServerGRPCWithOptions(t, tm, options...)
func withTenantedServerAndClient(t *testing.T, tm *tenancy.Manager, actualTest func(server *grpcServer, client *grpcClient)) {
server := initializeTenantedTestServerGRPC(t, tm)
client := newGRPCClient(t, server.lisAddr.String())
defer server.server.Stop()
defer client.conn.Close()
Expand Down

0 comments on commit 8786f23

Please sign in to comment.