Implement wait_for_network_idle

route · route · commit a2967cdc4b96 · 2020-01-21T19:56:25.000+03:00
diff --git a/lib/capybara/cuprite/driver.rb b/lib/capybara/cuprite/driver.rb
@@ -285,6 +285,10 @@ def clear_cookies
       browser.cookies.clear
     end
 
+    def wait_for_network_idle(**options)
+      browser.network.wait_for_idle(**options)
+    end
+
     def clear_memory_cache
       browser.network.clear(:cache)
     end
diff --git a/spec/integration/driver_spec.rb b/spec/integration/driver_spec.rb
@@ -1567,5 +1567,16 @@ def create_screenshot(file, *args)
       expect(@session.driver.frame_url).to end_with("/cuprite/frames")
       expect(@session.driver.current_url).to end_with("/cuprite/frames")
     end
+
+    it "waits for network idle" do
+      @session.visit "/cuprite/show_cookies"
+      expect(@session).not_to have_content("test_cookie")
+
+      @session.click_button "Set cookie slow"
+      @session.driver.wait_for_network_idle
+      @session.refresh
+
+      expect(@session).to have_content("test_cookie")
+    end
   end
 end
diff --git a/spec/support/test_app.rb b/spec/support/test_app.rb
@@ -58,6 +58,17 @@ def authorized?(login, password)
     request.cookies["capybara"]
   end
 
+  get "/cuprite/show_cookies" do
+    render_view "show_cookies"
+  end
+
+  get "/cuprite/set_cookie_slow" do
+    sleep 1
+    cookie_value = "test_cookie"
+    response.set_cookie("stealth", cookie_value)
+    "Cookie set to #{cookie_value}"
+  end
+
   get "/cuprite/slow" do
     sleep 0.2
     "slow page"
diff --git a/spec/support/views/show_cookies.erb b/spec/support/views/show_cookies.erb
@@ -0,0 +1,17 @@
+<!DOCTYPE html>
+<html>
+  <head>
+    <meta http-equiv="Content-type" content="text/html; charset=utf-8"/>
+    <script src="/cuprite/jquery.min.js" type="text/javascript" charset="utf-8"></script>
+    <script src="/cuprite/jquery-ui.min.js" type="text/javascript" charset="utf-8"></script>
+  </head>
+  <body>
+    <%= request.cookies["stealth"] %>
+    <button type="button" id="setCookie">Set cookie slow</button>
+    <script>
+      $("#setCookie").on("click", function(){
+        $.ajax("/cuprite/set_cookie_slow");
+      })
+    </script>
+  </body>
+</html>
