tests(Playwright): Full implementation of Page Object Models (POMs) to Playwright tests (#5608)

tests/utils/assertions.ts

@@ -1,6 +1,8 @@
 import { Page } from '@playwright/test';
 import { expect } from 'playwright-test-coverage';
 
+import { DOMOverlayPageObject } from '../pages';
+
 /**
  * Asserts the number of Modality Load Badges present on the page.
  */
@@ -11,5 +13,8 @@ export async function assertNumberOfModalityLoadBadges({
   page: Page;
   expectedCount: number;
 }) {
-  await expect(page.locator('css=div[data-cy^="ModalityLoadBadge-"]')).toHaveCount(expectedCount);
+  const domOverlayPageObject = new DOMOverlayPageObject(page);
+  const count = await domOverlayPageObject.viewport.getModalityLoadBadgeCount();
+
+  expect(count).toBe(expectedCount);
 }