Add thelper linter (#1541)

Author: kulti

.golangci.example.yml

@@ -300,6 +300,17 @@ linters-settings:
   testpackage:
     # regexp pattern to skip files
     skip-regexp: (export|internal)_test\.go
+  thelper:
+    # The following configurations enable all checks. It can be omitted because all checks are enabled by default.
+    # You can enable only required checks deleting unnecessary checks.
+    test:
+      first: true
+      name: true
+      begin: true
+    benchmark:
+      first: true
+      name: true
+      begin: true
   unparam:
     # Inspect exported functions, default is false. Set to true if no external program/library imports your code.
     # XXX: if you enable this setting, unparam will report a lot of false-positives in text editors:

go.mod

@@ -30,6 +30,7 @@ require (
 	github.com/jgautheron/goconst v0.0.0-20201117150253-ccae5bf973f3
 	github.com/jingyugao/rowserrcheck v0.0.0-20191204022205-72ab7603b68a
 	github.com/jirfag/go-printf-func-name v0.0.0-20191110105641-45db9963cdd3
+	github.com/kulti/thelper v0.1.0
 	github.com/kunwardeep/paralleltest v1.0.2
 	github.com/kyoh86/exportloopref v0.1.8
 	github.com/maratori/testpackage v1.0.1

go.sum

@@ -267,6 +267,7 @@ type LintersSettings struct {
 	Gofumpt     GofumptSettings
 	ErrorLint   ErrorLintSettings
 	Makezero    MakezeroSettings
+	Thelper     ThelperSettings
 
 	Custom map[string]CustomLinterSettings
 }
@@ -391,6 +392,19 @@ type MakezeroSettings struct {
 	Always bool
 }
 
+type ThelperSettings struct {
+	Test struct {
+		First bool `mapstructure:"first"`
+		Name  bool `mapstructure:"name"`
+		Begin bool `mapstructure:"begin"`
+	} `mapstructure:"test"`
+	Benchmark struct {
+		First bool `mapstructure:"first"`
+		Name  bool `mapstructure:"name"`
+		Begin bool `mapstructure:"begin"`
+	} `mapstructure:"benchmark"`
+}
+
 var defaultLintersSettings = LintersSettings{
 	Lll: LllSettings{
 		LineLength: 120,

pkg/config/config.go

@@ -0,0 +1,51 @@
+package golinters
+
+import (
+	"strings"
+
+	"github.com/kulti/thelper/pkg/analyzer"
+	"golang.org/x/tools/go/analysis"
+
+	"github.com/golangci/golangci-lint/pkg/config"
+	"github.com/golangci/golangci-lint/pkg/golinters/goanalysis"
+)
+
+func NewThelper(cfg *config.ThelperSettings) *goanalysis.Linter {
+	a := analyzer.NewAnalyzer()
+
+	cfgMap := map[string]map[string]interface{}{}
+	if cfg != nil {
+		var opts []string
+
+		if cfg.Test.Name {
+			opts = append(opts, "t_name")
+		}
+		if cfg.Test.Begin {
+			opts = append(opts, "t_begin")
+		}
+		if cfg.Test.First {
+			opts = append(opts, "t_first")
+		}
+
+		if cfg.Benchmark.Name {
+			opts = append(opts, "b_name")
+		}
+		if cfg.Benchmark.Begin {
+			opts = append(opts, "b_begin")
+		}
+		if cfg.Benchmark.First {
+			opts = append(opts, "b_first")
+		}
+
+		cfgMap[a.Name] = map[string]interface{}{
+			"checks": strings.Join(opts, ","),
+		}
+	}
+
+	return goanalysis.NewLinter(
+		"thelper",
+		"thelper detects golang test helpers without t.Helper() call and checks the consistency of test helpers",
+		[]*analysis.Analyzer{a},
+		cfgMap,
+	).WithLoadMode(goanalysis.LoadModeTypesInfo)
+}

pkg/golinters/thelper.go

@@ -91,11 +91,13 @@ func (m Manager) GetAllSupportedLinterConfigs() []*linter.Config {
 	var testpackageCfg *config.TestpackageSettings
 	var exhaustiveCfg *config.ExhaustiveSettings
 	var errorlintCfg *config.ErrorLintSettings
+	var thelperCfg *config.ThelperSettings
 	if m.cfg != nil {
 		govetCfg = &m.cfg.LintersSettings.Govet
 		testpackageCfg = &m.cfg.LintersSettings.Testpackage
 		exhaustiveCfg = &m.cfg.LintersSettings.Exhaustive
 		errorlintCfg = &m.cfg.LintersSettings.ErrorLint
+		thelperCfg = &m.cfg.LintersSettings.Thelper
 	}
 	const megacheckName = "megacheck"
 	lcs := []*linter.Config{
@@ -313,6 +315,10 @@ func (m Manager) GetAllSupportedLinterConfigs() []*linter.Config {
 			WithPresets(linter.PresetStyle).
 			WithLoadForGoAnalysis().
 			WithURL("https://github.com/tomarrell/wrapcheck"),
+		linter.NewConfig(golinters.NewThelper(thelperCfg)).
+			WithPresets(linter.PresetStyle).
+			WithLoadForGoAnalysis().
+			WithURL("https://github.com/kulti/thelper"),
 		linter.NewConfig(golinters.NewTparallel()).
 			WithPresets(linter.PresetStyle).
 			WithLoadForGoAnalysis().

pkg/lint/lintersdb/manager.go

@@ -0,0 +1,30 @@
+//args: -Ethelper
+package testdata
+
+import "testing"
+
+func thelperWithHelperAfterAssignment(t *testing.T) { // ERROR "test helper function should start from t.Helper()"
+	_ = 0
+	t.Helper()
+}
+
+func thelperWithNotFirst(s string, t *testing.T, i int) { // ERROR "parameter \*testing.T should be the first"
+	t.Helper()
+}
+
+func thelperWithIncorrectName(o *testing.T) { // ERROR "parameter \*testing.T should have name t"
+	o.Helper()
+}
+
+func bhelperWithHelperAfterAssignment(b *testing.B) { // ERROR "test helper function should start from b.Helper()"
+	_ = 0
+	b.Helper()
+}
+
+func bhelperWithNotFirst(s string, b *testing.B, i int) { // ERROR "parameter \*testing.B should be the first"
+	b.Helper()
+}
+
+func bhelperWithIncorrectName(o *testing.B) { // ERROR "parameter \*testing.B should have name b"
+	o.Helper()
+}