Umbraco-CMS/tests/Umbraco.Tests.UnitTests/Umbraco.Web.Common/Routing/TestRouteBuilder.cs

// Copyright (c) Umbraco.
// See LICENSE for more details.

using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Threading.Tasks;
using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Routing;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.Hosting.Internal;
using Microsoft.Extensions.Logging;
using Moq;

namespace Umbraco.Cms.Tests.UnitTests.Umbraco.Web.Common.Routing
{
    public class TestRouteBuilder : IEndpointRouteBuilder
    {
        private readonly ServiceProvider _serviceProvider;

        public TestRouteBuilder()
        {
            var services = new ServiceCollection();

            var diagnosticListener = new DiagnosticListener("UnitTests");
            services.AddSingleton<DiagnosticSource>(diagnosticListener);
            services.AddSingleton<DiagnosticListener>(diagnosticListener);

            services.AddLogging();
            services.AddMvc();
            services.AddSingleton<IHostApplicationLifetime>(x => new ApplicationLifetime(x.GetRequiredService<ILogger<ApplicationLifetime>>()));
            services.AddSignalR();
            _serviceProvider = services.BuildServiceProvider();
        }

        public ICollection<EndpointDataSource> DataSources { get; } = new List<EndpointDataSource>();

        public IServiceProvider ServiceProvider => _serviceProvider;

        public IApplicationBuilder CreateApplicationBuilder()
        {
            var mock = new Mock<IApplicationBuilder>();
            mock.Setup(x => x.Build()).Returns(httpContext => Task.CompletedTask);
            return mock.Object;
        }
    }
}