squidex/backend/tests/Squidex.Infrastructure.Tests/Log/BackgroundRequestLogStoreTests.cs

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

using System.Globalization;
using FakeItEasy;
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Options;
using Xunit;

namespace Squidex.Infrastructure.Log
{
    public class BackgroundRequestLogStoreTests
    {
        private readonly CancellationTokenSource cts = new CancellationTokenSource();
        private readonly CancellationToken ct;
        private readonly IRequestLogRepository requestLogRepository = A.Fake<IRequestLogRepository>();
        private readonly RequestLogStoreOptions options = new RequestLogStoreOptions();
        private readonly BackgroundRequestLogStore sut;

        public BackgroundRequestLogStoreTests()
        {
            ct = cts.Token;

            options.StoreEnabled = true;

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

            sut = new BackgroundRequestLogStore(Options.Create(options), requestLogRepository, log)
            {
                ForceWrite = true
            };
        }

        [Theory]
        [InlineData(true)]
        [InlineData(false)]
        public void Should_provide_disabled_from_options(bool enabled)
        {
            options.StoreEnabled = enabled;

            Assert.Equal(enabled, sut.IsEnabled);
        }

        [Fact]
        public async Task Should_forward_delete_call()
        {
            await sut.DeleteAsync("my-key", ct);

            A.CallTo(() => requestLogRepository.DeleteAsync("my-key", ct))
                .MustHaveHappened();
        }

        [Fact]
        public async Task Should_not_log_if_disabled()
        {
            options.StoreEnabled = false;

            for (var i = 0; i < 2500; i++)
            {
                await sut.LogAsync(new Request { Key = i.ToString(CultureInfo.InvariantCulture) }, ct);
            }

            sut.Next();
            sut.Dispose();

            // Wait for the timer to not trigger.
            await Task.Delay(500, ct);

            A.CallTo(() => requestLogRepository.InsertManyAsync(A<IEnumerable<Request>>._, A<CancellationToken>._))
                .MustNotHaveHappened();
        }

        [Fact]
        public async Task Should_provide_results_from_repository()
        {
            var key = "my-key";

            var dateFrom = DateTime.Today;
            var dateTo = dateFrom.AddDays(4);

            A.CallTo(() => requestLogRepository.QueryAllAsync(key, dateFrom, dateTo, ct))
                .Returns(AsyncEnumerable.Repeat(new Request { Key = key }, 1));

            var results = await sut.QueryAllAsync(key, dateFrom, dateTo, ct).ToListAsync(ct);

            Assert.NotEmpty(results);
        }

        [Fact]
        public async Task Should_not_provide_results_from_repository_if_disabled()
        {
            options.StoreEnabled = false;

            var key = "my-key";

            var dateFrom = DateTime.Today;
            var dateTo = dateFrom.AddDays(4);

            var results = await sut.QueryAllAsync(key, dateFrom, dateTo, ct).ToListAsync(ct);

            Assert.Empty(results);

            A.CallTo(() => requestLogRepository.QueryAllAsync(key, dateFrom, dateTo, ct))
                .MustNotHaveHappened();
        }

        [Fact]
        public async Task Should_write_logs_in_batches()
        {
            for (var i = 0; i < 2500; i++)
            {
                await sut.LogAsync(new Request { Key = i.ToString(CultureInfo.InvariantCulture) }, ct);
            }

            sut.Next();
            sut.Dispose();

            // Wait for the timer to trigger.
            await Task.Delay(500, ct);

            A.CallTo(() => requestLogRepository.InsertManyAsync(Batch("0", "999"), A<CancellationToken>._))
                .MustHaveHappened();

            A.CallTo(() => requestLogRepository.InsertManyAsync(Batch("1000", "1999"), A<CancellationToken>._))
                .MustHaveHappened();

            A.CallTo(() => requestLogRepository.InsertManyAsync(Batch("2000", "2499"), A<CancellationToken>._))
                .MustHaveHappened();
        }

        private static IEnumerable<Request> Batch(string from, string to)
        {
            return A<IEnumerable<Request>>.That.Matches(x => x.First().Key == from && x.Last().Key == to);
        }
    }
}