Umbraco-CMS/tests/Umbraco.Tests.UnitTests/AutoFixture/Customizations/UmbracoCustomizations.cs

using System;
using System.Linq;
using AutoFixture;
using AutoFixture.Kernel;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Identity;
using Microsoft.Extensions.Options;
using Moq;
using Umbraco.Cms.Core;
using Umbraco.Cms.Core.Configuration;
using Umbraco.Cms.Core.Configuration.Models;
using Umbraco.Cms.Core.Hosting;
using Umbraco.Cms.Core.Security;
using Umbraco.Cms.Core.Services;
using Umbraco.Cms.Infrastructure.Migrations.Install;
using Umbraco.Cms.Web.BackOffice.Controllers;
using Umbraco.Cms.Web.BackOffice.Install;
using Umbraco.Cms.Web.BackOffice.Routing;
using Umbraco.Cms.Web.Common.Security;

namespace Umbraco.Cms.Tests.UnitTests.AutoFixture.Customizations;

internal class UmbracoCustomizations : ICustomization
{
    public void Customize(IFixture fixture)
    {
        fixture.Customize<BackOfficeIdentityUser>(
            u => u.FromFactory<string, string, string>(
                (a, b, c) => BackOfficeIdentityUser.CreateNew(new GlobalSettings(), a, b, c)));

        fixture
            .Customize(new ConstructorCustomization(typeof(UsersController), new GreedyConstructorQuery()))
            .Customize(new ConstructorCustomization(typeof(InstallController), new GreedyConstructorQuery()))
            .Customize(new ConstructorCustomization(typeof(PreviewController), new GreedyConstructorQuery()))
            .Customize(new ConstructorCustomization(typeof(MemberController), new GreedyConstructorQuery()))
            .Customize(new ConstructorCustomization(typeof(BackOfficeController), new GreedyConstructorQuery()))
            .Customize(new ConstructorCustomization(typeof(BackOfficeUserManager), new GreedyConstructorQuery()))
            .Customize(new ConstructorCustomization(typeof(MemberManager), new GreedyConstructorQuery()))
            .Customize(new ConstructorCustomization(typeof(DatabaseSchemaCreatorFactory), new GreedyConstructorQuery()))
            .Customize(new ConstructorCustomization(typeof(BackOfficeServerVariables), new GreedyConstructorQuery()));

        // When requesting an IUserStore ensure we actually uses a IUserLockoutStore
        fixture.Customize<IUserStore<BackOfficeIdentityUser>>(cc =>
            cc.FromFactory(Mock.Of<IUserLockoutStore<BackOfficeIdentityUser>>));

        fixture.Customize<IUmbracoVersion>(
            u => u.FromFactory(
                () => new UmbracoVersion()));

        fixture.Customize<HostingSettings>(x =>
            x.With(settings => settings.ApplicationVirtualPath, string.Empty));

        fixture.Customize<BackOfficeAreaRoutes>(u => u.FromFactory(
            () => new BackOfficeAreaRoutes(
                Options.Create(new GlobalSettings()),
                Mock.Of<IHostingEnvironment>(x =>
                    x.ToAbsolute(It.IsAny<string>()) == "/umbraco" && x.ApplicationVirtualPath == string.Empty),
                Mock.Of<IRuntimeState>(x => x.Level == RuntimeLevel.Run),
                new UmbracoApiControllerTypeCollection(Enumerable.Empty<Type>))));

        fixture.Customize<PreviewRoutes>(u => u.FromFactory(
            () => new PreviewRoutes(
                Options.Create(new GlobalSettings()),
                Mock.Of<IHostingEnvironment>(x =>
                    x.ToAbsolute(It.IsAny<string>()) == "/umbraco" && x.ApplicationVirtualPath == string.Empty),
                Mock.Of<IRuntimeState>(x => x.Level == RuntimeLevel.Run))));

        var httpContextAccessor = new HttpContextAccessor { HttpContext = new DefaultHttpContext() };
        fixture.Customize<HttpContext>(x => x.FromFactory(() => httpContextAccessor.HttpContext));
        fixture.Customize<IHttpContextAccessor>(x => x.FromFactory(() => httpContextAccessor));

        fixture.Customize<WebRoutingSettings>(x =>
            x.With(settings => settings.UmbracoApplicationUrl, "http://localhost:5000"));
    }
}