1 files changed, 207 insertions, 0 deletions
diff --git a/vendor/github.com/sivchari/tenv/tenv.go b/vendor/github.com/sivchari/tenv/tenv.go
new file mode 100644
index 000000000..e35205226
--- /dev/null
+++ b/vendor/github.com/sivchari/tenv/tenv.go
@@ -0,0 +1,207 @@
+package tenv
+
+import (
+	"go/ast"
+	"strings"
+
+	"golang.org/x/tools/go/analysis"
+	"golang.org/x/tools/go/analysis/passes/inspect"
+	"golang.org/x/tools/go/ast/inspector"
+)
+
+const doc = "tenv is analyzer that detects using os.Setenv instead of t.Setenv since Go1.17"
+
+// Analyzer is tenv analyzer
+var Analyzer = &analysis.Analyzer{
+	Name: "tenv",
+	Doc:  doc,
+	Run:  run,
+	Requires: []*analysis.Analyzer{
+		inspect.Analyzer,
+	},
+}
+
+var (
+	A     = "all"
+	aflag bool
+)
+
+func init() {
+	Analyzer.Flags.BoolVar(&aflag, A, false, "the all option will run against all method in test file")
+}
+
+func run(pass *analysis.Pass) (interface{}, error) {
+	inspect := pass.ResultOf[inspect.Analyzer].(*inspector.Inspector)
+
+	nodeFilter := []ast.Node{
+		(*ast.FuncDecl)(nil),
+		(*ast.FuncLit)(nil),
+	}
+
+	inspect.Preorder(nodeFilter, func(n ast.Node) {
+		switch n := n.(type) {
+		case *ast.FuncDecl:
+			checkFuncDecl(pass, n, pass.Fset.File(n.Pos()).Name())
+		case *ast.FuncLit:
+			checkFuncLit(pass, n, pass.Fset.File(n.Pos()).Name())
+		}
+	})
+
+	return nil, nil
+}
+
+func checkFuncDecl(pass *analysis.Pass, f *ast.FuncDecl, fileName string) {
+	argName, ok := targetRunner(f.Type.Params.List, fileName)
+	if !ok {
+		return
+	}
+	checkStmts(pass, f.Body.List, f.Name.Name, argName)
+}
+
+func checkFuncLit(pass *analysis.Pass, f *ast.FuncLit, fileName string) {
+	argName, ok := targetRunner(f.Type.Params.List, fileName)
+	if !ok {
+		return
+	}
+	checkStmts(pass, f.Body.List, "anonymous function", argName)
+}
+
+func checkStmts(pass *analysis.Pass, stmts []ast.Stmt, funcName, argName string) {
+	for _, stmt := range stmts {
+		switch stmt := stmt.(type) {
+		case *ast.ExprStmt:
+			if !checkExprStmt(pass, stmt, funcName, argName) {
+				continue
+			}
+		case *ast.IfStmt:
+			if !checkIfStmt(pass, stmt, funcName, argName) {
+				continue
+			}
+		case *ast.AssignStmt:
+			if !checkAssignStmt(pass, stmt, funcName, argName) {
+				continue
+			}
+		}
+	}
+}
+
+func checkExprStmt(pass *analysis.Pass, stmt *ast.ExprStmt, funcName, argName string) bool {
+	callExpr, ok := stmt.X.(*ast.CallExpr)
+	if !ok {
+		return false
+	}
+	fun, ok := callExpr.Fun.(*ast.SelectorExpr)
+	if !ok {
+		return false
+	}
+	x, ok := fun.X.(*ast.Ident)
+	if !ok {
+		return false
+	}
+	targetName := x.Name + "." + fun.Sel.Name
+	if targetName == "os.Setenv" {
+		if argName == "" {
+			argName = "testing"
+		}
+		pass.Reportf(stmt.Pos(), "os.Setenv() can be replaced by `%s.Setenv()` in %s", argName, funcName)
+	}
+	return true
+}
+
+func checkIfStmt(pass *analysis.Pass, stmt *ast.IfStmt, funcName, argName string) bool {
+	assignStmt, ok := stmt.Init.(*ast.AssignStmt)
+	if !ok {
+		return false
+	}
+	rhs, ok := assignStmt.Rhs[0].(*ast.CallExpr)
+	if !ok {
+		return false
+	}
+	fun, ok := rhs.Fun.(*ast.SelectorExpr)
+	if !ok {
+		return false
+	}
+	x, ok := fun.X.(*ast.Ident)
+	if !ok {
+		return false
+	}
+	targetName := x.Name + "." + fun.Sel.Name
+	if targetName == "os.Setenv" {
+		if argName == "" {
+			argName = "testing"
+		}
+		pass.Reportf(stmt.Pos(), "os.Setenv() can be replaced by `%s.Setenv()` in %s", argName, funcName)
+	}
+	return true
+}
+
+func checkAssignStmt(pass *analysis.Pass, stmt *ast.AssignStmt, funcName, argName string) bool {
+	rhs, ok := stmt.Rhs[0].(*ast.CallExpr)
+	if !ok {
+		return false
+	}
+	fun, ok := rhs.Fun.(*ast.SelectorExpr)
+	if !ok {
+		return false
+	}
+	x, ok := fun.X.(*ast.Ident)
+	if !ok {
+		return false
+	}
+	targetName := x.Name + "." + fun.Sel.Name
+	if targetName == "os.Setenv" {
+		if argName == "" {
+			argName = "testing"
+		}
+		pass.Reportf(stmt.Pos(), "os.Setenv() can be replaced by `%s.Setenv()` in %s", argName, funcName)
+	}
+	return true
+}
+
+func targetRunner(params []*ast.Field, fileName string) (string, bool) {
+	for _, p := range params {
+		switch typ := p.Type.(type) {
+		case *ast.StarExpr:
+			if checkStarExprTarget(typ) {
+				argName := p.Names[0].Name
+				return argName, true
+			}
+		case *ast.SelectorExpr:
+			if checkSelectorExprTarget(typ) {
+				argName := p.Names[0].Name
+				return argName, true
+			}
+		}
+	}
+	if aflag && strings.HasSuffix(fileName, "_test.go") {
+		return "", true
+	}
+	return "", false
+}
+
+func checkStarExprTarget(typ *ast.StarExpr) bool {
+	selector, ok := typ.X.(*ast.SelectorExpr)
+	if !ok {
+		return false
+	}
+	x, ok := selector.X.(*ast.Ident)
+	if !ok {
+		return false
+	}
+	targetName := x.Name + "." + selector.Sel.Name
+	switch targetName {
+	case "testing.T", "testing.B", "testing.F":
+		return true
+	default:
+		return false
+	}
+}
+
+func checkSelectorExprTarget(typ *ast.SelectorExpr) bool {
+	x, ok := typ.X.(*ast.Ident)
+	if !ok {
+		return false
+	}
+	targetName := x.Name + "." + typ.Sel.Name
+	return targetName == "testing.TB"
+}