RackPeek/Tests/HardwareResources/Systems/RemoveSystemDriveUseCaseTests.cs

using System.ComponentModel.DataAnnotations;
using NSubstitute;
using RackPeek.Domain.Helpers;
using RackPeek.Domain.Resources.Hardware.Models;
using RackPeek.Domain.Resources.SystemResources;
using RackPeek.Domain.Resources.SystemResources.UseCases;

namespace Tests.HardwareResources.Systems;

public class RemoveSystemDriveUseCaseTests
{
    [Fact]
    public async Task ExecuteAsync_Removes_drive_when_found()
    {
        // Arrange
        var repo = Substitute.For<ISystemRepository>();
        var drive = new Drive { Type = "ssd", Size = 256 };
        var system = new SystemResource { Name = "sys1", Drives = new List<Drive> { drive } };

        repo.GetByNameAsync("sys1").Returns(system);

        var sut = new RemoveSystemDriveUseCase(repo);

        // Act
        await sut.ExecuteAsync("sys1", 0);

        // Assert
        Assert.Empty(system.Drives);

        await repo.Received(1).UpdateAsync(system);
    }

    [Fact]
    public async Task ExecuteAsync_Throws_when_system_not_found()
    {
        var repo = Substitute.For<ISystemRepository>();
        repo.GetByNameAsync("sys1").Returns((SystemResource?)null);

        var sut = new RemoveSystemDriveUseCase(repo);

        await Assert.ThrowsAsync<NotFoundException>(() => sut.ExecuteAsync("sys1", 0));
    }

    [Fact]
    public async Task ExecuteAsync_Throws_when_drive_not_found()
    {
        var repo = Substitute.For<ISystemRepository>();
        var system = new SystemResource { Name = "sys1", Drives = new List<Drive>() };

        repo.GetByNameAsync("sys1").Returns(system);

        var sut = new RemoveSystemDriveUseCase(repo);

        await Assert.ThrowsAsync<NotFoundException>(() => sut.ExecuteAsync("sys1", 0));
    }

    [Fact]
    public async Task ExecuteAsync_Throws_when_index_invalid()
    {
        var repo = Substitute.For<ISystemRepository>();

        repo.GetByNameAsync("sys1")
            .Returns(new SystemResource
            {
                Name = "sys1", Drives = new List<Drive> { new Drive { Type = "ssd", Size = 256 } }
            });

        var sut = new RemoveSystemDriveUseCase(repo);

        await Assert.ThrowsAsync<NotFoundException>(() => sut.ExecuteAsync("sys1", -1));
    }
}