Added check for puppeteer helper

codeceptjs · Jun 1, 2019 · 95b4591 · 95b4591
1 parent 3dd89ee
commit 95b4591
Showing 1 changed file with 13 additions and 9 deletions.
diff --git a/index.js b/index.js
@@ -75,17 +75,21 @@ class ResembleHelper extends Helper {
    * @returns {Promise<void} 
    */
   async screenshotElement(selector, name) {
-    const helper = this._getHelper();
-    const configuration = this.config;
+
+    if (this.helpers['Puppeteer']) {
+      const helper = this._getHelper();
+      const configuration = this.config;
 
-    await helper.waitForVisible(selector);
-    const els = await helper._locate(selector);
-    if (!els.length) throw new Error(`Element ${selector} couldn't be located`);
-    const el = els[0];
+      await helper.waitForVisible(selector);
+      const els = await helper._locate(selector);
+      if (!els.length) throw new Error(`Element ${selector} couldn't be located`);
+      const el = els[0];
 
-    await el.screenshot({
-      path: configuration.screenshotFolder + name + '.png'
-    });
+      await el.screenshot({
+        path: configuration.screenshotFolder + name + '.png'
+      });
+    }
+    else throw new Error("Method to be called only with Puppeteer. Other helpers not supported");
   }
 
   /**