From 733951f5fea480c9969ef368aa5e85fff6b5ab5e Mon Sep 17 00:00:00 2001 From: Artem Dvoretskii Date: Thu, 16 May 2024 01:00:35 +0300 Subject: [PATCH] fix lint errs --- internal/api/requestid/requestid.go | 1 + internal/api/requestid/requestid_test.go | 12 +++++++----- 2 files changed, 8 insertions(+), 5 deletions(-) diff --git a/internal/api/requestid/requestid.go b/internal/api/requestid/requestid.go index 9e98fbe..b39e369 100644 --- a/internal/api/requestid/requestid.go +++ b/internal/api/requestid/requestid.go @@ -9,6 +9,7 @@ package requestid import ( "context" + "github.com/gofrs/uuid/v5" "go.uber.org/zap" "google.golang.org/grpc" diff --git a/internal/api/requestid/requestid_test.go b/internal/api/requestid/requestid_test.go index 298787a..3424536 100644 --- a/internal/api/requestid/requestid_test.go +++ b/internal/api/requestid/requestid_test.go @@ -2,17 +2,18 @@ package requestid import ( "context" + "testing" + "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" "go.uber.org/zap" "google.golang.org/grpc/metadata" - "testing" ) func TestGenerator_InterceptorFunc(t *testing.T) { type args struct { - trust bool reqID string + trust bool empty bool } type want struct { @@ -76,11 +77,12 @@ func TestGenerator_InterceptorFunc(t *testing.T) { require.NotNil(t, interceptor) var ctx context.Context - if tt.args.reqID != "" { + switch { + case tt.args.reqID != "": ctx = metadata.NewIncomingContext(context.Background(), metadata.Pairs(xRequestIDField, tt.args.reqID)) - } else if tt.args.empty { + case tt.args.empty: ctx = metadata.NewIncomingContext(context.Background(), metadata.Pairs(xRequestIDField, "")) - } else { + default: ctx = context.Background() }