Files
community-rule/tests/unit/draftsMeRoute.test.ts
T
2026-04-22 19:15:04 -06:00

101 lines
3.0 KiB
TypeScript

import { NextRequest } from "next/server";
import { beforeEach, describe, expect, it, vi } from "vitest";
const isDatabaseConfiguredMock = vi.fn();
const getSessionUserMock = vi.fn();
const findUniqueMock = vi.fn();
vi.mock("../../lib/server/env", () => ({
isDatabaseConfigured: () => isDatabaseConfiguredMock(),
}));
vi.mock("../../lib/server/db", () => ({
prisma: {
ruleDraft: {
findUnique: (...args: unknown[]) => findUniqueMock(...args),
upsert: vi.fn(),
deleteMany: vi.fn(),
},
},
}));
vi.mock("../../lib/server/session", () => ({
getSessionUser: () => getSessionUserMock(),
}));
import { GET } from "../../app/api/drafts/me/route";
beforeEach(() => {
isDatabaseConfiguredMock.mockReset();
getSessionUserMock.mockReset();
findUniqueMock.mockReset();
});
describe("GET /api/drafts/me", () => {
it("returns 503 with the canonical shape when the database is not configured", async () => {
isDatabaseConfiguredMock.mockReturnValue(false);
const res = await GET(
new NextRequest("https://x.test/api/drafts/me"),
undefined,
);
expect(res.status).toBe(503);
const body = (await res.json()) as {
error: { code: string; message: string };
};
expect(body.error.code).toBe("db_unavailable");
expect(res.headers.get("x-request-id")).toBeTruthy();
});
it("returns 401 unauthorized with the canonical shape when no session user", async () => {
isDatabaseConfiguredMock.mockReturnValue(true);
getSessionUserMock.mockResolvedValueOnce(null);
const res = await GET(
new NextRequest("https://x.test/api/drafts/me"),
undefined,
);
expect(res.status).toBe(401);
const body = (await res.json()) as {
error: { code: string; message: string };
};
expect(body.error).toEqual({
code: "unauthorized",
message: "Unauthorized",
});
expect(res.headers.get("x-request-id")).toBeTruthy();
expect(findUniqueMock).not.toHaveBeenCalled();
});
it("forwards an incoming x-request-id on the response", async () => {
isDatabaseConfiguredMock.mockReturnValue(true);
getSessionUserMock.mockResolvedValueOnce(null);
const res = await GET(
new NextRequest("https://x.test/api/drafts/me", {
headers: { "x-request-id": "req_drafts-1" },
}),
undefined,
);
expect(res.headers.get("x-request-id")).toBe("req_drafts-1");
});
it("returns the draft when present", async () => {
isDatabaseConfiguredMock.mockReturnValue(true);
getSessionUserMock.mockResolvedValueOnce({
id: "u1",
email: "x@y.test",
});
findUniqueMock.mockResolvedValueOnce({
payload: { foo: 1 },
updatedAt: new Date("2026-01-01T00:00:00Z"),
});
const res = await GET(
new NextRequest("https://x.test/api/drafts/me"),
undefined,
);
expect(res.status).toBe(200);
const body = (await res.json()) as {
draft: { payload: { foo: number } } | null;
};
expect(body.draft?.payload).toEqual({ foo: 1 });
});
});