Umbraco-CMS/tests/Umbraco.Tests.Integration/Umbraco.Core/RuntimeStateTests.cs

using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Options;
using NUnit.Framework;
using Umbraco.Cms.Core;
using Umbraco.Cms.Core.Configuration.Models;
using Umbraco.Cms.Core.DependencyInjection;
using Umbraco.Cms.Core.IO;
using Umbraco.Cms.Core.Packaging;
using Umbraco.Cms.Core.PropertyEditors;
using Umbraco.Cms.Core.Services;
using Umbraco.Cms.Core.Strings;
using Umbraco.Cms.Infrastructure.Migrations;
using Umbraco.Cms.Infrastructure.Packaging;
using Umbraco.Cms.Tests.Common.Testing;
using Umbraco.Cms.Tests.Integration.Testing;
using Umbraco.Extensions;

namespace Umbraco.Cms.Tests.Integration.Umbraco.Core;

[TestFixture]
[UmbracoTest(Database = UmbracoTestOptions.Database.NewSchemaPerTest)]
public class RuntimeStateTests : UmbracoIntegrationTest
{
    private IRuntimeState RuntimeState => Services.GetRequiredService<IRuntimeState>();

    protected override void CustomTestSetup(IUmbracoBuilder builder)
    {
        var migrations = builder.PackageMigrationPlans();
        migrations.Clear();
        migrations.Add<TestMigrationPlan>();
    }

    [Test]
    public void GivenPackageMigrationsExist_WhenLatestStateIsRegistered_ThenLevelIsRun()
    {
        // Add the final state to the keyvalue storage
        var keyValueService = Services.GetRequiredService<IKeyValueService>();
        keyValueService.SetValue(
            Constants.Conventions.Migrations.KeyValuePrefix + TestMigrationPlan.TestMigrationPlanName,
            TestMigrationPlan.TestMigrationFinalState.ToString());

        RuntimeState.DetermineRuntimeLevel();

        Assert.AreEqual(RuntimeLevel.Run, RuntimeState.Level);
        Assert.AreEqual(RuntimeLevelReason.Run, RuntimeState.Reason);
    }

    [Test]
    public void GivenPackageMigrationsExist_WhenUnattendedMigrations_ThenLevelIsRun()
    {
        RuntimeState.DetermineRuntimeLevel();

        Assert.AreEqual(RuntimeLevel.Run, RuntimeState.Level);
        Assert.AreEqual(RuntimeLevelReason.UpgradePackageMigrations, RuntimeState.Reason);
    }

    [Test]
    public void GivenPackageMigrationsExist_WhenNotUnattendedMigrations_ThenLevelIsRun()
    {
        var unattendedOptions = Services.GetRequiredService<IOptions<UnattendedSettings>>();
        unattendedOptions.Value.PackageMigrationsUnattended = false;

        RuntimeState.DetermineRuntimeLevel();

        Assert.AreEqual(RuntimeLevel.Run, RuntimeState.Level);
        Assert.AreEqual(RuntimeLevelReason.Run, RuntimeState.Reason);
    }

    private class TestMigrationPlan : PackageMigrationPlan
    {
        public const string TestMigrationPlanName = "Test";

        public TestMigrationPlan() : base(TestMigrationPlanName)
        {
        }

        public static Guid TestMigrationFinalState => new("BB02C392-4007-4A6C-A550-28BA2FF7E43D");

        protected override void DefinePlan() => To<TestMigration>(TestMigrationFinalState);
    }

    private sealed class TestMigration : AsyncPackageMigrationBase
    {
        public TestMigration(
            IPackagingService packagingService,
            IMediaService mediaService,
            MediaFileManager mediaFileManager,
            MediaUrlGeneratorCollection mediaUrlGenerators,
            IShortStringHelper shortStringHelper,
            IContentTypeBaseServiceProvider contentTypeBaseServiceProvider,
            IMigrationContext context,
            IOptions<PackageMigrationSettings> options)
            : base(packagingService, mediaService, mediaFileManager, mediaUrlGenerators, shortStringHelper, contentTypeBaseServiceProvider, context, options)
        { }

        protected override Task MigrateAsync()
        {
            ImportPackage.FromEmbeddedResource<TestMigration>().Do();

            return Task.CompletedTask;
        }
    }
}