SilverMetal/windows/welcome/tests/SilverOS.Welcome.Tests/ApplyStepTests.cs

using Bunit;
using Moq;
using Microsoft.AspNetCore.Components;
using Microsoft.Extensions.DependencyInjection;
using SilverOS.Welcome.App.Components;
using SilverOS.Welcome.App.Components.Steps;
using SilverOS.Welcome.Core.Apply;
using SilverOS.Welcome.Core.Flavours;
using Xunit;

public class ApplyStepTests : TestContext
{
    [Fact]
    public async Task Calls_apply_with_the_wizard_selections()
    {
        var apply = new Mock<IApplyService>();
        var state = new WizardState
        {
            Flavour = new FlavourManifest { Id = "daily-driver", Hardening = new() { Modules = new[] { "00" } } },
            Username = "alice",
            Password = "pw",
            AdminPassword = "apw",
            BitLockerPin = "123456"
        };
        Services.AddSingleton(state);
        Services.AddSingleton(apply.Object);
        var cut = RenderComponent<ApplyStep>();
        await cut.InvokeAsync(() => cut.Instance.StartAsync());
        apply.Verify(a => a.RunAsync(
            It.Is<ApplyRequest>(r => r.Username == "alice" && r.Flavour.Id == "daily-driver"),
            It.IsAny<IProgress<ApplyProgress>>(),
            It.IsAny<CancellationToken>()), Times.Once);
    }

    [Fact]
    public async Task OnComplete_invoked_when_apply_succeeds()
    {
        var apply = new Mock<IApplyService>();
        apply.Setup(a => a.RunAsync(It.IsAny<ApplyRequest>(), It.IsAny<IProgress<ApplyProgress>>(), It.IsAny<CancellationToken>()))
             .Returns(Task.CompletedTask);
        var state = new WizardState
        {
            Flavour = new FlavourManifest { Id = "daily-driver", Hardening = new() { Modules = new[] { "00" } } },
            Username = "alice", Password = "pw", AdminPassword = "apw", BitLockerPin = "123456"
        };
        Services.AddSingleton(state);
        Services.AddSingleton(apply.Object);
        var completed = false;
        var cut = RenderComponent<ApplyStep>(p => p.Add(s => s.OnComplete, EventCallback.Factory.Create(this, () => { completed = true; })));
        await cut.InvokeAsync(() => cut.Instance.StartAsync());
        Assert.True(completed);
    }

    [Fact]
    public async Task Shows_error_and_retry_button_when_apply_fails()
    {
        var apply = new Mock<IApplyService>();
        apply.Setup(a => a.RunAsync(It.IsAny<ApplyRequest>(), It.IsAny<IProgress<ApplyProgress>>(), It.IsAny<CancellationToken>()))
             .ThrowsAsync(new InvalidOperationException("Module 03 failed"));
        var state = new WizardState
        {
            Flavour = new FlavourManifest { Id = "daily-driver", Hardening = new() { Modules = new[] { "00" } } },
            Username = "alice", Password = "pw", AdminPassword = "apw", BitLockerPin = "123456"
        };
        Services.AddSingleton(state);
        Services.AddSingleton(apply.Object);
        var cut = RenderComponent<ApplyStep>();
        await cut.InvokeAsync(() => cut.Instance.StartAsync());
        Assert.Contains("Module 03 failed", cut.Markup);
        Assert.NotNull(cut.Find(".btn-retry"));
    }
}