Umbraco-CMS/tests/Umbraco.Tests.Integration/Umbraco.Infrastructure/Persistence/Repositories/CacheInstructionRepositoryTest.cs

// Copyright (c) Umbraco.
// See LICENSE for more details.

using System;
using System.Collections.Generic;
using System.Linq;
using Microsoft.Extensions.Logging;
using NUnit.Framework;
using Umbraco.Cms.Core;
using Umbraco.Cms.Core.Models;
using Umbraco.Cms.Core.Persistence.Querying;
using Umbraco.Cms.Core.Scoping;
using Umbraco.Cms.Infrastructure.Persistence;
using Umbraco.Cms.Infrastructure.Persistence.Dtos;
using Umbraco.Cms.Infrastructure.Persistence.Repositories.Implement;
using Umbraco.Cms.Tests.Common.Testing;
using Umbraco.Cms.Tests.Integration.Testing;

namespace Umbraco.Cms.Tests.Integration.Umbraco.Infrastructure.Persistence.Repositories
{
    [TestFixture]
    [UmbracoTest(Database = UmbracoTestOptions.Database.NewSchemaPerTest, Logger = UmbracoTestOptions.Logger.Console)]
    public class CacheInstructionRepositoryTest : UmbracoIntegrationTest
    {
        private const string OriginIdentiy = "Test1";
        private const string Instructions = "{}";
        private readonly DateTime _date = new DateTime(2021, 7, 3, 10, 30, 0);
        private const int InstructionCount = 1;

        [Test]
        public void Can_Count_All()
        {
            const int Count = 5;

            IScopeProvider sp = ScopeProvider;
            using (IScope scope = ScopeProvider.CreateScope())
            {
                var repo = new CacheInstructionRepository((IScopeAccessor)sp);
                for (var i = 0; i < Count; i++)
                {
                    repo.Add(new CacheInstruction(0, _date, Instructions, OriginIdentiy, InstructionCount));
                }

                var count = repo.CountAll();

                Assert.That(count, Is.EqualTo(Count));
            }
        }

        [Test]
        public void Can_Count_Pending_Instructions()
        {
            IScopeProvider sp = ScopeProvider;
            using (IScope scope = ScopeProvider.CreateScope())
            {
                var repo = new CacheInstructionRepository((IScopeAccessor)sp);
                for (var i = 0; i < 5; i++)
                {
                    repo.Add(new CacheInstruction(0, _date, Instructions, OriginIdentiy, InstructionCount));
                }

                var count = repo.CountPendingInstructions(2);

                Assert.That(count, Is.EqualTo(3));
            }
        }

        [Test]
        public void Can_Check_Exists()
        {
            IScopeProvider sp = ScopeProvider;
            using (IScope scope = ScopeProvider.CreateScope())
            {
                var repo = new CacheInstructionRepository((IScopeAccessor)sp);

                var existsBefore = repo.Exists(1);

                repo.Add(new CacheInstruction(0, _date, Instructions, OriginIdentiy, InstructionCount));

                var existsAfter = repo.Exists(1);

                Assert.That(existsBefore, Is.False);
                Assert.That(existsAfter, Is.True);
            }
        }

        [Test]
        public void Can_Add_Cache_Instruction()
        {
            const string OriginIdentiy = "Test1";
            const string Instructions = "{}";
            var date = new DateTime(2021, 7, 3, 10, 30, 0);
            const int InstructionCount = 1;

            IScopeProvider sp = ScopeProvider;
            using (IScope scope = ScopeProvider.CreateScope())
            {
                var repo = new CacheInstructionRepository((IScopeAccessor)sp);
                repo.Add(new CacheInstruction(0, date, Instructions, OriginIdentiy, InstructionCount));

                List<CacheInstructionDto> dtos = scope.Database.Fetch<CacheInstructionDto>("WHERE id > -1");

                Assert.That(dtos.Any(), Is.True);

                CacheInstructionDto dto = dtos.First();
                Assert.That(dto.UtcStamp, Is.EqualTo(date));
                Assert.That(dto.Instructions, Is.EqualTo(Instructions));
                Assert.That(dto.OriginIdentity, Is.EqualTo(OriginIdentiy));
                Assert.That(dto.InstructionCount, Is.EqualTo(InstructionCount));
            }
        }

        [Test]
        public void Can_Get_Pending_Instructions()
        {
            IScopeProvider sp = ScopeProvider;
            using (IScope scope = ScopeProvider.CreateScope())
            {
                var repo = new CacheInstructionRepository((IScopeAccessor)sp);
                for (var i = 0; i < 5; i++)
                {
                    repo.Add(new CacheInstruction(0, _date, Instructions, OriginIdentiy, InstructionCount));
                }

                IEnumerable<CacheInstruction> instructions = repo.GetPendingInstructions(2, 2);

                Assert.That(instructions.Count(), Is.EqualTo(2));

                Assert.That(string.Join(",", instructions.Select(x => x.Id)), Is.EqualTo("3,4"));
            }
        }

        [Test]
        public void Can_Delete_Old_Instructions()
        {
            IScopeProvider sp = ScopeProvider;
            using (IScope scope = ScopeProvider.CreateScope())
            {
                var repo = new CacheInstructionRepository((IScopeAccessor)sp);
                for (var i = 0; i < 5; i++)
                {
                    DateTime date = i == 0 ? DateTime.UtcNow.AddDays(-2) : DateTime.UtcNow;
                    repo.Add(new CacheInstruction(0, date, Instructions, OriginIdentiy, InstructionCount));
                }

                repo.DeleteInstructionsOlderThan(DateTime.UtcNow.AddDays(-1));

                var count = repo.CountAll();
                Assert.That(count, Is.EqualTo(4));  // 5 have been added, 1 is older and deleted.
            }
        }
    }
}