fix test

aaronmcadam · Aug 5, 2024 · 177a61b · 177a61b
1 parent 3f24b76
commit 177a61b
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/tests/visitor-completes-quiz.spec.ts b/tests/visitor-completes-quiz.spec.ts
@@ -26,12 +26,23 @@ test("visitor accepted for treatment", async ({ page }) => {
       name: /which image best matches your hair loss/i,
     }),
   ).toBeVisible();
-
   await page
     .getByRole("button", { name: /temples/i })
     .first()
     .click();
+
+  await expect(
+    page.getByRole("heading", {
+      name: /which image best matches your hair loss/i,
+    }),
+  ).toBeVisible();
   await page.getByRole("button", { name: /no/i }).click();
+
+  await expect(
+    page.getByRole("heading", {
+      name: /breast cancer/i,
+    }),
+  ).toBeVisible();
   await page.getByRole("button", { name: /no/i }).click();
 
   await expect(