Umbraco-CMS/tests/Umbraco.Tests.UnitTests/Umbraco.Infrastructure/HostedServices/ServerRegistration/InstructionProcessTaskTests.cs

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

using System.Threading.Tasks;
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Options;
using Moq;
using NUnit.Framework;
using Umbraco.Cms.Core;
using Umbraco.Cms.Core.Configuration.Models;
using Umbraco.Cms.Core.Services;
using Umbraco.Cms.Core.Sync;
using Umbraco.Cms.Infrastructure.HostedServices.ServerRegistration;

namespace Umbraco.Cms.Tests.UnitTests.Umbraco.Infrastructure.HostedServices.ServerRegistration
{
    [TestFixture]
    public class InstructionProcessTaskTests
    {
        private Mock<IServerMessenger> _mockDatabaseServerMessenger;

        [TestCase(RuntimeLevel.Boot)]
        [TestCase(RuntimeLevel.Install)]
        [TestCase(RuntimeLevel.Unknown)]
        [TestCase(RuntimeLevel.Upgrade)]
        [TestCase(RuntimeLevel.BootFailed)]
        public async Task Does_Not_Execute_When_Runtime_State_Is_Not_Run(RuntimeLevel runtimeLevel)
        {
            InstructionProcessTask sut = CreateInstructionProcessTask(runtimeLevel: runtimeLevel);
            await sut.PerformExecuteAsync(null);
            VerifyMessengerNotSynced();
        }

        [Test]
        public async Task Executes_And_Touches_Server()
        {
            InstructionProcessTask sut = CreateInstructionProcessTask();
            await sut.PerformExecuteAsync(null);
            VerifyMessengerSynced();
        }

        private InstructionProcessTask CreateInstructionProcessTask(RuntimeLevel runtimeLevel = RuntimeLevel.Run)
        {
            var mockRunTimeState = new Mock<IRuntimeState>();
            mockRunTimeState.SetupGet(x => x.Level).Returns(runtimeLevel);

            var mockLogger = new Mock<ILogger<InstructionProcessTask>>();

            _mockDatabaseServerMessenger = new Mock<IServerMessenger>();

            var settings = new GlobalSettings();

            return new InstructionProcessTask(mockRunTimeState.Object, _mockDatabaseServerMessenger.Object, mockLogger.Object, Options.Create(settings));
        }

        private void VerifyMessengerNotSynced() => VerifyMessengerSyncedTimes(Times.Never());

        private void VerifyMessengerSynced() => VerifyMessengerSyncedTimes(Times.Once());

        private void VerifyMessengerSyncedTimes(Times times) => _mockDatabaseServerMessenger.Verify(x => x.Sync(), times);
    }
}