Add the HasCapabilities to the WebDriverBackedSelenium in case gettin…

…g the wrapped driver and querying that looks confusing.
nimeacuerdo · Mar 25, 2013 · 583e783 · 583e783
1 parent de430e6
commit 583e783
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/java/client/src/org/openqa/selenium/WebDriverBackedSelenium.java b/java/client/src/org/openqa/selenium/WebDriverBackedSelenium.java
@@ -22,7 +22,8 @@
 
 import org.openqa.selenium.internal.WrapsDriver;
 
-public class WebDriverBackedSelenium extends DefaultSelenium implements WrapsDriver {
+public class WebDriverBackedSelenium extends DefaultSelenium
+    implements HasCapabilities, WrapsDriver {
   public WebDriverBackedSelenium(Supplier<WebDriver> maker, String baseUrl) {
     super(new WebDriverCommandProcessor(baseUrl, maker));
   }
@@ -34,4 +35,14 @@ public WebDriverBackedSelenium(WebDriver baseDriver, String baseUrl) {
   public WebDriver getWrappedDriver() {
     return ((WrapsDriver) commandProcessor).getWrappedDriver();
   }
+
+  @Override
+  public Capabilities getCapabilities() {
+    WebDriver driver = getWrappedDriver();
+    if (driver instanceof HasCapabilities) {
+      return ((HasCapabilities) driver).getCapabilities();
+    }
+
+    return null;
+  }
 }