homeio

1import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
2
3describe("postgres pool", () => {
4  beforeEach(() => {
5    vi.resetModules();
6    delete (globalThis as { __homeioPgPool?: unknown }).__homeioPgPool;
7    delete (globalThis as { __homeioPgPoolErrorListenerAttached?: unknown }).__homeioPgPoolErrorListenerAttached;
8    delete (globalThis as { __homeioPgPoolInitLogged?: unknown }).__homeioPgPoolInitLogged;
9  });
10
11  afterEach(() => {
12    delete (globalThis as { __homeioPgPool?: unknown }).__homeioPgPool;
13    delete (globalThis as { __homeioPgPoolErrorListenerAttached?: unknown }).__homeioPgPoolErrorListenerAttached;
14    delete (globalThis as { __homeioPgPoolInitLogged?: unknown }).__homeioPgPoolInitLogged;
15  });
16
17  it("creates a pool using server env", async () => {
18    const poolInstance = { query: vi.fn(), on: vi.fn() };
19    const Pool = vi.fn(
20      class MockPool {
21        constructor() {
22          return poolInstance as unknown as MockPool;
23        }
24      },
25    );
26
27    vi.doMock("pg", () => ({ Pool }));
28    vi.doMock("@/lib/server/env", () => ({
29      serverEnv: {
30        NODE_ENV: "production",
31        DATABASE_URL: "postgresql://test:test@127.0.0.1:5432/home_server",
32        PG_MAX_CONNECTIONS: 3,
33        LOG_LEVEL: "info",
34        LOG_FILE_PATH: "logs/home-server.log",
35        LOG_TO_FILE: true,
36      },
37    }));
38
39    const { pgPool } = await import("@/lib/server/db/postgres");
40
41    expect(pgPool).toBe(poolInstance);
42    expect(Pool).toHaveBeenCalledWith(
43      expect.objectContaining({
44        connectionString: "postgresql://test:test@127.0.0.1:5432/home_server",
45        max: 3,
46      }),
47    );
48    expect(poolInstance.on).toHaveBeenCalledWith("error", expect.any(Function));
49  });
50
51  it("reuses global pool in non-production", async () => {
52    const existing = { query: vi.fn(), on: vi.fn() };
53    (globalThis as { __homeioPgPool?: unknown }).__homeioPgPool = existing;
54
55    const Pool = vi.fn(class MockPool {});
56    vi.doMock("pg", () => ({ Pool }));
57    vi.doMock("@/lib/server/env", () => ({
58      serverEnv: {
59        NODE_ENV: "development",
60        DATABASE_URL: "postgresql://test:test@127.0.0.1:5432/home_server",
61        PG_MAX_CONNECTIONS: 3,
62        LOG_LEVEL: "info",
63        LOG_FILE_PATH: "logs/home-server.log",
64        LOG_TO_FILE: true,
65      },
66    }));
67
68    const { pgPool } = await import("@/lib/server/db/postgres");
69
70    expect(pgPool).toBe(existing);
71    expect(Pool).not.toHaveBeenCalled();
72    expect(existing.on).toHaveBeenCalledWith("error", expect.any(Function));
73  });
74});
75