chore: added update email tests

apps/laboratory/tests/email.spec.ts

@@ -2,6 +2,11 @@ import { expect } from '@playwright/test'
 import { testMEmail } from './shared/fixtures/w3m-email-fixture'
 import { SECURE_WEBSITE_URL } from './shared/constants'
 
+const mailsacApiKey = process.env['MAILSAC_API_KEY']
+if (!mailsacApiKey) {
+  throw new Error('MAILSAC_API_KEY is not set')
+}
+
 testMEmail.beforeEach(async ({ modalValidator }) => {
   await modalValidator.expectConnected()
 })
@@ -48,3 +53,7 @@ testMEmail('it should disconnect correctly', async ({ modalPage, modalValidator
   await modalPage.disconnect()
   await modalValidator.expectDisconnected()
 })
+
+testMEmail('it should update email', async ({ modalPage }) => {
+  await modalPage.updateEmail(mailsacApiKey)
+})

apps/laboratory/tests/shared/pages/ModalPage.ts

@@ -13,6 +13,7 @@ export class ModalPage {
 
   private readonly connectButton: Locator
   private readonly url: string
+  private emailAddress = ''
 
   constructor(
     public readonly page: Page,
@@ -61,6 +62,8 @@ export class ModalPage {
   ): Promise<void> {
     await this.load()
 
+    this.emailAddress = emailAddress
+
     const email = new Email(mailsacApiKey)
 
     await email.deleteAllMessages(emailAddress)
@@ -90,9 +93,11 @@ export class ModalPage {
         otp = email.getOtpCodeFromBody(emailBody)
       }
     }
-    if (otp.replace(' ', '').length !== 6) {
+
+    if (otp === '') {
       otp = email.getOtpCodeFromBody(emailBody)
     }
+
     await this.enterOTP(otp)
   }
 
@@ -114,16 +119,15 @@ export class ModalPage {
     })
   }
 
-  async enterOTP(otp: string) {
-    await expect(this.page.getByText('Confirm Email')).toBeVisible({
+  async enterOTP(otp: string, headerTitle = 'Confirm Email') {
+    await expect(this.page.getByText(headerTitle)).toBeVisible({
       timeout: 10_000
     })
     await expect(this.page.getByText('Enter the code we sent')).toBeVisible({
       timeout: 10_000
     })
+
     const splitted = otp.split('')
-    // Remove empy space in OTP code 111 111
-    splitted.splice(3, 1)
 
     // eslint-disable-next-line no-plusplus
     for (let i = 0; i < splitted.length; i++) {
@@ -143,7 +147,7 @@ export class ModalPage {
       await input.fill(digit)
     }
 
-    await expect(this.page.getByText('Confirm Email')).not.toBeVisible()
+    await expect(this.page.getByText(headerTitle)).not.toBeVisible()
   }
 
   async disconnect() {
@@ -226,4 +230,48 @@ export class ModalPage {
   async closeModal() {
     await this.page.getByTestId('w3m-header-close')?.click?.()
   }
+
+  async updateEmail(mailsacApiKey: string) {
+    const email = new Email(mailsacApiKey)
+    const newEmailAddress = email.getEmailAddressToUse(1)
+
+    await this.page.getByTestId('account-button').click()
+    await this.page.getByTestId('w3m-account-email-update').click()
+    await this.page.getByTestId('wui-email-input').locator('input').focus()
+    await this.page.getByTestId('wui-email-input').locator('input').fill(newEmailAddress)
+
+    // Clear messages before putting new email
+    await email.deleteAllMessages(this.emailAddress)
+    await this.page.getByTestId('wui-email-input').locator('input').press('Enter')
+
+    // Wait until the next screen appears
+    await expect(this.page.getByText('Enter the code we sent')).toBeVisible({
+      timeout: 10_000
+    })
+    const confirmCurrentEmail = await this.page.getByText('Confirm Current Email').isVisible()
+    if (confirmCurrentEmail) {
+      await this.updateOtpFlow(this.emailAddress, mailsacApiKey, 'Confirm Current Email')
+    }
+
+    await this.updateOtpFlow(newEmailAddress, mailsacApiKey, 'Confirm New Email')
+
+    expect(
+      this.page.getByTestId('w3m-account-email-update'),
+      `Expected to go to the account screen after the update`
+    ).toBeVisible()
+  }
+
+  async updateOtpFlow(emailAddress: string, mailsacApiKey: string, headerTitle: string) {
+    const email = new Email(mailsacApiKey)
+
+    const messageId = await email.getLatestMessageId(emailAddress)
+
+    if (!messageId) {
+      throw new Error('No messageId found')
+    }
+    const emailBody = await email.getEmailBody(emailAddress, messageId)
+    const otp = email.getOtpCodeFromBody(emailBody)
+
+    await this.enterOTP(otp, headerTitle)
+  }
 }

apps/laboratory/tests/shared/utils/email.ts

@@ -70,7 +70,8 @@ export class Email {
   getOtpCodeFromBody(body: string): string {
     const match = body.match(OTP_CODE_REGEX)
     if (match) {
-      return match[0]
+      // Remove empty space in OTP code 111 111
+      return match[0].replace(' ', '')
     }
 
     throw new Error(`No code found in email: ${body}`)