RackPeek/Tests/Hardware/RemoveDriveUseCaseTests.cs

using NSubstitute;
using RackPeek.Domain.Resources.Hardware;
using RackPeek.Domain.Resources.Hardware.Models;
using RackPeek.Domain.Resources.Hardware.Server.Drive;
using Xunit;

namespace Tests.Hardware;

public class RemoveDriveUseCaseTests
{
    [Fact]
    public async Task ExecuteAsync_Removes_drive_when_index_is_valid()
    {
        // Arrange
        var repo = Substitute.For<IHardwareRepository>();
        var server = new Server
        {
            Name = "node01",
            Drives = new List<Drive>
            {
                new Drive { Type = "NVMe", Size = 2000 },
                new Drive { Type = "SATA", Size = 500 }
            }
        };

        // ❗ FIXED: return the server, not null
        repo.GetByNameAsync("node01").Returns(server);

        var sut = new RemoveDriveUseCase(repo);

        // Act
        await sut.ExecuteAsync("node01", index: 0);

        // Assert
        Assert.Single(server.Drives);
        Assert.Equal("SATA", server.Drives[0].Type);

        await repo.Received(1).UpdateAsync(Arg.Is<Server>(s =>
            s.Name == "node01" &&
            s.Drives.Count == 1 &&
            s.Drives[0].Type == "SATA"
        ));
    }

    [Fact]
    public async Task ExecuteAsync_Throws_if_index_out_of_range()
    {
        // Arrange
        var repo = Substitute.For<IHardwareRepository>();
        var server = new Server
        {
            Name = "node01",
            Drives = new List<Drive>
            {
                new Drive { Type = "NVMe", Size = 2000 }
            }
        };

        repo.GetByNameAsync("node01").Returns(server);

        var sut = new RemoveDriveUseCase(repo);

        // Act & Assert
        await Assert.ThrowsAsync<ArgumentOutOfRangeException>(async () =>
            await sut.ExecuteAsync("node01", index: 1)
        );

        await repo.DidNotReceive().UpdateAsync(Arg.Any<Server>());
    }

    [Fact]
    public async Task ExecuteAsync_Does_nothing_when_server_does_not_exist()
    {
        // Arrange
        var repo = Substitute.For<IHardwareRepository>();

        
        repo.GetByNameAsync("node01")
            .Returns((RackPeek.Domain.Resources.Hardware.Models.Hardware?)null);

        var sut = new RemoveDriveUseCase(repo);

        // Act
        await sut.ExecuteAsync("node01", index: 0);

        // Assert
        await repo.DidNotReceive().UpdateAsync(Arg.Any<Server>());
    }
}