Fix/await dom updated before axe scan (#398)

* Await DOM elements to be updated before running axeScan

* Add missing Intelligent case statement for getUrlMessage

---------

Co-authored-by: Wilson WeeSheng Khoo <[email protected]>

src/constants/common.ts

@@ -242,6 +242,7 @@ export const getUrlMessage = (scanner: ScannerTypes): string => {
   switch (scanner) {
     case ScannerTypes.WEBSITE:
     case ScannerTypes.CUSTOM:
+    case ScannerTypes.INTELLIGENT:
       return 'Please enter URL of website: ';
     case ScannerTypes.SITEMAP:
       return 'Please enter URL or file path to sitemap, or drag and drop a sitemap file here: ';

src/crawlers/commonCrawlerFunc.ts

@@ -158,6 +158,29 @@ export const runAxeScript = async (
   customFlowDetails,
   selectors = [],
 ) => {
+  // Checking for DOM mutations before proceeding to scan
+  await page.evaluate(() => {
+    return new Promise((resolve) => {
+      let timeout;
+
+      const observer = new MutationObserver(() => {
+        clearTimeout(timeout);
+
+        timeout = setTimeout(() => {
+          observer.disconnect();
+          resolve('DOM stabilized after mutations.');
+        }, 1000);
+      });
+
+      timeout = setTimeout(() => {
+        observer.disconnect();
+        resolve('No mutations detected, exit from idle state');
+      }, 1000);
+
+      observer.observe(document, { childList: true, subtree: true, attributes: true });
+    });
+  });
+
   await crawlee.playwrightUtils.injectFile(page, axeScript);
 
   const results = await page.evaluate(