session_repeated_tool_id_test.go RawHistoryFolder

1package core
2
3import (
4	"reflect"
5	"testing"
6
7	"github.com/bnema/zut/packages/provider"
8)
9
10func TestRepairToolUseResultPairsPreservesRepeatedIDsAcrossTurns(t *testing.T) {
11	messages := []provider.Message{
12		{Role: provider.RoleAssistant, Content: []provider.Content{provider.ToolCallBlock{ID: "reused", Name: "read"}}},
13		{Role: provider.RoleTool, Content: []provider.Content{provider.ToolResultBlock{CallID: "reused", Content: []provider.Content{provider.TextBlock{Text: "first"}}}}},
14		{Role: provider.RoleAssistant, Content: []provider.Content{provider.ToolCallBlock{ID: "reused", Name: "read"}}},
15		{Role: provider.RoleTool, Content: []provider.Content{provider.ToolResultBlock{CallID: "reused", Content: []provider.Content{provider.TextBlock{Text: "second"}}}}},
16	}
17	repaired := repairToolUseResultPairs(messages)
18	if !reflect.DeepEqual(repaired, messages) {
19		t.Fatal("session repair changed valid repeated-ID pairs")
20	}
21	if wire := provider.RepairOrphanedToolResults(repaired); !reflect.DeepEqual(wire, messages) {
22		t.Fatal("provider repair dropped a valid pair after session repair")
23	}
24}