| 1 | package provider |
| 2 | |
| 3 | import ( |
| 4 | "encoding/json" |
| 5 | "strings" |
| 6 | "testing" |
| 7 | ) |
| 8 | |
| 9 | func TestReadPauseRoundTripAndProviderIsolation(t *testing.T) { |
| 10 | pause := &ReadPause{ID: "turn-receipt", Reads: []PausedRead{{ReadID: "r", Path: "file", Reason: "page_budget"}}} |
| 11 | msg := Message{Role: RoleTool, ToolCallID: LocalOnlyToolID, Name: LocalOnlyToolName, LocalOnly: true, ReadPause: pause} |
| 12 | data, err := json.Marshal(msg) |
| 13 | if err != nil { |
| 14 | t.Fatal(err) |
| 15 | } |
| 16 | var restored Message |
| 17 | if err := json.Unmarshal(data, &restored); err != nil || restored.ReadPause == nil || restored.ReadPause.Reads[0].Reason != "page_budget" { |
| 18 | t.Fatal("lost pause on reload") |
| 19 | } |
| 20 | input := []Message{{Role: RoleUser, Content: "continue"}, restored, {Role: RoleAssistant, Content: "candidate", ReadPause: pause}} |
| 21 | visible, _ := json.Marshal(ModelMessages(input)) |
| 22 | if strings.Contains(string(visible), "turn-receipt") || strings.Contains(string(visible), "read_pause") { |
| 23 | t.Fatal("pause metadata leaked into provider request") |
| 24 | } |
| 25 | var old Message |
| 26 | if err := json.Unmarshal([]byte(`{"role":"assistant","content":"old"}`), &old); err != nil || old.ReadPause != nil { |
| 27 | t.Fatal("old message compatibility") |
| 28 | } |
| 29 | } |
| 30 |