enso/app/gui/integration-test/dashboard/actions/SetupPlanPageActions.ts

/** @file Actions for the second step of the "setup" page. */
import { PLAN_TO_UPGRADE_LABEL_ID } from '#/modules/payments/constants'
import { Plan } from 'enso-common/src/services/Backend'
import { TEXT } from '.'
import BaseActions from './BaseActions'
import SetupDonePageActions from './SetupDonePageActions'
import SetupOrganizationPageActions from './SetupOrganizationPageActions'

// ============================
// === SetupPlanPageActions ===
// ============================

/** Actions for the "select plan" step of the "setup" page. */
export default class SetupPlanPageActions extends BaseActions {
  /** Select a plan. */
  selectSoloPlan() {
    return this.step(`Select 'solo' plan`, async (page) => {
      await page.getByLabel(TEXT[PLAN_TO_UPGRADE_LABEL_ID[Plan.solo]]).click()
      await page
        .getByRole('group', { name: TEXT.licenseAgreementCheckbox })
        .getByText(TEXT.licenseAgreementCheckbox)
        .click()
      await page.getByText(TEXT.startTrial).click()
    }).into(SetupDonePageActions)
  }

  /** Select a plan that has teams. */
  selectTeamPlan(plan: Plan.enterprise | Plan.team, seats = 1, duration: 12 | 36 = 12) {
    return this.step(`Select '${plan}' plan`, async (page) => {
      await page.getByLabel(TEXT[PLAN_TO_UPGRADE_LABEL_ID[plan]]).click()
      await page
        .getByRole('group', { name: TEXT.licenseAgreementCheckbox })
        .getByText(TEXT.licenseAgreementCheckbox)
        .click()
      await page.getByLabel(TEXT.seats).getByRole('spinbutton').fill(String(seats))
      await page
        .getByLabel(TEXT.billingPeriod)
        .getByText(duration === 12 ? TEXT.billingPeriodOneYear : TEXT.billingPeriodThreeYears)
        .click()
      await page.getByText(TEXT.startTrial).click()
    }).into(SetupOrganizationPageActions)
  }

  /** Stay on the current (free) plan. */
  stayOnFreePlan() {
    return this.step(`Stay on current plan`, async (page) => {
      await page.getByText(TEXT.skip).click()
    }).into(SetupDonePageActions)
  }

  /** Stay on the current (paid) plan. */
  stayOnPaidPlan() {
    return this.step(`Stay on current plan`, async (page) => {
      await page.getByText(TEXT.skip).click()
    }).into(SetupOrganizationPageActions)
  }
}