squidex/backend/tests/Squidex.Infrastructure.Tests/EventSourcing/MongoEventStoreFixture.cs

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

using FakeItEasy;
using MongoDB.Driver;
using Squidex.Infrastructure.MongoDb;
using Squidex.Infrastructure.TestHelpers;
using Xunit;

#pragma warning disable MA0048 // File name must match type name

namespace Squidex.Infrastructure.EventSourcing
{
    public abstract class MongoEventStoreFixture : IAsyncLifetime
    {
        private readonly IMongoClient mongoClient;
        private readonly IMongoDatabase mongoDatabase;
        private readonly IEventNotifier notifier = A.Fake<IEventNotifier>();

        public MongoEventStore EventStore { get; }

        protected MongoEventStoreFixture(string connectionString)
        {
            mongoClient = new MongoClient(connectionString);
            mongoDatabase = mongoClient.GetDatabase(TestConfig.Configuration["mongodb:database"]);

            BsonJsonConvention.Register(TestUtils.DefaultOptions());

            EventStore = new MongoEventStore(mongoDatabase, notifier);
        }

        public Task InitializeAsync()
        {
            return EventStore.InitializeAsync(default);
        }

        public Task DisposeAsync()
        {
            return mongoClient.DropDatabaseAsync(mongoDatabase.DatabaseNamespace.DatabaseName);
        }
    }

    public sealed class MongoEventStoreDirectFixture : MongoEventStoreFixture
    {
        public MongoEventStoreDirectFixture()
            : base(TestConfig.Configuration["mongodb:configuration"])
        {
        }
    }

    public sealed class MongoEventStoreReplicaSetFixture : MongoEventStoreFixture
    {
        public MongoEventStoreReplicaSetFixture()
            : base(TestConfig.Configuration["mongodb:configurationReplica"])
        {
        }
    }
}