specs for ITR::Runner#start_coverage

spec/datadog/ci/itr/runner_spec.rb

@@ -4,13 +4,12 @@
 
 RSpec.describe Datadog::CI::ITR::Runner do
   let(:itr_enabled) { true }
+  let(:tracer_span) { Datadog::Tracing::SpanOperation.new("session") }
+  let(:test_session) { Datadog::CI::TestSession.new(tracer_span) }
 
   subject(:runner) { described_class.new(enabled: itr_enabled) }
 
   describe "#configure" do
-    let(:tracer_span) { Datadog::Tracing::SpanOperation.new("session") }
-    let(:test_session) { Datadog::CI::TestSession.new(tracer_span) }
-
     before do
       runner.configure(remote_configuration, test_session)
     end
@@ -63,4 +62,64 @@
       end
     end
   end
+
+  describe "#start_coverage" do
+    before do
+      runner.configure(remote_configuration, test_session)
+    end
+    context "when code coverage is disabled" do
+      let(:remote_configuration) { {"itr_enabled" => true, "code_coverage" => false, "tests_skipping" => false} }
+
+      it "does not start coverage" do
+        expect(runner).not_to receive(:coverage_collector)
+
+        runner.start_coverage
+        expect(runner.stop_coverage).to be_nil
+      end
+    end
+
+    context "when ITR is disabled" do
+      let(:remote_configuration) { {"itr_enabled" => false, "code_coverage" => false, "tests_skipping" => false} }
+
+      it "does not start coverage" do
+        expect(runner).not_to receive(:coverage_collector)
+
+        runner.start_coverage
+        expect(runner.stop_coverage).to be_nil
+      end
+    end
+
+    context "when code coverage is enabled" do
+      let(:remote_configuration) { {"itr_enabled" => true, "code_coverage" => true, "tests_skipping" => false} }
+
+      before do
+        skip("Code coverage is not supported in JRuby") if PlatformHelpers.jruby?
+      end
+
+      it "starts coverage" do
+        expect(runner).to receive(:coverage_collector).twice.and_call_original
+
+        runner.start_coverage
+        expect(1 + 1).to eq(2)
+        coverage = runner.stop_coverage
+        expect(coverage.size).to be > 0
+      end
+    end
+
+    context "when JRuby and code coverage is enabled" do
+      let(:remote_configuration) { {"itr_enabled" => true, "code_coverage" => true, "tests_skipping" => false} }
+
+      before do
+        skip("Skipped for CRuby") unless PlatformHelpers.jruby?
+      end
+
+      it "disables code coverage" do
+        expect(runner).not_to receive(:coverage_collector)
+        expect(runner.code_coverage?).to be(false)
+
+        runner.start_coverage
+        expect(runner.stop_coverage).to be_nil
+      end
+    end
+  end
 end