[rb] fix reset_driver! implementation

SeleniumHQ · Nov 29, 2022 · 769cd05 · 769cd05
1 parent 37498f8
commit 769cd05
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/rb/spec/integration/selenium/webdriver/spec_support/helpers.rb b/rb/spec/integration/selenium/webdriver/spec_support/helpers.rb
@@ -25,8 +25,8 @@ def driver
           GlobalTestEnv.driver_instance
         end
 
-        def reset_driver!(**opts)
-          GlobalTestEnv.reset_driver!(**opts)
+        def reset_driver!(**opts, &block)
+          GlobalTestEnv.reset_driver!(**opts, &block)
         end
 
         def quit_driver

diff --git a/rb/spec/integration/selenium/webdriver/spec_support/test_environment.rb b/rb/spec/integration/selenium/webdriver/spec_support/test_environment.rb
@@ -51,14 +51,14 @@ def browser
           driver == :remote ? ENV.fetch('WD_REMOTE_BROWSER', 'chrome').to_sym : driver
         end
 
-        def driver_instance(**opts)
-          @driver_instance || create_driver!(**opts)
+        def driver_instance(**opts, &block)
+          @driver_instance || create_driver!(**opts, &block)
         end
 
-        def reset_driver!(time: 0, **opts)
+        def reset_driver!(time: 0, **opts, &block)
           quit_driver
           sleep time
-          driver_instance(**opts)
+          driver_instance(**opts, &block)
         end
 
         def quit_driver