Added some unit tests

aspnet · Aug 20, 2017 · e519f74 · e519f74
1 parent 4f2e3b8
commit e519f74
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/test/Microsoft.AspNetCore.Hosting.Tests/WebHostBuilderTests.cs b/test/Microsoft.AspNetCore.Hosting.Tests/WebHostBuilderTests.cs
@@ -609,6 +609,33 @@ public void DefaultApplicationNameWithUseStartupOfT()
             }
         }
 
+        [Fact]
+        public void CustomApplicationName()
+        {
+            using (var host = new WebHostBuilder()
+                .UseServer(new TestServer())
+                .UseApplicationName("Test Application")
+                .UseStartup<StartupNoServices>()
+                .Build())
+            {
+                var hostingEnv = host.Services.GetService<IHostingEnvironment>();
+                Assert.Equal("Test Application", hostingEnv.ApplicationName);
+            }
+        }
+
+        [Fact]
+        public void DefaultApplicationNameWithInjectedIStartup()
+        {
+            using (var host = new WebHostBuilder()
+                .UseServer(new TestServer())
+                .ConfigureServices(s => s.AddSingleton<IStartup, StartupNoServices>())
+                .Build())
+            {
+                var hostingEnv = host.Services.GetService<IHostingEnvironment>();
+                Assert.Equal(Assembly.GetEntryAssembly().GetName().Name, hostingEnv.ApplicationName);
+            }
+        }
+
         [Fact]
         public void DefaultApplicationNameWithUseStartupOfType()
         {