Pass waits to bidi bridge

Author: aguspe

rb/lib/selenium/webdriver/common/driver.rb

@@ -320,8 +320,21 @@ def ref
       attr_reader :bridge
 
       def create_bridge(caps:, url:, http_client: nil)
-        klass = caps['webSocketUrl'] ? Remote::BiDiBridge : Remote::Bridge
-        klass.new(http_client: http_client, url: url).tap do |bridge|
+        if caps['webSocketUrl']
+          ws_options =
+            {
+              response_timeout: caps['wsResponseTimeout'],
+              response_interval: caps['wsResponseInterval']
+            }.compact
+
+          klass = Remote::BiDiBridge
+          bridge_opts = { http_client: http_client, url: url, ws_options: ws_options }
+        else
+          klass = Remote::Bridge
+          bridge_opts = { http_client: http_client, url: url }
+        end
+
+        klass.new(**bridge_opts).tap do |bridge|
           bridge.create_session(caps)
         end
       end

rb/lib/selenium/webdriver/remote/bidi_bridge.rb

@@ -23,10 +23,15 @@ module Remote
       class BiDiBridge < Bridge
         attr_reader :bidi
 
+        def initialize(url:, http_client: nil, ws_options: {})
+          super(url: url, http_client: http_client)
+          @ws_options = ws_options
+        end
+
         def create_session(capabilities)
           super
           socket_url = @capabilities[:web_socket_url]
-          @bidi = Selenium::WebDriver::BiDi.new(url: socket_url)
+          @bidi = Selenium::WebDriver::BiDi.new(url: socket_url, options: @ws_options)
         end
 
         def get(url)