Added ablility to set headless mode for webdriver with firefox too (#14)

Author: web-mi

README.md

@@ -45,7 +45,7 @@ exports.config = {
 ```
 
 * For Puppeteer, TestCafe, Nigthmare, Playwright: it enables `show: true`.
-* For WebDriver with Chrome browser: it adds `--headless` option to chrome options inside `desiredCapabilities`.
+* For WebDriver with Chrome or Firefox browser: it adds `--headless` option to chrome/firefox options inside `desiredCapabilities`.
 
 ### setHeadedWhen
 
@@ -80,7 +80,7 @@ exports.config = {
   helpers: {
     WebDriver: {
       // standard config goes here      
-    } 
+    },
     // or Puppeteer
     // or TestCafe,
     REST: {

hooks/setHeadedWhen.js

@@ -5,12 +5,12 @@ module.exports = function(when) {
 
   config.addHook(cfg => {
     if (!cfg.helpers) return;
-    if (cfg.helpers.Puppeteer) {      
+    if (cfg.helpers.Puppeteer) {
       cfg.helpers.Puppeteer.show = true;
     }
     if (cfg.helpers.Playwright) {
       cfg.helpers.Playwright.show = true;
-    }    
+    }
     if (cfg.helpers.Nightmare) {
       cfg.helpers.Nightmare.show = true;
     }
@@ -20,21 +20,38 @@ module.exports = function(when) {
     if (cfg.helpers.WebDriver) {
       if (cfg.helpers.WebDriver.browser === 'chrome') {
         if (
-          cfg.helpers.WebDriver.desiredCapabilities && 
+          cfg.helpers.WebDriver.desiredCapabilities &&
           cfg.helpers.WebDriver.desiredCapabilities.chromeOptions &&
           cfg.helpers.WebDriver.desiredCapabilities.chromeOptions.args
           ) {
             const args = cfg.helpers.WebDriver.desiredCapabilities.chromeOptions.args;
             cfg.helpers.WebDriver.desiredCapabilities.chromeOptions.args = args.filter(key => key !== '--headless')
         }
         if (
-          cfg.helpers.WebDriver.capabilities && 
+          cfg.helpers.WebDriver.capabilities &&
           cfg.helpers.WebDriver.capabilities.chromeOptions &&
           cfg.helpers.WebDriver.capabilities.chromeOptions.args
           ) {
             const args = cfg.helpers.WebDriver.capabilities.chromeOptions.args;
             cfg.helpers.WebDriver.capabilities.chromeOptions.args = args.filter(key => key !== '--headless')
         }
+      } else if (cfg.helpers.WebDriver.browser === 'firefox') {
+        if (
+            cfg.helpers.WebDriver.desiredCapabilities &&
+            cfg.helpers.WebDriver.desiredCapabilities.firefoxOptions &&
+            cfg.helpers.WebDriver.desiredCapabilities.firefoxOptions.args
+        ) {
+          const args = cfg.helpers.WebDriver.desiredCapabilities.firefoxOptions.args;
+          cfg.helpers.WebDriver.desiredCapabilities.firefoxOptions.args = args.filter(key => key !== '--headless')
+        }
+        if (
+            cfg.helpers.WebDriver.capabilities &&
+            cfg.helpers.WebDriver.capabilities.firefoxOptions &&
+            cfg.helpers.WebDriver.capabilities.firefoxOptions.args
+        ) {
+          const args = cfg.helpers.WebDriver.capabilities.firefoxOptions.args;
+          cfg.helpers.WebDriver.capabilities.firefoxOptions.args = args.filter(key => key !== '--headless')
+        }
 
       }
     }

hooks/setHeadlessWhen.js

@@ -11,7 +11,7 @@ module.exports = function(when) {
     }
     if (cfg.helpers.Playwright) {
       cfg.helpers.Playwright.show = false;
-    }    
+    }
     if (cfg.helpers.Nightmare) {
       cfg.helpers.Nightmare.show = false;
     }
@@ -25,7 +25,7 @@ module.exports = function(when) {
           cfg.helpers.WebDriver.desiredCapabilities || {},
           {
             chromeOptions: {
-              args: [ 
+              args: [
                 "--headless",
                 // Use --disable-gpu to avoid an error from a missing Mesa library, as per
                 // https://chromium.googlesource.com/chromium/src/+/lkgr/headless/README.md
@@ -34,6 +34,17 @@ module.exports = function(when) {
             }
           }
         )
+      } else if (cfg.helpers.WebDriver.browser === 'firefox') {
+        cfg.helpers.WebDriver.desiredCapabilities = merge(
+            cfg.helpers.WebDriver.desiredCapabilities || {},
+            {
+              firefoxOptions: {
+                args: [
+                  "--headless",
+                ]
+              }
+            }
+        )
       }
     }
   });