Map file path for fuzz tests

Author: lukaszcl

tools/flakeguard/reports/codebase_scanner.go

@@ -16,33 +16,39 @@ type TestFileMap map[string]string
 func ScanTestFiles(rootDir string) (TestFileMap, error) {
 	testFileMap := make(TestFileMap)
 
+	// Walk through the root directory to find test files
 	err := filepath.Walk(rootDir, func(path string, info os.FileInfo, err error) error {
 		if err != nil {
 			return err
 		}
+
+		// Skip files that are not Go test files
 		if !strings.HasSuffix(path, "_test.go") {
 			return nil
 		}
 
-		// Parse the file
+		// Parse the Go file
 		fset := token.NewFileSet()
 		node, err := parser.ParseFile(fset, path, nil, parser.AllErrors)
 		if err != nil {
 			return err
 		}
 
-		// Traverse the AST to find test functions
+		// Traverse the AST to find test or fuzz functions
 		ast.Inspect(node, func(n ast.Node) bool {
 			funcDecl, ok := n.(*ast.FuncDecl)
 			if !ok {
 				return true
 			}
-			if strings.HasPrefix(funcDecl.Name.Name, "Test") {
-				// Add both the package and test function to the map
+
+			// Match both "Test" and "Fuzz" prefixes
+			if strings.HasPrefix(funcDecl.Name.Name, "Test") || strings.HasPrefix(funcDecl.Name.Name, "Fuzz") {
+				// Add the function to the map
 				testFileMap[funcDecl.Name.Name] = path
 			}
 			return true
 		})
+
 		return nil
 	})