Umbraco-CMS/src/Umbraco.Tests.Integration/Persistence/Repositories/EntityRepositoryTest.cs

using System.Collections.Generic;
using System.Linq;
using NUnit.Framework;
using Umbraco.Core;
using Umbraco.Core.Models;
using Umbraco.Core.Models.Entities;
using Umbraco.Core.Persistence.Repositories.Implement;
using Umbraco.Core.Scoping;
using Umbraco.Core.Services;
using Umbraco.Tests.Common.Builders;
using Umbraco.Tests.Integration.Testing;
using Umbraco.Tests.TestHelpers.Entities;
using Umbraco.Tests.Testing;

namespace Umbraco.Tests.Integration.Persistence.Repositories
{
    [TestFixture]
    [UmbracoTest(Mapper = true, Database = UmbracoTestOptions.Database.NewSchemaPerTest)]
    public class EntityRepositoryTest : UmbracoIntegrationTest
    {
        private EntityRepository CreateRepository(IScopeAccessor scopeAccessor)
        {
            var entityRepository = new EntityRepository(scopeAccessor);
            return entityRepository;
        }

        [Test]
        public void Get_Paged_Mixed_Entities_By_Ids()
        {
            //Create content

            var contentService = GetRequiredService<IContentService>();
            var contentTypeService = GetRequiredService<IContentTypeService>();
            var createdContent = new List<IContent>();
            var contentType = ContentTypeBuilder.CreateBasicContentType("blah");
            contentTypeService.Save(contentType);
            for (var i = 0; i < 10; i++)
            {
                var c1 = ContentBuilder.CreateBasicContent(contentType);
                contentService.Save(c1);
                createdContent.Add(c1);
            }

            //Create media
            
            var mediaService = GetRequiredService<IMediaService>();
            var mediaTypeService = GetRequiredService<IMediaTypeService>();
            var createdMedia = new List<IMedia>();
            var imageType = MediaTypeBuilder.CreateImageMediaType("myImage");
            mediaTypeService.Save(imageType);
            for (var i = 0; i < 10; i++)
            {
                var c1 = MediaBuilder.CreateMediaImage(imageType, -1);
                mediaService.Save(c1);
                createdMedia.Add(c1);
            }

            // Create members

            var memberService = GetRequiredService<IMemberService>();
            var memberTypeService = GetRequiredService<IMemberTypeService>();
            var memberType = MemberTypeBuilder.CreateSimpleMemberType("simple");
            memberTypeService.Save(memberType);
            var createdMembers = MemberBuilder.CreateMultipleSimpleMembers(memberType, 10).ToList();
            memberService.Save(createdMembers);

            
            var provider = ScopeProvider;
            using (provider.CreateScope())
            {
                var repo = CreateRepository((IScopeAccessor)provider);

                var ids = createdContent.Select(x => x.Id).Concat(createdMedia.Select(x => x.Id)).Concat(createdMembers.Select(x => x.Id));

                var objectTypes = new[] { Constants.ObjectTypes.Document, Constants.ObjectTypes.Media, Constants.ObjectTypes.Member };

                var query = provider.SqlContext.Query<IUmbracoEntity>()
                    .WhereIn(e => e.Id, ids);

                var entities = repo.GetPagedResultsByQuery(query, objectTypes, 0, 20, out var totalRecords, null, null).ToList();
                
                Assert.AreEqual(20, entities.Count);
                Assert.AreEqual(30, totalRecords);

                //add the next page
                entities.AddRange(repo.GetPagedResultsByQuery(query, objectTypes, 1, 20, out totalRecords, null, null));

                Assert.AreEqual(30, entities.Count);
                Assert.AreEqual(30, totalRecords);

                var contentEntities = entities.OfType<IDocumentEntitySlim>().ToList();
                var mediaEntities = entities.OfType<IMediaEntitySlim>().ToList();
                var memberEntities = entities.OfType<IMemberEntitySlim>().ToList();

                Assert.AreEqual(10, contentEntities.Count);
                Assert.AreEqual(10, mediaEntities.Count);
                Assert.AreEqual(10, memberEntities.Count);
            }
        }
    }
}