driver.rb in datadog-ci-1.12.0

- old
+ new

@@ -1,11 +1,11 @@
 # frozen_string_literal: true
 
 require_relative "../patcher"
 
-require_relative "ext"
-require_relative "rum"
+require_relative "../../utils/rum"
+require_relative "../../ext/rum"
 require_relative "../../ext/test"
 
 module Datadog
   module CI
     module Contrib
@@ -20,13 +20,13 @@
             def quit
               return super unless datadog_configuration[:enabled]
 
               Datadog.logger.debug("[Selenium] Driver quit event")
 
-              RUM.stop_rum_session(@bridge)
+              Utils::RUM.stop_rum_session(@bridge, rum_flush_wait_millis: datadog_configuration[:rum_flush_wait_millis])
 
               Datadog.logger.debug("[Selenium] RUM session stopped, deleting cookie")
-              @bridge.manage.delete_cookie(Ext::COOKIE_TEST_EXECUTION_ID)
+              @bridge.manage.delete_cookie(CI::Ext::RUM::COOKIE_TEST_EXECUTION_ID)
             rescue ::Selenium::WebDriver::Error::WebDriverError => e
               Datadog.logger.debug("[Selenium] Error while quitting Selenium driver: #{e.message}")
             ensure
               super
             end