adas-core/adas-core.Test/Repositories/DischargeRepositoryTest.cs

using adas_core.Domain.Models;
using adas_core.Domain.Models.AppSettings;
using adas_core.Domain.Models.MongoModels;
using adas_core.Infrastructure.Repositories;
using adas_core.Test.Utilities;
using Microsoft.Extensions.Options;
using MongoDB.Bson;
using Options = Microsoft.Extensions.Options.Options;

namespace adas_core.Test.Repositories;

[TestFixture]
[Category("Integration")]
public class DischargeRepositoryTest
{
    [OneTimeSetUp]
    public async Task Init()
    {
        // Initialize options and repository
        _optionsApiSettings = Options.Create(_apiSettings);

        await IntegrationDb.Database.DropCollectionAsync("discharge");
        await IntegrationDb.Database.CreateCollectionAsync("discharge");

        _dischargeRepository = new DischargeRepository(_optionsApiSettings, IntegrationDb.Database);
    }

    private DischargeRepository _dischargeRepository;
    private readonly ApiSettings _apiSettings = new();
    private IOptions<ApiSettings> _optionsApiSettings;

    [Test]
    public async Task InsertOneAsync_ShouldInsertDischarge()
    {
        // Arrange
        var discharge = TestUtilities.CreateValidDischarge();

        // Act
        await _dischargeRepository.InsertOneAsync(discharge);

        var result = await _dischargeRepository.FindById(discharge.Id);
        Assert.That(result, Is.Not.Null);
    }

    [Test]
    public async Task Delete_ShouldDeleteDischarge()
    {
        // Arrange
        var discharge = TestUtilities.CreateValidDischarge();
        await _dischargeRepository.InsertOneAsync(discharge);

        // Act
        await _dischargeRepository.Delete(discharge.Id);

        var result = await _dischargeRepository.FindById(discharge.Id);
        Assert.That(result, Is.Null);
    }

    [Test]
    public async Task Update_DischargeSuccessfullyUpdated()
    {
        // Arrange
        var pocId = new ObjectId();
        var discharge = TestUtilities.CreateValidDischarge();
        discharge.PointOfCareId = pocId;
        await _dischargeRepository.InsertOneAsync(discharge);

        // Act
        discharge.Service = "UpdatedService";
        await _dischargeRepository.Update(discharge);

        // Retrieve the discharge from the database
        var updatedDischarge = await _dischargeRepository.GetDischargeByPointOfCareId(pocId);

        // Assert
        Assert.That(updatedDischarge, Is.Not.Null);
        Assert.That(updatedDischarge?.Service, Is.EqualTo("UpdatedService"));
    }

    // [Test]
    // public async Task UpdateUnit_UnitNameSuccessfullyUpdated()
    // {
    //     // Arrange
    //     var discharge = TestUtilities.CreateValidDischarge();
    //     await _dischargeRepository.InsertOneAsync(discharge);
    //
    //     // Act
    //     var newUnitName = "NewUnitName";
    //     await _dischargeRepository.UpdateUnit(discharge.Id, newUnitName);
    //
    //     // Retrieve the discharge from the database
    //     var updatedDischarge = await _dischargeRepository.FindById(discharge.Id);
    //
    //     // Assert
    //     Assert.That(updatedDischarge, Is.Not.Null);
    //     Assert.That(updatedDischarge?.PatientLocation, Is.Not.Null);
    //     Assert.That(updatedDischarge?.PatientLocation?.UnitName, Is.EqualTo(newUnitName));
    // }

    // [Test]
    // public async Task UpdatePatient_PatientSuccessfullyUpdated()
    // {
    //     // Arrange
    //     var discharge = TestUtilities.CreateValidDischarge();
    //     await _dischargeRepository.InsertOneAsync(discharge);
    //
    //     // Create a new patient object with updated information
    //     var updatedPatient = TestUtilities.CreateValidPatient();
    //
    //     // Act
    //     await _dischargeRepository.UpdatePatient(discharge.Id, updatedPatient);
    //
    //     // Retrieve the discharge from the database
    //     var updatedDischarge = await _dischargeRepository.FindById(discharge.Id);
    //
    //     // Assert
    //     Assert.That(updatedDischarge, Is.Not.Null);
    //     Assert.That(updatedDischarge?.Patient, Is.Not.Null);
    //     Assert.That(updatedDischarge?.Patient?.Id, Is.EqualTo(updatedPatient.Id));
    // }

    [Test]
    public async Task FindAll_ReturnsAllDischarges()
    {
        // Arrange: Prepare test data
        var expectedDischarges = new List<Discharge>
        {
            new() { Id = ObjectId.GenerateNewId(), DischargeDate = DateTime.UtcNow },
            new() { Id = ObjectId.GenerateNewId(), DischargeDate = DateTime.UtcNow },
            new() { Id = ObjectId.GenerateNewId(), DischargeDate = DateTime.UtcNow }
        };

        await _dischargeRepository.InsertManyAsync(expectedDischarges);

        // Act: Call the method under test
        var actualDischarges = await _dischargeRepository.FindAll();

        var discharges = actualDischarges.ToList();
        Assert.That(discharges, Is.Not.Null);
        Assert.That(discharges, Is.Not.Empty);
        Assert.That(discharges.Count(), Is.EqualTo(expectedDischarges.Count()));
    }

    [Test]
    public async Task FindById_ExistingId_ReturnsDischarge()
    {
        var id = ObjectId.GenerateNewId();
        var expectedDischarge = new Discharge { Id = id, DischargeDate = DateTime.UtcNow };
        await _dischargeRepository.InsertOneAsync(expectedDischarge);

        var actualDischarge = await _dischargeRepository.FindById(id);

        Assert.That(actualDischarge, Is.Not.Null);
        Assert.That(actualDischarge?.Id, Is.EqualTo(expectedDischarge.Id));
    }

    [Test]
    public async Task FindById_NonExistentId_ReturnsNull()
    {
        var id = ObjectId.GenerateNewId();

        var actualDischarge = await _dischargeRepository.FindById(id);

        Assert.That(actualDischarge, Is.Null);
    }

    // [Test]
    // public async Task FindByUnit_ExistingUnit_ReturnsDischarges()
    // {
    //     // Arrange: Prepare test data
    //     var unitName = "TestUnit";
    //     var expectedDischarges =
    //             new Discharge
    //                 { Id = ObjectId.GenerateNewId(), PatientLocation = new PatientLocation { UnitName = unitName } };
    //     await _dischargeRepository.InsertOneAsync(expectedDischarges);
    //
    //     // Act: Call the method under test
    //     var actualDischarges = await _dischargeRepository.FindByUnit(unitName);
    //
    //     // Assert: Verify the result
    //     Assert.That(actualDischarges, Is.Not.Null);
    //     Assert.That(actualDischarges?.ToList().First().PatientLocation?.UnitName, Is.EquivalentTo(expectedDischarges.PatientLocation.UnitName));
    // }

    [Test]
    public async Task FindByDestination_ExistingDestination_ReturnsDischarges()
    {
        // Arrange: Prepare test data
        var destination = "TestDestination";
        var expectedDischarges =
            new Discharge
                { Id = ObjectId.GenerateNewId(), Destination = destination };
        await _dischargeRepository.InsertOneAsync(expectedDischarges);

        // Act: Call the method under test
        var actualDischarges = await _dischargeRepository.FindByDestination(destination);

        // Assert: Verify the result
        var discharges = actualDischarges?.ToList() ?? [];
        Assert.That(discharges, Is.Not.Null);
        Assert.That(discharges.First().Id, Is.EqualTo(expectedDischarges.Id));
    }

    [Test]
    public async Task FindByDestination_NonExistentDestination_ReturnsEmptyList()
    {
        // Arrange: Use a non-existent destination
        var destination = "NonExistentDestination";

        // Act: Call the method under test
        var actualDischarges = await _dischargeRepository.FindByDestination(destination);

        // Assert: Verify that the result is an empty list
        Assert.That(actualDischarges, Is.Empty);
    }

    [Test]
    public async Task FindByPoCId_ExistingPoCId_ReturnsDischarges()
    {
        // Arrange: Prepare test data
        var pocId = ObjectId.GenerateNewId();
        var expectedDischarges =
            new Discharge
                { Id = ObjectId.GenerateNewId(), PointOfCareId = pocId };
        await _dischargeRepository.InsertOneAsync(expectedDischarges);

        // Act: Call the method under test
        var actualDischarges = await _dischargeRepository.FindByPoCId(pocId);

        // Assert: Verify the result
        var discharges = actualDischarges?.ToList() ?? [];
        Assert.That(discharges, Is.Not.Null);
        Assert.That(discharges.First().Id, Is.EqualTo(expectedDischarges.Id));
    }

    [Test]
    public async Task FindByPoCId_NonExistentPoCId_ReturnsEmptyList()
    {
        // Arrange: Use a non-existent PoCId
        var pocId = ObjectId.GenerateNewId();

        // Act: Call the method under test
        var actualDischarges = await _dischargeRepository.FindByPoCId(pocId);

        // Assert: Verify that the result is an empty list
        Assert.That(actualDischarges, Is.Empty);
    }

    [Test]
    public async Task FindByService_ExistingService_ReturnsDischarges()
    {
        // Arrange: Prepare test data
        var service = "TestService";
        var expectedDischarges =
            new Discharge
                { Id = ObjectId.GenerateNewId(), Service = service };
        await _dischargeRepository.InsertOneAsync(expectedDischarges);

        // Act: Call the method under test
        var actualDischarges = await _dischargeRepository.FindByService(service);

        // Assert: Verify the result
        var discharges = actualDischarges?.ToList() ?? [];
        Assert.That(discharges, Is.Not.Null);
        Assert.That(discharges.First().Id, Is.EqualTo(expectedDischarges.Id));
    }

    [Test]
    public async Task FindByService_NonExistentService_ReturnsEmptyList()
    {
        // Arrange: Use a non-existent service
        var service = "NonExistentService";

        // Act: Call the method under test
        var actualDischarges = await _dischargeRepository.FindByService(service);

        // Assert: Verify that the result is an empty list
        Assert.That(actualDischarges, Is.Empty);
    }

    // [Test]
    // public async Task GetDischargeByLocation_ExistingLocation_ReturnsDischarge()
    // {
    //     // Arrange: Prepare test data
    //     var location = new PatientLocation("TestUnit", "TestBed", "TestRoom");
    //     var expectedDischarges =
    //         new Discharge
    //             { Id = ObjectId.GenerateNewId(), PatientLocation = location};
    //     await _dischargeRepository.InsertOneAsync(expectedDischarges);
    //
    //     // Act: Call the method under test
    //     var actualDischarge = await _dischargeRepository.GetDischargeByLocation(location);
    //
    //     // Assert: Verify the result
    //     Assert.That(actualDischarge, Is.Not.Null);
    //     Assert.That(actualDischarge?.Id, Is.EqualTo(expectedDischarges.Id));
    // }

    [Test]
    public async Task GetDischargeByLocation_NonExistentLocation_ReturnsNull()
    {
        // Arrange: Use a non-existent location
        var location = new PatientLocation("NonExistentUnit", "NonExistentBed", "NonExistentRoom");

        // Act: Call the method under test
        var actualDischarge = await _dischargeRepository.GetDischargeByLocation(location);

        // Assert: Verify that the result is null
        Assert.That(actualDischarge, Is.Null);
    }

    [Test]
    public async Task GetDischargeByPointOfCareId_ExistingId_ReturnsDischarge()
    {
        // Arrange: Prepare test data
        var expectedId = ObjectId.GenerateNewId();
        var expectedDischarge =
            new Discharge
                { Id = ObjectId.GenerateNewId(), PointOfCareId = expectedId };
        await _dischargeRepository.InsertOneAsync(expectedDischarge);

        // Act: Call the method under test
        var actualDischarge = await _dischargeRepository.GetDischargeByPointOfCareId(expectedId);

        // Assert: Verify the result
        Assert.That(actualDischarge, Is.Not.Null);
        Assert.That(actualDischarge?.Id, Is.EqualTo(expectedDischarge.Id));
    }

    [Test]
    public async Task GetDischargeByPointOfCareId_NonExistentId_ReturnsNull()
    {
        // Arrange: Use a non-existent ID
        var nonExistentId = ObjectId.GenerateNewId();

        // Act: Call the method under test
        var actualDischarge = await _dischargeRepository.GetDischargeByPointOfCareId(nonExistentId);

        // Assert: Verify that the result is null
        Assert.That(actualDischarge, Is.Null);
    }

    [Test]
    public async Task GetByPatientId_ExistingPatientId_ReturnsDischarge()
    {
        // Arrange: Prepare test data
        var expectedId = ObjectId.GenerateNewId();
        var expectedDischarge =
            new Discharge
                { Id = ObjectId.GenerateNewId(), PatientId = expectedId };
        await _dischargeRepository.InsertOneAsync(expectedDischarge);

        // Act: Call the method under test
        var actualDischarge = await _dischargeRepository.GetByPatientId(expectedId);

        // Assert: Verify the result
        Assert.That(actualDischarge, Is.Not.Null);
        Assert.That(actualDischarge?.Id, Is.EqualTo(expectedDischarge.Id));
    }

    [Test]
    public async Task GetByPatientId_NonExistentPatientId_ReturnsNull()
    {
        // Arrange: Use a non-existent patient ID
        var nonExistentId = ObjectId.GenerateNewId();

        // Act: Call the method under test
        var actualDischarge = await _dischargeRepository.GetByPatientId(nonExistentId);

        // Assert: Verify that the result is null
        Assert.That(actualDischarge, Is.Null);
    }
}