92 lines
2.8 KiB
TypeScript
92 lines
2.8 KiB
TypeScript
import type { Page } from "@playwright/test";
|
|
import type { Booking, Prisma } from "@prisma/client";
|
|
import short from "short-uuid";
|
|
import { v5 as uuidv5 } from "uuid";
|
|
|
|
import dayjs from "@calcom/dayjs";
|
|
import { prisma } from "@calcom/prisma";
|
|
|
|
const translator = short();
|
|
|
|
type BookingFixture = ReturnType<typeof createBookingFixture>;
|
|
|
|
// creates a user fixture instance and stores the collection
|
|
export const createBookingsFixture = (page: Page) => {
|
|
const store = { bookings: [], page } as { bookings: BookingFixture[]; page: typeof page };
|
|
return {
|
|
create: async (
|
|
userId: number,
|
|
username: string | null,
|
|
eventTypeId = -1,
|
|
{
|
|
rescheduled = false,
|
|
paid = false,
|
|
status = "ACCEPTED",
|
|
attendees = {
|
|
create: {
|
|
email: "attendee@example.com",
|
|
name: "Attendee Example",
|
|
timeZone: "Europe/London",
|
|
},
|
|
},
|
|
}: Partial<Prisma.BookingCreateInput> = {},
|
|
startDateParam?: Date,
|
|
endDateParam?: Date
|
|
) => {
|
|
const startDate = startDateParam || dayjs().add(1, "day").toDate();
|
|
const seed = `${username}:${dayjs(startDate).utc().format()}:${new Date().getTime()}`;
|
|
const uid = translator.fromUUID(uuidv5(seed, uuidv5.URL));
|
|
const booking = await prisma.booking.create({
|
|
data: {
|
|
uid: uid,
|
|
title: "30min",
|
|
startTime: startDate,
|
|
endTime: endDateParam || dayjs().add(1, "day").add(30, "minutes").toDate(),
|
|
user: {
|
|
connect: {
|
|
id: userId,
|
|
},
|
|
},
|
|
attendees,
|
|
eventType: {
|
|
connect: {
|
|
id: eventTypeId,
|
|
},
|
|
},
|
|
rescheduled,
|
|
paid,
|
|
status,
|
|
},
|
|
});
|
|
const bookingFixture = createBookingFixture(booking, store.page);
|
|
store.bookings.push(bookingFixture);
|
|
return bookingFixture;
|
|
},
|
|
update: async (args: Prisma.BookingUpdateArgs) => await prisma.booking.update(args),
|
|
get: () => store.bookings,
|
|
delete: async (id: number) => {
|
|
await prisma.booking.delete({
|
|
where: { id },
|
|
});
|
|
store.bookings = store.bookings.filter((b) => b.id !== id);
|
|
},
|
|
};
|
|
};
|
|
|
|
// creates the single user fixture
|
|
const createBookingFixture = (booking: Booking, page: Page) => {
|
|
const store = { booking, page };
|
|
|
|
// self is a reflective method that return the Prisma object that references this fixture.
|
|
return {
|
|
id: store.booking.id,
|
|
uid: store.booking.uid,
|
|
self: async () =>
|
|
await prisma.booking.findUnique({
|
|
where: { id: store.booking.id },
|
|
include: { attendees: true, seatsReferences: true },
|
|
}),
|
|
delete: async () => await prisma.booking.delete({ where: { id: store.booking.id } }),
|
|
};
|
|
};
|