squidex/backend/tests/Squidex.Domain.Apps.Entities.Tests/Billing/UsageNotifierWorkerTest.cs

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

using FakeItEasy;
using NodaTime;
using Squidex.Domain.Apps.Core.TestHelpers;
using Squidex.Domain.Apps.Entities.Apps;
using Squidex.Domain.Apps.Entities.Notifications;
using Squidex.Domain.Apps.Entities.TestHelpers;
using Squidex.Infrastructure;
using Squidex.Infrastructure.TestHelpers;
using Squidex.Shared.Users;
using Xunit;

namespace Squidex.Domain.Apps.Entities.Billing
{
    public class UsageNotifierWorkerTest
    {
        private readonly TestState<UsageNotifierWorker.State> state = new TestState<UsageNotifierWorker.State>("Default");
        private readonly IClock clock = A.Fake<IClock>();
        private readonly IAppProvider appProvider = A.Fake<IAppProvider>();
        private readonly IUserNotifications notificationSender = A.Fake<IUserNotifications>();
        private readonly IUserResolver userResolver = A.Fake<IUserResolver>();
        private readonly IAppEntity app = Mocks.App(NamedId.Of(DomainId.NewGuid(), "my-app"));
        private readonly UsageNotifierWorker sut;
        private Instant time = SystemClock.Instance.GetCurrentInstant();

        public UsageNotifierWorkerTest()
        {
            A.CallTo(() => appProvider.GetAppAsync(app.Id, true, default))
                .Returns(app);

            A.CallTo(() => clock.GetCurrentInstant())
                .ReturnsLazily(() => time);

            A.CallTo(() => notificationSender.IsActive)
                .Returns(true);

            sut = new UsageNotifierWorker(state.PersistenceFactory, appProvider, notificationSender, userResolver)
            {
                Clock = clock
            };
        }

        [Fact]
        public async Task Should_load_on_initialize()
        {
            await sut.InitializeAsync(default);

            A.CallTo(() => state.Persistence.ReadAsync(EtagVersion.Any, default))
                .MustHaveHappened();
        }

        [Fact]
        public async Task Should_not_send_notification_if_not_active()
        {
            SetupUser("1", null);
            SetupUser("2", null);

            var message = new UsageTrackingCheck
            {
                AppId = app.Id,
                Usage = 1000,
                UsageLimit = 3000,
                Users = new[] { "1", "2" }
            };

            await sut.HandleAsync(message, default);

            A.CallTo(() => notificationSender.SendUsageAsync(A<IUser>._, A<IAppEntity>._, A<long>._, A<long>._, A<CancellationToken>._))
                .MustNotHaveHappened();
        }

        [Fact]
        public async Task Should_notify_found_users()
        {
            var user1 = SetupUser("1", "user1@email.com");
            var user2 = SetupUser("2", "user2@email.com");
            var user3 = SetupUser("3", null);

            var message = new UsageTrackingCheck
            {
                AppId = app.Id,
                Usage = 1000,
                UsageLimit = 3000,
                Users = new[] { "1", "2", "3" }
            };

            await sut.HandleAsync(message, default);

            A.CallTo(() => notificationSender.SendUsageAsync(user1!, app, 1000, 3000, default))
                .MustHaveHappened();

            A.CallTo(() => notificationSender.SendUsageAsync(user2!, app, 1000, 3000, default))
                .MustHaveHappened();

            A.CallTo(() => notificationSender.SendUsageAsync(user3!, app, 1000, 3000, default))
                .MustNotHaveHappened();
        }

        [Fact]
        public async Task Should_not_notify_again()
        {
            var user = SetupUser("1", "user1@email.com");

            var message = new UsageTrackingCheck
            {
                AppId = app.Id,
                Usage = 1000,
                UsageLimit = 3000,
                Users = new[] { "1" }
            };

            await sut.HandleAsync(message, default);
            await sut.HandleAsync(message, default);

            A.CallTo(() => notificationSender.SendUsageAsync(user!, app, 1000, 3000, default))
                .MustHaveHappenedOnceExactly();
        }

        [Fact]
        public async Task Should_send_again_after_3_days()
        {
            var user = SetupUser("1", "user1@email.com");

            var message = new UsageTrackingCheck
            {
                AppId = app.Id,
                Usage = 1000,
                UsageLimit = 3000,
                Users = new[] { "1" }
            };

            await sut.HandleAsync(message, default);

            time = time.Plus(Duration.FromDays(3));

            await sut.HandleAsync(message, default);

            A.CallTo(() => notificationSender.SendUsageAsync(user!, app, 1000, 3000, default))
                .MustHaveHappenedTwiceExactly();
        }

        [Theory]
        [InlineData(1)]
        [InlineData(6)]
        [InlineData(12)]
        [InlineData(24)]
        [InlineData(48)]
        public async Task Should_not_notify_again_after_few_hours(int hours)
        {
            var user = SetupUser("1", "user1@email.com");

            var message = new UsageTrackingCheck
            {
                AppId = app.Id,
                Usage = 1000,
                UsageLimit = 3000,
                Users = new[] { "1" }
            };

            await sut.HandleAsync(message, default);

            time = time.Plus(Duration.FromHours(hours));

            await sut.HandleAsync(message, default);

            A.CallTo(() => notificationSender.SendUsageAsync(user!, app, 1000, 3000, default))
                .MustHaveHappenedOnceExactly();
        }

        private IUser? SetupUser(string id, string? email)
        {
            if (email != null)
            {
                var user = UserMocks.User(id, email);

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

                return user;
            }
            else
            {
                A.CallTo(() => userResolver.FindByIdOrEmailAsync(id, default))
                    .Returns(Task.FromResult<IUser?>(null));

                return null;
            }
        }
    }
}