(Hopefully) fixed Swagger file generation bug

dukeofharen · Jul 17, 2018 · ac6a917 · ac6a917
1 parent a4c9a35
commit ac6a917
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 16 deletions.
diff --git a/src/HttPlaceholder.SwaggerGenerator/Program.cs b/src/HttPlaceholder.SwaggerGenerator/Program.cs
@@ -20,7 +20,8 @@ static async Task Main(string[] args)
          var startup = new Startup();
          var testServer = new TestServer(
            new WebHostBuilder()
-            .UseStartup<Startup>());
+            .ConfigureServices(services => Startup.ConfigureServicesStatic(services))
+            .Configure(appBuilder => Startup.ConfigureStatic(appBuilder, null, false)));
          var client = testServer.CreateClient();
 
          // Retrieve the Swagger URL.

diff --git a/src/HttPlaceholder/Startup.cs b/src/HttPlaceholder/Startup.cs
@@ -15,6 +15,34 @@ namespace HttPlaceholder
    public class Startup
    {
       public void ConfigureServices(IServiceCollection services)
+      {
+         ConfigureServicesStatic(services);
+      }
+
+      public void Configure(IApplicationBuilder app, IHostingEnvironment env)
+      {
+         ConfigureStatic(app, env, true);
+      }
+
+      public static void ConfigureStatic(IApplicationBuilder app, IHostingEnvironment env, bool preloadStubs)
+      {
+         app.UseMiddleware<StubHandlingMiddleware>();
+         app.UseMvc();
+         app.UseSwagger();
+         app.UseSwaggerUI(c =>
+         {
+            c.SwaggerEndpoint("/swagger/v1/swagger.json", "HttPlaceholder API V1");
+         });
+
+         if (preloadStubs)
+         {
+            // Check if the stubs can be loaded.
+            var stubContainer = app.ApplicationServices.GetService<IStubContainer>();
+            Task.Run(() => stubContainer.GetStubsAsync()).Wait();
+         }
+      }
+
+      public static void ConfigureServicesStatic(IServiceCollection services)
       {
          services.AddMvc(options =>
          {
@@ -34,20 +62,5 @@ public void ConfigureServices(IServiceCollection services)
          DependencyRegistration.RegisterDependencies(services);
          Services.DependencyRegistration.RegisterDependencies(services);
       }
-
-      public void Configure(IApplicationBuilder app, IHostingEnvironment env)
-      {
-         app.UseMiddleware<StubHandlingMiddleware>();
-         app.UseMvc();
-         app.UseSwagger();
-         app.UseSwaggerUI(c =>
-         {
-            c.SwaggerEndpoint("/swagger/v1/swagger.json", "HttPlaceholder API V1");
-         });
-
-         // Check if the stubs can be loaded.
-         var stubContainer = app.ApplicationServices.GetService<IStubContainer>();
-         Task.Run(() => stubContainer.GetStubsAsync()).Wait();
-      }
    }
 }