Merge pull request #24 from robertgates55/electron-fix

Updated the electron driver so that it works

Author: robertgates55

pom.xml

@@ -5,13 +5,18 @@
 import org.openqa.selenium.remote.DesiredCapabilities;
 import org.openqa.selenium.remote.RemoteWebDriver;
 
+import java.net.MalformedURLException;
+import java.net.URL;
 import java.util.HashMap;
 import java.util.Map;
 
 import static com.frameworkium.config.SystemProperty.APP_PATH;
+import static com.frameworkium.config.SystemProperty.GRID_URL;
 
 public class ElectronImpl extends DriverType {
 
+    private static URL remoteURL;
+
     @Override
     public DesiredCapabilities getDesiredCapabilities() {
         Map<String, String> chromeOptions = new HashMap<>();
@@ -20,17 +25,25 @@ public DesiredCapabilities getDesiredCapabilities() {
         } else {
             chromeOptions.put("binary", APP_PATH.getValue());
         }
-        DesiredCapabilities desiredCapabilities = new DesiredCapabilities();
-        desiredCapabilities.setCapability("host", "localhost");
-        desiredCapabilities.setCapability("port", 9515);
+        try {
+            if (GRID_URL.isSpecified()) {
+                remoteURL = new URL(GRID_URL.getValue());
+            }
+            else {
+                remoteURL = new URL("http://localhost:9515");
+            }
+        }
+        catch(MalformedURLException e) {
+            throw new RuntimeException(e);
+        }
+        DesiredCapabilities desiredCapabilities = DesiredCapabilities.chrome();
         desiredCapabilities.setCapability("browserName", "chrome");
         desiredCapabilities.setCapability("chromeOptions", chromeOptions);
         return desiredCapabilities;
     }
 
     @Override
     public WebDriver getWebDriverObject(DesiredCapabilities capabilities) {
-        return new RemoteWebDriver(capabilities);
+        return new RemoteWebDriver(remoteURL ,capabilities);
     }
-
 }