require: fix invalid examples in doc comments (codegen)

_codegen/main.go

@@ -107,7 +107,8 @@ func parseTemplates() (*template.Template, *template.Template, error) {
 		funcTemplate = string(f)
 	}
 	tmpl, err := template.New("function").Funcs(template.FuncMap{
-		"replace": strings.ReplaceAll,
+		"replace":               strings.ReplaceAll,
+		"requireCommentParseIf": requireCommentParseIf,
 	}).Parse(funcTemplate)
 	if err != nil {
 		return nil, nil, err
@@ -298,6 +299,50 @@ func (f *testFunc) CommentWithoutT(receiver string) string {
 	return strings.Replace(f.Comment(), search, replace, -1)
 }
 
+func requireCommentParseIf(s string) string {
+	lines := strings.Split(s, "\n")
+	out := make([]string, 0, len(lines))
+	rePrefix := regexp.MustCompile(`//[ \t]*`)
+	ifDepth := 0
+	existsIfRequire := false
+	SomeFunctionLineIdx := 0
+
+	for i, line := range lines {
+		commentPrefix := rePrefix.FindString(line)
+		comment := strings.TrimSpace(line[2:])
+
+		if strings.HasSuffix(comment, "SomeFunction()") {
+			SomeFunctionLineIdx = i
+		}
+
+		if ifDepth > 0 && strings.HasPrefix(comment, "}") {
+			ifDepth = ifDepth - 1
+			continue
+		}
+
+		if strings.HasPrefix(comment, "if require.") && strings.HasSuffix(comment, "{") {
+			ifDepth = ifDepth + 1
+			existsIfRequire = true
+			comment = strings.TrimPrefix(comment, "if ")
+			comment = strings.TrimSpace(comment)
+			comment = strings.TrimSuffix(comment, "{")
+		}
+
+		if ifDepth > 0 {
+			commentPrefix = "//\t"
+		}
+
+		out = append(out, commentPrefix+comment)
+	}
+
+	if existsIfRequire && SomeFunctionLineIdx != 0 {
+		comment := out[SomeFunctionLineIdx][2:]
+		out[SomeFunctionLineIdx] = "//\t" + strings.TrimSpace(comment)
+	}
+
+	return strings.Join(out, "\n")
+}
+
 // Standard header https://go.dev/s/generatedcode.
 var headerTemplate = `// Code generated with github.com/stretchr/testify/_codegen; DO NOT EDIT.
 

require/require.go.tmpl

@@ -1,4 +1,4 @@
-{{ replace .Comment "assert." "require."}}
+{{ replace .Comment "assert." "require." | requireCommentParseIf }}
 func {{.DocInfo.Name}}(t TestingT, {{.Params}}) {
 	if h, ok := t.(tHelper); ok { h.Helper() }
 	if assert.{{.DocInfo.Name}}(t, {{.ForwardedParams}}) { return }