[rb] allow setting safari technology preview after creating options instance

titusfortner · titusfortner · commit 04d571231191 · 2022-12-24T21:44:05.000-06:00
diff --git a/rb/lib/selenium/webdriver/safari/options.rb b/rb/lib/selenium/webdriver/safari/options.rb
@@ -35,6 +35,10 @@ def add_option(name, value = nil)
           super
         end
 
+        def as_json(*)
+          @options[:browser_name] = Safari.technology_preview? ? "Safari Technology Preview" : 'safari'
+          super
+        end
       end # Options
     end # Safari
   end # WebDriver
diff --git a/rb/spec/integration/selenium/webdriver/safari/driver_spec.rb b/rb/spec/integration/selenium/webdriver/safari/driver_spec.rb
@@ -27,6 +27,28 @@ module Safari
           driver.permissions = {'getUserMedia' => false}
           expect(driver.permissions).to eq('getUserMedia' => false)
         end
+
+        context 'setting technology preview' do
+          before(:all) { quit_driver }
+
+          after do
+            Service.driver_path = nil
+            Safari.use_technology_preview = nil
+          end
+
+          it 'sets before options', exclusive: {browser: :safari} do
+            Safari.technology_preview!
+            local_driver = WebDriver.for :safari
+            expect(local_driver.capabilities.browser_name).to eq 'Safari Technology Preview'
+          end
+
+          it 'sets after options' do
+            options = Options.safari
+            Safari.technology_preview!
+            local_driver = WebDriver.for :safari, options: options
+            expect(local_driver.capabilities.browser_name).to eq 'Safari Technology Preview'
+          end
+        end
       end
     end # Safari
   end # WebDriver
