using System;
using MongoDB.Driver;
using MongoDbGenericRepository;
using MongoDbGenericRepository.DataAccess.Create;
using MongoDbGenericRepository.DataAccess.Delete;
using MongoDbGenericRepository.DataAccess.Index;
using MongoDbGenericRepository.DataAccess.Read;

namespace CoreUnitTests.Infrastructure;

public class TestKeyedMongoRepository<TKey> : BaseMongoRepository<TKey>
    where TKey : IEquatable<TKey>
{
    public TestKeyedMongoRepository(IMongoDatabase mongoDatabase)
        : base(mongoDatabase)
    {
    }

    public void SetIndexHandler(IMongoDbIndexHandler indexHandler) => MongoDbIndexHandler = indexHandler;

    public void SetDbCreator(IMongoDbCreator creator) => MongoDbCreator = creator;

    public void SetReader(IMongoDbReader reader) => MongoDbReader = reader;

    public void SetEraser(IMongoDbEraser eraser) => MongoDbEraser = eraser;
}