Support ForStmt (#36)

* support for stmt

Signed-off-by: sivchari <shibuuuu5@gmail.com>

* new release

Signed-off-by: sivchari <shibuuuu5@gmail.com>

---------

Signed-off-by: sivchari <shibuuuu5@gmail.com>

release/RELEASE

@@ -1,4 +1,4 @@
-tag: v1.8.1
+tag: v1.9.1
 prerelease: false
 
 releaseNoteGenerator:

tenv.go

@@ -1,7 +1,6 @@
 package tenv
 
 import (
-	"fmt"
 	"go/ast"
 	"strings"
 
@@ -82,6 +81,8 @@ func checkStmts(pass *analysis.Pass, stmts []ast.Stmt, funcName, argName string)
 			if !checkAssignStmt(pass, stmt, funcName, argName) {
 				continue
 			}
+		case *ast.ForStmt:
+			checkForStmt(pass, stmt, funcName, argName)
 		}
 	}
 }
@@ -95,7 +96,6 @@ func checkExprStmt(pass *analysis.Pass, stmt *ast.ExprStmt, funcName, argName st
 	if !ok {
 		return false
 	}
-	fmt.Println(pass.TypesInfo.Types[fun.Sel].Value)
 	x, ok := fun.X.(*ast.Ident)
 	if !ok {
 		return false
@@ -160,6 +160,10 @@ func checkAssignStmt(pass *analysis.Pass, stmt *ast.AssignStmt, funcName, argNam
 	return true
 }
 
+func checkForStmt(pass *analysis.Pass, stmt *ast.ForStmt, funcName, argName string) {
+	checkStmts(pass, stmt.Body.List, funcName, argName)
+}
+
 func targetRunner(params []*ast.Field, fileName string) (string, bool) {
 	for _, p := range params {
 		switch typ := p.Type.(type) {

testdata/src/a/a_test.go

@@ -1,6 +1,7 @@
 package a
 
 import (
+	"fmt"
 	"os"
 	"testing"
 )
@@ -66,3 +67,31 @@ func TestEmpty(t *testing.T) {
 func TestEmptyTB(t *testing.T) {
 	func(testing.TB) {}(t)
 }
+
+func TestTDD(t *testing.T) {
+	for _, tt := range []struct {
+		name string
+	}{
+		{"test"},
+	} {
+		t.Run(tt.name, func(t *testing.T) {
+			os.Setenv("a", "b")        // want "os\\.Setenv\\(\\) can be replaced by `t\\.Setenv\\(\\)` in anonymous function"
+			err := os.Setenv("a", "b") // want "os\\.Setenv\\(\\) can be replaced by `t\\.Setenv\\(\\)` in anonymous function"
+			_ = err
+			if err := os.Setenv("a", "b"); err != nil { // want "os\\.Setenv\\(\\) can be replaced by `t\\.Setenv\\(\\)` in anonymous function"
+				_ = err
+			}
+		})
+	}
+}
+
+func TestLoop(t *testing.T) {
+	for i := 0; i < 3; i++ {
+		os.Setenv(fmt.Sprintf("a%d", i), "b")        // want "os\\.Setenv\\(\\) can be replaced by `t\\.Setenv\\(\\)` in TestLoop"
+		err := os.Setenv(fmt.Sprintf("a%d", i), "b") // want "os\\.Setenv\\(\\) can be replaced by `t\\.Setenv\\(\\)` in TestLoop"
+		_ = err
+		if err := os.Setenv(fmt.Sprintf("a%d", i), "b"); err != nil { // want "os\\.Setenv\\(\\) can be replaced by `t\\.Setenv\\(\\)` in TestLoop"
+			_ = err
+		}
+	}
+}