Skip to content

Commit a517a20

Browse files
committed
add a unit test
1 parent 3aa629e commit a517a20

File tree

1 file changed

+23
-0
lines changed

1 file changed

+23
-0
lines changed

tracer/test/Datadog.Trace.Tests/TracerManagerFactoryTests.cs

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -103,6 +103,29 @@ public void RemoteConfigIsDisabledInAzureAppServices()
103103
_manager.TracerFlareManager.Should().BeOfType<NullTracerFlareManager>();
104104
}
105105

106+
[Theory]
107+
[InlineData(true)]
108+
[InlineData(false)]
109+
public void DiscoveryServiceCanBeDisabled(bool enabled)
110+
{
111+
var source = CreateConfigurationSource((ConfigurationKeys.AgentFeaturePollingEnabled, enabled.ToString()));
112+
var settings = new TracerSettings(source);
113+
114+
settings.AgentFeaturePollingEnabled.Should().Be(enabled);
115+
116+
var factory = new TracerManagerFactory();
117+
var discoveryService = factory.GetDiscoveryService(settings);
118+
119+
if (enabled)
120+
{
121+
discoveryService.Should().BeOfType<DiscoveryService>();
122+
}
123+
else
124+
{
125+
discoveryService.Should().BeSameAs(NullDiscoveryService.Instance);
126+
}
127+
}
128+
106129
private static TracerManager CreateTracerManager(TracerSettings settings)
107130
{
108131
return new TracerManagerFactory().CreateTracerManager(

0 commit comments

Comments
 (0)