homeio

1import { beforeEach, describe, expect, it, vi } from "vitest";
2
3vi.mock("@/lib/server/db/drizzle", () => ({
4  db: {
5    select: vi.fn(),
6  },
7}));
8
9import { db } from "@/lib/server/db/drizzle";
10import { hasAnyUsers } from "@/lib/server/modules/auth/repository";
11
12function makeSelectChain(rows: unknown[]) {
13  const chain: Record<string, unknown> = {};
14  for (const method of ["from", "where", "orderBy", "limit", "innerJoin"]) {
15    chain[method] = vi.fn().mockReturnValue(chain);
16  }
17  chain.then = (resolve: (v: unknown) => unknown) => Promise.resolve(rows).then(resolve);
18  chain.catch = (reject: (r: unknown) => unknown) => Promise.resolve(rows).catch(reject);
19  chain.finally = (cb: () => void) => Promise.resolve(rows).finally(cb);
20  return chain;
21}
22
23describe("auth repository", () => {
24  beforeEach(() => {
25    vi.mocked(db.select).mockReset();
26  });
27
28  it("returns false when the users table is empty", async () => {
29    vi.mocked(db.select).mockReturnValue(
30      makeSelectChain([{ total: 0 }]) as never,
31    );
32
33    await expect(hasAnyUsers()).resolves.toBe(false);
34    expect(db.select).toHaveBeenCalledTimes(1);
35  });
36
37  it("returns true when the users table has at least one row", async () => {
38    vi.mocked(db.select).mockReturnValue(
39      makeSelectChain([{ total: 1 }]) as never,
40    );
41
42    await expect(hasAnyUsers()).resolves.toBe(true);
43  });
44});
45