Cleanuparr/code/backend/Cleanuparr.Infrastructure.Tests/Features/Arr/ArrClientFactoryTests.cs

using Cleanuparr.Domain.Enums;
using Cleanuparr.Infrastructure.Features.Arr;
using Cleanuparr.Infrastructure.Features.Arr.Interfaces;
using Moq;
using Xunit;

namespace Cleanuparr.Infrastructure.Tests.Features.Arr;

public class ArrClientFactoryTests
{
    private readonly Mock<ISonarrClient> _sonarrClientMock;
    private readonly Mock<IRadarrClient> _radarrClientMock;
    private readonly Mock<ILidarrClient> _lidarrClientMock;
    private readonly Mock<IReadarrClient> _readarrClientMock;
    private readonly Mock<IWhisparrV2Client> _whisparrClientMock;
    private readonly Mock<IWhisparrV3Client> _whisparrV3ClientMock;
    private readonly ArrClientFactory _factory;

    public ArrClientFactoryTests()
    {
        _sonarrClientMock = new Mock<ISonarrClient>();
        _radarrClientMock = new Mock<IRadarrClient>();
        _lidarrClientMock = new Mock<ILidarrClient>();
        _readarrClientMock = new Mock<IReadarrClient>();
        _whisparrClientMock = new Mock<IWhisparrV2Client>();
        _whisparrV3ClientMock = new Mock<IWhisparrV3Client>();

        _factory = new ArrClientFactory(
            _sonarrClientMock.Object,
            _radarrClientMock.Object,
            _lidarrClientMock.Object,
            _readarrClientMock.Object,
            _whisparrClientMock.Object,
            _whisparrV3ClientMock.Object
        );
    }

    #region GetClient Tests

    [Fact]
    public void GetClient_Sonarr_ReturnsSonarrClient()
    {
        // Act
        var result = _factory.GetClient(InstanceType.Sonarr, 0);

        // Assert
        Assert.Same(_sonarrClientMock.Object, result);
    }

    [Fact]
    public void GetClient_Radarr_ReturnsRadarrClient()
    {
        // Act
        var result = _factory.GetClient(InstanceType.Radarr, 0);

        // Assert
        Assert.Same(_radarrClientMock.Object, result);
    }

    [Fact]
    public void GetClient_Lidarr_ReturnsLidarrClient()
    {
        // Act
        var result = _factory.GetClient(InstanceType.Lidarr, 0);

        // Assert
        Assert.Same(_lidarrClientMock.Object, result);
    }

    [Fact]
    public void GetClient_Readarr_ReturnsReadarrClient()
    {
        // Act
        var result = _factory.GetClient(InstanceType.Readarr, 0);

        // Assert
        Assert.Same(_readarrClientMock.Object, result);
    }

    [Fact]
    public void GetClient_Whisparr_ReturnsWhisparrClient()
    {
        // Act
        var result = _factory.GetClient(InstanceType.Whisparr, 2);

        // Assert
        Assert.Same(_whisparrClientMock.Object, result);
    }

    [Fact]
    public void GetClient_WhisparrV3_ReturnsWhisparrClient()
    {
        // Act
        var result = _factory.GetClient(InstanceType.Whisparr, 3);
        
        // Assert
        Assert.Same(_whisparrV3ClientMock.Object, result);
    }

    [Fact]
    public void GetClient_UnsupportedType_ThrowsNotImplementedException()
    {
        // Arrange
        var unsupportedType = (InstanceType)999;

        // Act & Assert
        var exception = Assert.Throws<NotImplementedException>(() => _factory.GetClient(unsupportedType, It.IsAny<float>()));
        Assert.Contains("not yet supported", exception.Message);
        Assert.Contains("999", exception.Message);
    }

    [Theory]
    [MemberData(nameof(InstancesData))]
    public void GetClient_AllSupportedTypes_ReturnsNonNullClient(InstanceType instanceType, float? version)
    {
        // Act
        var result = _factory.GetClient(instanceType, version ?? 0f);

        // Assert
        Assert.NotNull(result);
        Assert.IsAssignableFrom<IArrClient>(result);
    }

    [Theory]
    [MemberData(nameof(InstancesData))]
    public void GetClient_CalledMultipleTimes_ReturnsSameInstance(InstanceType instanceType, float? version)
    {
        // Act
        var result1 = _factory.GetClient(instanceType, version ?? 0f);
        var result2 = _factory.GetClient(instanceType, version ?? 0f);

        // Assert
        Assert.Same(result1, result2);
    }
    
    public static IEnumerable<object?[]> InstancesData =>
    [
        [InstanceType.Sonarr, null],
        [InstanceType.Radarr, null],
        [InstanceType.Lidarr, null],
        [InstanceType.Readarr, null],
        [InstanceType.Whisparr, 2f],
        [InstanceType.Whisparr, 3f]
    ];

    #endregion
}