Added conditional for WebDriver in 'screenshotElement' method. (#2)

index.js

@@ -94,8 +94,17 @@ class ResembleHelper extends Helper {
       await el.screenshot({
         path: configuration.screenshotFolder + name + '.png'
       });
-    }
-    else throw new Error("Method only works with Puppeteer");
+    } else if (this.helpers['WebDriver']) {
+      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 el.saveScreenshot(configuration.screenshotFolder + name + '.png');
+  }
+    else throw new Error("Method only works with Puppeteer and WebDriver helpers.");
   }
 
   /**

package.json

@@ -1,6 +1,6 @@
 {
   "name": "codeceptjs-resemblehelper",
-  "version": "1.6.0",
+  "version": "1.7.0",
   "description": "Resemble Js helper for CodeceptJS, with Support for Webdriver, Puppeteer & Appium",
   "repository": {
     "type": "git",