cexll/myclaude
1
0
Fork 0
mirror of https://github.com/cexll/myclaude.git synced 2026-02-14 03:31:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

test(ParseJSONStream): 增加对超大单行文本和非字符串文本的处理测试

Browse Source
This commit is contained in:
freespace8
2025-11-28 15:10:47 +08:00
parent aea19f0e1f
commit 3df4fec6dd
1 changed files with 22 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
codex-wrapper/main_test.go
View File

@@ -330,12 +330,16 @@ func TestNormalizeText(t *testing.T) {
} }
func TestParseJSONStream(t *testing.T) { func TestParseJSONStream(t *testing.T) {
tests := []struct { type testCase struct {
name string name string
input string input string
wantMessage string wantMessage string
wantThreadID string wantThreadID string
}{ }
longText := strings.Repeat("a", 2*1024*1024) // >1MB agent_message payload
tests := []testCase{
{ {
name: "thread started and agent message", name: "thread started and agent message",
input: `{"type":"thread.started","thread_id":"abc-123"} input: `{"type":"thread.started","thread_id":"abc-123"}
@@ -364,6 +368,12 @@ func TestParseJSONStream(t *testing.T) {
wantMessage: "Valid", wantMessage: "Valid",
wantThreadID: "", wantThreadID: "",
}, },
{
name: "super long single line (>1MB)",
input: `{"type":"item.completed","item":{"type":"agent_message","text":"` + longText + `"}}`,
wantMessage: longText,
wantThreadID: "",
},
{ {
name: "empty input", name: "empty input",
input: "", input: "",
@@ -371,23 +381,25 @@ func TestParseJSONStream(t *testing.T) {
wantThreadID: "", wantThreadID: "",
}, },
{ {
name: "invalid JSON (skipped)", name: "item completed with nil item",
input: "not valid json\n{\"type\":\"thread.started\",\"thread_id\":\"xyz\"}", input: strings.Join([]string{
`{"type":"thread.started","thread_id":"nil-item-thread"}`,
`{"type":"item.completed","item":null}`,
}, "\n"),
wantMessage: "", wantMessage: "",
wantThreadID: "xyz", wantThreadID: "nil-item-thread",
}, },
{ {
name: "blank lines ignored", name: "agent message with non-string text",
input: "\n\n{\"type\":\"thread.started\",\"thread_id\":\"test\"}\n\n", input: `{"type":"item.completed","item":{"type":"agent_message","text":12345}}`,
wantMessage: "", wantMessage: "",
wantThreadID: "test", wantThreadID: "",
}, },
} }
for _, tt := range tests { for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) { t.Run(tt.name, func(t *testing.T) {
r := strings.NewReader(tt.input) gotMessage, gotThreadID := parseJSONStream(strings.NewReader(tt.input))
gotMessage, gotThreadID := parseJSONStream(r)
if gotMessage != tt.wantMessage { if gotMessage != tt.wantMessage {
t.Errorf("parseJSONStream() message = %q, want %q", gotMessage, tt.wantMessage) t.Errorf("parseJSONStream() message = %q, want %q", gotMessage, tt.wantMessage)