diff --git a/lib/datadog/ci/contrib/rspec/runner.rb b/lib/datadog/ci/contrib/rspec/runner.rb index d4600a1d..c9f04513 100644 --- 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 index 720b9cfc..713d195b 100644 --- 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 index 3d369772..01433d41 100644 --- 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 index 399049a6..82fe6d6d 100644 --- 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