From 08e1bbf8aa2dbf278cc56655572e07ea66bd758c Mon Sep 17 00:00:00 2001
From: John Hildenbiddle <john@hildenbiddle.com>
Date: Tue, 7 Jul 2020 13:20:49 -0500
Subject: [PATCH] Add docsifyURL and waitForSelector options

---
 tests/helpers/docsifyInit.js | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/tests/helpers/docsifyInit.js b/tests/helpers/docsifyInit.js
index f3bf6479..68d24e24 100644
--- a/tests/helpers/docsifyInit.js
+++ b/tests/helpers/docsifyInit.js
@@ -14,6 +14,8 @@ async function docsifyInit(page, options) {
     script: '',
     scriptURLs: [],
     url: `http://${serverIP}:${serverPort}`,
+    docsifyURL: '/lib/docsify.js',
+    waitForSelector: '#main',
   };
   const settings = {
     ...defaults,
@@ -73,6 +75,11 @@ async function docsifyInit(page, options) {
   for (const url of settings.scriptURLs) {
     await page.addScriptTag({ url });
   }
+
+  // Detect docsify "complete" by waiting for specified element
+  if (settings.waitForSelector) {
+    await page.waitForSelector(settings.waitForSelector);
+  }
 }
 
 module.exports = docsifyInit;