import type { Page, WorkerInfo } from "@playwright/test"; import type Prisma from "@prisma/client"; import { MembershipRole, Prisma as PrismaType } from "@prisma/client"; import { hashSync as hash } from "bcryptjs"; import dayjs from "@calcom/dayjs"; import { DEFAULT_SCHEDULE, getAvailabilityFromSchedule } from "@calcom/lib/availability"; import { prisma } from "@calcom/prisma"; import type { TimeZoneEnum } from "./types"; // Don't import hashPassword from app as that ends up importing next-auth and initializing it before NEXTAUTH_URL can be updated during tests. export function hashPassword(password: string) { const hashedPassword = hash(password, 12); return hashedPassword; } type UserFixture = ReturnType; const userIncludes = PrismaType.validator()({ eventTypes: true, credentials: true, routingForms: true, }); const userWithEventTypes = PrismaType.validator()({ include: userIncludes, }); const seededForm = { id: "948ae412-d995-4865-875a-48302588de03", name: "Seeded Form - Pro", }; type UserWithIncludes = PrismaType.UserGetPayload; const createTeamAndAddUser = async ( { user }: { user: { id: number; role?: MembershipRole } }, workerInfo: WorkerInfo ) => { const team = await prisma.team.create({ data: { name: "", slug: `team-${workerInfo.workerIndex}-${Date.now()}`, }, }); const { role = MembershipRole.OWNER, id: userId } = user; await prisma.membership.create({ data: { teamId: team.id, userId, role: role, accepted: true, }, }); return team; }; // creates a user fixture instance and stores the collection export const createUsersFixture = (page: Page, workerInfo: WorkerInfo) => { const store = { users: [], page } as { users: UserFixture[]; page: typeof page }; return { create: async ( opts?: CustomUserOpts | null, scenario: { seedRoutingForms?: boolean; hasTeam?: true; } = {} ) => { const _user = await prisma.user.create({ data: createUser(workerInfo, opts), }); let defaultEventTypes: SupportedTestEventTypes[] = [ { title: "30 min", slug: "30-min", length: 30 }, { title: "Paid", slug: "paid", length: 30, price: 1000 }, { title: "Opt in", slug: "opt-in", requiresConfirmation: true, length: 30 }, ]; if (opts?.eventTypes) defaultEventTypes = defaultEventTypes.concat(opts.eventTypes); for (const eventTypeData of defaultEventTypes) { eventTypeData.owner = { connect: { id: _user.id } }; eventTypeData.users = { connect: { id: _user.id } }; await prisma.eventType.create({ data: eventTypeData, }); } if (scenario.seedRoutingForms) { await prisma.app_RoutingForms_Form.create({ data: { routes: [ { id: "8a898988-89ab-4cde-b012-31823f708642", action: { type: "eventTypeRedirectUrl", value: "pro/30min" }, queryValue: { id: "8a898988-89ab-4cde-b012-31823f708642", type: "group", children1: { "8988bbb8-0123-4456-b89a-b1823f70c5ff": { type: "rule", properties: { field: "c4296635-9f12-47b1-8153-c3a854649182", value: ["event-routing"], operator: "equal", valueSrc: ["value"], valueType: ["text"], }, }, }, }, }, { id: "aa8aaba9-cdef-4012-b456-71823f70f7ef", action: { type: "customPageMessage", value: "Custom Page Result" }, queryValue: { id: "aa8aaba9-cdef-4012-b456-71823f70f7ef", type: "group", children1: { "b99b8a89-89ab-4cde-b012-31823f718ff5": { type: "rule", properties: { field: "c4296635-9f12-47b1-8153-c3a854649182", value: ["custom-page"], operator: "equal", valueSrc: ["value"], valueType: ["text"], }, }, }, }, }, { id: "a8ba9aab-4567-489a-bcde-f1823f71b4ad", action: { type: "externalRedirectUrl", value: "https://google.com" }, queryValue: { id: "a8ba9aab-4567-489a-bcde-f1823f71b4ad", type: "group", children1: { "998b9b9a-0123-4456-b89a-b1823f7232b9": { type: "rule", properties: { field: "c4296635-9f12-47b1-8153-c3a854649182", value: ["external-redirect"], operator: "equal", valueSrc: ["value"], valueType: ["text"], }, }, }, }, }, { id: "aa8ba8b9-0123-4456-b89a-b182623406d8", action: { type: "customPageMessage", value: "Multiselect chosen" }, queryValue: { id: "aa8ba8b9-0123-4456-b89a-b182623406d8", type: "group", children1: { "b98a8abb-cdef-4012-b456-718262343d27": { type: "rule", properties: { field: "d4292635-9f12-17b1-9153-c3a854649182", value: [["Option-2"]], operator: "multiselect_equals", valueSrc: ["value"], valueType: ["multiselect"], }, }, }, }, }, { id: "898899aa-4567-489a-bcde-f1823f708646", action: { type: "customPageMessage", value: "Fallback Message" }, isFallback: true, queryValue: { id: "898899aa-4567-489a-bcde-f1823f708646", type: "group" }, }, ], fields: [ { id: "c4296635-9f12-47b1-8153-c3a854649182", type: "text", label: "Test field", required: true, }, { id: "d4292635-9f12-17b1-9153-c3a854649182", type: "multiselect", label: "Multi Select", identifier: "multi", selectText: "Option-1\nOption-2", required: false, }, ], user: { connect: { id: _user.id, }, }, name: seededForm.name, }, }); } const user = await prisma.user.findUniqueOrThrow({ where: { id: _user.id }, include: userIncludes, }); if (scenario.hasTeam) { const team = await createTeamAndAddUser({ user: { id: user.id, role: "OWNER" } }, workerInfo); await prisma.eventType.create({ data: { team: { connect: { id: team.id, }, }, users: { connect: { id: _user.id, }, }, owner: { connect: { id: _user.id, }, }, title: "Team Event - 30min", slug: "team-event-30min", length: 30, }, }); } const userFixture = createUserFixture(user, store.page!); store.users.push(userFixture); return userFixture; }, get: () => store.users, logout: async () => { await page.goto("/auth/logout"); }, deleteAll: async () => { const ids = store.users.map((u) => u.id); await prisma.user.deleteMany({ where: { id: { in: ids } } }); store.users = []; }, delete: async (id: number) => { await prisma.user.delete({ where: { id } }); store.users = store.users.filter((b) => b.id !== id); }, }; }; type JSONValue = string | number | boolean | { [x: string]: JSONValue } | Array; // creates the single user fixture const createUserFixture = (user: UserWithIncludes, page: Page) => { const store = { user, page }; // self is a reflective method that return the Prisma object that references this fixture. const self = async () => (await prisma.user.findUnique({ where: { id: store.user.id }, include: { eventTypes: true } }))!; return { id: user.id, username: user.username, eventTypes: user.eventTypes, routingForms: user.routingForms, self, login: async () => login({ ...(await self()), password: user.username }, store.page), logout: async () => { await page.goto("/auth/logout"); }, getPaymentCredential: async () => getPaymentCredential(store.page), // ths is for developemnt only aimed to inject debugging messages in the metadata field of the user debug: async (message: string | Record) => { await prisma.user.update({ where: { id: store.user.id }, data: { metadata: { debug: message } } }); }, delete: async () => (await prisma.user.delete({ where: { id: store.user.id } }))!, }; }; type SupportedTestEventTypes = PrismaType.EventTypeCreateInput & { _bookings?: PrismaType.BookingCreateInput[]; }; type CustomUserOptsKeys = "username" | "password" | "completedOnboarding" | "locale" | "name"; type CustomUserOpts = Partial> & { timeZone?: TimeZoneEnum; eventTypes?: SupportedTestEventTypes[]; }; // creates the actual user in the db. const createUser = (workerInfo: WorkerInfo, opts?: CustomUserOpts | null): PrismaType.UserCreateInput => { // build a unique name for our user const uname = `${opts?.username || "user"}-${workerInfo.workerIndex}-${Date.now()}`; return { username: uname, name: opts?.name, email: `${uname}@example.com`, password: hashPassword(uname), emailVerified: new Date(), completedOnboarding: opts?.completedOnboarding ?? true, timeZone: opts?.timeZone ?? dayjs.tz.guess(), locale: opts?.locale ?? "en", schedules: opts?.completedOnboarding ?? true ? { create: { name: "Working Hours", availability: { createMany: { data: getAvailabilityFromSchedule(DEFAULT_SCHEDULE), }, }, }, } : undefined, }; }; // login using a replay of an E2E routine. export async function login( user: Pick & Partial>, page: Page ) { // get locators const loginLocator = page.locator("[data-testid=login-form]"); const emailLocator = loginLocator.locator("#email"); const passwordLocator = loginLocator.locator("#password"); const signInLocator = loginLocator.locator('[type="submit"]'); //login await page.goto("/"); await emailLocator.fill(user.email ?? `${user.username}@example.com`); // eslint-disable-next-line @typescript-eslint/no-non-null-assertion await passwordLocator.fill(user.password ?? user.username!); await signInLocator.click(); // Moving away from waiting 2 seconds, as it is not a reliable way to expect session to be started await page.waitForLoadState("networkidle"); } export async function getPaymentCredential(page: Page) { await page.goto("/apps/stripe"); /** We start the Stripe flow */ await Promise.all([ page.waitForNavigation({ url: "https://connect.stripe.com/oauth/v2/authorize?*" }), page.click('[data-testid="install-app-button"]'), ]); await Promise.all([ page.waitForNavigation({ url: "/apps/installed/payment?hl=stripe" }), /** We skip filling Stripe forms (testing mode only) */ page.click('[id="skip-account-app"]'), ]); }