homeio

1import { NextRequest, NextResponse } from "next/server";
2import { describe, expect, it, vi } from "vitest";
3
4vi.mock("@/lib/server/modules/system/power-schedule", () => ({
5  deleteScheduledRebootArtifacts: vi.fn(),
6}));
7
8vi.mock("@/lib/server/modules/system/power-service", () => ({
9  deleteFactoryResetArtifacts: vi.fn(),
10  scheduleFactoryReset: vi.fn(),
11}));
12
13import { POST } from "@/app/api/v1/system/power/factory-reset/route";
14import { requireApiSession } from "@/lib/server/modules/auth/api";
15import { deleteScheduledRebootArtifacts } from "@/lib/server/modules/system/power-schedule";
16import {
17  deleteFactoryResetArtifacts,
18  scheduleFactoryReset,
19} from "@/lib/server/modules/system/power-service";
20
21describe("POST /api/v1/system/power/factory-reset", () => {
22  it("returns 202 and schedules factory reset for authenticated users", async () => {
23    vi.mocked(deleteScheduledRebootArtifacts).mockResolvedValueOnce(undefined);
24    vi.mocked(deleteFactoryResetArtifacts).mockResolvedValueOnce(undefined);
25    vi.mocked(scheduleFactoryReset).mockResolvedValueOnce(undefined);
26
27    const request = new NextRequest("http://localhost/api/v1/system/power/factory-reset", {
28      method: "POST",
29      headers: {
30        cookie: "homeio_session=session-token",
31      },
32    });
33
34    const response = await POST(request);
35    const json = (await response.json()) as {
36      data: { action: string; accepted: boolean };
37    };
38
39    expect(response.status).toBe(202);
40    expect(json.data).toEqual({ action: "factory-reset", accepted: true });
41    expect(deleteScheduledRebootArtifacts).toHaveBeenCalledOnce();
42    expect(deleteFactoryResetArtifacts).toHaveBeenCalledOnce();
43    expect(scheduleFactoryReset).toHaveBeenCalledOnce();
44  });
45
46  it("returns 401 when the session is missing or invalid", async () => {
47    vi.mocked(requireApiSession).mockResolvedValueOnce({
48      session: null,
49      response: NextResponse.json({ error: "Unauthorized" }, { status: 401 }),
50    });
51
52    const request = new NextRequest("http://localhost/api/v1/system/power/factory-reset", {
53      method: "POST",
54    });
55
56    const response = await POST(request);
57
58    expect(response.status).toBe(401);
59    expect(scheduleFactoryReset).not.toHaveBeenCalled();
60  });
61
62  it("returns 500 when factory reset scheduling fails", async () => {
63    vi.mocked(deleteScheduledRebootArtifacts).mockResolvedValueOnce(undefined);
64    vi.mocked(deleteFactoryResetArtifacts).mockResolvedValueOnce(undefined);
65    vi.mocked(scheduleFactoryReset).mockRejectedValueOnce(new Error("spawn failed"));
66
67    const request = new NextRequest("http://localhost/api/v1/system/power/factory-reset", {
68      method: "POST",
69      headers: {
70        cookie: "homeio_session=session-token",
71      },
72    });
73
74    const response = await POST(request);
75    const json = (await response.json()) as { error: string };
76
77    expect(response.status).toBe(500);
78    expect(json.error).toBe("Failed to schedule factory reset");
79  });
80});
81