aspnetcore/test/Microsoft.AspNetCore.Routing.Tests/LinkGeneratorTestBase.cs

// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.

using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Http.Features;
using Microsoft.AspNetCore.Routing.Internal;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging.Abstractions;
using Microsoft.Extensions.ObjectPool;
using Microsoft.Extensions.Options;
using System;
using System.Collections.Generic;

namespace Microsoft.AspNetCore.Routing
{
    public abstract class LinkGeneratorTestBase
    {
        protected HttpContext CreateHttpContext(object ambientValues = null)
        {
            var httpContext = new DefaultHttpContext();

            var context = new EndpointSelectorContext
            {
                RouteValues = new RouteValueDictionary(ambientValues)
            };

            httpContext.Features.Set<IEndpointFeature>(context);
            httpContext.Features.Set<IRouteValuesFeature>(context);
            return httpContext;
        }

        protected ServiceCollection GetBasicServices()
        {
            var services = new ServiceCollection();
            services.AddSingleton<ObjectPoolProvider, DefaultObjectPoolProvider>();
            services.AddOptions();
            services.AddRouting();
            services.AddLogging();
            return services;
        }

        protected virtual void AddAdditionalServices(IServiceCollection services)
        {
        }

        private protected DefaultLinkGenerator CreateLinkGenerator(params Endpoint[] endpoints)
        {
            return CreateLinkGenerator(configureServices: null, endpoints);
        }

        private protected DefaultLinkGenerator CreateLinkGenerator(
            Action<IServiceCollection> configureServices,
            params Endpoint[] endpoints)
        {
            return CreateLinkGenerator(configureServices, new[] { new DefaultEndpointDataSource(endpoints ?? Array.Empty<Endpoint>()) });
        }

        private protected DefaultLinkGenerator CreateLinkGenerator(EndpointDataSource[] dataSources)
        {
            return CreateLinkGenerator(configureServices: null, dataSources);
        }

        private protected DefaultLinkGenerator CreateLinkGenerator(
            Action<IServiceCollection> configureServices,
            EndpointDataSource[] dataSources)
        {
            var services = GetBasicServices();
            AddAdditionalServices(services);
            configureServices?.Invoke(services);

            services.Configure<RouteOptions>(o =>
            {
                if (dataSources != null)
                {
                    foreach (var dataSource in dataSources)
                    {
                        o.EndpointDataSources.Add(dataSource);
                    }
                }
            });

            var serviceProvider = services.BuildServiceProvider();
            var routeOptions = serviceProvider.GetRequiredService<IOptions<RouteOptions>>();

            return new DefaultLinkGenerator(
                new DefaultParameterPolicyFactory(routeOptions, serviceProvider),
                new CompositeEndpointDataSource(routeOptions.Value.EndpointDataSources),
                new DefaultObjectPool<UriBuildingContext>(new UriBuilderContextPooledObjectPolicy()),
                routeOptions,
                NullLogger<DefaultLinkGenerator>.Instance,
                serviceProvider);
        }
    }
}