HtmlPage getBody returns the body and not the frameset

HtmlUnit · Nov 2, 2024 · 5ad1198 · 5ad1198
1 parent 8607b1a
commit 5ad1198
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/src/main/java/org/htmlunit/javascript/host/dom/Document.java b/src/main/java/org/htmlunit/javascript/host/dom/Document.java
@@ -73,9 +73,11 @@
 import org.htmlunit.html.HtmlAnchor;
 import org.htmlunit.html.HtmlArea;
 import org.htmlunit.html.HtmlAttributeChangeEvent;
+import org.htmlunit.html.HtmlBody;
 import org.htmlunit.html.HtmlElement;
 import org.htmlunit.html.HtmlEmbed;
 import org.htmlunit.html.HtmlForm;
+import org.htmlunit.html.HtmlFrameSet;
 import org.htmlunit.html.HtmlImage;
 import org.htmlunit.html.HtmlPage;
 import org.htmlunit.html.HtmlRb;
@@ -825,10 +827,21 @@ public HTMLCollection getApplets() {
     public HTMLElement getBody() {
         final Page page = getPage();
         if (page instanceof HtmlPage) {
-            final HtmlElement body = ((HtmlPage) page).getBody();
+            final HtmlPage htmlPage = (HtmlPage) page;
+            final HtmlElement body = htmlPage.getBody();
             if (body != null) {
                 return body.getScriptableObject();
             }
+
+            // strange but this returns the frameset element
+            final DomElement doc = htmlPage.getDocumentElement();
+            if (doc != null) {
+                for (final DomNode node : doc.getChildren()) {
+                    if (node instanceof HtmlFrameSet) {
+                        return (HTMLElement) node.getScriptableObject();
+                    }
+                }
+            }
         }
         return null;
     }