| 1 | package doctor |
| 2 | |
| 3 | import ( |
| 4 | "archive/zip" |
| 5 | "encoding/json" |
| 6 | "io" |
| 7 | "os" |
| 8 | "path/filepath" |
| 9 | "sort" |
| 10 | "strings" |
| 11 | "testing" |
| 12 | "time" |
| 13 | |
| 14 | "reasonix/internal/agent" |
| 15 | "reasonix/internal/store" |
| 16 | ) |
| 17 | |
| 18 | func TestWriteSessionBundleIncludesRecoveryChain(t *testing.T) { |
| 19 | home := t.TempDir() |
| 20 | t.Setenv("REASONIX_HOME", home) |
| 21 | |
| 22 | dir := filepath.Join(home, "projects", "workspace", "sessions") |
| 23 | if err := os.MkdirAll(dir, 0o755); err != nil { |
| 24 | t.Fatal(err) |
| 25 | } |
| 26 | parent := filepath.Join(dir, "parent-session.jsonl") |
| 27 | recovery := filepath.Join(dir, "parent-session-recovery-deadbeef.jsonl") |
| 28 | writeFile(t, parent, `{"role":"user","content":"parent"}`+"\n") |
| 29 | writeFile(t, recovery, `{"role":"user","content":"recovery"}`+"\n") |
| 30 | if err := agent.SaveBranchMeta(parent, agent.BranchMeta{ID: "parent-session"}); err != nil { |
| 31 | t.Fatal(err) |
| 32 | } |
| 33 | if err := agent.SaveBranchMeta(recovery, agent.BranchMeta{ |
| 34 | ID: "parent-session-recovery-deadbeef", |
| 35 | ParentID: "parent-session", |
| 36 | Recovered: true, |
| 37 | RecoveryReason: "snapshot conflict", |
| 38 | RecoveryDepth: 1, |
| 39 | }); err != nil { |
| 40 | t.Fatal(err) |
| 41 | } |
| 42 | writeFile(t, store.SessionEventLog(recovery), `{"type":"replace"}`+"\n") |
| 43 | writeFile(t, store.SessionEventLogDamaged(recovery), `{"damaged_tail":true}`+"\ntorn"+"\n") |
| 44 | writeFile(t, store.SessionEventIndex(recovery), `{"log_size":19}`+"\n") |
| 45 | writeFile(t, store.SessionConflictLog(recovery), `{"outcome":"forked_recovery_branch"}`+"\n") |
| 46 | writeFile(t, store.SessionLeaseInfo(recovery), `{"pid":1234}`+"\n") |
| 47 | |
| 48 | out := filepath.Join(t.TempDir(), "diag.zip") |
| 49 | got, err := WriteSessionBundle(SessionBundleOptions{ |
| 50 | Version: "test-version", |
| 51 | SessionRef: store.SessionMeta(recovery), |
| 52 | OutputPath: out, |
| 53 | Now: time.Unix(1700000000, 0), |
| 54 | }) |
| 55 | if err != nil { |
| 56 | t.Fatalf("WriteSessionBundle: %v", err) |
| 57 | } |
| 58 | if got.Path != out { |
| 59 | t.Fatalf("bundle path = %q, want %q", got.Path, out) |
| 60 | } |
| 61 | if got.SessionPath != recovery { |
| 62 | t.Fatalf("session path = %q, want %q", got.SessionPath, recovery) |
| 63 | } |
| 64 | |
| 65 | files := zipFiles(t, out) |
| 66 | for _, want := range []string{ |
| 67 | "doctor.json", |
| 68 | "manifest.json", |
| 69 | "sessions/parent-session-recovery-deadbeef/parent-session-recovery-deadbeef.jsonl", |
| 70 | "sessions/parent-session-recovery-deadbeef/parent-session-recovery-deadbeef.jsonl.meta", |
| 71 | "sessions/parent-session-recovery-deadbeef/parent-session-recovery-deadbeef.events.jsonl", |
| 72 | "sessions/parent-session-recovery-deadbeef/parent-session-recovery-deadbeef.events.jsonl.damaged", |
| 73 | "sessions/parent-session-recovery-deadbeef/parent-session-recovery-deadbeef.event-index.json", |
| 74 | "sessions/parent-session-recovery-deadbeef/parent-session-recovery-deadbeef.conflicts.jsonl", |
| 75 | "sessions/parent-session-recovery-deadbeef/parent-session-recovery-deadbeef.jsonl.lease.json", |
| 76 | "sessions/parent-session/parent-session.jsonl", |
| 77 | "sessions/parent-session/parent-session.jsonl.meta", |
| 78 | } { |
| 79 | if _, ok := files[want]; !ok { |
| 80 | t.Fatalf("zip missing %s; entries=%v", want, keys(files)) |
| 81 | } |
| 82 | } |
| 83 | |
| 84 | var manifest SessionBundleManifest |
| 85 | if err := json.Unmarshal(files["manifest.json"], &manifest); err != nil { |
| 86 | t.Fatalf("manifest JSON: %v", err) |
| 87 | } |
| 88 | if strings.Contains(string(files["manifest.json"]), home) { |
| 89 | t.Fatalf("manifest leaked REASONIX_HOME path:\n%s", files["manifest.json"]) |
| 90 | } |
| 91 | if manifest.Version != "test-version" { |
| 92 | t.Fatalf("manifest version = %q", manifest.Version) |
| 93 | } |
| 94 | if !strings.Contains(manifest.RequestedRef, "<REASONIX_HOME>") { |
| 95 | t.Fatalf("requested ref = %q, want redacted REASONIX_HOME path", manifest.RequestedRef) |
| 96 | } |
| 97 | if len(manifest.Sessions) != 2 { |
| 98 | t.Fatalf("manifest sessions = %+v, want recovery plus parent", manifest.Sessions) |
| 99 | } |
| 100 | if manifest.Sessions[0].BranchID != "parent-session-recovery-deadbeef" || manifest.Sessions[0].ParentID != "parent-session" { |
| 101 | t.Fatalf("recovery manifest entry = %+v", manifest.Sessions[0]) |
| 102 | } |
| 103 | } |
| 104 | |
| 105 | func writeFile(t *testing.T, path, data string) { |
| 106 | t.Helper() |
| 107 | if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil { |
| 108 | t.Fatal(err) |
| 109 | } |
| 110 | if err := os.WriteFile(path, []byte(data), 0o644); err != nil { |
| 111 | t.Fatal(err) |
| 112 | } |
| 113 | } |
| 114 | |
| 115 | func zipFiles(t *testing.T, path string) map[string][]byte { |
| 116 | t.Helper() |
| 117 | zr, err := zip.OpenReader(path) |
| 118 | if err != nil { |
| 119 | t.Fatal(err) |
| 120 | } |
| 121 | defer zr.Close() |
| 122 | out := map[string][]byte{} |
| 123 | for _, f := range zr.File { |
| 124 | rc, err := f.Open() |
| 125 | if err != nil { |
| 126 | t.Fatal(err) |
| 127 | } |
| 128 | data, err := io.ReadAll(rc) |
| 129 | _ = rc.Close() |
| 130 | if err != nil { |
| 131 | t.Fatal(err) |
| 132 | } |
| 133 | out[f.Name] = data |
| 134 | } |
| 135 | return out |
| 136 | } |
| 137 | |
| 138 | func keys(m map[string][]byte) []string { |
| 139 | out := make([]string, 0, len(m)) |
| 140 | for k := range m { |
| 141 | out = append(out, k) |
| 142 | } |
| 143 | sort.Strings(out) |
| 144 | return out |
| 145 | } |
| 146 |