From 23c4652c9c5d1d972f23091d767bf6142ef53d6e Mon Sep 17 00:00:00 2001
From: Michael Telatynski <7t3chguy@gmail.com>
Date: Mon, 27 Jan 2025 13:01:30 +0000
Subject: [PATCH] Iterate

Signed-off-by: Michael Telatynski <7t3chguy@gmail.com>
---
 playwright/e2e/oidc/index.ts          | 10 +++-------
 playwright/e2e/register/email.spec.ts |  7 +++----
 2 files changed, 6 insertions(+), 11 deletions(-)

diff --git a/playwright/e2e/oidc/index.ts b/playwright/e2e/oidc/index.ts
index 9afa93fd70..5efc245309 100644
--- a/playwright/e2e/oidc/index.ts
+++ b/playwright/e2e/oidc/index.ts
@@ -27,15 +27,11 @@ export async function registerAccountMas(
     await page.getByRole("textbox", { name: "Confirm Password" }).fill(password);
     await page.getByRole("button", { name: "Continue" }).click();
 
-    let summary: MailpitMessagesSummaryResponse;
+    let code: string;
     await expect(async () => {
-        summary = await mailpit.listMessages();
-        expect(summary.messages_count).toBe(1);
+        const text = await mailpit.renderMessageText();
+        [, code] = text.match(/Your verification code to confirm this email address is: (\d{6})/);
     }).toPass();
-    expect(summary.messages[0].To[0].Address).toEqual(email);
-    const [, code] = summary.messages[0].Snippet.match(
-        /Your verification code to confirm this email address is: (\d{6})/,
-    );
 
     await page.getByRole("textbox", { name: "6-digit code" }).fill(code);
     await page.getByRole("button", { name: "Continue" }).click();
diff --git a/playwright/e2e/register/email.spec.ts b/playwright/e2e/register/email.spec.ts
index 5125c70f64..5ef45ec8fd 100644
--- a/playwright/e2e/register/email.spec.ts
+++ b/playwright/e2e/register/email.spec.ts
@@ -51,10 +51,9 @@ test.describe("Email Registration", async () => {
 
             await expect(page.getByText("An error was encountered when sending the email")).not.toBeVisible();
 
-            const summary = await mailpitClient.listMessages();
-            expect(summary.messages_count).toBe(1);
-            expect(summary.messages[0].To[0].Address).toEqual("alice@email.com");
-            const [emailLink] = summary.messages[0].Snippet.match(/http.+/);
+            const { Links: links } = await mailpitClient.linkCheck();
+            expect(links).toHaveLength(1);
+            const emailLink = links[0].URL;
             await request.get(emailLink); // "Click" the link in the email
 
             await expect(page.getByText("Welcome alice")).toBeVisible();