capybara_screenshot.rb in rspeckled-2.1.0

- old
+ new

@@ -9,9 +9,10 @@
 
 ::Capybara::Screenshot.register_filename_prefix_formatter(:rspec) do |example|
   example.full_description.downcase.tr(' ', '-').gsub(%r{^.*/spec/}, '')
 end
 
-::Capybara::Screenshot.register_driver(:chrome)           { |driver, path| driver.browser.save_screenshot(path) }
-::Capybara::Screenshot.register_driver(:headless_chrome)  { |driver, path| driver.browser.save_screenshot(path) }
-::Capybara::Screenshot.register_driver(:firefox)          { |driver, path| driver.browser.save_screenshot(path) }
-::Capybara::Screenshot.register_driver(:headless_firefox) { |driver, path| driver.browser.save_screenshot(path) }
+%i{chrome headless_chrome firefox headless_firefox}.each do |browser|
+  ::Capybara::Screenshot.register_driver(browser) do |driver, path|
+    driver.browser.save_screenshot(path)
+  end
+end