Implementing FindsByClassName in HtmlUnitDriver using CSS instead of …

…XPath
SeleniumHQ · Jan 1, 2014 · e01454b · e01454b
1 parent e2b3aae
commit e01454b
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/java/client/src/org/openqa/selenium/htmlunit/HtmlUnitDriver.java b/java/client/src/org/openqa/selenium/htmlunit/HtmlUnitDriver.java
@@ -86,6 +86,7 @@
 import org.openqa.selenium.interactions.HasInputDevices;
 import org.openqa.selenium.interactions.Keyboard;
 import org.openqa.selenium.interactions.Mouse;
+import org.openqa.selenium.internal.FindsByClassName;
 import org.openqa.selenium.internal.FindsByCssSelector;
 import org.openqa.selenium.internal.FindsById;
 import org.openqa.selenium.internal.FindsByLinkText;
@@ -119,7 +120,7 @@
 
 public class HtmlUnitDriver implements WebDriver, JavascriptExecutor,
     FindsById, FindsByLinkText, FindsByXPath, FindsByName, FindsByCssSelector,
-    FindsByTagName, HasCapabilities, HasInputDevices {
+    FindsByTagName, FindsByClassName, HasCapabilities, HasInputDevices {
 
   private WebClient webClient;
   private WebWindow currentWindow;
@@ -914,6 +915,16 @@ public List<WebElement> findElementsById(String id) {
     return findElementsByXPath("//*[@id='" + id + "']");
   }
 
+  @Override
+  public WebElement findElementByClassName(String className) {
+    return findElementByCssSelector("." + className);
+  }
+
+  @Override
+  public List<WebElement> findElementsByClassName(String className) {
+    return findElementsByCssSelector("." + className);
+  }
+
   public WebElement findElementByCssSelector(String using) {
     if (!(lastPage() instanceof HtmlPage)) {
       throw new NoSuchElementException("Unable to locate element using css: " + lastPage());