Fix IsGeneratedFile

gostaticanalysis · Nov 23, 2020 · f81035d · f81035d
1 parent ecea7d7
commit f81035d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/file.go b/file.go
@@ -23,8 +23,8 @@ var genCommentRegexp = regexp.MustCompile(`^// Code generated .* DO NOT EDIT\.$`
 // IsGeneratedFile reports whether the file has been generated automatically.
 // If file is nil, IsGeneratedFile will return false.
 func IsGeneratedFile(file *ast.File) bool {
-	if file == nil || file.Doc == nil {
+	if file == nil || len(file.Comments) == 0 {
 		return false
 	}
-	return genCommentRegexp.MatchString(file.Doc.List[0].Text)
+	return genCommentRegexp.MatchString(file.Comments[0].List[0].Text)
 }
diff --git a/file_test.go b/file_test.go
@@ -20,6 +20,7 @@ func TestIsGeneratedFile(t *testing.T) {
 		"true":  {"// Code generated by test; DO NOT EDIT.", true},
 		"false": {"//Code generated by test; DO NOT EDIT.", false},
 		"empty": {"", false},
+		"blank": {"// Code generated by test; DO NOT EDIT.\n", true},
 	}
 
 	for name, tt := range cases {