wip test session to squash

lib/datadog/ci/context/global.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module Datadog
+  module CI
+    module Context
+      class Global
+        def initialize
+          @mutex = Mutex.new
+          @test_session = nil
+        end
+
+        def activate_test_session!(test_session)
+          @mutex.synchronize do
+            raise "Nested test sessions are not supported. Currently active test session: #{@test_session}" unless @test_session.nil?
+
+            @test_session = test_session
+          end
+        end
+
+        def deactivate_test_session!(test_session)
+          @mutex.synchronize do
+            return if @test_session.nil?
+
+            if @test_session == test_session
+              @test_session = nil
+            else
+              raise "Trying to deactivate test session #{test_session}, but currently active test session is #{@test_session}"
+            end
+          end
+        end
+      end
+    end
+  end
+end

lib/datadog/ci/ext/app_types.rb

@@ -5,6 +5,7 @@ module CI
     module Ext
       module AppTypes
         TYPE_TEST = "test"
+        TYPE_TEST_SESSION = "test_session_end"
       end
     end
   end

lib/datadog/ci/recorder.rb

@@ -12,6 +12,7 @@
 
 require_relative "span"
 require_relative "test"
+require_relative "test_session"
 
 module Datadog
   module CI
@@ -22,6 +23,24 @@ class Recorder
       def initialize
         @environment_tags = Ext::Environment.tags(ENV).freeze
         @local_context = Context::Local.new
+        @global_context = Context::Global.new
+      end
+
+      def start_test_session(service_name: nil, tags: {})
+        span_options = {
+          service: service_name,
+          span_type: Ext::AppTypes::TYPE_TEST_SESSION
+        }
+
+        tracer_span = Datadog::Tracing.trace("test.session", **span_options)
+        trace = Datadog::Tracing.active_trace
+
+        set_trace_origin(trace)
+
+        test_session = build_test_session(tracer_span, tags)
+        @global_context.activate_test_session!(test_session)
+
+        test_session
       end
 
       # Creates a new span for a CI test
@@ -93,6 +112,18 @@ def set_trace_origin(trace)
         trace.origin = Ext::Test::CONTEXT_ORIGIN if trace
       end
 
+      def build_test_session(tracer_span, tags)
+        test_session = TestSession.new(tracer_span)
+
+        test_session.set_default_tags
+        test_session.set_environment_runtime_tags
+
+        test_session.set_tags(tags)
+        test_session.set_tags(environment_tags)
+
+        test_session
+      end
+
       def build_test(tracer_span, tags)
         test = Test.new(tracer_span)
 

lib/datadog/ci/test.rb

@@ -5,7 +5,6 @@
 module Datadog
   module CI
     # Represents a single part of a test run.
-    # Could be a session, suite, test, or any custom span.
     #
     # @public_api
     class Test < Span

lib/datadog/ci/test_session.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+require_relative "span"
+
+module Datadog
+  module CI
+    # Represents the whole test command process.
+    # This object can be shared between multiple threads.
+    #
+    # @public_api
+    class TestSession < Span
+      def initialize(tracer_span)
+        super
+
+        @mutex = Mutex.new
+      end
+    end
+  end
+end

lib/datadog/ci/test_visibility/serializers/test_v1.rb

@@ -37,7 +37,7 @@ def content_map_size
           end
 
           def type
-            "test"
+            Ext::AppTypes::TYPE_TEST
           end
 
           def name

sig/datadog/ci/ext/app_types.rbs

@@ -2,7 +2,8 @@ module Datadog
   module CI
     module Ext
       module AppTypes
-        TYPE_TEST: String
+        TYPE_TEST: "test"
+        TYPE_TEST_SESSION: "test_session_end"
       end
     end
   end

sig/datadog/ci/test_session.rbs

@@ -0,0 +1,6 @@
+module Datadog
+  module CI
+    class TestSession < Span
+    end
+  end
+end

sig/datadog/ci/test_visibility/serializers/test_v1.rbs

@@ -11,7 +11,7 @@ module Datadog
           def content_fields: () -> Array[String | Hash[String, String]]
           def content_map_size: () -> Integer
 
-          def type: () -> "test"
+          def type: () -> ::String
 
           def name: () -> ::String