Fixed unit test failure from shared reference. (#308)

* Fixed unit test failure from shared reference.
microsoft · Jan 5, 2017 · 0d8ab7f · 0d8ab7f
1 parent 20d3166
commit 0d8ab7f
Showing 1 changed file with 13 additions and 6 deletions.
diff --git a/...oft.ApplicationInsights.AspNetCore.Tests/Extensions/ApplicationInsightsExtensionsTests.cs b/...oft.ApplicationInsights.AspNetCore.Tests/Extensions/ApplicationInsightsExtensionsTests.cs
@@ -237,14 +237,21 @@ public static void RegistersTelemetryClientToGetTelemetryConfigurationFromContai
                 IServiceProvider serviceProvider = services.BuildServiceProvider();
                 var configuration = serviceProvider.GetTelemetryConfiguration();
                 configuration.InstrumentationKey = Guid.NewGuid().ToString();
-                configuration.TelemetryChannel = telemetryChannel;
+                ITelemetryChannel oldChannel = configuration.TelemetryChannel;
+                try
+                {
+                    configuration.TelemetryChannel = telemetryChannel;
 
-                var telemetryClient = serviceProvider.GetRequiredService<TelemetryClient>();
-                telemetryClient.TrackEvent("myevent");
+                    var telemetryClient = serviceProvider.GetRequiredService<TelemetryClient>();
+                    telemetryClient.TrackEvent("myevent");
 
-                // We want to check that configuration from contaier was used but configuration is a private field so we check
-                // instrumentation key instead
-                Assert.Equal(configuration.InstrumentationKey, sentTelemetry.Context.InstrumentationKey);
+                    // We want to check that configuration from container was used but configuration is a private field so we check instrumentation key instead.
+                    Assert.Equal(configuration.InstrumentationKey, sentTelemetry.Context.InstrumentationKey);
+                }
+                finally
+                {
+                    configuration.TelemetryChannel = oldChannel;
+                }
             }
 
             [Fact]