squidex/backend/tests/Squidex.Domain.Apps.Entities.Tests/Teams/DomainObject/TeamDomainObjectTests.cs

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

using FakeItEasy;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using Squidex.Domain.Apps.Core.Apps;
using Squidex.Domain.Apps.Core.TestHelpers;
using Squidex.Domain.Apps.Entities.Billing;
using Squidex.Domain.Apps.Entities.Teams.Commands;
using Squidex.Domain.Apps.Entities.TestHelpers;
using Squidex.Domain.Apps.Events.Teams;
using Squidex.Infrastructure;
using Squidex.Shared.Users;
using Xunit;

namespace Squidex.Domain.Apps.Entities.Teams.DomainObject
{
    public class TeamDomainObjectTests : HandlerTestBase<TeamDomainObject.State>
    {
        private readonly IBillingPlans billingPlans = A.Fake<IBillingPlans>();
        private readonly IBillingManager billingManager = A.Fake<IBillingManager>();
        private readonly IUser user;
        private readonly IUserResolver userResolver = A.Fake<IUserResolver>();
        private readonly Plan planPaid = new Plan { Id = "premium" };
        private readonly Plan planFree = new Plan { Id = "free" };
        private readonly string contributorId = DomainId.NewGuid().ToString();
        private readonly string name = "My Team";
        private readonly DomainId teamId = DomainId.NewGuid();
        private readonly TeamDomainObject sut;

        protected override DomainId Id
        {
            get => teamId;
        }

        public TeamDomainObjectTests()
        {
            user = UserMocks.User(contributorId);

            A.CallTo(() => userResolver.FindByIdOrEmailAsync(contributorId, default))
                .Returns(user);

            A.CallTo(() => billingPlans.GetFreePlan())
                .Returns(planFree);

            A.CallTo(() => billingPlans.GetPlan(planFree.Id))
                .Returns(planFree);

            A.CallTo(() => billingPlans.GetPlan(planPaid.Id))
                .Returns(planPaid);

            A.CallTo(() => billingManager.MustRedirectToPortalAsync(Actor.Identifier, A<ITeamEntity>._, A<string>._, default))
                .Returns(Task.FromResult<Uri?>(null));

            var serviceProvider =
                new ServiceCollection()
                    .AddSingleton(billingPlans)
                    .AddSingleton(billingManager)
                    .AddSingleton(userResolver)
                    .BuildServiceProvider();

            var log = A.Fake<ILogger<TeamDomainObject>>();

#pragma warning disable MA0056 // Do not call overridable members in constructor
            sut = new TeamDomainObject(Id, PersistenceFactory, log, serviceProvider);
#pragma warning restore MA0056 // Do not call overridable members in constructor
        }

        [Fact]
        public async Task Create_should_create_events_and_set_intitial_state()
        {
            var command = new CreateTeam { Name = name };

            var actual = await PublishAsync(command);

            actual.ShouldBeEquivalent(sut.Snapshot);

            Assert.Equal(name, sut.Snapshot.Name);

            LastEvents
                .ShouldHaveSameEvents(
                    CreateTeamEvent(new TeamCreated { Name = name }),
                    CreateTeamEvent(new TeamContributorAssigned { ContributorId = Actor.Identifier, Role = Role.Owner })
                );
        }

        [Fact]
        public async Task Create_should_not_assign_client_as_contributor()
        {
            var command = new CreateTeam { Name = name, Actor = ActorClient };

            var actual = await PublishAsync(command);

            actual.ShouldBeEquivalent(sut.Snapshot);

            Assert.Equal(name, sut.Snapshot.Name);

            LastEvents
                .ShouldHaveSameEvents(
                    CreateTeamEvent(new TeamCreated { Name = name }, true) // Must be with client actor.
                );
        }

        [Fact]
        public async Task Update_should_create_events_and_update_label_and_description()
        {
            var command = new UpdateTeam { Name = "Changed Name" };

            await ExecuteCreateAsync();

            var actual = await PublishIdempotentAsync(command);

            actual.ShouldBeEquivalent(sut.Snapshot);

            Assert.Equal(command.Name, sut.Snapshot.Name);

            LastEvents
                .ShouldHaveSameEvents(
                    CreateTeamEvent(new TeamUpdated { Name = command.Name })
                );
        }

        [Fact]
        public async Task ChangePlan_should_create_events_and_update_plan()
        {
            var command = new ChangePlan { PlanId = planPaid.Id };

            A.CallTo(() => billingManager.MustRedirectToPortalAsync(Actor.Identifier, A<ITeamEntity>._, planPaid.Id, default))
                .Returns(Task.FromResult<Uri?>(null));

            await ExecuteCreateAsync();

            var actual = await PublishIdempotentAsync(command);

            actual.ShouldBeEquivalent(new PlanChangedResult(planPaid.Id));

            Assert.Equal(planPaid.Id, sut.Snapshot.Plan!.PlanId);

            LastEvents
                .ShouldHaveSameEvents(
                    CreateTeamEvent(new TeamPlanChanged { PlanId = planPaid.Id })
                );

            A.CallTo(() => billingManager.MustRedirectToPortalAsync(Actor.Identifier, A<ITeamEntity>._, planPaid.Id, default))
                .MustHaveHappened();

            A.CallTo(() => billingManager.SubscribeAsync(Actor.Identifier, A<ITeamEntity>._, planPaid.Id, default))
                .MustHaveHappened();
        }

        [Fact]
        public async Task ChangePlan_from_callback_should_create_events_and_update_plan()
        {
            var command = new ChangePlan { PlanId = planPaid.Id, FromCallback = true };

            await ExecuteCreateAsync();

            var actual = await PublishIdempotentAsync(command);

            actual.ShouldBeEquivalent(new PlanChangedResult(planPaid.Id));

            Assert.Equal(planPaid.Id, sut.Snapshot.Plan!.PlanId);

            LastEvents
                .ShouldHaveSameEvents(
                    CreateTeamEvent(new TeamPlanChanged { PlanId = planPaid.Id })
                );

            A.CallTo(() => billingManager.MustRedirectToPortalAsync(A<string>._, A<ITeamEntity>._, A<string?>._, A<CancellationToken>._))
                .MustNotHaveHappened();

            A.CallTo(() => billingManager.SubscribeAsync(A<string>._, A<ITeamEntity>._, A<string?>._, A<CancellationToken>._))
                .MustNotHaveHappened();
        }

        [Fact]
        public async Task ChangePlan_from_callback_should_reset_plan_for_free_plan()
        {
            var command = new ChangePlan { PlanId = planFree.Id, FromCallback = true };

            await ExecuteCreateAsync();
            await ExecuteChangePlanAsync();

            var actual = await PublishIdempotentAsync(command);

            actual.ShouldBeEquivalent(new PlanChangedResult(planFree.Id, true));

            Assert.Null(sut.Snapshot.Plan);

            LastEvents
                .ShouldHaveSameEvents(
                    CreateTeamEvent(new TeamPlanReset())
                );

            A.CallTo(() => billingManager.MustRedirectToPortalAsync(A<string>._, A<ITeamEntity>._, A<string?>._, A<CancellationToken>._))
                .MustHaveHappenedOnceExactly();

            A.CallTo(() => billingManager.UnsubscribeAsync(A<string>._, A<ITeamEntity>._, A<CancellationToken>._))
                .MustNotHaveHappened();
        }

        [Fact]
        public async Task ChangePlan_should_reset_plan_for_free_plan()
        {
            var command = new ChangePlan { PlanId = planFree.Id };

            await ExecuteCreateAsync();
            await ExecuteChangePlanAsync();

            var actual = await PublishIdempotentAsync(command);

            actual.ShouldBeEquivalent(new PlanChangedResult(planFree.Id, true));

            Assert.Null(sut.Snapshot.Plan);

            LastEvents
                .ShouldHaveSameEvents(
                    CreateTeamEvent(new TeamPlanReset())
                );

            A.CallTo(() => billingManager.MustRedirectToPortalAsync(Actor.Identifier, A<ITeamEntity>._, planPaid.Id, default))
                .MustHaveHappenedOnceExactly();

            A.CallTo(() => billingManager.UnsubscribeAsync(A<string>._, A<ITeamEntity>._, A<CancellationToken>._))
                .MustHaveHappened();
        }

        [Fact]
        public async Task ChangePlan_should_not_make_update_for_redirect_actual()
        {
            var command = new ChangePlan { PlanId = planPaid.Id };

            A.CallTo(() => billingManager.MustRedirectToPortalAsync(Actor.Identifier, A<ITeamEntity>._, planPaid.Id, default))
                .Returns(new Uri("http://squidex.io"));

            await ExecuteCreateAsync();

            var actual = await PublishIdempotentAsync(command);

            actual.ShouldBeEquivalent(new PlanChangedResult(planPaid.Id, false, new Uri("http://squidex.io")));

            Assert.Null(sut.Snapshot.Plan);
        }

        [Fact]
        public async Task ChangePlan_should_not_call_billing_manager_for_callback()
        {
            var command = new ChangePlan { PlanId = planPaid.Id, FromCallback = true };

            await ExecuteCreateAsync();

            var actual = await PublishIdempotentAsync(command);

            actual.ShouldBeEquivalent(new PlanChangedResult(planPaid.Id));

            Assert.Equal(planPaid.Id, sut.Snapshot.Plan?.PlanId);

            A.CallTo(() => billingManager.MustRedirectToPortalAsync(Actor.Identifier, A<ITeamEntity>._, planPaid.Id, A<CancellationToken>._))
                .MustNotHaveHappened();

            A.CallTo(() => billingManager.SubscribeAsync(Actor.Identifier, A<ITeamEntity>._, planPaid.Id, A<CancellationToken>._))
                .MustNotHaveHappened();
        }

        [Fact]
        public async Task AssignContributor_should_create_events_and_add_contributor()
        {
            var command = new AssignContributor { ContributorId = contributorId };

            await ExecuteCreateAsync();

            var actual = await PublishIdempotentAsync(command);

            actual.ShouldBeEquivalent(sut.Snapshot);

            Assert.Equal(command.Role, sut.Snapshot.Contributors[contributorId]);

            LastEvents
                .ShouldHaveSameEvents(
                    CreateTeamEvent(new TeamContributorAssigned { ContributorId = contributorId, Role = command.Role, IsAdded = true })
                );
        }

        [Fact]
        public async Task RemoveContributor_should_create_events_and_remove_contributor()
        {
            var command = new RemoveContributor { ContributorId = contributorId };

            await ExecuteCreateAsync();
            await ExecuteAssignContributorAsync();

            var actual = await PublishAsync(command);

            actual.ShouldBeEquivalent(sut.Snapshot);

            Assert.False(sut.Snapshot.Contributors.ContainsKey(contributorId));

            LastEvents
                .ShouldHaveSameEvents(
                    CreateTeamEvent(new TeamContributorRemoved { ContributorId = contributorId })
                );
        }

        private Task ExecuteCreateAsync()
        {
            return PublishAsync(new CreateTeam { Name = name });
        }

        private Task ExecuteAssignContributorAsync()
        {
            return PublishAsync(new AssignContributor { ContributorId = contributorId });
        }

        private Task ExecuteChangePlanAsync()
        {
            return PublishAsync(new ChangePlan { PlanId = planPaid.Id });
        }

        private T CreateTeamEvent<T>(T @event, bool fromClient = false) where T : TeamEvent
        {
            @event.TeamId = teamId;

            return CreateEvent(@event, fromClient);
        }

        private T CreateTeamCommand<T>(T command) where T : TeamCommand
        {
            command.TeamId = teamId;

            return CreateCommand(command);
        }

        private Task<object> PublishIdempotentAsync(TeamCommand command)
        {
            return PublishIdempotentAsync(sut, CreateTeamCommand(command));
        }

        private async Task<object> PublishAsync(TeamCommand command)
        {
            var actual = await sut.ExecuteAsync(CreateTeamCommand(command), default);

            return actual.Payload;
        }
    }
}