Add Phase 2 integration tests for SessionDO internal endpoints

packages/control-plane/test/integration/events-messages-list.test.ts

@@ -0,0 +1,169 @@
+import { describe, it, expect } from "vitest";
+import { initSession, seedEvents } from "./helpers";
+
+describe("GET /internal/events", () => {
+  it("lists events with default pagination", async () => {
+    const { stub } = await initSession();
+    const baseTime = Date.now();
+
+    await seedEvents(
+      stub,
+      Array.from({ length: 5 }, (_, i) => ({
+        id: `evt-list-${i}`,
+        type: "tool_call",
+        data: JSON.stringify({ type: "tool_call", tool: "read_file", callId: `c-${i}` }),
+        createdAt: baseTime + i,
+      }))
+    );
+
+    const res = await stub.fetch("http://internal/internal/events?type=tool_call");
+    expect(res.status).toBe(200);
+
+    const body = await res.json<{
+      events: Array<{ id: string; type: string }>;
+      hasMore: boolean;
+    }>();
+
+    const seeded = body.events.filter((e) => e.id.startsWith("evt-list-"));
+    expect(seeded).toHaveLength(5);
+    expect(body.hasMore).toBe(false);
+  });
+
+  it("respects limit parameter", async () => {
+    const { stub } = await initSession();
+    const baseTime = Date.now();
+
+    await seedEvents(
+      stub,
+      Array.from({ length: 10 }, (_, i) => ({
+        id: `evt-lim-${i}`,
+        type: "tool_result",
+        data: JSON.stringify({ type: "tool_result", callId: `c-${i}`, result: "ok" }),
+        createdAt: baseTime + i,
+      }))
+    );
+
+    const res = await stub.fetch("http://internal/internal/events?type=tool_result&limit=3");
+    expect(res.status).toBe(200);
+
+    const body = await res.json<{
+      events: Array<{ id: string }>;
+      hasMore: boolean;
+      cursor: string;
+    }>();
+
+    expect(body.events).toHaveLength(3);
+    expect(body.hasMore).toBe(true);
+    expect(body.cursor).toBeDefined();
+  });
+
+  it("cursor pagination returns next page without overlap", async () => {
+    const { stub } = await initSession();
+    const baseTime = Date.now();
+
+    await seedEvents(
+      stub,
+      Array.from({ length: 7 }, (_, i) => ({
+        id: `evt-page-${i}`,
+        type: "error",
+        data: JSON.stringify({ type: "error", message: `error-${i}` }),
+        createdAt: baseTime + i,
+      }))
+    );
+
+    // Page 1
+    const res1 = await stub.fetch("http://internal/internal/events?type=error&limit=3");
+    const page1 = await res1.json<{
+      events: Array<{ id: string }>;
+      cursor: string;
+      hasMore: boolean;
+    }>();
+    expect(page1.events).toHaveLength(3);
+    expect(page1.hasMore).toBe(true);
+
+    // Page 2
+    const res2 = await stub.fetch(
+      `http://internal/internal/events?type=error&limit=3&cursor=${page1.cursor}`
+    );
+    const page2 = await res2.json<{
+      events: Array<{ id: string }>;
+      hasMore: boolean;
+    }>();
+
+    // No overlap between pages
+    const page1Ids = new Set(page1.events.map((e) => e.id));
+    for (const event of page2.events) {
+      expect(page1Ids.has(event.id)).toBe(false);
+    }
+  });
+
+  it("filters events by type", async () => {
+    const { stub } = await initSession();
+    const baseTime = Date.now();
+
+    await seedEvents(stub, [
+      {
+        id: "evt-filter-tc",
+        type: "tool_call",
+        data: JSON.stringify({ type: "tool_call", tool: "write_file" }),
+        createdAt: baseTime,
+      },
+      {
+        id: "evt-filter-tr",
+        type: "tool_result",
+        data: JSON.stringify({ type: "tool_result", callId: "c1", result: "done" }),
+        createdAt: baseTime + 1,
+      },
+      {
+        id: "evt-filter-tc2",
+        type: "tool_call",
+        data: JSON.stringify({ type: "tool_call", tool: "read_file" }),
+        createdAt: baseTime + 2,
+      },
+    ]);
+
+    const res = await stub.fetch("http://internal/internal/events?type=tool_call");
+    const body = await res.json<{ events: Array<{ id: string; type: string }> }>();
+
+    const seeded = body.events.filter((e) => e.id.startsWith("evt-filter-tc"));
+    expect(seeded).toHaveLength(2);
+    for (const event of seeded) {
+      expect(event.type).toBe("tool_call");
+    }
+  });
+});
+
+describe("GET /internal/messages", () => {
+  it("lists messages with status filter", async () => {
+    const { stub } = await initSession();
+
+    // Enqueue two prompts
+    const res1 = await stub.fetch("http://internal/internal/prompt", {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ content: "First prompt", authorId: "user-1", source: "web" }),
+    });
+    const { messageId: msgId1 } = await res1.json<{ messageId: string }>();
+
+    const res2 = await stub.fetch("http://internal/internal/prompt", {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ content: "Second prompt", authorId: "user-1", source: "web" }),
+    });
+    const { messageId: msgId2 } = await res2.json<{ messageId: string }>();
+
+    // Check that messages are listed
+    const listRes = await stub.fetch("http://internal/internal/messages");
+    expect(listRes.status).toBe(200);
+
+    const body = await listRes.json<{
+      messages: Array<{ id: string; content: string; status: string }>;
+      hasMore: boolean;
+    }>();
+
+    expect(body.messages.length).toBeGreaterThanOrEqual(2);
+    const ids = body.messages.map((m) => m.id);
+    expect(ids).toContain(msgId1);
+    expect(ids).toContain(msgId2);
+  });
+});

packages/control-plane/test/integration/helpers.ts

@@ -0,0 +1,103 @@
+import { env, runInDurableObject } from "cloudflare:test";
+import type { SessionDO } from "../../src/session/durable-object";
+
+/**
+ * Create a fresh DO, call /internal/init, return the stub and id.
+ */
+export async function initSession(overrides?: {
+  sessionName?: string;
+  repoOwner?: string;
+  repoName?: string;
+  repoId?: number;
+  title?: string;
+  model?: string;
+  userId?: string;
+  githubLogin?: string;
+}) {
+  const id = env.SESSION.newUniqueId();
+  const stub = env.SESSION.get(id);
+  const defaults = {
+    sessionName: `test-${Date.now()}`,
+    repoOwner: "acme",
+    repoName: "web-app",
+    repoId: 12345,
+    userId: "user-1",
+    ...overrides,
+  };
+  const res = await stub.fetch("http://internal/internal/init", {
+    method: "POST",
+    headers: { "Content-Type": "application/json" },
+    body: JSON.stringify(defaults),
+  });
+  if (res.status !== 200) throw new Error(`Init failed: ${res.status}`);
+  return { stub, id };
+}
+
+/**
+ * Query the DO's SQLite via runInDurableObject.
+ */
+export async function queryDO<T>(
+  stub: DurableObjectStub,
+  sql: string,
+  ...params: unknown[]
+): Promise<T[]> {
+  return runInDurableObject(stub, (instance: SessionDO) => {
+    return instance.ctx.storage.sql.exec(sql, ...params).toArray() as T[];
+  });
+}
+
+/**
+ * Seed events directly into DO SQLite.
+ */
+export async function seedEvents(
+  stub: DurableObjectStub,
+  events: Array<{
+    id: string;
+    type: string;
+    data: string;
+    messageId?: string;
+    createdAt: number;
+  }>
+): Promise<void> {
+  await runInDurableObject(stub, (instance: SessionDO) => {
+    for (const e of events) {
+      instance.ctx.storage.sql.exec(
+        "INSERT INTO events (id, type, data, message_id, created_at) VALUES (?, ?, ?, ?, ?)",
+        e.id,
+        e.type,
+        e.data,
+        e.messageId ?? null,
+        e.createdAt
+      );
+    }
+  });
+}
+
+/**
+ * Seed a message directly into DO SQLite.
+ */
+export async function seedMessage(
+  stub: DurableObjectStub,
+  msg: {
+    id: string;
+    authorId: string;
+    content: string;
+    source: string;
+    status: string;
+    createdAt: number;
+    startedAt?: number;
+  }
+): Promise<void> {
+  await runInDurableObject(stub, (instance: SessionDO) => {
+    instance.ctx.storage.sql.exec(
+      "INSERT INTO messages (id, author_id, content, source, status, created_at, started_at) VALUES (?, ?, ?, ?, ?, ?, ?)",
+      msg.id,
+      msg.authorId,
+      msg.content,
+      msg.source,
+      msg.status,
+      msg.createdAt,
+      msg.startedAt ?? null
+    );
+  });
+}