mirror of
https://github.com/chenhg5/cc-connect.git
synced 2026-07-03 12:28:10 +08:00
20268b2b17
ListSessions, GetSessionHistory, DeleteSession, and patchSessionSource all ignored the Agent.codexHome config field and only checked the CODEX_HOME environment variable. When codex_home was set in config.toml, StartSession correctly wrote sessions to the custom directory, but the listing functions searched the default ~/.codex/sessions/ — making all sessions invisible. - Add resolveCodexHomeDir helper (explicit config > env > default) - Thread codexHome through listCodexSessions, findSessionFile, getSessionHistory, and patchSessionSource - Extract CODEX_HOME from session extraEnv for post-exit patching Made-with: Cursor
76 lines
2.2 KiB
Go
76 lines
2.2 KiB
Go
package codex
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestPatchSessionSource(t *testing.T) {
|
|
tmpDir := t.TempDir()
|
|
|
|
sessionID := "test-session-abc123"
|
|
sessionsDir := filepath.Join(tmpDir, ".codex", "sessions")
|
|
if err := os.MkdirAll(sessionsDir, 0o755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
fname := filepath.Join(sessionsDir, "rollout-"+sessionID+".jsonl")
|
|
line1 := `{"timestamp":"2026-01-01T00:00:00Z","type":"session_meta","payload":{"id":"` + sessionID + `","source":"exec","originator":"codex_exec","cwd":"/tmp"}}`
|
|
line2 := `{"timestamp":"2026-01-01T00:00:01Z","type":"response_item","payload":{"role":"user"}}`
|
|
content := line1 + "\n" + line2 + "\n"
|
|
|
|
if err := os.WriteFile(fname, []byte(content), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
codexHome := filepath.Join(tmpDir, ".codex")
|
|
|
|
patchSessionSource(sessionID, codexHome)
|
|
|
|
data, err := os.ReadFile(fname)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
lines := strings.SplitN(string(data), "\n", 2)
|
|
|
|
if !strings.Contains(lines[0], `"source":"cli"`) {
|
|
t.Errorf("expected source:cli, got first line: %s", lines[0])
|
|
}
|
|
if !strings.Contains(lines[0], `"originator":"codex_cli_rs"`) {
|
|
t.Errorf("expected originator:codex_cli_rs, got first line: %s", lines[0])
|
|
}
|
|
if strings.Contains(lines[0], `"source":"exec"`) {
|
|
t.Error("source:exec was not replaced")
|
|
}
|
|
|
|
// Second line should be untouched
|
|
if !strings.HasPrefix(lines[1], `{"timestamp":"2026-01-01T00:00:01Z"`) {
|
|
t.Errorf("second line was corrupted: %s", lines[1])
|
|
}
|
|
}
|
|
|
|
func TestPatchSessionSource_Idempotent(t *testing.T) {
|
|
tmpDir := t.TempDir()
|
|
sessionID := "test-idempotent-xyz"
|
|
sessionsDir := filepath.Join(tmpDir, ".codex", "sessions")
|
|
os.MkdirAll(sessionsDir, 0o755)
|
|
|
|
fname := filepath.Join(sessionsDir, "rollout-"+sessionID+".jsonl")
|
|
line1 := `{"type":"session_meta","payload":{"id":"` + sessionID + `","source":"cli","originator":"codex_cli_rs"}}`
|
|
if err := os.WriteFile(fname, []byte(line1+"\n"), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
codexHome := filepath.Join(tmpDir, ".codex")
|
|
|
|
patchSessionSource(sessionID, codexHome)
|
|
|
|
data, _ := os.ReadFile(fname)
|
|
if string(data) != line1+"\n" {
|
|
t.Errorf("file was modified when it shouldn't have been")
|
|
}
|
|
}
|