fix(codegen): preserve *assert.CollectT in Require

Update replace funcmap to replace "assert." with "require." while preserving
"*assert.CollectT" references. This prevents unintended replacement of
pointer references to CollectT, ensuring correct code generation.

Signed-off-by: a2not <[email protected]>

Author: a2not

_codegen/main.go

@@ -106,9 +106,7 @@ func parseTemplates() (*template.Template, *template.Template, error) {
 		}
 		funcTemplate = string(f)
 	}
-	tmpl, err := template.New("function").Funcs(template.FuncMap{
-		"replace": strings.ReplaceAll,
-	}).Parse(funcTemplate)
+	tmpl, err := template.New("function").Parse(funcTemplate)
 	if err != nil {
 		return nil, nil, err
 	}
@@ -298,6 +296,18 @@ func (f *testFunc) CommentWithoutT(receiver string) string {
 	return strings.Replace(f.Comment(), search, replace, -1)
 }
 
+func (f *testFunc) Replace(comment, search, replace string) string {
+	// replace strings, but preserve "*assert.CollectT"
+	const (
+		assertCollectT            = "*assert.CollectT"
+		assertCollectTPlaceholder = "__COLLECT_T_PLACEHOLDER__"
+	)
+
+	protected := strings.ReplaceAll(comment, assertCollectT, assertCollectTPlaceholder)
+	result := strings.ReplaceAll(protected, search, replace)
+	return strings.ReplaceAll(result, assertCollectTPlaceholder, assertCollectT)
+}
+
 // Standard header https://go.dev/s/generatedcode.
 var headerTemplate = `// Code generated with github.com/stretchr/testify/_codegen; DO NOT EDIT.
 

require/require.go

@@ -429,7 +429,7 @@ func Eventually(t TestingT, condition func() bool, waitFor time.Duration, tick t
 //		time.Sleep(8*time.Second)
 //		externalValue = true
 //	}()
-//	require.EventuallyWithT(t, func(c *require.CollectT) {
+//	require.EventuallyWithT(t, func(c *assert.CollectT) {
 //		// add assertions as needed; any assertion failure will fail the current tick
 //		require.True(c, externalValue, "expected 'externalValue' to be true")
 //	}, 10*time.Second, 1*time.Second, "external state has not changed to 'true'; still false")
@@ -457,7 +457,7 @@ func EventuallyWithT(t TestingT, condition func(collect *assert.CollectT), waitF
 //		time.Sleep(8*time.Second)
 //		externalValue = true
 //	}()
-//	require.EventuallyWithTf(t, func(c *require.CollectT, "error message %s", "formatted") {
+//	require.EventuallyWithTf(t, func(c *assert.CollectT, "error message %s", "formatted") {
 //		// add assertions as needed; any assertion failure will fail the current tick
 //		require.True(c, externalValue, "expected 'externalValue' to be true")
 //	}, 10*time.Second, 1*time.Second, "external state has not changed to 'true'; still false")

require/require.go.tmpl

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