#52 Add set_content method

route · route · commit 13ab43d4983d · 2020-02-21T22:37:00.000+01:00
diff --git a/README.md b/README.md
@@ -705,6 +705,12 @@ browser.evaluate("window.__injected") # => 42
 #### frames
 #### main_frame
 #### frame_by
+#### set_content(html)
+
+Sets a content of given frame
+
+  * html `String`
+
 
 Play around inside given frame
 
diff --git a/lib/ferrum/browser.rb b/lib/ferrum/browser.rb
@@ -17,7 +17,7 @@ class Browser
     delegate %i[default_context] => :contexts
     delegate %i[targets create_target create_page page pages windows] => :default_context
     delegate %i[goto back forward refresh reload stop
-                at_css at_xpath css xpath current_url title body doctype
+                at_css at_xpath css xpath current_url title body doctype set_content
                 headers cookies network
                 mouse keyboard
                 screenshot pdf viewport_size
diff --git a/lib/ferrum/frame.rb b/lib/ferrum/frame.rb
@@ -35,6 +35,15 @@ def main?
       @parent_id.nil?
     end
 
+    def set_content(html)
+      evaluate_async(%(
+        document.open();
+        document.write(arguments[0]);
+        document.close();
+        arguments[1](true);
+      ), @page.timeout, html)
+    end
+
     def execution_id?(execution_id)
       @execution_id == execution_id
     end
diff --git a/lib/ferrum/page.rb b/lib/ferrum/page.rb
@@ -31,7 +31,7 @@ def reset
 
     extend Forwardable
     delegate %i[at_css at_xpath css xpath
-                current_url current_title url title body doctype
+                current_url current_title url title body doctype set_content
                 execution_id evaluate evaluate_on evaluate_async execute
                 add_script_tag add_style_tag] => :main_frame
 
diff --git a/spec/frame_spec.rb b/spec/frame_spec.rb
@@ -235,6 +235,12 @@ module Ferrum
         expect(frame.url).to end_with("/ferrum/slow")
         expect(browser.current_url).to end_with("/ferrum/frames")
       end
+
+      it "can set page content" do
+        browser.set_content(%(<html><head></head><body>Voila!</body></html>))
+
+        expect(browser.body).to include("Voila!")
+      end
     end
   end
 end
