squidex/backend/tests/Squidex.Domain.Apps.Entities.Tests/Assets/DomainObject/Guards/GuardAssetTests.cs

// ==========================================================================
//  Squidex Headless CMS
// ==========================================================================
//  Copyright (c) Squidex UG (haftungsbeschraenkt)
//  All rights reserved. Licensed under the MIT license.
// ==========================================================================

using System.Security.Claims;
using Microsoft.Extensions.DependencyInjection;
using Squidex.Domain.Apps.Core.Assets;
using Squidex.Domain.Apps.Core.TestHelpers;
using Squidex.Domain.Apps.Entities.Assets.Commands;
using Squidex.Domain.Apps.Entities.Contents;
using Squidex.Domain.Apps.Entities.Contents.Repositories;
using Squidex.Domain.Apps.Entities.TestHelpers;
using Squidex.Infrastructure;
using Squidex.Infrastructure.Validation;

namespace Squidex.Domain.Apps.Entities.Assets.DomainObject.Guards;

public class GuardAssetTests : GivenContext, IClassFixture<TranslationsFixture>
{
    private readonly IAssetQueryService assetQuery = A.Fake<IAssetQueryService>();
    private readonly IContentRepository contentRepository = A.Fake<IContentRepository>();

    [Fact]
    public async Task Should_throw_exception_if_moving_to_invalid_folder()
    {
        var parentId = DomainId.NewGuid();

        var operation = Operation(CreateAsset());

        A.CallTo(() => assetQuery.FindAssetFolderAsync(AppId.Id, parentId, CancellationToken))
            .Returns([]);

        await ValidationAssert.ThrowsAsync(() => operation.MustMoveToValidFolder(parentId, CancellationToken),
            new ValidationError("Asset folder does not exist.", "ParentId"));
    }

    [Fact]
    public async Task Should_not_throw_exception_if_moving_to_valid_folder()
    {
        var parentId = DomainId.NewGuid();

        var operation = Operation(CreateAsset());

        A.CallTo(() => assetQuery.FindAssetFolderAsync(AppId.Id, parentId, CancellationToken))
            .Returns([CreateAssetFolder()]);

        await operation.MustMoveToValidFolder(parentId, CancellationToken);
    }

    [Fact]
    public async Task Should_not_throw_exception_if_moving_to_same_folder()
    {
        var parentId = DomainId.NewGuid();

        var operation = Operation(CreateAsset() with { ParentId = parentId });

        await operation.MustMoveToValidFolder(parentId, CancellationToken);

        A.CallTo(() => assetQuery.FindAssetFolderAsync(AppId.Id, parentId, A<CancellationToken>._))
            .MustNotHaveHappened();
    }

    [Fact]
    public async Task Should_not_throw_exception_if_moving_to_root()
    {
        var parentId = DomainId.Empty;

        var operation = Operation(CreateAsset().WithId(parentId));

        await operation.MustMoveToValidFolder(parentId, CancellationToken);

        A.CallTo(() => assetQuery.FindAssetFolderAsync(AppId.Id, parentId, A<CancellationToken>._))
            .MustNotHaveHappened();
    }

    [Fact]
    public async Task Should_throw_exception_if_referenced()
    {
        var operation = Operation(CreateAsset());

        A.CallTo(() => contentRepository.HasReferrersAsync(App, operation.CommandId, SearchScope.All, CancellationToken))
            .Returns(true);

        await Assert.ThrowsAsync<DomainException>(() => operation.CheckReferrersAsync(CancellationToken));
    }

    [Fact]
    public async Task Should_not_throw_exception_if_not_referenced()
    {
        var operation = Operation(CreateAsset());

        A.CallTo(() => contentRepository.HasReferrersAsync(App, operation.CommandId, SearchScope.All, CancellationToken))
            .Returns(true);

        await Assert.ThrowsAsync<DomainException>(() => operation.CheckReferrersAsync(CancellationToken));
    }

    private AssetOperation Operation(Asset asset)
    {
        return Operation(asset, Mocks.FrontendUser());
    }

    private AssetOperation Operation(Asset asset, ClaimsPrincipal? currentUser)
    {
        var serviceProvider =
            new ServiceCollection()
                .AddSingleton(contentRepository)
                .AddSingleton(assetQuery)
                .BuildServiceProvider();

        return new AssetOperation(serviceProvider, () => asset)
        {
            App = App,
            CommandId = asset.Id,
            Command = new CreateAsset { User = currentUser, Actor = User },
        };
    }
}