wip broken

lib/datadog/ci/context/global.rb

@@ -3,12 +3,17 @@
 module Datadog
   module CI
     module Context
+      # This context is shared between threads and represents the current test session.
       class Global
         def initialize
           @mutex = Mutex.new
           @test_session = nil
         end
 
+        def active_test_session
+          @test_session
+        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?
@@ -17,16 +22,8 @@ def activate_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
+        def deactivate_test_session!
+          @mutex.synchronize { @test_session = nil }
         end
       end
     end

lib/datadog/ci/recorder.rb

@@ -8,6 +8,7 @@
 require_relative "ext/test"
 require_relative "ext/environment"
 
+require_relative "context/global"
 require_relative "context/local"
 
 require_relative "span"
@@ -96,6 +97,10 @@ def active_test
         @local_context.active_test
       end
 
+      def active_test_session
+        @global_context.active_test_session
+      end
+
       def deactivate_test(test)
         @local_context.deactivate_test!(test)
       end

sig/datadog/ci/context/global.rbs

@@ -0,0 +1,19 @@
+module Datadog
+  module CI
+    module Context
+      class Global
+        @mutex: Thread::Mutex
+
+        @test_session: Datadog::CI::TestSession?
+
+        def initialize: () -> void
+
+        def active_test_session: () -> Datadog::CI::TestSession?
+
+        def activate_test_session!: (Datadog::CI::TestSession test_session) -> void
+
+        def deactivate_test_session!: () -> void
+      end
+    end
+  end
+end

sig/datadog/ci/recorder.rbs

@@ -3,13 +3,16 @@ module Datadog
     class Recorder
       @environment_tags: Hash[String, String]
       @local_context: Datadog::CI::Context::Local
+      @global_context: Datadog::CI::Context::Global
 
       attr_reader environment_tags: Hash[String, String]
 
       def trace_test: (String span_name, ?service_name: String?, ?operation_name: String, ?tags: Hash[untyped, untyped]) ?{ (Datadog::CI::Test span) -> untyped } -> untyped
 
       def trace: (String span_type, String span_name, ?tags: Hash[untyped, untyped]) ?{ (Datadog::CI::Span span) -> untyped } -> untyped
 
+      def start_test_session: (?service_name: String?, ?tags: Hash[untyped, untyped]) -> Datadog::CI::TestSession
+
       def active_test: () -> Datadog::CI::Test?
 
       def active_span: () -> Datadog::CI::Span?
@@ -24,6 +27,8 @@ module Datadog
 
       def build_test: (Datadog::Tracing::SpanOperation tracer_span, Hash[untyped, untyped] tags) -> Datadog::CI::Test
 
+      def build_test_session: (Datadog::Tracing::SpanOperation tracer_span, Hash[untyped, untyped] tags) -> Datadog::CI::TestSession
+
       def build_span: (Datadog::Tracing::SpanOperation tracer_span, Hash[untyped, untyped] tags) -> Datadog::CI::Span
     end
   end

sig/datadog/ci/test_session.rbs

@@ -1,6 +1,7 @@
 module Datadog
   module CI
     class TestSession < Span
+      @mutex: Thread::Mutex
     end
   end
 end

spec/datadog/ci/context/global_spec.rb

@@ -0,0 +1,51 @@
+RSpec.describe Datadog::CI::Context::Global do
+  subject { described_class.new }
+
+  let(:tracer_span) { double(Datadog::Tracing::SpanOperation, name: "test.session") }
+  let(:session) { Datadog::CI::TestSession.new(tracer_span) }
+
+  describe "#activate_test_session!" do
+    context "when a test session is already active" do
+      before do
+        subject.activate_test_session!(Datadog::CI::TestSession.new(tracer_span))
+      end
+
+      it "raises an error" do
+        expect { subject.activate_test_session!(session) }.to(
+          raise_error(
+            RuntimeError,
+            "Nested test sessions are not supported. Currently active test session: " \
+            "#{session}"
+          )
+        )
+      end
+    end
+
+    context "when no test session is active" do
+      it "activates the test session" do
+        subject.activate_test_session!(session)
+        expect(subject.active_test_session).to be(session)
+      end
+    end
+  end
+
+  describe "#deactivate_test_session!" do
+    context "when a test session is active" do
+      before do
+        subject.activate_test_session!(session)
+      end
+
+      it "deactivates the test session" do
+        subject.deactivate_test_session!
+        expect(subject.active_test_session).to be_nil
+      end
+    end
+
+    context "when no test session is active" do
+      it "does nothing" do
+        subject.deactivate_test_session!
+        expect(subject.active_test_session).to be_nil
+      end
+    end
+  end
+end

spec/datadog/ci/recorder_spec.rb

@@ -251,4 +251,30 @@ def expect_initialized_span
       it { is_expected.to be_nil }
     end
   end
+
+  describe "#start_test_session" do
+    subject(:start_test_session) { recorder.start_test_session(service_name: service) }
+
+    let(:session_operation_name) { "test.session" }
+
+    let(:span_op) { Datadog::Tracing::SpanOperation.new(session_operation_name) }
+    let(:ci_test_session) { instance_double(Datadog::CI::TestSession) }
+
+    before do
+      allow(Datadog::Tracing)
+        .to receive(:trace)
+        .with(
+          session_operation_name,
+          {
+            span_type: Datadog::CI::Ext::AppTypes::TYPE_TEST_SESSION,
+            service: service
+          }
+        )
+        .and_return(span_op)
+
+      allow(Datadog::CI::TestSession).to receive(:new).with(span_op).and_return(ci_test_session)
+    end
+
+    it { is_expected.to be(ci_test_session) }
+  end
 end