1package provider
2
3import (
4 "reflect"
5 "testing"
6)
7
8func pairingResult(id, text string) Message {
9 return Message{Role: RoleTool, Content: []Content{ToolResultBlock{CallID: id, Content: []Content{TextBlock{Text: text}}}}}
10}
11
12func pairingCall(id string) Message {
13 return Message{Role: RoleAssistant, Content: []Content{ToolCallBlock{ID: id, Name: "read"}}}
14}
15
16func TestRepairOrphanedToolResultsPreservesRepeatedToolCallPairs(t *testing.T) {
17 msgs := []Message{pairingCall("reused"), pairingResult("reused", "first"), pairingCall("reused"), pairingResult("reused", "second")}
18 got := RepairOrphanedToolResults(msgs)
19 if !reflect.DeepEqual(got, msgs) {
20 t.Fatalf("got %#v, want both repeated-ID pairs preserved", got)
21 }
22}
23
24func TestAnthropicBuildRequestPreservesRepeatedToolCallPairs(t *testing.T) {
25 c := NewAnthropic("test-key", "").(*anthropicClient)
26 wire, err := c.buildRequest(Request{
27 Model: "claude-sonnet-4-5",
28 Messages: []Message{
29 pairingCall("reused"), pairingResult("reused", "first"),
30 pairingCall("reused"), pairingResult("reused", "second"),
31 },
32 })
33 if err != nil {
34 t.Fatal(err)
35 }
36 if len(wire.Messages) != 4 {
37 t.Fatalf("messages=%d, want both repeated-ID tool result pairs", len(wire.Messages))
38 }
39 for _, index := range []int{1, 3} {
40 if len(wire.Messages[index].Content) != 1 {
41 t.Fatalf("message %d content=%d, want 1 tool result", index, len(wire.Messages[index].Content))
42 }
43 if _, ok := wire.Messages[index].Content[0].(anthToolResultBlock); !ok {
44 t.Fatalf("message %d content type=%T, want anthToolResultBlock", index, wire.Messages[index].Content[0])
45 }
46 }
47}
48
49func TestRepairOrphanedToolResultsDropsLeadingOrphanBeforeReusedID(t *testing.T) {
50 user := Message{Role: RoleUser, Content: []Content{TextBlock{Text: "next turn"}}}
51 messages := []Message{pairingResult("reused", "orphan"), user, pairingCall("reused"), pairingResult("reused", "valid")}
52 want := messages[1:]
53 if got := RepairOrphanedToolResults(messages); !reflect.DeepEqual(got, want) {
54 t.Fatalf("got %#v, want leading orphan removed and later valid pair retained", got)
55 }
56}
57
58func TestRepairOrphanedToolResultsDropsExcessAndOrphans(t *testing.T) {
59 msgs := []Message{
60 pairingCall("a"), pairingResult("a", "one"), pairingResult("a", "dup"), pairingResult("orphan", "x"),
61 }
62 got := RepairOrphanedToolResults(msgs)
63 if len(got) != 2 {
64 t.Fatalf("messages=%d, want 2", len(got))
65 }
66 if tr := got[1].Content[0].(ToolResultBlock); tr.Content[0].(TextBlock).Text != "one" {
67 t.Fatalf("kept wrong result: %#v", tr)
68 }
69}
70
71func TestRepairOrphanedToolResultsKeepsMixedTextAndDoesNotMutate(t *testing.T) {
72 msgs := []Message{
73 pairingCall("a"),
74 {Role: RoleTool, Content: []Content{
75 TextBlock{Text: "note"},
76 ToolResultBlock{CallID: "a"},
77 ToolResultBlock{CallID: "a"},
78 ToolResultBlock{CallID: "orphan"},
79 }},
80 }
81 orig := []Message{msgs[0], {Role: RoleTool, Content: append([]Content(nil), msgs[1].Content...)}}
82 got := RepairOrphanedToolResults(msgs)
83 want := []Content{TextBlock{Text: "note"}, ToolResultBlock{CallID: "a"}}
84 if len(got) != 2 || !reflect.DeepEqual(got[1].Content, want) {
85 t.Fatalf("got %#v, want %#v", got, want)
86 }
87 if !reflect.DeepEqual(msgs, orig) {
88 t.Fatalf("input mutated: %#v", msgs)
89 }
90}