// ========================================================================== // Squidex Headless CMS // ========================================================================== // Copyright (c) Squidex UG (haftungsbeschränkt) // All rights reserved. Licensed under the MIT license. // ========================================================================== using System; using System.Collections.Generic; using System.Linq; using System.Threading.Tasks; using FakeItEasy; using Squidex.Domain.Apps.Entities.Assets.Repositories; using Squidex.Domain.Apps.Entities.TestHelpers; using Squidex.Infrastructure; using Squidex.Infrastructure.Queries; using Xunit; namespace Squidex.Domain.Apps.Entities.Assets.Queries { public class AssetQueryServiceTests { private readonly IAssetEnricher assetEnricher = A.Fake(); private readonly IAssetRepository assetRepository = A.Fake(); private readonly NamedId appId = NamedId.Of(Guid.NewGuid(), "my-app"); private readonly Context requestContext; private readonly AssetQueryParser queryParser = A.Fake(); private readonly AssetQueryService sut; public AssetQueryServiceTests() { requestContext = new Context(Mocks.FrontendUser(), Mocks.App(appId)); A.CallTo(() => queryParser.ParseQuery(requestContext, A.Ignored)) .Returns(new ClrQuery()); sut = new AssetQueryService(assetEnricher, assetRepository, queryParser); } [Fact] public async Task Should_find_asset_by_id_and_enrich_it() { var found = new AssetEntity { Id = Guid.NewGuid() }; var enriched = new AssetEntity(); A.CallTo(() => assetRepository.FindAssetAsync(found.Id, false)) .Returns(found); A.CallTo(() => assetEnricher.EnrichAsync(found, requestContext)) .Returns(enriched); var result = await sut.FindAssetAsync(requestContext, found.Id); Assert.Same(enriched, result); } [Fact] public async Task Should_find_assets_by_hash_and_and_enrich_it() { var found = new AssetEntity { Id = Guid.NewGuid() }; var enriched = new AssetEntity(); A.CallTo(() => assetRepository.QueryByHashAsync(appId.Id, "hash")) .Returns(new List { found }); A.CallTo(() => assetEnricher.EnrichAsync(A>.That.IsSameSequenceAs(found), requestContext)) .Returns(new List { enriched }); var result = await sut.QueryByHashAsync(requestContext, appId.Id, "hash"); Assert.Same(enriched, result.Single()); } [Fact] public async Task Should_load_assets_from_ids_and_resolve_tags() { var found1 = new AssetEntity { Id = Guid.NewGuid() }; var found2 = new AssetEntity { Id = Guid.NewGuid() }; var enriched1 = new AssetEntity(); var enriched2 = new AssetEntity(); var ids = HashSet.Of(found1.Id, found2.Id); A.CallTo(() => assetRepository.QueryAsync(appId.Id, A>.That.IsSameSequenceAs(ids))) .Returns(ResultList.CreateFrom(8, found1, found2)); A.CallTo(() => assetEnricher.EnrichAsync(A>.That.IsSameSequenceAs(found1, found2), requestContext)) .Returns(new List { enriched1, enriched2 }); var result = await sut.QueryAsync(requestContext, Q.Empty.WithIds(ids)); Assert.Equal(8, result.Total); Assert.Equal(new[] { enriched1, enriched2 }, result.ToArray()); } [Fact] public async Task Should_load_assets_with_query_and_resolve_tags() { var found1 = new AssetEntity { Id = Guid.NewGuid() }; var found2 = new AssetEntity { Id = Guid.NewGuid() }; var enriched1 = new AssetEntity(); var enriched2 = new AssetEntity(); A.CallTo(() => assetRepository.QueryAsync(appId.Id, A.Ignored)) .Returns(ResultList.CreateFrom(8, found1, found2)); A.CallTo(() => assetEnricher.EnrichAsync(A>.That.IsSameSequenceAs(found1, found2), requestContext)) .Returns(new List { enriched1, enriched2 }); var result = await sut.QueryAsync(requestContext, Q.Empty); Assert.Equal(8, result.Total); Assert.Equal(new[] { enriched1, enriched2 }, result.ToArray()); } } }