calendar/packages/embeds/embed-core/playwright/lib/testUtils.ts

import type { Page, Frame } from "@playwright/test";
import { expect } from "@playwright/test";

import prisma from "@calcom/prisma";

export async function getQueuedFormResponse(queuedFormResponseId: string) {
  return prisma.app_RoutingForms_QueuedFormResponse.findFirst({
    where: {
      id: queuedFormResponseId,
    },
    include: {
      actualResponse: true,
    },
  });
}

export async function getAllFormResponses(formId: string) {
  return prisma.app_RoutingForms_FormResponse.findMany({
    where: {
      formId: formId,
    },
    orderBy: {
      createdAt: "desc",
    },
  });
}

export async function getLatestQueuedFormResponse({ formId }: { formId: string }) {
  return prisma.app_RoutingForms_QueuedFormResponse.findFirst({
    where: {
      formId: formId,
    },
    orderBy: {
      createdAt: "desc",
    },
  });
}

export const deleteAllBookingsByEmail = async (email: string) =>
  await prisma.booking.deleteMany({
    where: {
      attendees: {
        some: {
          email: email,
        },
      },
    },
  });

export const getBooking = async (bookingId: string) => {
  const booking = await prisma.booking.findUnique({
    where: {
      uid: bookingId,
    },
    include: {
      attendees: true,
    },
  });
  if (!booking) {
    throw new Error("Booking not found");
  }
  return booking;
};

/**
 * @deprecated use ensureEmbedIframe instead.
 */
export const getEmbedIframe = async ({
  calNamespace,
  page,
  pathname,
}: {
  calNamespace: string;
  page: Page;
  pathname: string;
}) => {
  await page.waitForFunction(
    () => {
      const iframe = document.querySelector<HTMLIFrameElement>(".cal-embed");
      // eslint-disable-next-line @typescript-eslint/ban-ts-comment
      // @ts-ignore
      return iframe && iframe.contentWindow && window.iframeReady;
    },
    { polling: 500 }
  );
  const embedIframe = page.frame(`cal-embed=${calNamespace}`);
  if (!embedIframe) {
    return null;
  }
  const u = new URL(embedIframe.url());
  if (u.pathname === `${pathname}/embed`) {
    return embedIframe;
  }
  console.log(`Embed iframe url pathname match. Expected: "${pathname}/embed"`, `Actual: ${u.pathname}`);
  return null;
};

export const ensureEmbedIframe = async ({
  calNamespace,
  page,
  pathname,
}: {
  calNamespace: string;
  page: Page;
  pathname: string;
}) => {
  const embedIframe = await getEmbedIframe({ calNamespace, page, pathname });
  if (!embedIframe) {
    throw new Error("Embed iframe not found");
  }
  return embedIframe;
};

async function selectFirstAvailableTimeSlotNextMonth(frame: Frame, page: Page) {
  await frame.click('[data-testid="incrementMonth"]');

  // @TODO: Find a better way to make test wait for full month change render to end
  // so it can click up on the right day, also when done, resolve other todos as well
  // The problem is that the Month Text changes instantly but we don't know when the corresponding dates are visible

  // Waiting for full month increment
  await frame.waitForTimeout(1000);
  // expect(await page.screenshot()).toMatchSnapshot("availability-page-2.png");
  // TODO: Find out why the first day is always booked on tests
  await frame.locator('[data-testid="day"][data-disabled="false"]').nth(1).click();
  await frame.click('[data-testid="time"]');
}

export async function bookFirstEvent(username: string, frame: Frame, page: Page) {
  // Click first event type on Profile Page
  await frame.click('[data-testid="event-type-link"]');
  await frame.waitForURL((url) => {
    // Wait for reaching the event page
    const matches = url.pathname.match(new RegExp(`/${username}/(.+)$`));
    if (!matches || !matches[1]) {
      return false;
    }
    if (matches[1] === "embed") {
      return false;
    }
    return true;
  });

  // Let current month dates fully render.
  // There is a bug where if we don't let current month fully render and quickly click go to next month, current month gets rendered
  // This doesn't seem to be replicable with the speed of a person, only during automation.
  // It would also allow correct snapshot to be taken for current month.
  await frame.waitForTimeout(1000);
  // expect(await page.screenshot()).toMatchSnapshot("availability-page-1.png");
  // Remove /embed from the end if present.
  return bookEvent({ frame, page });
}

export async function bookEvent({ frame, page }: { frame: Frame; page: Page }) {
  const eventSlug = new URL(frame.url()).pathname.replace(/\/embed$/, "");
  await selectFirstAvailableTimeSlotNextMonth(frame, page);
  // expect(await page.screenshot()).toMatchSnapshot("booking-page.png");
  // --- fill form
  await frame.fill('[name="name"]', "Embed User");
  await frame.fill('[name="email"]', "embed-user@example.com");
  const responsePromise = page.waitForResponse("**/api/book/event");
  await frame.press('[name="email"]', "Enter");
  const response = await responsePromise;
  const booking = (await response.json()) as { uid: string; eventSlug: string };
  expect(response.status()).toBe(200);
  booking.eventSlug = eventSlug;
  return booking;
}

export async function rescheduleEvent(username: string, frame: Frame, page: Page) {
  await selectFirstAvailableTimeSlotNextMonth(frame, page);
  // --- fill form
  await frame.press('[name="email"]', "Enter");
  const responsePromise = page.waitForResponse("**/api/book/event");
  await frame.click("[data-testid=confirm-reschedule-button]");
  const response = await responsePromise;
  expect(response.status()).toBe(200);
  const responseObj = await response.json();
  const booking = responseObj.uid;
  return booking;
}
export async function installAppleCalendar(page: Page) {
  await page.goto("/apps/categories/calendar");
  await page.click('[data-testid="app-store-app-card-apple-calendar"]');
  await page.waitForURL("/apps/apple-calendar");
  await page.click('[data-testid="install-app-button"]');
}

export async function assertNoRequestIsBlocked(page: Page) {
  page.on("requestfailed", (request) => {
    const error = request.failure()?.errorText;
    // Identifies that the request is blocked by the browser due to COEP restrictions
    if (error?.includes("ERR_BLOCKED_BY_RESPONSE")) {
      throw new Error(`Request Blocked: ${request.url()}. Error: ${error}`);
    }
  });
}

export async function expectEmbedIFrameToBeVisible({
  calNamespace,
  page,
}: {
  calNamespace: string;
  page: Page;
}) {
  const iframe = page.locator(`[name="cal-embed=${calNamespace}"]`);
  await expect(iframe).toBeVisible();
}

export async function expectActualFormResponseConnectedToQueuedFormResponse({
  queuedFormResponse,
  page,
  numberOfExpectedSetFieldValues,
}: {
  queuedFormResponse: { id: string };
  page: Page;
}) {
  const responsePromise = page.waitForResponse("**/queued-response");
  const response = await responsePromise;
  expect(response.status()).toBe(200);

  const queuedFormResponseFromDb = await getQueuedFormResponse(queuedFormResponse.id);

  expect(queuedFormResponseFromDb?.actualResponse?.id).toBeDefined();
  const responseFromDb = queuedFormResponseFromDb?.actualResponse?.response;
  expect(responseFromDb).toBeDefined();
  const valuesFromResponse = Object.values(responseFromDb).map((item) => item.value);
  const valuesSetInResponse = valuesFromResponse.filter((value) => !!value);
  // We are unable to verify the exact response values because we don't directly have the form field identifiers in responseFromDB and would require correlating that with the actual Form from DB
  // So, for now we just verify the number of values set
  // There are 5 values that are submitted when CTA is clicked.
  // TODO: We should be able to verify the exact response values by correlating the form field identifiers with the actual Form from DB
  expect(valuesSetInResponse.length).toBe(numberOfExpectedSetFieldValues);
}

export async function cancelBookingThroughEmbed(bookingUid: string, frame: Frame, page: Page) {
  await frame.waitForSelector('[data-testid="cancel_reason"]');

  await frame.fill('[data-testid="cancel_reason"]', "Test cancellation from embed");

  const responsePromise = page.waitForResponse("**/api/cancel");
  await frame.click('[data-testid="confirm_cancel"]');

  const response = await responsePromise;
  expect(response.status()).toBe(200);

  await expect(frame.locator('[data-testid="cancelled-headline"]')).toBeVisible();

  return response;
}