test: 新增 eventRouter 和 router 的單元測試

Author: c-c-epr

test/eventRouter.test.ts

@@ -0,0 +1,72 @@
+import { beforeEach, describe, expect, it, vi } from "vitest";
+
+const mocks = vi.hoisted(() => ({
+  sendMessage: vi.fn(),
+  markAsRead: vi.fn().mockResolvedValue(undefined),
+  loadStart: vi.fn().mockResolvedValue(undefined),
+}));
+
+vi.mock("../src/utils/eventRoutes", () => ({
+  sendMessage: mocks.sendMessage,
+  markAsRead: mocks.markAsRead,
+  loadStart: mocks.loadStart,
+}));
+
+import { eventRouter } from "../src/eventRouter";
+
+describe("eventRouter", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it("handles 'Hi' message", async () => {
+    const event = {
+      type: "message",
+      source: { userId: "U123" },
+      message: { type: "text", text: "Hi", markAsReadToken: "mark-token" },
+      replyToken: "reply-token",
+    };
+    const ctx = { waitUntil: vi.fn() } as any;
+
+    await eventRouter(event, "accessToken", ctx);
+
+    expect(ctx.waitUntil).toHaveBeenCalledTimes(1);
+    expect(mocks.markAsRead).toHaveBeenCalledWith("accessToken", "mark-token");
+    expect(mocks.loadStart).toHaveBeenCalledWith("accessToken", "U123", 5);
+    expect(mocks.sendMessage).toHaveBeenCalledWith("accessToken", "reply-token", [
+      { type: "text", text: "Hi" },
+    ]);
+  });
+
+  it("handles 'hi' message", async () => {
+    const event = {
+      type: "message",
+      source: { userId: "U123" },
+      message: { type: "text", text: "hi", markAsReadToken: "mark-token" },
+      replyToken: "reply-token",
+    };
+    const ctx = { waitUntil: vi.fn() } as any;
+
+    await eventRouter(event, "accessToken", ctx);
+
+    expect(mocks.sendMessage).toHaveBeenCalledWith("accessToken", "reply-token", [
+      { type: "text", text: "Hi" },
+    ]);
+  });
+
+  it("handles default text message", async () => {
+    const event = {
+      type: "message",
+      source: { userId: "U123" },
+      message: { type: "text", text: "unknown", markAsReadToken: "mark-token" },
+      replyToken: "reply-token",
+    };
+    const ctx = { waitUntil: vi.fn() } as any;
+
+    await eventRouter(event, "accessToken", ctx);
+
+    expect(mocks.sendMessage).toHaveBeenCalledWith("accessToken", "reply-token", [
+      { type: "text", text: "Hello from Cloudflare Workers!" },
+    ]);
+  });
+});

test/router.test.ts

@@ -0,0 +1,52 @@
+import { beforeEach, describe, expect, it, vi } from "vitest";
+import { router } from "../src/router";
+
+const mockEnv = {
+  LINE_CHANNEL_SECRET: { get: vi.fn() },
+  LINE_CHANNEL_ACCESS_TOKEN: { get: vi.fn() },
+};
+
+describe("router", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+    mockEnv.LINE_CHANNEL_SECRET.get.mockResolvedValue("secret");
+    mockEnv.LINE_CHANNEL_ACCESS_TOKEN.get.mockResolvedValue("accessToken");
+  });
+
+  it("returns 405 for non-POST methods", async () => {
+    const request = new Request("https://example.com", { method: "GET" });
+    const response = await router(request, mockEnv as any, {} as any);
+
+    expect(response.status).toBe(405);
+    expect(await response.text()).toBe("Method Not Allowed");
+  });
+
+  it("returns 400 if signature is missing", async () => {
+    const request = new Request("https://example.com", { method: "POST" });
+    const response = await router(request, mockEnv as any, {} as any);
+
+    expect(response.status).toBe(400);
+    expect(await response.text()).toBe("Missing signature");
+  });
+
+  it("returns 401 if signature is invalid", async () => {
+    const request = new Request("https://example.com", {
+      method: "POST",
+      headers: { "x-line-signature": "invalid" },
+      body: JSON.stringify({}),
+    });
+    const response = await router(request, mockEnv as any, {} as any);
+
+    expect(response.status).toBe(401);
+    expect(await response.text()).toBe("Invalid signature");
+  });
+
+  it("returns 500 if server is misconfigured", async () => {
+    mockEnv.LINE_CHANNEL_SECRET.get.mockResolvedValue(null);
+    const request = new Request("https://example.com", { method: "POST" });
+    const response = await router(request, mockEnv as any, {} as any);
+
+    expect(response.status).toBe(500);
+    expect(await response.text()).toBe("Server misconfigured");
+  });
+});