Add a test

microsoft · May 3, 2017 · 4dc70d9 · 4dc70d9
1 parent 3d3c311
commit 4dc70d9
Showing 1 changed file with 50 additions and 0 deletions.
diff --git a/...oft.ApplicationInsights.AspNetCore.Tests/Extensions/ApplicationInsightsExtensionsTests.cs b/...oft.ApplicationInsights.AspNetCore.Tests/Extensions/ApplicationInsightsExtensionsTests.cs
@@ -8,7 +8,9 @@ namespace Microsoft.Extensions.DependencyInjection.Test
     using System.Diagnostics;
     using System.Linq;
     using System.Reflection;
+    using Logging;
     using Microsoft.ApplicationInsights;
+    using Microsoft.ApplicationInsights.AspNetCore.Logging;
     using Microsoft.ApplicationInsights.AspNetCore.TelemetryInitializers;
     using Microsoft.ApplicationInsights.AspNetCore.Tests;
     using Microsoft.ApplicationInsights.Channel;
@@ -446,6 +448,38 @@ private static int GetTelemetryProcessorsCountInConfiguration<T>(TelemetryConfig
             {
                 return telemetryConfiguration.TelemetryProcessors.Where(processor => processor.GetType() == typeof(T)).Count();
             }
+
+            [Fact]
+            public static void LoggerCallbackIsInvoked()
+            {
+                var services = new ServiceCollection();
+                services.AddSingleton<ApplicationInsightsLoggerEvents>();
+                var serviceProvider = services.BuildServiceProvider();
+
+                var loggerProvider = new MockLoggingFactory();
+
+                bool firstLoggerCallback = false;
+                bool secondLoggerCallback = false;
+
+                loggerProvider.AddApplicationInsights(serviceProvider, (s, level) => true, () => firstLoggerCallback = true);
+                loggerProvider.AddApplicationInsights(serviceProvider, (s, level) => true, () => secondLoggerCallback = true);
+
+                Assert.True(firstLoggerCallback);
+                Assert.False(secondLoggerCallback);
+            }
+
+            [Fact]
+            public static void NullLoggerCallbackAlowed()
+            {
+                var services = new ServiceCollection();
+                services.AddSingleton<ApplicationInsightsLoggerEvents>();
+                var serviceProvider = services.BuildServiceProvider();
+
+                var loggerProvider = new MockLoggingFactory();
+
+                loggerProvider.AddApplicationInsights(serviceProvider, (s, level) => true, null);
+                loggerProvider.AddApplicationInsights(serviceProvider, (s, level) => true, null);
+            }
         }
 
         public static class AddApplicationInsightsSettings
@@ -521,5 +555,21 @@ public static ServiceCollection CreateServicesAndAddApplicationinsightsTelemetry
             }
             return services;
         }
+
+        private class MockLoggingFactory : ILoggerFactory
+        {
+            public void Dispose()
+            {
+            }
+
+            public ILogger CreateLogger(string categoryName)
+            {
+                return null;
+            }
+
+            public void AddProvider(ILoggerProvider provider)
+            {
+            }
+        }
     }
 }