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

using System.Diagnostics;
using System.Linq;
using NUnit.Framework;
using Umbraco.Cms.Core.Services;
using Umbraco.Cms.Infrastructure.Persistence.Dtos;
using Umbraco.Cms.Infrastructure.Persistence.Repositories.Implement;
using Umbraco.Cms.Tests.Common.Builders;
using Umbraco.Cms.Tests.Common.Testing;
using Umbraco.Cms.Tests.Integration.Testing;
using Umbraco.Extensions;

namespace Umbraco.Cms.Tests.Integration.Umbraco.Infrastructure.Persistence.Repositories
{
    /// <remarks>
    /// v9 -> Tests.Integration
    /// </remarks>
    [TestFixture]
    [UmbracoTest(Database = UmbracoTestOptions.Database.NewSchemaPerTest)]
    public class DocumentVersionRepositoryTest : UmbracoIntegrationTest
    {
        public IFileService FileService => GetRequiredService<IFileService>();
        public IContentTypeService ContentTypeService => GetRequiredService<IContentTypeService>();
        public IContentService ContentService => GetRequiredService<IContentService>();

        [Test]
        public void GetDocumentVersionsEligibleForCleanup_Always_ExcludesActiveVersions()
        {
            var contentType = ContentTypeBuilder.CreateSimpleContentType("umbTextpage", "Textpage");
            FileService.SaveTemplate(contentType.DefaultTemplate);
            ContentTypeService.Save(contentType);

            var content = ContentBuilder.CreateSimpleContent(contentType);

            ContentService.SaveAndPublish(content);
            // At this point content has 2 versions, a draft version and a published version.

            ContentService.SaveAndPublish(content);
            // At this point content has 3 versions, a historic version, a draft version and a published version.

            using (ScopeProvider.CreateScope())
            {
                var sut = new DocumentVersionRepository(ScopeAccessor);
                var results = sut.GetDocumentVersionsEligibleForCleanup();

                Assert.Multiple(() =>
                {
                    Assert.AreEqual(1, results.Count);
                    Assert.AreEqual(1, results.First().VersionId);
                });
            }
        }

        [Test]
        public void GetDocumentVersionsEligibleForCleanup_Always_ExcludesPinnedVersions()
        {
            var contentType = ContentTypeBuilder.CreateSimpleContentType("umbTextpage", "Textpage");
            FileService.SaveTemplate(contentType.DefaultTemplate);
            ContentTypeService.Save(contentType);

            var content = ContentBuilder.CreateSimpleContent(contentType);

            ContentService.SaveAndPublish(content);
            // At this point content has 2 versions, a draft version and a published version.
            ContentService.SaveAndPublish(content);
            ContentService.SaveAndPublish(content);
            ContentService.SaveAndPublish(content);
            // At this point content has 5 versions, 3 historic versions, a draft version and a published version.

            var allVersions = ContentService.GetVersions(content.Id);
            Debug.Assert(allVersions.Count() == 5); // Sanity check

            using (var scope = ScopeProvider.CreateScope())
            {
                scope.Database.Update<ContentVersionDto>("set preventCleanup = 1 where id in (1,3)");

                var sut = new DocumentVersionRepository(ScopeAccessor);
                var results = sut.GetDocumentVersionsEligibleForCleanup();

                Assert.Multiple(() =>
                {
                    Assert.AreEqual(1, results.Count);

                    // We pinned 1 & 3
                    // 4 is current
                    // 5 is published
                    // So all that is left is 2
                    Assert.AreEqual(2, results.First().VersionId);
                });
            }
        }

        [Test]
        public void DeleteVersions_Always_DeletesSpecifiedVersions()
        {
            var contentType = ContentTypeBuilder.CreateSimpleContentType("umbTextpage", "Textpage");
            FileService.SaveTemplate(contentType.DefaultTemplate);
            ContentTypeService.Save(contentType);

            var content = ContentBuilder.CreateSimpleContent(contentType);

            ContentService.SaveAndPublish(content);
            ContentService.SaveAndPublish(content);
            ContentService.SaveAndPublish(content);
            ContentService.SaveAndPublish(content);
            using (var scope = ScopeProvider.CreateScope())
            {
                var query = scope.SqlContext.Sql();

                query.Select<ContentVersionDto>()
                    .From<ContentVersionDto>();

                var sut = new DocumentVersionRepository(ScopeAccessor);
                sut.DeleteVersions(new []{1,2,3});

                var after = scope.Database.Fetch<ContentVersionDto>(query);

                Assert.Multiple(() =>
                {
                    Assert.AreEqual(2, after.Count);
                    Assert.True(after.All(x => x.Id > 3));
                });
            }
        }
    }
}