apply testing.F

sivchari · Apr 2, 2022 · 60539c5 · 60539c5
1 parent a4468e3
commit 60539c5
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 1 deletion.
diff --git a/tenv.go b/tenv.go
@@ -182,7 +182,7 @@ func checkStarExprTarget(typ *ast.StarExpr) bool {
 	}
 	targetName := x.Name + "." + selector.Sel.Name
 	switch targetName {
-	case "testing.T", "testing.B":
+	case "testing.T", "testing.B", "testing.F":
 		return true
 	default:
 		return false

diff --git a/testdata/src/a/a.go b/testdata/src/a/a.go
@@ -46,3 +46,12 @@ func TBF(tb testing.TB) {
 		_ = err
 	}
 }
+
+func FF(f *testing.F) {
+	os.Setenv("a", "b")        // want "os\\.Setenv\\(\\) can be replaced by `f\\.Setenv\\(\\)` in FF"
+	err := os.Setenv("a", "b") // want "os\\.Setenv\\(\\) can be replaced by `f\\.Setenv\\(\\)` in FF"
+	_ = err
+	if err := os.Setenv("a", "b"); err != nil { // want "os\\.Setenv\\(\\) can be replaced by `f\\.Setenv\\(\\)` in FF"
+		_ = err
+	}
+}
diff --git a/testdata/src/a/a_test.go b/testdata/src/a/a_test.go
@@ -46,3 +46,12 @@ func TB(tb testing.TB) {
 		_ = err
 	}
 }
+
+func FuzzF(f *testing.F) {
+	os.Setenv("a", "b")        // want "os\\.Setenv\\(\\) can be replaced by `f\\.Setenv\\(\\)` in FuzzF"
+	err := os.Setenv("a", "b") // want "os\\.Setenv\\(\\) can be replaced by `f\\.Setenv\\(\\)` in FuzzF"
+	_ = err
+	if err := os.Setenv("a", "b"); err != nil { // want "os\\.Setenv\\(\\) can be replaced by `f\\.Setenv\\(\\)` in FuzzF"
+		_ = err
+	}
+}