try to use default constructor if no caps are provided

searls · Jan 13, 2014 · adf2d60 · adf2d60
1 parent 26a954c
commit adf2d60
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/main/java/com/github/searls/jasmine/driver/WebDriverFactory.java b/src/main/java/com/github/searls/jasmine/driver/WebDriverFactory.java
@@ -57,10 +57,17 @@ private Class<? extends WebDriver> getWebDriverClass() throws Exception {
 
   private Constructor<? extends WebDriver> getWebDriverConstructor() throws Exception {
     Class<? extends WebDriver> webDriverClass = getWebDriverClass();
+    boolean hasCapabilities = !webDriverCapabilities.isEmpty();
     try {
-      return webDriverClass.getConstructor(Capabilities.class);
-    } catch (Exception exception) {
+      if (hasCapabilities) {
+        return webDriverClass.getConstructor(Capabilities.class);
+      }
       return webDriverClass.getConstructor();
+    } catch (Exception exception) {
+      if (hasCapabilities) {
+        return webDriverClass.getConstructor();
+      }
+      return webDriverClass.getConstructor(Capabilities.class);
     }
   }