lib/browsed/firefox.rb in browsed-0.2.4 vs lib/browsed/firefox.rb in browsed-0.2.5

@@ -1,12 +1,13 @@
 module Browsed
   module Firefox
     
     private
-      def register_firefox_driver(options: {}, timeout: 60, debug: false)
+      def register_firefox_driver(options: {})
+        open_timeout            =   options.fetch(:open_timeout, 60)
+        read_timeout            =   options.fetch(:read_timeout, 60)
         browser_id              =   options.fetch(:browser_id, nil)
-        headless                =   options.fetch(:headless, false)
         download_path           =   options.fetch(:download_path, self.configuration.download_path)
         private_browsing        =   options.fetch(:private_browsing, false)
         
         profile                 =   Selenium::WebDriver::Firefox::Profile.new
         
@@ -29,13 +30,14 @@
         
         id                      =   SecureRandom.hex[0..15]
         profile                 =   firefox_proxy_options(profile)
         options                 =   Selenium::WebDriver::Firefox::Options.new(profile: profile)
         options.args           <<   "--browser_id=#{id}" unless browser_id.to_s.empty?
-        options.args           <<   "--headless"         if headless
+        options.args           <<   "--headless"         if headless?
         
         Capybara.register_driver self.driver do |app|
-          Capybara::Selenium::Driver.new(app, browser: :firefox, options: options)
+          client                =   Selenium::WebDriver::Remote::Http::Default.new(open_timeout: open_timeout, read_timeout: read_timeout)
+          Capybara::Selenium::Driver.new(app, browser: :firefox, http_client: client, options: options)
         end
         
         return id
       end