ClaudeMailbox/node/tests/db.test.ts

import { describe, it, expect, afterEach, beforeEach } from "vitest";
import { mkdtempSync, rmSync } from "node:fs";
import { tmpdir } from "node:os";
import { join } from "node:path";
import { MailboxStore } from "../src/db.js";

let dir: string;
let dbPath: string;

beforeEach(() => {
  dir = mkdtempSync(join(tmpdir(), "claude-mailbox-test-"));
  dbPath = join(dir, "test.db");
});

afterEach(() => {
  rmSync(dir, { recursive: true, force: true });
});

describe("schema", () => {
  it("creates fresh tables and is idempotent on re-open", () => {
    const a = new MailboxStore(dbPath);
    a.upsertMailbox("alice");
    a.close();

    const b = new MailboxStore(dbPath);
    const list = b.listMailboxes();
    expect(list.map((m) => m.name)).toEqual(["alice"]);
    b.close();
  });
});

describe("send / peek / check round-trip", () => {
  it("delivers a message exactly once", () => {
    const store = new MailboxStore(dbPath);
    try {
      const result = store.send("alice", "bob", "hello bob");
      expect(result.id).toBeGreaterThan(0);

      const peek1 = store.peek("bob");
      expect(peek1.pending).toBe(1);
      expect(peek1.oldestAt).toBeInstanceOf(Date);

      const pulled = store.checkInbox("bob");
      expect(pulled).toHaveLength(1);
      expect(pulled[0]!.from_mailbox).toBe("alice");
      expect(pulled[0]!.body).toBe("hello bob");

      const peek2 = store.peek("bob");
      expect(peek2.pending).toBe(0);
      expect(peek2.oldestAt).toBeNull();

      const empty = store.checkInbox("bob");
      expect(empty).toEqual([]);
    } finally {
      store.close();
    }
  });

  it("checkInbox returns all pending in order and marks them delivered atomically", () => {
    const store = new MailboxStore(dbPath);
    try {
      for (let i = 0; i < 10; i++) {
        store.send("alice", "bob", `msg ${i}`);
      }
      const first = store.checkInbox("bob");
      expect(first).toHaveLength(10);
      expect(first.map((m) => m.body)).toEqual(
        Array.from({ length: 10 }, (_, i) => `msg ${i}`),
      );
      const second = store.checkInbox("bob");
      expect(second).toEqual([]);
    } finally {
      store.close();
    }
  });
});

describe("listMailboxes", () => {
  it("returns mailboxes alphabetically with pendingForYou for the caller", () => {
    const store = new MailboxStore(dbPath);
    try {
      store.send("alice", "bob", "x");
      store.send("alice", "bob", "y");
      store.send("carol", "bob", "z");

      const fromBob = store.listMailboxes("bob");
      expect(fromBob.map((m) => m.name)).toEqual(["alice", "bob", "carol"]);
      const bobRow = fromBob.find((m) => m.name === "bob");
      expect(bobRow?.pendingForYou).toBe(3);
    } finally {
      store.close();
    }
  });
});