Updated the electron driver so that it works

Frameworkium · Nov 17, 2015 · 1f052c2 · 1f052c2
1 parent 4e86aeb
commit 1f052c2
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/main/java/com/frameworkium/config/drivers/ElectronImpl.java b/src/main/java/com/frameworkium/config/drivers/ElectronImpl.java
@@ -14,23 +14,23 @@ public class ElectronImpl extends DriverType {
 
     @Override
     public DesiredCapabilities getDesiredCapabilities() {
-        Map<String, String> chromeOptions = new HashMap<>();
+
+        ChromeOptions options = new ChromeOptions();
         if (!APP_PATH.isSpecified()) {
             logger.error("App path must be specified when using Electron!");
         } else {
-            chromeOptions.put("binary", APP_PATH.getValue());
+            options.setBinary(APP_PATH.getValue());
         }
-        DesiredCapabilities desiredCapabilities = new DesiredCapabilities();
+        DesiredCapabilities desiredCapabilities = new DesiredCapabilities().chrome();
         desiredCapabilities.setCapability("host", "localhost");
         desiredCapabilities.setCapability("port", 9515);
-        desiredCapabilities.setCapability("browserName", "chrome");
-        desiredCapabilities.setCapability("chromeOptions", chromeOptions);
+        desiredCapabilities.setCapability("chromeOptions", options);
         return desiredCapabilities;
     }
 
     @Override
     public WebDriver getWebDriverObject(DesiredCapabilities capabilities) {
-        return new RemoteWebDriver(capabilities);
+        return new ChromeDriver(capabilities);
     }
 
 }