Umbraco-CMS/tests/Umbraco.Tests.UnitTests/Umbraco.Cms.Api.Management/Services/Flags/IsProtectedFlagProviderTests.cs

using NUnit.Framework;
using Umbraco.Cms.Api.Management.Services.Flags;
using Umbraco.Cms.Api.Management.ViewModels.Document.Collection;
using Umbraco.Cms.Api.Management.ViewModels.Document.Item;
using Umbraco.Cms.Api.Management.ViewModels.Tree;

namespace Umbraco.Cms.Tests.UnitTests.Umbraco.Cms.Api.Management.Services.Flags;

[TestFixture]
internal class IsProtectedFlagProviderTests
{
    [Test]
    public void IsProtectedFlagProvider_Can_Provide_Tree_Flags()
    {
        var sut = new IsProtectedFlagProvider();
        Assert.IsTrue(sut.CanProvideFlags<DocumentTreeItemResponseModel>());
    }

    [Test]
    public void IsProtectedFlagProvider_Can_Provide_Collection_Flags()
    {
        var sut = new IsProtectedFlagProvider();
        Assert.IsTrue(sut.CanProvideFlags<DocumentCollectionResponseModel>());
    }

    [Test]
    public void IsProtectedFlagProvider_Can_Provide_Plain_Flags()
    {
        var sut = new IsProtectedFlagProvider();
        Assert.IsTrue(sut.CanProvideFlags<DocumentItemResponseModel>());
    }

    [Test]
    public async Task IsProtectedFlagProvider_Should_Populate_Tree_Flags()
    {
        var sut = new IsProtectedFlagProvider();

        var viewModels = new List<DocumentTreeItemResponseModel>
        {
            new(),
            new() { IsProtected = true },
        };

        await sut.PopulateFlagsAsync(viewModels);

        Assert.AreEqual(viewModels[0].Flags.Count(), 0);
        Assert.AreEqual(viewModels[1].Flags.Count(), 1);

        var flagModel = viewModels[1].Flags.First();
        Assert.AreEqual("Umb.IsProtected", flagModel.Alias);
    }

    [Test]
    public async Task IsProtectedFlagProvider_Should_Populate_Collection_Flags()
    {
        var sut = new IsProtectedFlagProvider();

        var viewModels = new List<DocumentCollectionResponseModel>
        {
            new(),
            new() { IsProtected = true },
        };

        await sut.PopulateFlagsAsync(viewModels);

        Assert.AreEqual(viewModels[0].Flags.Count(), 0);
        Assert.AreEqual(viewModels[1].Flags.Count(), 1);

        var flagModel = viewModels[1].Flags.First();
        Assert.AreEqual("Umb.IsProtected", flagModel.Alias);
    }

    [Test]
    public async Task IsProtectedFlagProvider_Should_Populate_Plain_Flags()
    {
        var sut = new IsProtectedFlagProvider();

        var viewModels = new List<DocumentItemResponseModel>
        {
            new(),
            new() { IsProtected = true },
        };

        await sut.PopulateFlagsAsync(viewModels);

        Assert.AreEqual(viewModels[0].Flags.Count(), 0);
        Assert.AreEqual(viewModels[1].Flags.Count(), 1);

        var flagModel = viewModels[1].Flags.First();
        Assert.AreEqual("Umb.IsProtected", flagModel.Alias);
    }
}