mirror of
https://github.com/tiennm99/goclaw.git
synced 2026-09-20 04:23:29 +00:00
viettranx
6e94c5fdf8
- Fix TrimLeft → TrimPrefix to strip only one underscore separator - Add registryName to indexedResult for parallel path cache - Use registryName for bootstrapToolAllowlist and loopDetector checks - Add server-side sanitization for ToolCallPrefix input - Remove dead chatMessages alias and unrelated ParseStripAssistantPrefill - Remove orphaned stripAssistantPrefill i18n keys - Add 13 unit tests for StripToolPrefix
39 lines
1.3 KiB
Go
39 lines
1.3 KiB
Go
package tools
|
|
|
|
import "testing"
|
|
|
|
func TestStripToolPrefix(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
tmpl string
|
|
input string
|
|
expect string
|
|
}{
|
|
// Template-based ({tool_name} placeholder)
|
|
{"template match", "proxy_{tool_name}", "proxy_exec", "exec"},
|
|
{"template with suffix", "{tool_name}_v2", "exec_v2", "exec"},
|
|
{"template prefix+suffix", "pre_{tool_name}_suf", "pre_exec_suf", "exec"},
|
|
{"template no match", "proxy_{tool_name}", "other_exec", "other_exec"},
|
|
{"template empty result", "proxy_{tool_name}", "proxy_", "proxy_"},
|
|
{"template exact placeholder", "{tool_name}", "exec", "exec"},
|
|
|
|
// Literal prefix
|
|
{"literal prefix with separator", "proxy_", "proxy_exec", "exec"},
|
|
{"literal prefix no match", "proxy_", "other_exec", "other_exec"},
|
|
{"literal prefix underscore join", "proxy", "proxy_exec", "exec"},
|
|
{"literal prefix single underscore strip", "proxy", "proxy__exec", "_exec"},
|
|
{"literal prefix empty after strip", "proxy_", "proxy_", "proxy_"},
|
|
{"literal name shorter than prefix", "proxy_", "pro", "pro"},
|
|
{"literal name equals prefix", "proxy", "proxy", "proxy"},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got := StripToolPrefix(tt.tmpl, tt.input)
|
|
if got != tt.expect {
|
|
t.Errorf("StripToolPrefix(%q, %q) = %q, want %q", tt.tmpl, tt.input, got, tt.expect)
|
|
}
|
|
})
|
|
}
|
|
}
|