Extracted common attribute route logic in ApplicationBuilderMock (#326)

Author: ivaylokenov

src/MyTested.AspNetCore.Mvc.Abstractions/Internal/Application/ApplicationBuilderMock.cs

@@ -222,12 +222,6 @@ private void ExtractEndpointRoutes(Func<RequestDelegate, RequestDelegate> middle
                 }
             }
 
-            var attributeRoutingType = WebFramework.Internals.AttributeRouting;
-            var createAttributeMegaRouteMethod = attributeRoutingType.GetMethod("CreateAttributeMegaRoute");
-            var router = (IRouter)createAttributeMegaRouteMethod.Invoke(null, new[] { this.ApplicationServices });
-
-            routeBuilder.Routes.Insert(0, router);
-
             var routes = routeBuilder.Routes;
 
             for (int i = 0; i < routes.Count; i++)

src/MyTested.AspNetCore.Mvc.Abstractions/Internal/Application/TestApplicationRouter.cs

@@ -43,17 +43,23 @@ private static void PrepareApplicationAndRouting()
 
             AdditionalRouting?.Invoke(routeBuilder);
 
-            if (StartupType == null || routeBuilder.Routes.Count == 0)
+            var routeBuilderRoutes = routeBuilder.Routes;
+
+            if (StartupType == null || routeBuilderRoutes.Count == 0)
             {
                 routeBuilder.MapRoute(
                     name: "default",
                     template: "{controller=Home}/{action=Index}/{id?}");
+            }
 
-                var attributeRoutingType = WebFramework.Internals.AttributeRouting;
+            var attributeRoutingType = WebFramework.Internals.AttributeRouting;
+
+            if (routeBuilderRoutes[0].GetType() != attributeRoutingType)
+            {
                 var createAttributeMegaRouteMethod = attributeRoutingType.GetMethod("CreateAttributeMegaRoute");
                 var router = (IRouter)createAttributeMegaRouteMethod.Invoke(null, new[] { serviceProvider });
 
-                routeBuilder.Routes.Insert(0, router);
+                routeBuilderRoutes.Insert(0, router);
             }
 
             router = routeBuilder.Build();