homeio

1import { describe, expect, it, vi } from "vitest";
2
3vi.mock("@/lib/server/modules/terminal/service", () => ({
4  executeTerminalCommand: vi.fn(),
5  TerminalServiceError: class TerminalServiceError extends Error {
6    code: string;
7    statusCode: number;
8
9    constructor(
10      message: string,
11      options?: { code?: string; statusCode?: number },
12    ) {
13      super(message);
14      this.code = options?.code ?? "execution_failed";
15      this.statusCode = options?.statusCode ?? 500;
16    }
17  },
18}));
19
20import { POST } from "@/app/api/v1/terminal/execute/route";
21import {
22  executeTerminalCommand,
23  TerminalServiceError,
24} from "@/lib/server/modules/terminal/service";
25
26describe("POST /api/v1/terminal/execute", () => {
27  it("returns terminal execution result", async () => {
28    vi.mocked(executeTerminalCommand).mockResolvedValueOnce({
29      cwd: "/home/homeio",
30      lines: [
31        {
32          type: "output",
33          content: "hello",
34        },
35      ],
36      exitCode: 0,
37      durationMs: 5,
38    });
39
40    const request = new Request("http://localhost/api/v1/terminal/execute", {
41      method: "POST",
42      headers: {
43        "Content-Type": "application/json",
44      },
45      body: JSON.stringify({
46        command: "echo hello",
47      }),
48    });
49
50    const response = await POST(request);
51    const json = (await response.json()) as {
52      data: { cwd: string };
53    };
54
55    expect(response.status).toBe(200);
56    expect(json.data.cwd).toBe("/home/homeio");
57  });
58
59  it("maps typed service errors", async () => {
60    vi.mocked(executeTerminalCommand).mockRejectedValueOnce(
61      new TerminalServiceError("forbidden", {
62        code: "forbidden_command",
63        statusCode: 403,
64      }),
65    );
66
67    const request = new Request("http://localhost/api/v1/terminal/execute", {
68      method: "POST",
69      headers: {
70        "Content-Type": "application/json",
71      },
72      body: JSON.stringify({
73        command: "rm -rf /",
74      }),
75    });
76
77    const response = await POST(request);
78    const json = (await response.json()) as { code: string };
79
80    expect(response.status).toBe(403);
81    expect(json.code).toBe("forbidden_command");
82  });
83});
84