squidex/backend/tests/Squidex.Domain.Apps.Entities.Tests/Contents/MongoDb/ContentsQueryFixture.cs

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

using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using FakeItEasy;
using LoremNET;
using MongoDB.Bson;
using MongoDB.Driver;
using Newtonsoft.Json;
using Squidex.Domain.Apps.Core;
using Squidex.Domain.Apps.Core.Contents;
using Squidex.Domain.Apps.Core.Schemas;
using Squidex.Domain.Apps.Core.TestHelpers;
using Squidex.Domain.Apps.Entities.Apps;
using Squidex.Domain.Apps.Entities.MongoDb.Contents;
using Squidex.Domain.Apps.Entities.Schemas;
using Squidex.Domain.Apps.Entities.TestHelpers;
using Squidex.Infrastructure;
using Squidex.Infrastructure.Json.Objects;
using Squidex.Infrastructure.MongoDb;

namespace Squidex.Domain.Apps.Entities.Contents.MongoDb
{
    public sealed class ContentsQueryFixture
    {
        private readonly Random random = new Random();
        private readonly int numValues = 10000;
        private readonly IMongoClient mongoClient = new MongoClient("mongodb://localhost");
        private readonly IMongoDatabase mongoDatabase;

        public MongoContentRepository ContentRepository { get; }

        public NamedId<DomainId>[] AppIds { get; } =
        {
            NamedId.Of(DomainId.Create("3b5ba909-e5a5-4858-9d0d-df4ff922d452"), "my-app1"),
            NamedId.Of(DomainId.Create("4b3672c1-97c6-4e0b-a067-71e9e9a29db9"), "my-app1")
        };

        public NamedId<DomainId>[] SchemaIds { get; } =
        {
            NamedId.Of(DomainId.Create("3b5ba909-e5a5-4858-9d0d-df4ff922d452"), "my-schema1"),
            NamedId.Of(DomainId.Create("4b3672c1-97c6-4e0b-a067-71e9e9a29db9"), "my-schema2"),
            NamedId.Of(DomainId.Create("76357c9b-0514-4377-9fcc-a632e7ef960d"), "my-schema3"),
            NamedId.Of(DomainId.Create("164c451e-e5a8-41f8-8aaf-e4b56603d7e7"), "my-schema4"),
            NamedId.Of(DomainId.Create("741e902c-fdfa-41ad-8e5a-b7cb9d6e3d94"), "my-schema5")
        };

        public ContentsQueryFixture()
        {
            mongoDatabase = mongoClient.GetDatabase("QueryTests");

            SetupJson();

            var appProvider = CreateAppProvider();

            ContentRepository =
                new MongoContentRepository(
                    mongoDatabase,
                    appProvider, false);

            Task.Run(async () =>
            {
                await Task.WhenAll(
                    SetupAsync(ContentRepository, mongoDatabase));
            }).Wait();
        }

        private async Task SetupAsync(MongoContentRepository contentRepository, IMongoDatabase database)
        {
            await contentRepository.InitializeAsync();

            await database.RunCommandAsync<BsonDocument>("{ profile : 0 }");
            await database.DropCollectionAsync("system.profile");

            var collections = contentRepository.GetInternalCollections();

            foreach (var collection in collections)
            {
                var contentCount = await collection.Find(new BsonDocument()).CountDocumentsAsync();

                if (contentCount == 0)
                {
                    var batch = new List<MongoContentEntity>();

                    async Task ExecuteBatchAsync(MongoContentEntity? entity)
                    {
                        if (entity != null)
                        {
                            batch.Add(entity);
                        }

                        if ((entity == null || batch.Count >= 1000) && batch.Count > 0)
                        {
                            await collection.InsertManyAsync(batch);

                            batch.Clear();
                        }
                    }

                    foreach (var appId in AppIds)
                    {
                        foreach (var schemaId in SchemaIds)
                        {
                            for (var i = 0; i < numValues; i++)
                            {
                                var data =
                                    new ContentData()
                                        .AddField("field1",
                                            new ContentFieldData()
                                                .AddInvariant(JsonValue.Create(i)))
                                        .AddField("field2",
                                            new ContentFieldData()
                                                .AddInvariant(JsonValue.Create(Lorem.Paragraph(200, 20))));

                                var content = new MongoContentEntity
                                {
                                    DocumentId = DomainId.NewGuid(),
                                    AppId = appId,
                                    Data = data,
                                    IndexedAppId = appId.Id,
                                    IndexedSchemaId = schemaId.Id,
                                    IsDeleted = false,
                                    SchemaId = schemaId,
                                    Status = Status.Published
                                };

                                await ExecuteBatchAsync(content);
                            }
                        }
                    }

                    await ExecuteBatchAsync(null);
                }
            }

            await database.RunCommandAsync<BsonDocument>("{ profile : 2 }");
        }

        private static IAppProvider CreateAppProvider()
        {
            var appProvider = A.Fake<IAppProvider>();

            A.CallTo(() => appProvider.GetSchemaAsync(A<DomainId>._, A<DomainId>._, false))
                .ReturnsLazily(x => Task.FromResult<ISchemaEntity?>(CreateSchema(x.GetArgument<DomainId>(0)!, x.GetArgument<DomainId>(1)!)));

            return appProvider;
        }

        private static void SetupJson()
        {
            var jsonSerializer = JsonSerializer.Create(TestUtils.DefaultSerializerSettings);

            BsonJsonConvention.Register(jsonSerializer);
        }

        public DomainId RandomAppId()
        {
            return AppIds[random.Next(0, AppIds.Length)].Id;
        }

        public IAppEntity RandomApp()
        {
            return CreateApp(RandomAppId());
        }

        public DomainId RandomSchemaId()
        {
            return SchemaIds[random.Next(0, SchemaIds.Length)].Id;
        }

        public ISchemaEntity RandomSchema()
        {
            return CreateSchema(RandomAppId(), RandomSchemaId());
        }

        public string RandomValue()
        {
            return random.Next(0, numValues).ToString();
        }

        private static IAppEntity CreateApp(DomainId appId)
        {
            return Mocks.App(NamedId.Of(appId, "my-app"));
        }

        private static ISchemaEntity CreateSchema(DomainId appId, DomainId schemaId)
        {
            var schemaDef =
                new Schema("my-schema")
                    .AddField(Fields.Number(1, "value", Partitioning.Invariant));

            var schema =
                Mocks.Schema(
                    NamedId.Of(appId, "my-app"),
                    NamedId.Of(schemaId, "my-schema"),
                    schemaDef);

            return schema;
        }
    }
}