fix: tests

35C4n0r · 35C4n0r · commit e1f136d934b4 · 2025-08-15T19:15:50.000+05:30
diff --git a/lib/msgfmt/msgfmt_test.go b/lib/msgfmt/msgfmt_test.go
@@ -188,7 +188,7 @@ func TestRemoveUserInput(t *testing.T) {
 			assert.NoError(t, err)
 			expected, err := testdataDir.ReadFile(path.Join(dir, c.Name(), "expected.txt"))
 			assert.NoError(t, err)
-			assert.Equal(t, string(expected), RemoveUserInput(string(msg), string(userInput)))
+			assert.Equal(t, string(expected), RemoveUserInput(string(msg), string(userInput), AgentTypeCustom))
 		})
 	}
 }

Original file line number	Diff line number	Diff line change
`@@ -188,7 +188,7 @@ func TestRemoveUserInput(t *testing.T) {`
`188`	`188`	`assert.NoError(t, err)`
`189`	`189`	`expected, err := testdataDir.ReadFile(path.Join(dir, c.Name(), "expected.txt"))`
`190`	`190`	`assert.NoError(t, err)`
`191`		`- assert.Equal(t, string(expected), RemoveUserInput(string(msg), string(userInput)))`
	`191`	`+ assert.Equal(t, string(expected), RemoveUserInput(string(msg), string(userInput), AgentTypeCustom))`
`192`	`192`	`})`
`193`	`193`	`}`
`194`	`194`	`}`