Add additional tests for CI Visibility test behaviour

DataDog · Jan 6, 2025 · 2635b99 · 2635b99
1 parent 9194a87
commit 2635b99
Show file tree

Hide file tree

Showing 2 changed files with 104 additions and 0 deletions.
diff --git a/tracer/test/Datadog.Trace.Tests/Configuration/CIVisibilitySettingsTests.cs b/tracer/test/Datadog.Trace.Tests/Configuration/CIVisibilitySettingsTests.cs
@@ -14,6 +14,8 @@ namespace Datadog.Trace.Tests.Configuration
 {
     public class CIVisibilitySettingsTests : SettingsTestsBase
     {
+        private static readonly string ExpectedExcludedSession = "/session/FakeSessionIdForPollingPurposes".ToUpperInvariant();
+
         [Theory]
         [MemberData(nameof(BooleanTestCases), null)]
         public void Enabled(string value, bool? expected)
@@ -185,5 +187,106 @@ public void ForceAgentsEvpProxy(string value, string expected)
 
             settings.ForceAgentsEvpProxy.Should().Be(expected);
         }
+
+        [Theory]
+        [InlineData("some-service", "true")]
+        [InlineData(null, "false")]
+        public void AddsUserProvidedTestServiceTagToGlobalTags(string serviceName, string expectedTag)
+        {
+            var source = CreateConfigurationSource((ConfigurationKeys.ServiceName, serviceName));
+
+            var ciVisSettings = new CIVisibilitySettings(source, NullConfigurationTelemetry.Instance);
+            var tracerSettings = ciVisSettings.InitializeTracerSettings([source]);
+
+            tracerSettings.GlobalTags.Should()
+                          .ContainKey(Datadog.Trace.Ci.Tags.CommonTags.UserProvidedTestServiceTag)
+                          .WhoseValue.Should()
+                          .Be(expectedTag);
+        }
+
+        [Fact]
+        public void ServiceNameIsNormalized()
+        {
+            var originalName = "My Service Name!";
+            var normalizedName = "my_service_name";
+
+            var source = CreateConfigurationSource((ConfigurationKeys.ServiceName, originalName));
+
+            var ciVisSettings = new CIVisibilitySettings(source, NullConfigurationTelemetry.Instance);
+            var tracerSettings = ciVisSettings.InitializeTracerSettings([source]);
+
+            tracerSettings.ServiceName.Should().Be(normalizedName);
+        }
+
+        [Fact]
+        public void AddsFakeSessionToExcludedHttpClientUrls()
+        {
+            var source = CreateConfigurationSource();
+
+            var ciVisSettings = new CIVisibilitySettings(source, NullConfigurationTelemetry.Instance);
+            var tracerSettings = ciVisSettings.InitializeTracerSettings([source]);
+
+            tracerSettings.HttpClientExcludedUrlSubstrings
+                          .Should()
+                          .Contain(ExpectedExcludedSession);
+        }
+
+        [Fact]
+        public void AddsFakeSessionToExcludedHttpClientUrls_WhenUrlsAlreadyExist()
+        {
+            var source = CreateConfigurationSource(
+                (ConfigurationKeys.HttpClientExcludedUrlSubstrings, "/some-url/path"));
+
+            var ciVisSettings = new CIVisibilitySettings(source, NullConfigurationTelemetry.Instance);
+            var tracerSettings = ciVisSettings.InitializeTracerSettings([source]);
+
+            tracerSettings.HttpClientExcludedUrlSubstrings
+                          .Should()
+                          .Contain(ExpectedExcludedSession);
+        }
+
+        [Fact]
+        public void AddsFakeSessionToExcludedHttpClientUrls_WhenRunningInAas()
+        {
+            var source = CreateConfigurationSource(
+                (ConfigurationKeys.AzureAppService.AzureAppServicesContextKey, "true"),
+                (ConfigurationKeys.HttpClientExcludedUrlSubstrings, "/some-url/path"));
+
+            var ciVisSettings = new CIVisibilitySettings(source, NullConfigurationTelemetry.Instance);
+            var tracerSettings = ciVisSettings.InitializeTracerSettings([source]);
+
+            tracerSettings.HttpClientExcludedUrlSubstrings
+                          .Should()
+                          .Contain(ExpectedExcludedSession);
+        }
+
+        [Fact]
+        public void WhenLogsEnabled_AddsDirectSubmission()
+        {
+            var source = CreateConfigurationSource(
+                (ConfigurationKeys.CIVisibility.Logs, "true"));
+
+            var ciVisSettings = new CIVisibilitySettings(source, NullConfigurationTelemetry.Instance);
+            var tracerSettings = ciVisSettings.InitializeTracerSettings([source]);
+
+            tracerSettings.LogSubmissionSettings
+                          .EnabledIntegrationNames
+                          .Should()
+                          .Contain(nameof(IntegrationId.XUnit));
+        }
+
+        [Fact]
+        public void WhenLogsNotEnabled_DoesNotAddDirectSubmission()
+        {
+            var source = CreateConfigurationSource();
+
+            var ciVisSettings = new CIVisibilitySettings(source, NullConfigurationTelemetry.Instance);
+            var tracerSettings = ciVisSettings.InitializeTracerSettings([source]);
+
+            tracerSettings.LogSubmissionSettings
+                          .EnabledIntegrationNames
+                          .Should()
+                          .NotContain(nameof(IntegrationId.XUnit));
+        }
     }
 }
diff --git a/tracer/test/Datadog.Trace.Tests/Configuration/TracerSettingsTests.cs b/tracer/test/Datadog.Trace.Tests/Configuration/TracerSettingsTests.cs
@@ -217,6 +217,7 @@ public void Environment(string value, string expected)
 
         [Theory]
         [InlineData("test", null, null, "test")]
+        [InlineData("My Service Name!", null, null, "My Service Name!")]
         [InlineData("test", null, "ignored_otel", "test")]
         [InlineData("test", "error", "ignored_otel", "test")]
         [InlineData(null, "test", null, "test")]