li
32a5043bdb
- Output Session-ID to stderr immediately when backend reports session_started, so Claude can capture it even if task times out - Fix Command: display line to show actual task text instead of misleading -p - for Gemini backend (geminiDirect logic) - Bump codeagent-wrapper 5.8.0 → 5.9.0 - Bump EXPECTED_BINARY_VERSION to 5.9.0
116 lines
3.3 KiB
Go
116 lines
3.3 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestBackendParseJSONStream_UnknownEventsAreSilent(t *testing.T) {
|
|
input := strings.Join([]string{
|
|
`{"type":"turn.started"}`,
|
|
`{"type":"assistant","text":"hi"}`,
|
|
`{"type":"user","text":"yo"}`,
|
|
`{"type":"item.completed","item":{"type":"agent_message","text":"ok"}}`,
|
|
}, "\n")
|
|
|
|
var infos []string
|
|
infoFn := func(msg string) { infos = append(infos, msg) }
|
|
|
|
message, threadID := parseJSONStreamInternal(strings.NewReader(input), nil, infoFn, nil, nil)
|
|
if message != "ok" {
|
|
t.Fatalf("message=%q, want %q (infos=%v)", message, "ok", infos)
|
|
}
|
|
if threadID != "" {
|
|
t.Fatalf("threadID=%q, want empty (infos=%v)", threadID, infos)
|
|
}
|
|
|
|
for _, msg := range infos {
|
|
if strings.Contains(msg, "Agent event:") {
|
|
t.Fatalf("unexpected log for unknown event: %q", msg)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestParseJSONStreamInternalWithContent_EmitsProgressLines(t *testing.T) {
|
|
input := strings.Join([]string{
|
|
`{"type":"thread.started","thread_id":"tid-123"}`,
|
|
`{"type":"turn.started"}`,
|
|
`{"type":"item.completed","item":{"type":"reasoning","text":"Checking files and APIs"}}`,
|
|
`{"type":"item.completed","item":{"type":"mcp_tool_call"}}`,
|
|
`{"type":"item.completed","item":{"type":"command_execution","command":"echo hi","aggregated_output":"hi\n","exit_code":0}}`,
|
|
`{"type":"item.completed","item":{"type":"agent_message","text":"Done with changes"}}`,
|
|
`{"type":"turn.completed","thread_id":"tid-123"}`,
|
|
}, "\n")
|
|
|
|
var progress []string
|
|
message, threadID := parseJSONStreamInternalWithContent(
|
|
strings.NewReader(input),
|
|
nil,
|
|
func(string) {},
|
|
nil,
|
|
nil,
|
|
nil,
|
|
func(line string) { progress = append(progress, line) },
|
|
nil,
|
|
)
|
|
|
|
if message != "Done with changes" {
|
|
t.Fatalf("message=%q, want %q", message, "Done with changes")
|
|
}
|
|
if threadID != "tid-123" {
|
|
t.Fatalf("threadID=%q, want %q", threadID, "tid-123")
|
|
}
|
|
|
|
joined := strings.Join(progress, "\n")
|
|
for _, want := range []string{
|
|
"[PROGRESS] session_started id=tid-123",
|
|
"[PROGRESS] turn_started",
|
|
"[PROGRESS] reasoning text=\"Checking files and APIs\"",
|
|
"[PROGRESS] mcp_call",
|
|
"[PROGRESS] cmd_done cmd=\"echo hi\" exit=0",
|
|
"[PROGRESS] message text=\"Done with changes\"",
|
|
"[PROGRESS] turn_completed total_events=7",
|
|
} {
|
|
if !strings.Contains(joined, want) {
|
|
t.Fatalf("missing progress %q in %q", want, joined)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestSafeProgressSnippet_UsesRuneSafeTruncation(t *testing.T) {
|
|
got := safeProgressSnippet("中文测试进度输出", 5)
|
|
if got != "中文..." {
|
|
t.Fatalf("got %q, want %q", got, "中文...")
|
|
}
|
|
|
|
got = safeProgressSnippet("中文", 2)
|
|
if got != "中文" {
|
|
t.Fatalf("got %q, want %q", got, "中文")
|
|
}
|
|
}
|
|
|
|
func TestFormatProgressLine_HandlesNilFields(t *testing.T) {
|
|
if got := formatProgressLine("turn_started", nil); got != "turn_started" {
|
|
t.Fatalf("got %q, want %q", got, "turn_started")
|
|
}
|
|
}
|
|
|
|
func TestParseArgs_ParsesProgressFlag(t *testing.T) {
|
|
oldArgs := os.Args
|
|
defer func() { os.Args = oldArgs }()
|
|
|
|
os.Args = []string{"codeagent-wrapper", "--progress", "task body", "/tmp/work"}
|
|
cfg, err := parseArgs()
|
|
if err != nil {
|
|
t.Fatalf("parseArgs error: %v", err)
|
|
}
|
|
if !cfg.Progress {
|
|
t.Fatalf("expected Progress=true")
|
|
}
|
|
if cfg.Task != "task body" || cfg.WorkDir != "/tmp/work" {
|
|
t.Fatalf("unexpected cfg: %+v", cfg)
|
|
}
|
|
}
|
|
|