squidex/backend/tests/Squidex.Domain.Apps.Entities.Tests/TestHelpers/Mocks.cs

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

using System.Security.Claims;
using FakeItEasy;
using Squidex.Domain.Apps.Core;
using Squidex.Domain.Apps.Core.Apps;
using Squidex.Domain.Apps.Core.Schemas;
using Squidex.Domain.Apps.Entities.Apps;
using Squidex.Domain.Apps.Entities.Schemas;
using Squidex.Domain.Apps.Entities.Teams;
using Squidex.Infrastructure;
using Squidex.Infrastructure.Security;
using Squidex.Shared;
using Squidex.Shared.Identity;

namespace Squidex.Domain.Apps.Entities.TestHelpers
{
    public static class Mocks
    {
        public static IAppEntity App(NamedId<DomainId> appId, params Language[] languages)
        {
            var config = LanguagesConfig.English;

            foreach (var language in languages)
            {
                config = config.Set(language);
            }

            var app = A.Fake<IAppEntity>();

            A.CallTo(() => app.Id).Returns(appId.Id);
            A.CallTo(() => app.Name).Returns(appId.Name);
            A.CallTo(() => app.Languages).Returns(config);
            A.CallTo(() => app.UniqueId).Returns(appId.Id);

            return app;
        }

        public static ISchemaEntity Schema(NamedId<DomainId> appId, NamedId<DomainId> schemaId, Schema? schemaDef = null)
        {
            var schema = A.Fake<ISchemaEntity>();

            schemaDef ??= new Schema(schemaId.Name);

            A.CallTo(() => schema.Id).Returns(schemaId.Id);
            A.CallTo(() => schema.AppId).Returns(appId);
            A.CallTo(() => schema.SchemaDef).Returns(schemaDef);
            A.CallTo(() => schema.UniqueId).Returns(DomainId.Combine(appId, schemaId.Id));

            return schema;
        }

        public static ITeamEntity Team(DomainId teamId, string teamName = "my-team", string contributor = "user")
        {
            var team = A.Fake<ITeamEntity>();

            A.CallTo(() => team.Id).Returns(teamId);
            A.CallTo(() => team.UniqueId).Returns(teamId);
            A.CallTo(() => team.Name).Returns(teamName);
            A.CallTo(() => team.Contributors).Returns(Contributors.Empty.Assign(contributor, Role.Owner));

            return team;
        }

        public static ClaimsPrincipal ApiUser(string? role = null)
        {
            return CreateUser(role, "api");
        }

        public static ClaimsPrincipal FrontendUser(string? role = null, string? permission = null)
        {
            return CreateUser(role, DefaultClients.Frontend, permission);
        }

        private static ClaimsPrincipal CreateUser(string? role, string client, string? permission = null)
        {
            var claimsIdentity = new ClaimsIdentity();
            var claimsPrincipal = new ClaimsPrincipal(claimsIdentity);

            claimsIdentity.AddClaim(new Claim(OpenIdClaims.ClientId, client));

            if (role != null)
            {
                claimsIdentity.AddClaim(new Claim(ClaimTypes.Role, role));
            }

            if (permission != null)
            {
                claimsIdentity.AddClaim(new Claim(SquidexClaimTypes.Permissions, permission));
            }

            return claimsPrincipal;
        }
    }
}