feat(atlas): update resolvers and index for lineage-aware session resolution

- Update recent-model-resolver for session origin awareness - Update resolve-active-boulder-session with lineage support - Add comprehensive test coverage for boulder session resolution - Add fallback tests for recent model resolver - Update index tests for new lineage tracking 🤖 Generated with assistance of OhMyOpenCode
2026-04-05 17:15:17 +09:00
parent d12c74120d
commit d1be22fb1b
6 changed files with 281 additions and 43 deletions
@@ -0,0 +1,99 @@
+import { afterEach, beforeEach, describe, expect, test } from "bun:test"
+import { existsSync, mkdirSync, rmSync, writeFileSync } from "node:fs"
+import { tmpdir } from "node:os"
+import { join } from "node:path"
+import { randomUUID } from "node:crypto"
+import { clearBoulderState, writeBoulderState } from "../../features/boulder-state"
+import { resolveActiveBoulderSession } from "./resolve-active-boulder-session"
+
+describe("resolveActiveBoulderSession", () => {
+  let testDirectory = ""
+
+  beforeEach(() => {
+    testDirectory = join(tmpdir(), `resolve-active-boulder-${randomUUID()}`)
+    if (!existsSync(testDirectory)) {
+      mkdirSync(testDirectory, { recursive: true })
+    }
+    clearBoulderState(testDirectory)
+  })
+
+  afterEach(() => {
+    clearBoulderState(testDirectory)
+    if (existsSync(testDirectory)) {
+      rmSync(testDirectory, { recursive: true, force: true })
+    }
+  })
+
+  test("returns null for unrelated session even when active boulder plan is complete", async () => {
+    // given
+    const planPath = join(testDirectory, "complete-plan.md")
+    writeFileSync(planPath, "# Plan\n- [x] Task 1\n", "utf-8")
+    writeBoulderState(testDirectory, {
+      active_plan: planPath,
+      started_at: "2026-01-02T10:00:00Z",
+      session_ids: ["ses_tracked"],
+      session_origins: { ses_tracked: "direct" },
+      plan_name: "complete-plan",
+    })
+
+    // when
+    const result = await resolveActiveBoulderSession({
+      client: { session: { get: async () => ({ data: {} }) } } as never,
+      directory: testDirectory,
+      sessionID: "ses_unrelated",
+    })
+
+    // then
+    expect(result).toBeNull()
+  })
+
+  test("returns tracked direct session for incomplete boulder plan", async () => {
+    // given
+    const planPath = join(testDirectory, "incomplete-plan.md")
+    writeFileSync(planPath, "# Plan\n- [ ] Task 1\n", "utf-8")
+    writeBoulderState(testDirectory, {
+      active_plan: planPath,
+      started_at: "2026-01-02T10:00:00Z",
+      session_ids: ["ses_tracked"],
+      session_origins: { ses_tracked: "direct" },
+      plan_name: "incomplete-plan",
+    })
+
+    // when
+    const result = await resolveActiveBoulderSession({
+      client: { session: { get: async () => ({ data: {} }) } } as never,
+      directory: testDirectory,
+      sessionID: "ses_tracked",
+    })
+
+    // then
+    expect(result).not.toBeNull()
+    expect(result?.progress.isComplete).toBe(false)
+    expect(result?.boulderState.session_ids).toContain("ses_tracked")
+  })
+
+  test("returns tracked appended session for incomplete boulder plan", async () => {
+    // given
+    const planPath = join(testDirectory, "appended-incomplete-plan.md")
+    writeFileSync(planPath, "# Plan\n- [ ] Task 1\n", "utf-8")
+    writeBoulderState(testDirectory, {
+      active_plan: planPath,
+      started_at: "2026-01-02T10:00:00Z",
+      session_ids: ["ses_root", "ses_appended"],
+      session_origins: { ses_root: "direct", ses_appended: "appended" },
+      plan_name: "appended-incomplete-plan",
+    })
+
+    // when
+    const result = await resolveActiveBoulderSession({
+      client: { session: { get: async () => ({ data: {} }) } } as never,
+      directory: testDirectory,
+      sessionID: "ses_appended",
+    })
+
+    // then
+    expect(result).not.toBeNull()
+    expect(result?.progress.isComplete).toBe(false)
+    expect(result?.boulderState.session_ids).toContain("ses_appended")
+  })
+})