Umbraco-CMS/tests/Umbraco.Tests.UnitTests/Umbraco.Core/Scheduling/ContentVersionCleanupTest.cs

using System;
using System.Threading.Tasks;
using AutoFixture.NUnit3;
using Microsoft.Extensions.Options;
using Moq;
using NUnit.Framework;
using Umbraco.Cms.Core;
using Umbraco.Cms.Core.Configuration.Models;
using Umbraco.Cms.Core.Runtime;
using Umbraco.Cms.Core.Services;
using Umbraco.Cms.Core.Sync;
using Umbraco.Cms.Infrastructure.HostedServices;
using Umbraco.Cms.Tests.UnitTests.AutoFixture;

namespace Umbraco.Tests.Scheduling
{
    [TestFixture]
    class ContentVersionCleanupTest
    {
        [Test, AutoMoqData]
        public async Task ContentVersionCleanup_WhenNotEnabled_DoesNotCleanupWillRepeat(
            [Frozen] Mock<IOptionsMonitor<ContentSettings>> settings,
            [Frozen] Mock<IMainDom> mainDom,
            [Frozen] Mock<IServerRoleAccessor> serverRoleAccessor,
            [Frozen] Mock<IRuntimeState> runtimeState,
            [Frozen] Mock<IContentVersionService> cleanupService,
            ContentVersionCleanup sut)
        {
            settings.Setup(x => x.CurrentValue).Returns(new ContentSettings()
            {
                ContentVersionCleanupPolicy = new ContentVersionCleanupPolicySettings()
                {
                    EnableCleanup = false
                }
            });
            runtimeState.Setup(x => x.Level).Returns(RuntimeLevel.Run);
            mainDom.Setup(x => x.IsMainDom).Returns(true);
            serverRoleAccessor.Setup(x => x.CurrentServerRole).Returns(ServerRole.SchedulingPublisher);

            await sut.PerformExecuteAsync(null);

            cleanupService.Verify(x => x.PerformContentVersionCleanup(It.IsAny<DateTime>()), Times.Never);
        }

        [Test, AutoMoqData]
        public async Task ContentVersionCleanup_RuntimeLevelNotRun_DoesNotCleanupWillRepeat(
            [Frozen] Mock<IOptionsMonitor<ContentSettings>> settings,
            [Frozen] Mock<IMainDom> mainDom,
            [Frozen] Mock<IServerRoleAccessor> serverRoleAccessor,
            [Frozen] Mock<IRuntimeState> runtimeState,
            [Frozen] Mock<IContentVersionService> cleanupService,
            ContentVersionCleanup sut)
        {
            settings.Setup(x => x.CurrentValue).Returns(new ContentSettings()
            {
                ContentVersionCleanupPolicy = new ContentVersionCleanupPolicySettings()
                {
                    EnableCleanup = true
                }
            });
            runtimeState.Setup(x => x.Level).Returns(RuntimeLevel.Unknown);
            mainDom.Setup(x => x.IsMainDom).Returns(true);
            serverRoleAccessor.Setup(x => x.CurrentServerRole).Returns(ServerRole.SchedulingPublisher);

            await sut.PerformExecuteAsync(null);

            cleanupService.Verify(x => x.PerformContentVersionCleanup(It.IsAny<DateTime>()), Times.Never);
        }

        [Test, AutoMoqData]
        public async Task  ContentVersionCleanup_ServerRoleUnknown_DoesNotCleanupWillRepeat(
            [Frozen] Mock<IOptionsMonitor<ContentSettings>> settings,
            [Frozen] Mock<IMainDom> mainDom,
            [Frozen] Mock<IServerRoleAccessor> serverRoleAccessor,
            [Frozen] Mock<IRuntimeState> runtimeState,
            [Frozen] Mock<IContentVersionService> cleanupService,
            ContentVersionCleanup sut)
        {
            settings.Setup(x => x.CurrentValue).Returns(new ContentSettings()
            {
                ContentVersionCleanupPolicy = new ContentVersionCleanupPolicySettings()
                {
                    EnableCleanup = true
                }
            });
            runtimeState.Setup(x => x.Level).Returns(RuntimeLevel.Run);
            mainDom.Setup(x => x.IsMainDom).Returns(true);
            serverRoleAccessor.Setup(x => x.CurrentServerRole).Returns(ServerRole.Unknown);

            await sut.PerformExecuteAsync(null);

            cleanupService.Verify(x => x.PerformContentVersionCleanup(It.IsAny<DateTime>()), Times.Never);
        }

        [Test, AutoMoqData]
        public async Task ContentVersionCleanup_NotMainDom_DoesNotCleanupWillNotRepeat(
            [Frozen] Mock<IOptionsMonitor<ContentSettings>> settings,
            [Frozen] Mock<IMainDom> mainDom,
            [Frozen] Mock<IServerRoleAccessor> serverRoleAccessor,
            [Frozen] Mock<IRuntimeState> runtimeState,
            [Frozen] Mock<IContentVersionService> cleanupService,
            ContentVersionCleanup sut)
        {
            settings.Setup(x => x.CurrentValue).Returns(new ContentSettings()
            {
                ContentVersionCleanupPolicy = new ContentVersionCleanupPolicySettings()
                {
                    EnableCleanup = true
                }
            });

            runtimeState.Setup(x => x.Level).Returns(RuntimeLevel.Run);
            mainDom.Setup(x => x.IsMainDom).Returns(false);
            serverRoleAccessor.Setup(x => x.CurrentServerRole).Returns(ServerRole.SchedulingPublisher);

            await sut.PerformExecuteAsync(null);

            cleanupService.Verify(x => x.PerformContentVersionCleanup(It.IsAny<DateTime>()), Times.Never);
        }

        [Test, AutoMoqData]
        public async Task ContentVersionCleanup_Enabled_DelegatesToCleanupService(
            [Frozen] Mock<IOptionsMonitor<ContentSettings>> settings,
            [Frozen] Mock<IMainDom> mainDom,
            [Frozen] Mock<IServerRoleAccessor> serverRoleAccessor,
            [Frozen] Mock<IRuntimeState> runtimeState,
            [Frozen] Mock<IContentVersionService> cleanupService,
            ContentVersionCleanup sut)
        {
            settings.Setup(x => x.CurrentValue).Returns(new ContentSettings()
            {
                ContentVersionCleanupPolicy = new ContentVersionCleanupPolicySettings()
                {
                    EnableCleanup = true
                }
            });

            runtimeState.Setup(x => x.Level).Returns(RuntimeLevel.Run);
            mainDom.Setup(x => x.IsMainDom).Returns(true);
            serverRoleAccessor.Setup(x => x.CurrentServerRole).Returns(ServerRole.SchedulingPublisher);


            await sut.PerformExecuteAsync(null);

            cleanupService.Verify(x => x.PerformContentVersionCleanup(It.IsAny<DateTime>()), Times.Once);
        }
    }
}