homeio

1import { describe, expect, it, vi } from "vitest";
2import { closeServerGracefully } from "@/lib/server/http/graceful-shutdown";
3
4describe("closeServerGracefully", () => {
5  it("closes idle connections and resolves when the server closes cleanly", async () => {
6    const closeIdleConnections = vi.fn();
7    const close = vi.fn((callback: () => void) => {
8      callback();
9      return {} as never;
10    });
11
12    await closeServerGracefully({
13      close,
14      closeIdleConnections,
15    });
16
17    expect(close).toHaveBeenCalledTimes(1);
18    expect(closeIdleConnections).toHaveBeenCalledTimes(1);
19  });
20
21  it("forces remaining connections closed when the shutdown timeout expires", async () => {
22    vi.useFakeTimers();
23
24    const closeIdleConnections = vi.fn();
25    const closeAllConnections = vi.fn();
26    const close = vi.fn(() => ({} as never));
27
28    const shutdownPromise = closeServerGracefully(
29      {
30        close,
31        closeIdleConnections,
32        closeAllConnections,
33      },
34      50,
35    );
36
37    await vi.advanceTimersByTimeAsync(50);
38    await shutdownPromise;
39
40    expect(close).toHaveBeenCalledTimes(1);
41    expect(closeIdleConnections).toHaveBeenCalledTimes(2);
42    expect(closeAllConnections).toHaveBeenCalledTimes(1);
43
44    vi.useRealTimers();
45  });
46});
47