DRY up code

site-prism · Jan 29, 2024 · 1d4b146 · 1d4b146
1 parent 128a773
commit 1d4b146
Showing 1 changed file with 8 additions and 9 deletions.
diff --git a/features/support/env.rb b/features/support/env.rb
@@ -19,16 +19,15 @@
 SimpleCov.start if defined?(SimpleCov) && RUBY_VERSION < '3.1'
 
 browser = ENV.fetch('BROWSER', 'chrome').to_sym
-options =
-  if browser == :chrome
-    AutomationHelpers::Drivers::V4::Options.for(:chrome).tap do |opts|
-      opts.add_argument('--no-sandbox')
-      opts.add_argument('--disable-dev-shm-usage')
-      opts.add_argument('--disable-gpu')
-    end
-  else
-    AutomationHelpers::Drivers::V4::Options.for(:firefox)
+options = AutomationHelpers::Drivers::V4::Options.for(browser)
+
+if browser == :chrome
+  options.tap do |opts|
+    opts.add_argument('--no-sandbox')
+    opts.add_argument('--disable-dev-shm-usage')
+    opts.add_argument('--disable-gpu')
   end
+end
 
 Capybara.register_driver :site_prism do |app|
   Capybara::Selenium::Driver.new(