| 1 | package provider |
| 2 | |
| 3 | import ( |
| 4 | "encoding/json" |
| 5 | "reflect" |
| 6 | "testing" |
| 7 | ) |
| 8 | |
| 9 | func TestReadCompletionAndDiagnosticStayHostOnly(t *testing.T) { |
| 10 | receipt := &ReadCompletion{ID: "run", Reads: []CompletedRead{{ReadID: "r", Path: "file", Snapshot: "s", Intent: "range", Verdict: "partial_read_sufficient", Covered: [][2]int{{0, 2}}}}} |
| 11 | message := Message{Role: RoleTool, Name: LocalOnlyToolName, ToolCallID: LocalOnlyToolID, LocalOnly: true, ReadCompletion: receipt} |
| 12 | b, err := json.Marshal(message) |
| 13 | if err != nil { |
| 14 | t.Fatal(err) |
| 15 | } |
| 16 | var restored Message |
| 17 | if err := json.Unmarshal(b, &restored); err != nil || !reflect.DeepEqual(restored.ReadCompletion, receipt) { |
| 18 | t.Fatalf("round trip: %s %v", b, err) |
| 19 | } |
| 20 | base := []Message{{Role: RoleUser, Content: "continue"}, {Role: RoleAssistant, Content: "done"}} |
| 21 | withMeta := append([]Message(nil), base...) |
| 22 | withMeta[1].ReadCompletion = receipt |
| 23 | withMeta[1].ToolDiagnostic = json.RawMessage(`{"code":"WRITE_EVIDENCE_MISSING","path":"private"}`) |
| 24 | withMeta = append(withMeta, restored) |
| 25 | if got := ModelMessages(withMeta); !reflect.DeepEqual(got, base) { |
| 26 | t.Fatalf("provider metadata leaked: %+v", got) |
| 27 | } |
| 28 | var legacy Message |
| 29 | if err := json.Unmarshal([]byte(`{"role":"assistant","content":"legacy"}`), &legacy); err != nil || legacy.ReadCompletion != nil || legacy.ToolDiagnostic != nil { |
| 30 | t.Fatal("legacy message changed") |
| 31 | } |
| 32 | } |
| 33 |