[rb] create user-friendly method for enabling bidi

rb/lib/selenium/webdriver/common/options.rb

@@ -71,6 +71,8 @@ def set_capabilities
       def initialize(**opts)
         self.class.set_capabilities
 
+        opts[:web_socket_url] = opts.delete(:bidi) if opts.key?(:bidi)
+
         @options = opts
         @options[:browser_name] = self.class::BROWSER
       end
@@ -91,6 +93,14 @@ def add_option(name, value = nil)
         @options[name] = value
       end
 
+      def enable_bidi!
+        @options[:web_socket_url] = true
+      end
+
+      def bidi?
+        !!@options[:web_socket_url]
+      end
+
       def ==(other)
         return false unless other.is_a? self.class
 

rb/sig/lib/selenium/webdriver/common/options.rbs

@@ -39,6 +39,10 @@ module Selenium
 
       def ==: (untyped other) -> bool
 
+      def bidi?: -> bool
+
+      def enable_bidi!: -> bool
+
       alias eql? ==
 
       def as_json: (*untyped) -> untyped

rb/spec/integration/selenium/webdriver/chrome/options_spec.rb

@@ -43,6 +43,38 @@ module Chrome
             expect(ua).to eq('foo;bar')
           end
         end
+
+        it 'enables bidi' do
+          quit_driver
+
+          options = Selenium::WebDriver::Options.chrome
+          expect(options.web_socket_url).to be_nil
+          expect(options.bidi?).to be false
+
+          options.enable_bidi!
+          expect(options.web_socket_url).to be true
+          expect(options.bidi?).to be true
+
+          driver = Selenium::WebDriver.for :chrome, options: options
+
+          expect(driver.capabilities.web_socket_url).to be_a String
+
+          driver.quit
+        end
+
+        it 'enables BiDi on initialization' do
+          quit_driver
+
+          options = Selenium::WebDriver::Options.chrome(bidi: true)
+          expect(options.web_socket_url).to be true
+          expect(options.bidi?).to be true
+
+          driver = Selenium::WebDriver.for :chrome, options: options
+
+          expect(driver.capabilities.web_socket_url).to be_a String
+
+          driver.quit
+        end
       end
     end # Chrome
   end # WebDriver

rb/spec/integration/selenium/webdriver/edge/options_spec.rb

@@ -43,6 +43,38 @@ module Edge
             expect(ua).to eq('foo;bar')
           end
         end
+
+        it 'enables bidi' do
+          quit_driver
+
+          options = Selenium::WebDriver::Options.chrome
+          expect(options.web_socket_url).to be_nil
+          expect(options.bidi?).to be false
+
+          options.enable_bidi!
+          expect(options.web_socket_url).to be true
+          expect(options.bidi?).to be true
+
+          driver = Selenium::WebDriver.for :chrome, options: options
+
+          expect(driver.capabilities.web_socket_url).to be_a String
+
+          driver.quit
+        end
+
+        it 'enables BiDi on initialization' do
+          quit_driver
+
+          options = Selenium::WebDriver::Options.edge(bidi: true)
+          expect(options.web_socket_url).to be true
+          expect(options.bidi?).to be true
+
+          driver = Selenium::WebDriver.for :edge, options: options
+
+          expect(driver.capabilities.web_socket_url).to be_a String
+
+          driver.quit
+        end
       end
     end # Edge
   end # WebDriver

rb/spec/unit/selenium/webdriver/chrome/options_spec.rb

@@ -141,6 +141,18 @@ module Chrome
           end
         end
 
+        describe '#enable_bidi!' do
+          it 'allows setting and querying bidi' do
+            expect(options.web_socket_url).to be_nil
+            expect(options.bidi?).to be false
+
+            options.enable_bidi!
+
+            expect(options.bidi?).to be true
+            expect(options.web_socket_url).to be true
+          end
+        end
+
         describe '#add_extension' do
           it 'adds an extension' do
             allow(File).to receive(:file?).and_return(true)