Files
calendar/apps/web/test/lib/plainTiers.test.ts
T
2025-03-25 00:18:26 +00:00

93 lines
2.8 KiB
TypeScript

/* eslint-disable @typescript-eslint/no-explicit-any */
import prismaMock from "../../../../tests/libs/__mocks__/prismaMock";
import type { Params } from "app/_types";
import type { NextRequest } from "next/server";
import { describe, it, expect, vi, beforeAll, beforeEach } from "vitest";
import { getServerSession } from "@calcom/features/auth/lib/getServerSession";
import { IS_PLAIN_CHAT_ENABLED } from "@calcom/lib/constants";
import { POST } from "../../app/api/plain-hash/route";
// Mock NextResponse
vi.mock("next/server", () => ({
NextResponse: {
json: (data: any, init?: ResponseInit) => ({
json: async () => data,
status: init?.status || 200,
}),
},
}));
// Mock session
vi.mock("@calcom/features/auth/lib/getServerSession", () => ({
getServerSession: vi.fn(),
}));
vi.mock("next/headers", () => ({
headers: () => new Headers(),
cookies: () => ({ get: () => null }),
}));
vi.mock("@lib/buildLegacyCtx", () => ({
buildLegacyRequest: vi.fn().mockReturnValue({
headers: new Headers(),
cookies: {},
}),
}));
vi.mock("app/api/defaultResponderForAppDir", () => ({
defaultResponderForAppDir: (handler: () => Promise<Response>) => handler,
}));
vi.mock("@calcom/prisma", () => ({
default: prismaMock,
}));
describe("Plain Integration API", () => {
beforeAll(() => {
process.env.PLAIN_CHAT_HMAC_SECRET_KEY = "test-secret";
process.env.NEXT_PUBLIC_PLAIN_CHAT_ID = "test-chat-id";
// Force the feature to be enabled for tests
vi.mock("@calcom/lib/constants", () => ({
IS_PLAIN_CHAT_ENABLED: true,
}));
});
beforeEach(() => {
vi.resetAllMocks();
});
// Skip all tests if the feature is disabled
if (!IS_PLAIN_CHAT_ENABLED) {
it("skips tests when Plain Chat is disabled", () => {
console.log("Plain Chat is disabled, skipping tests");
});
return;
}
const tiers = [
{ tier: "free", teams: [] },
{ tier: "teams", teams: [{ team: { id: 2 } }] },
{ tier: "enterprise", teams: [{ team: { id: 3, isOrganization: true } }] },
];
it.each(tiers)("should return $tier tier", async ({ tier, teams }) => {
const userData = { id: 1, email: `${tier}@example.com`, name: `${tier} User` };
(getServerSession as any).mockResolvedValue({ user: userData });
prismaMock.user.findUnique.mockResolvedValue({ ...userData, teams });
const params = Promise.resolve<Params>({});
const data = await (await POST({} as NextRequest, { params })).json();
expect(data).toMatchObject({ userTier: tier, email: userData.email, fullName: userData.name });
});
it("should return 401 when no session exists", async () => {
(getServerSession as any).mockResolvedValue(null);
const params = Promise.resolve<Params>({});
expect((await POST({} as NextRequest, { params })).status).toBe(401);
});
});