using System;
using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using System.Threading;
using System.Threading.Tasks;
using AutoFixture;
using CoreUnitTests.Infrastructure.Model;
using FluentAssertions;
using MongoDB.Driver;
using MongoDbGenericRepository;
using Moq;
using Xunit;

namespace CoreUnitTests.DataAccessTests.MongoDbReaderTests;

public class GetAllAsyncTests : BaseReaderTests
{
    [Fact]
    public async Task WithFilter_GetsMatchingDocuments()
    {
        // Arrange
        var collection = MockOf<IMongoCollection<TestDocument>>();
        var documents = Fixture.CreateMany<TestDocument>().ToList();
        Expression<Func<TestDocument, bool>> filter = x => x.Id == documents[0].Id;
        var (context, cursor) = SetupAsyncGet(documents, collection);


        // Act
        var result = await Sut.GetAllAsync<TestDocument, Guid>(filter);

        // Assert
        context.Verify(x => x.GetCollection<TestDocument>(null), Times.Once);
        cursor.Verify(x => x.Current, Times.Exactly(documents.Count));
        cursor.Verify(x => x.MoveNextAsync(CancellationToken.None), Times.Exactly(documents.Count + 1));
        result.Should().NotBeNull();
        result.Should().OnlyContain(x => documents.Contains(x));
    }

    [Fact]
    public async Task WithFilterAndCancellationToken_GetsMatchingDocuments()
    {
        // Arrange
        var collection = MockOf<IMongoCollection<TestDocument>>();
        var documents = Fixture.CreateMany<TestDocument>().ToList();
        var token = new CancellationToken(false);
        Expression<Func<TestDocument, bool>> filter = x => x.Id == documents[0].Id;
        var (context, cursor) = SetupAsyncGet(documents, collection);

        // Act
        var result = await Sut.GetAllAsync<TestDocument, Guid>(filter, cancellationToken: token);

        // Assert
        context.Verify(x => x.GetCollection<TestDocument>(null), Times.Once);
        cursor.Verify(x => x.Current, Times.Exactly(documents.Count));
        cursor.Verify(x => x.MoveNextAsync(token), Times.Exactly(documents.Count + 1));
        result.Should().NotBeNull();
        result.Should().OnlyContain(x => documents.Contains(x));
    }

    [Fact]
    public async Task WithFilterAndPartitionKey_GetsMatchingDocuments()
    {
        // Arrange
        var collection = MockOf<IMongoCollection<TestDocument>>();
        var documents = Fixture.CreateMany<TestDocument>().ToList();
        var partitionKey = Fixture.Create<string>();
        Expression<Func<TestDocument, bool>> filter = x => x.Id == documents[0].Id;
        var (context, cursor) = SetupAsyncGet(documents, collection, partitionKey);

        // Act
        var result = await Sut.GetAllAsync<TestDocument, Guid>(filter, partitionKey);

        // Assert
        context.Verify(x => x.GetCollection<TestDocument>(partitionKey), Times.Once);
        cursor.Verify(x => x.Current, Times.Exactly(documents.Count));
        cursor.Verify(x => x.MoveNextAsync(CancellationToken.None), Times.Exactly(documents.Count + 1));
        result.Should().NotBeNull();
        result.Should().OnlyContain(x => documents.Contains(x));
    }

    [Fact]
    public async Task WithFilterAndPartitionKeyAndCancellationToken_GetsMatchingDocuments()
    {
        // Arrange
        var collection = MockOf<IMongoCollection<TestDocument>>();
        var documents = Fixture.CreateMany<TestDocument>().ToList();
        var partitionKey = Fixture.Create<string>();
        var token = new CancellationToken(false);
        Expression<Func<TestDocument, bool>> filter = x => x.Id == documents[0].Id;
        var (context, cursor) = SetupAsyncGet(documents, collection, partitionKey);

        // Act
        var result = await Sut.GetAllAsync<TestDocument, Guid>(filter, partitionKey, token);

        // Assert
        context.Verify(x => x.GetCollection<TestDocument>(partitionKey), Times.Once);
        cursor.Verify(x => x.Current, Times.Exactly(documents.Count));
        cursor.Verify(x => x.MoveNextAsync(token), Times.Exactly(documents.Count + 1));
        result.Should().NotBeNull();
        result.Should().OnlyContain(x => documents.Contains(x));
    }

    [Fact]
    public async Task WithCondition_GetsMatchingDocuments()
    {
        // Arrange
        var collection = MockOf<IMongoCollection<TestDocument>>();
        var documents = Fixture.CreateMany<TestDocument>().ToList();
        var condition = Builders<TestDocument>.Filter.Eq("Id", documents[0].Id);
        var (context, cursor) = SetupAsyncGet(documents, collection);

        // Act
        var result = await Sut.GetAllAsync<TestDocument, Guid>(condition);

        // Assert
        context.Verify(x => x.GetCollection<TestDocument>(null), Times.Once);
        cursor.Verify(x => x.Current, Times.Exactly(documents.Count));
        cursor.Verify(x => x.MoveNextAsync(CancellationToken.None), Times.Exactly(documents.Count + 1));
        result.Should().NotBeNull();
        result.Should().OnlyContain(x => documents.Contains(x));
    }

    [Fact]
    public async Task WithConditionAndCancellationToken_GetsMatchingDocuments()
    {
        // Arrange
        var collection = MockOf<IMongoCollection<TestDocument>>();
        var documents = Fixture.CreateMany<TestDocument>().ToList();
        var condition = Builders<TestDocument>.Filter.Eq("Id", documents[0].Id);
        var token = new CancellationToken(false);
        var (context, cursor) = SetupAsyncGet(documents, collection);

        // Act
        var result = await Sut.GetAllAsync<TestDocument, Guid>(condition, cancellationToken: token);

        // Assert
        context.Verify(x => x.GetCollection<TestDocument>(null), Times.Once);
        cursor.Verify(x => x.Current, Times.Exactly(documents.Count));
        cursor.Verify(x => x.MoveNextAsync(token), Times.Exactly(documents.Count + 1));
        result.Should().NotBeNull();
        result.Should().OnlyContain(x => documents.Contains(x));
    }

    [Fact]
    public async Task WithConditionAndPartitionKey_GetsMatchingDocuments()
    {
        // Arrange
        var collection = MockOf<IMongoCollection<TestDocument>>();
        var documents = Fixture.CreateMany<TestDocument>().ToList();
        var condition = Builders<TestDocument>.Filter.Eq("Id", documents[0].Id);
        var partitionKey = Fixture.Create<string>();
        var (context, cursor) = SetupAsyncGet(documents, collection, partitionKey);

        // Act
        var result = await Sut.GetAllAsync<TestDocument, Guid>(condition, partitionKey: partitionKey);

        // Assert
        context.Verify(x => x.GetCollection<TestDocument>(partitionKey), Times.Once);
        cursor.Verify(x => x.MoveNextAsync(CancellationToken.None), Times.Exactly(documents.Count + 1));
        cursor.Verify(x => x.Current, Times.Exactly(documents.Count));
        result.Should().NotBeNull();
        result.Should().OnlyContain(x => documents.Contains(x));
    }

    [Fact]
    public async Task WithConditionAndPartitionKeyAndCancellationToken_GetsMatchingDocuments()
    {
        // Arrange
        var collection = MockOf<IMongoCollection<TestDocument>>();
        var documents = Fixture.CreateMany<TestDocument>().ToList();
        var condition = Builders<TestDocument>.Filter.Eq("Id", documents[0].Id);
        var partitionKey = Fixture.Create<string>();
        var token = new CancellationToken(false);
        var (context, cursor) = SetupAsyncGet(documents, collection, partitionKey);

        // Act
        var result = await Sut.GetAllAsync<TestDocument, Guid>(condition, partitionKey: partitionKey, cancellationToken: token);

        // Assert
        context.Verify(x => x.GetCollection<TestDocument>(partitionKey), Times.Once);
        cursor.Verify(x => x.Current, Times.Exactly(documents.Count));
        cursor.Verify(x => x.MoveNextAsync(token), Times.Exactly(documents.Count + 1));
        result.Should().NotBeNull();
        result.Should().OnlyContain(x => documents.Contains(x));
    }

    [Fact]
    public async Task WithConditionAndFindOptions_GetsMatchingDocuments()
    {
        // Arrange
        var collection = MockOf<IMongoCollection<TestDocument>>();
        var documents = Fixture.CreateMany<TestDocument>().ToList();
        var options = Fixture
            .Build<FindOptions>()
            .Without(x => x.Comment)
            .Without(x => x.Hint)
            .Create();
        var condition = Builders<TestDocument>.Filter.Eq("Id", documents[0].Id);
        var (context, cursor) = SetupAsyncGet(documents, collection);

        // Act
        var result = await Sut.GetAllAsync<TestDocument, Guid>(condition, options);

        // Assert
        context.Verify(x => x.GetCollection<TestDocument>(null), Times.Once);
        cursor.Verify(x => x.Current, Times.Exactly(documents.Count));
        cursor.Verify(x => x.MoveNextAsync(CancellationToken.None), Times.Exactly(documents.Count + 1));
        result.Should().NotBeNull();
        result.Should().OnlyContain(x => documents.Contains(x));
    }

    [Fact]
    public async Task WithConditionAndFindOptionsAndCancellationToken_GetsMatchingDocuments()
    {
        // Arrange
        var collection = MockOf<IMongoCollection<TestDocument>>();
        var documents = Fixture.CreateMany<TestDocument>().ToList();
        var options = Fixture
            .Build<FindOptions>()
            .Without(x => x.Comment)
            .Without(x => x.Hint)
            .Create();
        var condition = Builders<TestDocument>.Filter.Eq("Id", documents[0].Id);
        var token = new CancellationToken(false);
        var (context, cursor) = SetupAsyncGet(documents, collection);

        // Act
        var result = await Sut.GetAllAsync<TestDocument, Guid>(condition, options, cancellationToken: token);

        // Assert
        context.Verify(x => x.GetCollection<TestDocument>(null), Times.Once);
        cursor.Verify(x => x.Current, Times.Exactly(documents.Count));
        cursor.Verify(x => x.MoveNextAsync(token), Times.Exactly(documents.Count + 1));
        result.Should().NotBeNull();
        result.Should().OnlyContain(x => documents.Contains(x));
    }

    [Fact]
    public async Task WithConditionAndFindOptionsAndPartitionKeyAndCancellationToken_GetsMatchingDocuments()
    {
        // Arrange
        var collection = MockOf<IMongoCollection<TestDocument>>();
        var documents = Fixture.CreateMany<TestDocument>().ToList();
        var options = Fixture
            .Build<FindOptions>()
            .Without(x => x.Comment)
            .Without(x => x.Hint)
            .Create();
        var condition = Builders<TestDocument>.Filter.Eq("Id", documents[0].Id);
        var token = new CancellationToken(false);
        var partitionKey = Fixture.Create<string>();
        var (context, cursor) = SetupAsyncGet(documents, collection, partitionKey);

        // Act
        var result = await Sut.GetAllAsync<TestDocument, Guid>(condition, options, partitionKey, token);

        // Assert
        context.Verify(x => x.GetCollection<TestDocument>(partitionKey), Times.Once);
        cursor.Verify(x => x.Current, Times.Exactly(documents.Count));
        cursor.Verify(x => x.MoveNextAsync(token), Times.Exactly(documents.Count + 1));
        result.Should().NotBeNull();
        result.Should().OnlyContain(x => documents.Contains(x));
    }

    private (Mock<IMongoDbContext>, Mock<IAsyncCursor<TDocument>>) SetupAsyncGet<TDocument>(
        List<TDocument> documents,
        Mock<IMongoCollection<TDocument>> collection,
        string partitionKey = null)
    {
        var asyncCursor = SetupAsyncCursor(documents);

        SetupFindAsync(collection, asyncCursor);

        var context = MockOf<IMongoDbContext>();

        context
            .Setup(x => x.GetCollection<TDocument>(partitionKey))
            .Returns(collection.Object);

        return (context, asyncCursor);
    }
}