From 050a618215339ba7ed5a27b898c3c672a99a449a Mon Sep 17 00:00:00 2001
From: Toussaint Louis <66522867+lotoussa@users.noreply.github.com>
Date: Mon, 4 Nov 2024 14:56:28 +0200
Subject: [PATCH] Tests(DPxAI): update test for class-it

---
 dom/class-it_test.js | 25 +++++++++++++++++--------
 1 file changed, 17 insertions(+), 8 deletions(-)

diff --git a/dom/class-it_test.js b/dom/class-it_test.js
index f84a44592..1112c78ce 100644
--- a/dom/class-it_test.js
+++ b/dom/class-it_test.js
@@ -26,11 +26,20 @@ tests.push(async ({ page, eq }) => {
 })
 
 tests.push(async ({ page, eq }) => {
-  // check that the targetted elements have the correct class names
-  await eq.$('p#eye-left', { className: 'eye' })
-  await eq.$('p#eye-right', { className: 'eye' })
-  await eq.$('div#arm-left', { className: 'arm body-member' })
-  await eq.$('div#arm-right', { className: 'arm body-member' })
-  await eq.$('div#leg-left', { className: 'leg body-member' })
-  await eq.$('div#leg-right', { className: 'leg body-member' })
-})
+  // Helper function to check if an element has the expected classes
+  async function hasClasses(selector, expectedClasses) {
+    const element = await page.$(selector);
+    const className = await element.evaluate(el => el.className);
+    const classList = className.split(' ').sort();
+    const expectedClassList = expectedClasses.split(' ').sort();
+    return JSON.stringify(classList) === JSON.stringify(expectedClassList);
+  }
+
+  // Check that the target elements have the correct classes
+  eq(await hasClasses('p#eye-left', 'eye'), true);
+  eq(await hasClasses('p#eye-right', 'eye'), true);
+  eq(await hasClasses('div#arm-left', 'arm body-member'), true);
+  eq(await hasClasses('div#arm-right', 'arm body-member'), true);
+  eq(await hasClasses('div#leg-left', 'leg body-member'), true);
+  eq(await hasClasses('div#leg-right', 'leg body-member'), true);
+});