squidex/backend/tests/Squidex.Domain.Apps.Entities.Tests/Assets/Queries/ScriptAssetTests.cs

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

using Squidex.Domain.Apps.Core.Assets;
using Squidex.Domain.Apps.Core.Scripting;
using Squidex.Domain.Apps.Entities.Assets.Queries.Steps;
using Squidex.Domain.Apps.Entities.TestHelpers;
using Squidex.Infrastructure;

namespace Squidex.Domain.Apps.Entities.Assets.Queries;

public class ScriptAssetTests : GivenContext
{
    private readonly IScriptEngine scriptEngine = A.Fake<IScriptEngine>();
    private readonly ScriptAsset sut;

    public ScriptAssetTests()
    {
        sut = new ScriptAsset(scriptEngine);
    }

    [Fact]
    public async Task Should_not_call_script_engine_if_no_script_configured()
    {
        var asset = new AssetEntity();

        await sut.EnrichAsync(ApiContext, new[] { asset }, CancellationToken);

        A.CallTo(() => scriptEngine.ExecuteAsync(A<AssetScriptVars>._, A<string>._, ScriptOptions(), A<CancellationToken>._))
            .MustNotHaveHappened();
    }

    [Fact]
    public async Task Should_not_call_script_engine_for_frontend_user()
    {
        A.CallTo(() => App.AssetScripts)
            .Returns(new AssetScripts { Query = "my-query" });

        var asset = new AssetEntity();

        await sut.EnrichAsync(FrontendContext, new[] { asset }, CancellationToken);

        A.CallTo(() => scriptEngine.ExecuteAsync(A<AssetScriptVars>._, A<string>._, ScriptOptions(), A<CancellationToken>._))
            .MustNotHaveHappened();
    }

    [Fact]
    public async Task Should_call_script_engine()
    {
        A.CallTo(() => App.AssetScripts)
            .Returns(new AssetScripts { Query = "my-query" });

        var asset = new AssetEntity { Id = DomainId.NewGuid() };

        await sut.EnrichAsync(ApiContext, new[] { asset }, CancellationToken);

        A.CallTo(() => scriptEngine.ExecuteAsync(
                A<AssetScriptVars>.That.Matches(x =>
                    Equals(x["assetId"], asset.Id) &&
                    Equals(x["appId"], AppId.Id) &&
                    Equals(x["appName"], AppId.Name) &&
                    Equals(x["user"], ApiContext.UserPrincipal)),
                "my-query",
                ScriptOptions(), CancellationToken))
            .MustHaveHappened();
    }

    [Fact]
    public async Task Should_make_test_with_pre_query_script()
    {
        A.CallTo(() => App.AssetScripts)
            .Returns(new AssetScripts { Query = "my-query", QueryPre = "my-pre-query" });

        var asset = new AssetEntity { Id = DomainId.NewGuid() };

        await sut.EnrichAsync(ApiContext, new[] { asset }, CancellationToken);

        A.CallTo(() => scriptEngine.ExecuteAsync(
                A<AssetScriptVars>.That.Matches(x =>
                    Equals(x.GetValue<object>("assetId"), null) &&
                    Equals(x["appId"], AppId.Id) &&
                    Equals(x["appName"], AppId.Name) &&
                    Equals(x["user"], ApiContext.UserPrincipal)),
                "my-pre-query",
                ScriptOptions(), CancellationToken))
            .MustHaveHappened();

        A.CallTo(() => scriptEngine.ExecuteAsync(
                A<AssetScriptVars>.That.Matches(x =>
                    Equals(x.GetValue<object>("assetId"), asset.Id) &&
                    Equals(x["appId"], AppId.Id) &&
                    Equals(x["appName"], AppId.Name) &&
                    Equals(x["user"], ApiContext.UserPrincipal)),
                "my-query",
                ScriptOptions(), CancellationToken))
            .MustHaveHappened();
    }

    private static ScriptOptions ScriptOptions()
    {
        return A<ScriptOptions>.That.Matches(x => x.AsContext);
    }
}