wehub-resource-sync
57 lines
1.7 KiB
Go
57 lines
1.7 KiB
Go
package provider
|
|
|
|
import "testing"
|
|
|
|
// TestSanitizeDuplicateToolCallIDsKeepsEachResult probes whether a malformed
|
|
// history with two tool calls sharing one id round-trips both results. The map
|
|
// keyed on id collapses them, so the loop guard's safety net silently drops one.
|
|
func TestSanitizeDuplicateToolCallIDsKeepsEachResult(t *testing.T) {
|
|
msgs := []Message{
|
|
{Role: RoleUser, Content: "go"},
|
|
{Role: RoleAssistant, ToolCalls: []ToolCall{
|
|
{ID: "dup", Name: "read_file"},
|
|
{ID: "dup", Name: "grep"},
|
|
}},
|
|
{Role: RoleTool, ToolCallID: "dup", Name: "read_file", Content: "FILE-RESULT"},
|
|
{Role: RoleTool, ToolCallID: "dup", Name: "grep", Content: "GREP-RESULT"},
|
|
}
|
|
out := SanitizeToolPairing(msgs)
|
|
|
|
var got []string
|
|
for _, m := range out {
|
|
if m.Role == RoleTool {
|
|
got = append(got, m.Content)
|
|
}
|
|
}
|
|
if len(got) != 2 {
|
|
t.Fatalf("want 2 tool results, got %d: %v", len(got), got)
|
|
}
|
|
if got[0] == got[1] {
|
|
t.Errorf("both tool results collapsed to the same content %q — one was lost", got[0])
|
|
}
|
|
}
|
|
|
|
// TestSanitizeEmptyToolCallIDs probes two calls with empty ids — same collapse
|
|
// risk, and the placeholder/backfill path keys on "" too.
|
|
func TestSanitizeEmptyToolCallIDsKeepsEachResult(t *testing.T) {
|
|
msgs := []Message{
|
|
{Role: RoleAssistant, ToolCalls: []ToolCall{
|
|
{ID: "", Name: "a"},
|
|
{ID: "", Name: "b"},
|
|
}},
|
|
{Role: RoleTool, ToolCallID: "", Name: "a", Content: "RESULT-A"},
|
|
{Role: RoleTool, ToolCallID: "", Name: "b", Content: "RESULT-B"},
|
|
}
|
|
out := SanitizeToolPairing(msgs)
|
|
|
|
var got []string
|
|
for _, m := range out {
|
|
if m.Role == RoleTool {
|
|
got = append(got, m.Content)
|
|
}
|
|
}
|
|
if len(got) != 2 || got[0] == got[1] {
|
|
t.Errorf("empty-id pair collapsed: %v", got)
|
|
}
|
|
}
|