[JENKINS-31631] Handle YUI submit <button> clicks

(cherry picked from commit 7fdb589)
jenkinsci · Nov 24, 2015 · 5defc34 · 5defc34
1 parent 24785d5
commit 5defc34
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 3 deletions.
diff --git a/test/src/main/java/com/gargoylesoftware/htmlunit/html/HtmlElementUtil.java b/test/src/main/java/com/gargoylesoftware/htmlunit/html/HtmlElementUtil.java
@@ -28,6 +28,9 @@
 import com.gargoylesoftware.htmlunit.WebClientUtil;
 
 import java.io.IOException;
+import java.util.Arrays;
+import java.util.HashSet;
+import java.util.Set;
 
 /**
  * {@link HtmlElement} helper methods.
@@ -58,4 +61,16 @@ public static Page click(HtmlElement element) throws IOException {
             WebClientUtil.waitForJSExec(webClient);
         }
     }
+
+    /**
+     * Does the supplied element define the specified HTML "class" name.
+     * @param element The element to check.
+     * @param className The HTML "class" name to check for.
+     * @return {@code true} if the element defines the specified class, otherwise {@code false}.
+     */
+    public static boolean hasClassName(HtmlElement element, String className) {
+        String classAttribute = element.getAttribute("class");
+        Set<String> classes = new HashSet<>(Arrays.asList(classAttribute.split(" ")));
+        return classes.contains(className);
+    }
 }
diff --git a/test/src/main/java/org/jvnet/hudson/test/JenkinsRule.java b/test/src/main/java/org/jvnet/hudson/test/JenkinsRule.java
@@ -37,6 +37,7 @@
 import com.gargoylesoftware.htmlunit.html.DomNode;
 import com.gargoylesoftware.htmlunit.html.HtmlButton;
 import com.gargoylesoftware.htmlunit.html.HtmlElement;
+import com.gargoylesoftware.htmlunit.html.HtmlElementUtil;
 import com.gargoylesoftware.htmlunit.html.HtmlForm;
 import com.gargoylesoftware.htmlunit.html.HtmlImage;
 import com.gargoylesoftware.htmlunit.html.HtmlInput;
@@ -1342,9 +1343,12 @@ public HtmlPage submit(HtmlForm form) throws Exception {
      */
     public HtmlPage submit(HtmlForm form, String name) throws Exception {
         for( HtmlElement e : form.getHtmlElementsByTagName("button")) {
-            HtmlElement p = (HtmlElement)e.getParentNode().getParentNode();
-            if(e instanceof HtmlButton && p.getAttribute("name").equals(name)) {
-                return (HtmlPage)HtmlFormUtil.submit(form, (HtmlButton) e);
+            HtmlElement p = (HtmlElement)e.getParentNode().getParentNode();                        
+            if (p.getAttribute("name").equals(name) && HtmlElementUtil.hasClassName(p, "yui-submit-button")) {
+                // For YUI handled submit buttons, just do a click.
+                return (HtmlPage) HtmlElementUtil.click(e);
+            } else if (e.getAttribute("name").equals(name)) {
+                return (HtmlPage) HtmlFormUtil.submit(form, e);
             }
         }
         throw new AssertionError("No such submit button with the name "+name);