add total_tests_count field for TestSession and set it from RSpec ins…

…trumentation
DataDog · Sep 9, 2024 · cba919e · cba919e
1 parent a3d14f3
commit cba919e
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 0 deletions.
diff --git a/lib/datadog/ci/contrib/rspec/runner.rb b/lib/datadog/ci/contrib/rspec/runner.rb
@@ -25,6 +25,7 @@ def run_specs(*args)
                 },
                 service: datadog_configuration[:service_name]
               )
+              test_session&.total_tests_count = ::RSpec.world.example_count
 
               test_module = test_visibility_component.start_test_module(Ext::FRAMEWORK)
 

diff --git a/lib/datadog/ci/test_session.rb b/lib/datadog/ci/test_session.rb
@@ -12,6 +12,8 @@ module CI
     #
     # @public_api
     class TestSession < ConcurrentSpan
+      attr_accessor :total_tests_count
+
       # Finishes the current test session.
       # @return [void]
       def finish

diff --git a/sig/datadog/ci/test_session.rbs b/sig/datadog/ci/test_session.rbs
@@ -1,6 +1,7 @@
 module Datadog
   module CI
     class TestSession < ConcurrentSpan
+      attr_accessor total_tests_count: Integer
       @inheritable_tags: Hash[untyped, untyped]
 
       def inheritable_tags: () -> Hash[untyped, untyped]

diff --git a/vendor/rbs/rspec/0/rspec.rbs b/vendor/rbs/rspec/0/rspec.rbs
@@ -50,4 +50,5 @@ end
 
 class RSpec::Core::World
   def wants_to_quit: () -> bool
+  def example_count: () -> Integer
 end
-Original file line number
+Diff line change
@@ Expand Up / @@ -25,6 +25,7 @@ def run_specs(*args) @@
                     },
                     service: datadog_configuration[:service_name]
                   )
+                  test_session&.total_tests_count = ::RSpec.world.example_count
                   test_module = test_visibility_component.start_test_module(Ext::FRAMEWORK)
@@ Expand Down @@