Detect and execute Fuzz targets with seed inputs.

nevkontakte · Aug 1, 2022 · b19a3f1 · b19a3f1
1 parent 4845bda
commit b19a3f1
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 3 deletions.
diff --git a/internal/testmain/testmain.go b/internal/testmain/testmain.go
@@ -69,6 +69,7 @@ type TestMain struct {
 	Package    *build.PackageData
 	Tests      []TestFunc
 	Benchmarks []TestFunc
+	Fuzz       []TestFunc
 	Examples   []ExampleFunc
 	TestMain   *TestFunc
 }
@@ -133,6 +134,11 @@ func (tm *TestMain) scanFile(f *ast.File, loc FuncLocation) error {
 				Location: loc,
 				Name:     name,
 			})
+		case isTest(name, "Fuzz"):
+			tm.Fuzz = append(tm.Fuzz, TestFunc{
+				Location: loc,
+				Name:     name,
+			})
 		}
 	}
 
@@ -275,8 +281,11 @@ var benchmarks = []testing.InternalBenchmark{
 {{- end}}
 }
 
-// TODO(nevkontakte): Extract fuzz targets from the source.
-var fuzzTargets = []testing.InternalFuzzTarget{}
+var fuzzTargets = []testing.InternalFuzzTarget{
+{{- range .Fuzz}}
+	{"{{.Name}}", {{.Location}}.{{.Name}}},
+{{- end}}
+}
 
 var examples = []testing.InternalExample{
 {{- range .Examples }}

diff --git a/internal/testmain/testmain_test.go b/internal/testmain/testmain_test.go
@@ -36,6 +36,10 @@ func TestScan(t *testing.T) {
 			{Location: LocInPackage, Name: "BenchmarkXxx"},
 			{Location: LocExternal, Name: "BenchmarkYyy"},
 		},
+		Fuzz: []TestFunc{
+			{Location: LocInPackage, Name: "FuzzXxx"},
+			{Location: LocExternal, Name: "FuzzYyy"},
+		},
 		Examples: []ExampleFunc{
 			{Location: LocInPackage, Name: "ExampleXxx"},
 			{Location: LocExternal, Name: "ExampleYyy", Output: "hello\n"},
@@ -71,6 +75,10 @@ func TestSynthesize(t *testing.T) {
 					{Location: LocInPackage, Name: "BenchmarkXxx"},
 					{Location: LocExternal, Name: "BenchmarkYyy"},
 				},
+				Fuzz: []TestFunc{
+					{Location: LocInPackage, Name: "FuzzXxx"},
+					{Location: LocExternal, Name: "FuzzYyy"},
+				},
 				Examples: []ExampleFunc{
 					{Location: LocInPackage, Name: "ExampleXxx", EmptyOutput: true},
 					{Location: LocExternal, Name: "ExampleYyy", EmptyOutput: true},
@@ -134,7 +142,10 @@ var benchmarks = []testing.InternalBenchmark{
 	{"BenchmarkYyy", _xtest.BenchmarkYyy},
 }
 
-var fuzzTargets = []testing.InternalFuzzTarget{}
+var fuzzTargets = []testing.InternalFuzzTarget{
+	{"FuzzXxx", _test.FuzzXxx},
+	{"FuzzYyy", _xtest.FuzzYyy},
+}
 
 var examples = []testing.InternalExample{
 	{"ExampleXxx", _test.ExampleXxx, "", false},