Skip to content

Commit

Permalink
PMM-7 Lint fix. (#2717)
Browse files Browse the repository at this point in the history
* Fix lint.

* Changes.

* Changes.

* PMM-7 Small fix.

* PMM-7 Fix.

* Revert interceptors until refactor.

* Fix

* Fix.

* Fix.

---------

Co-authored-by: Nurlan Moldomurov <[email protected]>
  • Loading branch information
JiriCtvrtka and BupycHuk authored Feb 14, 2024
1 parent 58e8c44 commit e9aee30
Show file tree
Hide file tree
Showing 16 changed files with 103 additions and 98 deletions.
8 changes: 4 additions & 4 deletions admin/commands/summary.go
Original file line number Diff line number Diff line change
Expand Up @@ -357,12 +357,12 @@ type SummaryCommand struct {
Pprof bool `name:"pprof" help:"Include performance profiling data"`
}

func (cmd *SummaryCommand) makeArchive(ctx context.Context, globals *flags.GlobalFlags) (err error) {
func (cmd *SummaryCommand) makeArchive(ctx context.Context, globals *flags.GlobalFlags) error {
var f *os.File
var err error

if f, err = os.Create(cmd.Filename); err != nil {
err = errors.WithStack(err)
return //nolint:nakedret
return errors.WithStack(err)
}

defer func() {
Expand All @@ -389,7 +389,7 @@ func (cmd *SummaryCommand) makeArchive(ctx context.Context, globals *flags.Globa
addServerData(ctx, zipW, cmd.Pprof)
}

return //nolint:nakedret
return nil
}

// RunCmdWithContext runs summary command.
Expand Down
9 changes: 4 additions & 5 deletions agent/agents/process/process_logger.go
Original file line number Diff line number Diff line change
Expand Up @@ -55,23 +55,22 @@ func newProcessLogger(l *logrus.Entry, lines int, redactWords []string) *process

// Write implements io.Writer.
// This method is thread-safe.
func (pl *processLogger) Write(p []byte) (n int, err error) { //nolint:nonamedreturns
func (pl *processLogger) Write(p []byte) (int, error) {
pl.m.Lock()
defer pl.m.Unlock()

b := bytes.NewBuffer(pl.buf)
n, err = b.Write(p)
n, err := b.Write(p)
if err != nil {
return //nolint:nakedret
return n, err
}

var line string
for {
line, err = b.ReadString('\n')
if err != nil {
pl.buf = []byte(line)
err = nil
return //nolint:nakedret
return n, nil //nolint:nilerr
}
line = strings.TrimSuffix(line, "\n")
if pl.replacer != nil {
Expand Down
8 changes: 5 additions & 3 deletions agent/client/channel/channel_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -49,10 +49,12 @@ func (s *testServer) Connect(stream agentpb.Agent_ConnectServer) error {

var _ agentpb.AgentServer = (*testServer)(nil)

//nolint:nakedret
func setup(t *testing.T, connect func(agentpb.Agent_ConnectServer) error, expected ...error) (channel *Channel, cc *grpc.ClientConn, teardown func()) {
func setup(t *testing.T, connect func(agentpb.Agent_ConnectServer) error, expected ...error) (*Channel, *grpc.ClientConn, func()) {
t.Helper()

var channel *Channel
var cc *grpc.ClientConn
var teardown func()
// logrus.SetLevel(logrus.DebugLevel)

// start server with given connect handler
Expand Down Expand Up @@ -101,7 +103,7 @@ func setup(t *testing.T, connect func(agentpb.Agent_ConnectServer) error, expect
require.NoError(t, <-serveError)
}

return
return channel, cc, teardown
}

func TestAgentRequestWithTruncatedInvalidUTF8(t *testing.T) {
Expand Down
14 changes: 8 additions & 6 deletions agent/config/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -186,7 +186,9 @@ func getFromCmdLine(cfg *Config, l *logrus.Entry) (string, error) {
}

// get is Get for unit tests: it parses args instead of command-line.
func get(args []string, cfg *Config, l *logrus.Entry) (configFileF string, err error) { //nolint:nonamedreturns,cyclop
func get(args []string, cfg *Config, l *logrus.Entry) (string, error) { //nolint:cyclop
var configFileF string
var err error
// tweak configuration on exit to cover all return points
defer func() {
if cfg == nil {
Expand Down Expand Up @@ -302,25 +304,25 @@ func get(args []string, cfg *Config, l *logrus.Entry) (configFileF string, err e
// parse command-line flags and environment variables
app, cfgFileF := Application(cfg)
if _, err = app.Parse(args); err != nil {
return //nolint:nakedret
return configFileF, err
}
if *cfgFileF == "" {
return //nolint:nakedret
return configFileF, err
}

if configFileF, err = filepath.Abs(*cfgFileF); err != nil {
return //nolint:nakedret
return configFileF, err
}
l.Infof("Loading configuration file %s.", configFileF)
fileCfg, err := loadFromFile(configFileF)
if err != nil {
return //nolint:nakedret
return configFileF, err
}

// re-parse flags into configuration from file
app, _ = Application(fileCfg)
if _, err = app.Parse(args); err != nil {
return //nolint:nakedret
return configFileF, err
}

*cfg = *fileCfg
Expand Down
8 changes: 6 additions & 2 deletions managed/models/action_helpers_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -118,8 +118,12 @@ func TestCleanupResults(t *testing.T) {
require.NoError(t, sqlDB.Close())
}()

setup := func(t *testing.T) (q *reform.Querier, teardown func(t *testing.T)) {
setup := func(t *testing.T) (*reform.Querier, func(t *testing.T)) {
t.Helper()

var q *reform.Querier
var teardown func(t *testing.T)

db := reform.NewDB(sqlDB, postgresql.Dialect, reform.NewPrintfLogger(t.Logf))
tx, err := db.Begin()
require.NoError(t, err)
Expand Down Expand Up @@ -158,7 +162,7 @@ func TestCleanupResults(t *testing.T) {
t.Helper()
require.NoError(t, tx.Rollback())
}
return //nolint:nakedret
return q, teardown
}

t.Run("CheckActionResultByID", func(t *testing.T) {
Expand Down
8 changes: 4 additions & 4 deletions managed/models/agent_helpers_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -45,12 +45,12 @@ func TestAgentHelpers(t *testing.T) {
require.NoError(t, sqlDB.Close())
}()

setup := func(t *testing.T) (q *reform.Querier, teardown func(t *testing.T)) {
setup := func(t *testing.T) (*reform.Querier, func(t *testing.T)) {
t.Helper()
db := reform.NewDB(sqlDB, postgresql.Dialect, reform.NewPrintfLogger(t.Logf))
tx, err := db.Begin()
require.NoError(t, err)
q = tx.Querier
q := tx.Querier

for _, str := range []reform.Struct{
&models.Node{
Expand Down Expand Up @@ -188,11 +188,11 @@ func TestAgentHelpers(t *testing.T) {
require.NoError(t, q.Insert(str))
}

teardown = func(t *testing.T) {
teardown := func(t *testing.T) {
t.Helper()
require.NoError(t, tx.Rollback())
}
return //nolint:nakedret
return q, teardown
}

t.Run("AgentsForNode", func(t *testing.T) {
Expand Down
8 changes: 4 additions & 4 deletions managed/models/agent_model_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -384,12 +384,12 @@ func TestExporterURL(t *testing.T) {
require.NoError(t, sqlDB.Close())
}()

setup := func(t *testing.T) (q *reform.Querier, teardown func(t *testing.T)) {
setup := func(t *testing.T) (*reform.Querier, func(t *testing.T)) {
t.Helper()
db := reform.NewDB(sqlDB, postgresql.Dialect, reform.NewPrintfLogger(t.Logf))
tx, err := db.Begin()
require.NoError(t, err)
q = tx.Querier
q := tx.Querier

for _, str := range []reform.Struct{
&models.Node{
Expand Down Expand Up @@ -504,11 +504,11 @@ func TestExporterURL(t *testing.T) {
require.NoError(t, q.Insert(str), "failed to INSERT %+v", str)
}

teardown = func(t *testing.T) {
teardown := func(t *testing.T) {
t.Helper()
require.NoError(t, tx.Rollback())
}
return //nolint:nakedret
return q, teardown
}

t.Run("ExporterURL", func(t *testing.T) {
Expand Down
8 changes: 4 additions & 4 deletions managed/models/dsn_helpers_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -43,12 +43,12 @@ func TestFindDSNByServiceID(t *testing.T) {
require.NoError(t, sqlDB.Close())
}()

setup := func(t *testing.T) (q *reform.Querier, teardown func(t *testing.T)) {
setup := func(t *testing.T) (*reform.Querier, func(t *testing.T)) {
t.Helper()
db := reform.NewDB(sqlDB, postgresql.Dialect, reform.NewPrintfLogger(t.Logf))
tx, err := db.Begin()
require.NoError(t, err)
q = tx.Querier
q := tx.Querier

for _, str := range []reform.Struct{
&models.Node{
Expand Down Expand Up @@ -165,11 +165,11 @@ func TestFindDSNByServiceID(t *testing.T) {
require.NoError(t, q.Insert(str))
}

teardown = func(t *testing.T) {
teardown := func(t *testing.T) {
t.Helper()
require.NoError(t, tx.Rollback())
}
return //nolint:nakedret
return q, teardown
}

t.Run("FindDSNByServiceIDandPMMAgentIDWithNoAgent", func(t *testing.T) {
Expand Down
8 changes: 4 additions & 4 deletions managed/models/kubernetes_helpers_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -44,12 +44,12 @@ func TestKubernetesHelpers(t *testing.T) {
require.NoError(t, sqlDB.Close())
})

setup := func(t *testing.T) (q *reform.Querier, teardown func(t *testing.T)) {
setup := func(t *testing.T) (*reform.Querier, func(t *testing.T)) {
t.Helper()
db := reform.NewDB(sqlDB, postgresql.Dialect, reform.NewPrintfLogger(t.Logf))
tx, err := db.Begin()
require.NoError(t, err)
q = tx.Querier
q := tx.Querier

for _, str := range []reform.Struct{
&models.KubernetesCluster{
Expand Down Expand Up @@ -82,11 +82,11 @@ func TestKubernetesHelpers(t *testing.T) {
require.NoError(t, q.Insert(str), "failed to INSERT %+v", str)
}

teardown = func(t *testing.T) {
teardown := func(t *testing.T) {
t.Helper()
require.NoError(t, tx.Rollback())
}
return //nolint:nakedret
return q, teardown
}

t.Run("FindAllKubernetesClusters", func(t *testing.T) {
Expand Down
8 changes: 4 additions & 4 deletions managed/models/node_helpers_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -43,12 +43,12 @@ func TestNodeHelpers(t *testing.T) {
require.NoError(t, sqlDB.Close())
}()

setup := func(t *testing.T) (q *reform.Querier, teardown func(t *testing.T)) {
setup := func(t *testing.T) (*reform.Querier, func(t *testing.T)) {
t.Helper()
db := reform.NewDB(sqlDB, postgresql.Dialect, reform.NewPrintfLogger(t.Logf))
tx, err := db.Begin()
require.NoError(t, err)
q = tx.Querier
q := tx.Querier

for _, str := range []reform.Struct{
&models.Node{
Expand Down Expand Up @@ -111,11 +111,11 @@ func TestNodeHelpers(t *testing.T) {
require.NoError(t, q.Insert(str), "failed to INSERT %+v", str)
}

teardown = func(t *testing.T) {
teardown := func(t *testing.T) {
t.Helper()
require.NoError(t, tx.Rollback())
}
return //nolint:nakedret
return q, teardown
}

t.Run("CreateNode", func(t *testing.T) {
Expand Down
8 changes: 4 additions & 4 deletions managed/models/service_helpers_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -43,12 +43,12 @@ func TestServiceHelpers(t *testing.T) {
require.NoError(t, sqlDB.Close())
}()

setup := func(t *testing.T) (q *reform.Querier, teardown func(t *testing.T)) {
setup := func(t *testing.T) (*reform.Querier, func(t *testing.T)) {
t.Helper()
db := reform.NewDB(sqlDB, postgresql.Dialect, reform.NewPrintfLogger(t.Logf))
tx, err := db.Begin()
require.NoError(t, err)
q = tx.Querier
q := tx.Querier

for _, str := range []reform.Struct{
&models.Node{
Expand Down Expand Up @@ -140,11 +140,11 @@ func TestServiceHelpers(t *testing.T) {
require.NoError(t, q.Insert(str))
}

teardown = func(t *testing.T) {
teardown := func(t *testing.T) {
t.Helper()
require.NoError(t, tx.Rollback())
}
return //nolint:nakedret
return q, teardown
}

t.Run("FindServices", func(t *testing.T) {
Expand Down
16 changes: 8 additions & 8 deletions managed/services/management/dbaas/components_service_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -53,18 +53,18 @@ func TestComponentService(t *testing.T) {
kubeConfig = "{}"
)

setup := func(t *testing.T) (ctx context.Context, cs dbaasv1beta1.ComponentsServer, dbaasClient *mockDbaasClient, kubeClient *mockKubernetesClient,
kubeStorage *mockKubeStorageManager,
setup := func(t *testing.T) (context.Context, dbaasv1beta1.ComponentsServer, *mockDbaasClient, *mockKubernetesClient,
*mockKubeStorageManager,
) {
t.Helper()

ctx = logger.Set(context.Background(), t.Name())
ctx := logger.Set(context.Background(), t.Name())
uuid.SetRand(&tests.IDReader{})

sqlDB := testdb.Open(t, models.SetupFixtures, nil)
db := reform.NewDB(sqlDB, postgresql.Dialect, nil)

dbaasClient = &mockDbaasClient{}
dbaasClient := &mockDbaasClient{}

kubernetesCluster, err := models.CreateKubernetesCluster(db.Querier, &models.CreateKubernetesClusterParams{
KubernetesClusterName: clusterName,
Expand All @@ -80,12 +80,12 @@ func TestComponentService(t *testing.T) {
})

vsc := NewVersionServiceClient(versionServiceURL)
kubeStorage = &mockKubeStorageManager{}
kubeClient = &mockKubernetesClient{}
kubeStorage := &mockKubeStorageManager{}
kubeClient := &mockKubernetesClient{}
kubeClient.On("GetServerVersion").Return(nil, nil)
cs = NewComponentsService(db, dbaasClient, vsc, kubeStorage)
cs := NewComponentsService(db, dbaasClient, vsc, kubeStorage)

return //nolint:nakedret
return ctx, cs, dbaasClient, kubeClient, kubeStorage
}

t.Run("PXC", func(t *testing.T) {
Expand Down
Loading

0 comments on commit e9aee30

Please sign in to comment.