[rb] convert RSpec after hook to test listener

Turns out that when using :aggregate_failures, the exception isn't populated yet in the after hook

Author: titusfortner

rb/spec/integration/selenium/webdriver/spec_helper.rb

@@ -29,11 +29,33 @@
 
 GlobalTestEnv = WebDriver::SpecSupport::TestEnvironment.new
 
+class SeleniumTestListener
+  # formatter complains if there is a return in this method
+  # rubocop:disable Style/GuardClause
+  def example_finished(notification)
+    if notification.exception
+      if WebDriver.logger.debug?
+        driver = GlobalTestEnv.driver_instance
+        session_id = driver.instance_variable_get(:@bridge).session_id
+
+        FileUtils.mkdir_p('bazel-logs')
+        driver.save_screenshot("bazel-logs/#{session_id}.png")
+        WebDriver.logger.info("Page Source: #{driver.page_source}")
+      end
+
+      GlobalTestEnv.reset_driver! unless notification.exception.is_a?(RSpec::Expectations::ExpectationNotMetError)
+    end
+  end
+  # rubocop:enable Style/GuardClause
+end
+
 RSpec.configure do |c|
   c.define_derived_metadata do |meta|
     meta[:aggregate_failures] = true
   end
 
+  c.reporter.register_listener(SeleniumTestListener.new, :example_finished)
+
   c.include(WebDriver::SpecSupport::Helpers)
 
   c.before(:suite) do
@@ -62,15 +84,6 @@
     results = guards.disposition
     send(*results) if results
   end
-
-  c.after do |example|
-    if example.exception && WebDriver.logger.debug?
-      driver.save_screenshot("bazel-logs/#{driver.session_id}.png")
-      WebDriver.logger.info("Page Source: #{driver.page_source}")
-    end
-
-    reset_driver! if example.exception || example.execution_result.pending_exception
-  end
 end
 
 WebDriver::Platform.exit_hook { GlobalTestEnv.quit }