| 1 | package agent |
| 2 | |
| 3 | import ( |
| 4 | "path/filepath" |
| 5 | "testing" |
| 6 | |
| 7 | "reasonix/internal/provider" |
| 8 | ) |
| 9 | |
| 10 | func TestUpdateSessionListingProjectionIfCurrentRejectsAdvancedGeneration(t *testing.T) { |
| 11 | dir := t.TempDir() |
| 12 | path := filepath.Join(dir, "20260101-000000-deepseek-chat.jsonl") |
| 13 | writeSessionFile(t, path, []provider.Message{{Role: provider.RoleUser, Content: "old question"}}) |
| 14 | _, oldState, _, err := LoadSessionDisplayMessages(path) |
| 15 | if err != nil { |
| 16 | t.Fatal(err) |
| 17 | } |
| 18 | if err := SaveBranchMeta(path, BranchMeta{ID: BranchID(path), Revision: 1, ContentDigest: oldState.DigestHex}); err != nil { |
| 19 | t.Fatal(err) |
| 20 | } |
| 21 | _, oldState, _, err = LoadSessionDisplayMessages(path) |
| 22 | if err != nil || !oldState.RevisionKnown { |
| 23 | t.Fatalf("load old state = %+v err=%v", oldState, err) |
| 24 | } |
| 25 | |
| 26 | writeSessionFile(t, path, []provider.Message{{Role: provider.RoleUser, Content: "new question"}}) |
| 27 | _, newState, _, err := LoadSessionDisplayMessages(path) |
| 28 | if err != nil { |
| 29 | t.Fatal(err) |
| 30 | } |
| 31 | if err := UpdateBranchMeta(path, false, func(meta *BranchMeta) error { |
| 32 | meta.Revision = 2 |
| 33 | meta.ContentDigest = newState.DigestHex |
| 34 | meta.Preview = "new question" |
| 35 | meta.Turns = 1 |
| 36 | meta.SchemaVersion = BranchMetaCountsVersion |
| 37 | stampSessionListingProjection(meta) |
| 38 | return nil |
| 39 | }); err != nil { |
| 40 | t.Fatal(err) |
| 41 | } |
| 42 | applied, err := UpdateSessionListingProjectionIfCurrent(path, "", "", "old question", 1, false, oldState) |
| 43 | if err != nil || applied { |
| 44 | t.Fatalf("stale projection applied=%v err=%v", applied, err) |
| 45 | } |
| 46 | meta, ok, err := LoadBranchMeta(path) |
| 47 | if err != nil || !ok || meta.Revision != 2 || meta.Preview != "new question" || meta.ContentDigest != newState.DigestHex { |
| 48 | t.Fatalf("advanced metadata changed: ok=%v err=%v meta=%+v", ok, err, meta) |
| 49 | } |
| 50 | } |
| 51 | |
| 52 | func TestUpdateSessionListingProjectionIfCurrentRepairsMatchingLegacySession(t *testing.T) { |
| 53 | dir := t.TempDir() |
| 54 | path := filepath.Join(dir, "20260101-000000-deepseek-chat.jsonl") |
| 55 | writeSessionFile(t, path, []provider.Message{{Role: provider.RoleUser, Content: "legacy question"}}) |
| 56 | if err := SaveBranchMeta(path, BranchMeta{ID: BranchID(path)}); err != nil { |
| 57 | t.Fatal(err) |
| 58 | } |
| 59 | _, state, _, err := LoadSessionDisplayMessages(path) |
| 60 | if err != nil || state.RevisionKnown { |
| 61 | t.Fatalf("load legacy state = %+v err=%v", state, err) |
| 62 | } |
| 63 | applied, err := UpdateSessionListingProjectionIfCurrent(path, "", "", "legacy question", 1, false, state) |
| 64 | if err != nil || !applied { |
| 65 | t.Fatalf("legacy projection applied=%v err=%v", applied, err) |
| 66 | } |
| 67 | meta, ok, err := LoadBranchMeta(path) |
| 68 | if err != nil || !ok || meta.Preview != "legacy question" || meta.Turns != 1 || meta.SchemaVersion != BranchMetaCountsVersion { |
| 69 | t.Fatalf("legacy projection missing: ok=%v err=%v meta=%+v", ok, err, meta) |
| 70 | } |
| 71 | } |
| 72 | |
| 73 | func TestUpdateSessionListingProjectionIfCurrentStampsMatchingGeneration(t *testing.T) { |
| 74 | dir := t.TempDir() |
| 75 | path := filepath.Join(dir, "20260101-000000-deepseek-chat.jsonl") |
| 76 | writeSessionFile(t, path, []provider.Message{{Role: provider.RoleUser, Content: "current question"}}) |
| 77 | _, state, _, err := LoadSessionDisplayMessages(path) |
| 78 | if err != nil { |
| 79 | t.Fatal(err) |
| 80 | } |
| 81 | if err := SaveBranchMeta(path, BranchMeta{ID: BranchID(path), Revision: 3, ContentDigest: state.DigestHex}); err != nil { |
| 82 | t.Fatal(err) |
| 83 | } |
| 84 | _, state, _, err = LoadSessionDisplayMessages(path) |
| 85 | if err != nil || !state.RevisionKnown { |
| 86 | t.Fatalf("load current state = %+v err=%v", state, err) |
| 87 | } |
| 88 | applied, err := UpdateSessionListingProjectionIfCurrent(path, "", "", "current question", 1, false, state) |
| 89 | if err != nil || !applied { |
| 90 | t.Fatalf("matching projection applied=%v err=%v", applied, err) |
| 91 | } |
| 92 | meta, ok, err := LoadBranchMeta(path) |
| 93 | if err != nil || !ok || meta.ListingRevision != 3 || meta.ListingContentDigest != state.DigestHex || meta.Preview != "current question" { |
| 94 | t.Fatalf("matching projection not stamped: ok=%v err=%v meta=%+v", ok, err, meta) |
| 95 | } |
| 96 | } |
| 97 |