fix: Setting content directly does not allow for use of DOM queries (#310)

route · web-flow · commit 277f04ebc618 · 2022-11-20T21:18:06.000+05:00
fixes #197
diff --git a/lib/ferrum/frame.rb b/lib/ferrum/frame.rb
@@ -113,6 +113,7 @@ def content=(html)
         document.close();
         arguments[1](true);
       ), @page.timeout, html)
+      @page.document_node_id
     end
     alias set_content content=
 
diff --git a/lib/ferrum/page.rb b/lib/ferrum/page.rb
@@ -325,6 +325,10 @@ def use_authorized_proxy?
       use_proxy? && @proxy_user && @proxy_password
     end
 
+    def document_node_id
+      command("DOM.getDocument", depth: 0).dig("root", "nodeId")
+    end
+
     private
 
     def subscribe
@@ -441,10 +445,6 @@ def combine_url!(url_or_path)
       (nil_or_relative ? @browser.base_url.join(url.to_s) : url).to_s
     end
 
-    def document_node_id
-      command("DOM.getDocument", depth: 0).dig("root", "nodeId")
-    end
-
     def ws_url
       "ws://#{@browser.process.host}:#{@browser.process.port}/devtools/page/#{@target_id}"
     end
diff --git a/spec/frame_spec.rb b/spec/frame_spec.rb
@@ -181,9 +181,10 @@
   end
 
   it "can set page content" do
-    browser.content = "<html><head></head><body>Voila!</body></html>"
+    browser.content = "<html><head></head><body>Voila! <a href='#'>Link</a></body></html>"
 
     expect(browser.body).to include("Voila!")
+    expect(browser.at_css("a").text).to eq("Link")
   end
 
   it "gets page doctype" do
