openAgent

1import { describe, expect } from "bun:test"
2import { asc } from "drizzle-orm"
3import { Effect } from "effect"
4import { Database } from "@opencode-ai/core/database/database"
5import { LayerNode } from "@opencode-ai/core/effect/layer-node"
6import { AppNodeBuilder } from "@opencode-ai/core/effect/app-node-builder"
7import { EventV2 } from "@opencode-ai/core/event"
8import { Project } from "@opencode-ai/core/project"
9import { ProjectTable } from "@opencode-ai/core/project/sql"
10import { AbsolutePath } from "@opencode-ai/core/schema"
11import { SessionV2 } from "@opencode-ai/core/session"
12import { SessionTable, TodoTable } from "@opencode-ai/core/session/sql"
13import { SessionTodo } from "@opencode-ai/core/session/todo"
14import { testEffect } from "./lib/effect"
15 
16const it = testEffect(AppNodeBuilder.build(LayerNode.group([Database.node, EventV2.node, SessionTodo.node])))
17const sessionID = SessionV2.ID.make("ses_todo_test")
18 
19const setup = Effect.gen(function* () {
20  const { db } = yield* Database.Service
21  yield* db
22    .insert(ProjectTable)
23    .values({ id: Project.ID.global, worktree: AbsolutePath.make("/project"), sandboxes: [] })
24    .run()
25    .pipe(Effect.orDie)
26  yield* db
27    .insert(SessionTable)
28    .values({
29      id: sessionID,
30      project_id: Project.ID.global,
31      slug: "todo",
32      directory: "/project",
33      title: "todo",
34      version: "test",
35    })
36    .run()
37    .pipe(Effect.orDie)
38})
39 
40describe("SessionTodo", () => {
41  it.effect("replaces persisted todos in order and publishes updates", () =>
42    Effect.gen(function* () {
43      yield* setup
44      const { db } = yield* Database.Service
45      const events = yield* EventV2.Service
46      const todos = yield* SessionTodo.Service
47      const published = new Array<EventV2.Payload>()
48      const unsubscribe = yield* events.listen((event) =>
49        Effect.sync(() => {
50          if (event.type === SessionTodo.Event.Updated.type) published.push(event)
51        }),
52      )
53      yield* Effect.addFinalizer(() => unsubscribe)
54 
55      yield* todos.update({
56        sessionID,
57        todos: [
58          { content: "second", status: "pending", priority: "low" },
59          { content: "first", status: "in_progress", priority: "high" },
60        ],
61      })
62      expect(yield* todos.get(sessionID)).toEqual([
63        { content: "second", status: "pending", priority: "low" },
64        { content: "first", status: "in_progress", priority: "high" },
65      ])
66      expect(
67        (yield* db.select().from(TodoTable).orderBy(asc(TodoTable.position)).all().pipe(Effect.orDie)).map((row) => ({
68          content: row.content,
69          position: row.position,
70        })),
71      ).toEqual([
72        { content: "second", position: 0 },
73        { content: "first", position: 1 },
74      ])
75 
76      yield* todos.update({ sessionID, todos: [{ content: "replacement", status: "completed", priority: "medium" }] })
77      expect(yield* todos.get(sessionID)).toEqual([{ content: "replacement", status: "completed", priority: "medium" }])
78 
79      yield* todos.update({ sessionID, todos: [] })
80      expect(yield* todos.get(sessionID)).toEqual([])
81      expect(published.map((event) => event.data)).toEqual([
82        {
83          sessionID,
84          todos: [
85            { content: "second", status: "pending", priority: "low" },
86            { content: "first", status: "in_progress", priority: "high" },
87          ],
88        },
89        { sessionID, todos: [{ content: "replacement", status: "completed", priority: "medium" }] },
90        { sessionID, todos: [] },
91      ])
92    }),
93  )
94})
95