Merge pull request #65 from automate-website/#64-chrome-headless

Added support for Chrome headless (#64)
automate-website · Jul 9, 2017 · 2aed3b7 · 2aed3b7
2 parents dc6359c + 36e55c5
commit 2aed3b7
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 12 deletions.
diff --git a/src/main/java/website/automate/jwebrobot/executor/WebDriverProvider.java b/src/main/java/website/automate/jwebrobot/executor/WebDriverProvider.java
@@ -11,17 +11,22 @@ public class WebDriverProvider {
 
     public WebDriver createInstance(Type type) {
         switch (type) {
-            case FIREFOX:
-                return new FirefoxDriver();
             case CHROME:
                 return new ChromeDriver();
+            case CHROME_HEADLESS:
+                ChromeOptions options = new ChromeOptions();
+                options.addArguments("--headless", "--disable-gpu");
+                ChromeDriver chromeDriver = new ChromeDriver(options);
+
+                return chromeDriver;
             case OPERA:
                 return new OperaDriver(getOperaCapabilities());
+            case FIREFOX:
             default:
-                throw new RuntimeException("Unsupported web driver type.");
+                return new FirefoxDriver();
         }
     }
-    
+
     private DesiredCapabilities getOperaCapabilities(){
         DesiredCapabilities capabilities = DesiredCapabilities.operaBlink();
         String operaBinaryPath = System.getProperty("webdriver.opera.binary");
@@ -34,16 +39,21 @@ private DesiredCapabilities getOperaCapabilities(){
     }
 
     public enum Type {
-        FIREFOX,
-        CHROME,
-        OPERA;
+        FIREFOX("firefox"),
+        CHROME("chrome"),
+        CHROME_HEADLESS("chrome-headless"),
+        OPERA("opera");
+
+        private final String canonicalName;
+
+        Type(String canonicalName) {
+            this.canonicalName = canonicalName;
+        }
 
         public static Type fromString(String typeString) {
-            if (typeString != null) {
-                for (Type type : Type.values()) {
-                    if (typeString.equalsIgnoreCase(type.name())) {
-                        return type;
-                    }
+            for (Type type : Type.values()) {
+                if (type.canonicalName.equalsIgnoreCase(typeString)) {
+                    return type;
                 }
             }
 

diff --git a/src/test/java/website/automate/jwebrobot/ChromeHeadlessTests.java b/src/test/java/website/automate/jwebrobot/ChromeHeadlessTests.java
@@ -0,0 +1,4 @@
+package website.automate.jwebrobot;
+
+public interface ChromeHeadlessTests extends ChromeTests {
+}
diff --git a/src/test/java/website/automate/jwebrobot/JWebRobotIT.java b/src/test/java/website/automate/jwebrobot/JWebRobotIT.java
@@ -64,4 +64,14 @@ public void simpleScenarioShouldBeExecutedWithChrome() {
             "-browser", "chrome"
         });
     }
+
+    @Ignore
+    @Test
+    @Category(ChromeHeadlessTests.class)
+    public void simpleScenarioShouldBeExecutedWithChromeHeadless() {
+        JWebRobot.main(new String [] {
+            "-" + SCENARIO_PATH_PARAM_NAME, ROOT_PACKAGE_DIRECTORY_PATH + "wikipedia-test.yaml",
+            "-browser", "chrome-headless"
+        });
+    }
 }