Minecraft-Console-Client/MinecraftClient.Tests/RestartCoordinatorTests.cs

namespace MinecraftClient.Tests;

public sealed class RestartCoordinatorTests
{
    [Fact]
    public async Task ReplacesQueuedSameAttemptAndQueuesNewerAttempt()
    {
        var firstStarted = new TaskCompletionSource(TaskCreationOptions.RunContinuationsAsynchronously);
        var releaseFirst = new TaskCompletionSource(TaskCreationOptions.RunContinuationsAsynchronously);
        var replacementCompleted = new TaskCompletionSource(TaskCreationOptions.RunContinuationsAsynchronously);
        var secondCompleted = new TaskCompletionSource(TaskCreationOptions.RunContinuationsAsynchronously);
        var executedAccounts = new List<string>();

        using var coordinator = new RestartCoordinator(
            async (request, cancellationToken) =>
            {
                if (request.ConnectionAttempt == 9)
                {
                    firstStarted.SetResult();
                    await releaseFirst.Task.WaitAsync(cancellationToken);
                }
                else if (request.ConnectionAttempt == 10)
                {
                    executedAccounts.Add(request.SettingsSnapshot?.Account.Login ?? string.Empty);
                    replacementCompleted.SetResult();
                }
                else if (request.ConnectionAttempt == 11)
                {
                    secondCompleted.SetResult();
                }
            },
            exception => throw new Xunit.Sdk.XunitException(exception.ToString()));

        Assert.True(coordinator.TrySchedule(new RestartRequest(9, TimeSpan.Zero, true)));
        await firstStarted.Task.WaitAsync(TimeSpan.FromSeconds(5));

        Assert.True(coordinator.TrySchedule(new RestartRequest(10, TimeSpan.Zero, true, CreateSettingsSnapshot("first"))));
        Assert.True(coordinator.TrySchedule(new RestartRequest(10, TimeSpan.Zero, true, CreateSettingsSnapshot("replacement"))));
        Assert.True(coordinator.TrySchedule(new RestartRequest(11, TimeSpan.Zero, true)));
        Assert.True(coordinator.HasScheduledRestart(11));

        releaseFirst.SetResult();
        await replacementCompleted.Task.WaitAsync(TimeSpan.FromSeconds(5));
        await secondCompleted.Task.WaitAsync(TimeSpan.FromSeconds(5));

        Assert.Equal(["replacement"], executedAccounts);
    }

    [Fact]
    public void RejectsStaleAttempt()
    {
        using var coordinator = new RestartCoordinator(
            (_, _) => Task.CompletedTask,
            exception => throw new Xunit.Sdk.XunitException(exception.ToString()));

        Assert.True(coordinator.TrySchedule(new RestartRequest(20, TimeSpan.Zero, true)));
        Assert.False(coordinator.TrySchedule(new RestartRequest(19, TimeSpan.Zero, true)));
    }

    [Fact]
    public async Task RejectsCompletedAttempt()
    {
        var completed = new TaskCompletionSource(TaskCreationOptions.RunContinuationsAsynchronously);
        using var coordinator = new RestartCoordinator(
            (_, _) =>
            {
                completed.SetResult();
                return Task.CompletedTask;
            },
            exception => throw new Xunit.Sdk.XunitException(exception.ToString()));

        Assert.True(coordinator.TrySchedule(new RestartRequest(20, TimeSpan.Zero, true)));
        await completed.Task.WaitAsync(TimeSpan.FromSeconds(5));
        Assert.True(SpinWait.SpinUntil(() => !coordinator.HasScheduledRestart(20), TimeSpan.FromSeconds(5)));

        Assert.False(coordinator.TrySchedule(new RestartRequest(20, TimeSpan.Zero, true)));
    }

    [Fact]
    public void TerminalStopRejectsFurtherRestarts()
    {
        using var coordinator = new RestartCoordinator(
            (_, _) => Task.CompletedTask,
            exception => throw new Xunit.Sdk.XunitException(exception.ToString()));

        coordinator.Stop();

        Assert.False(coordinator.TrySchedule(new RestartRequest(1, TimeSpan.Zero, true)));
        Assert.False(coordinator.HasScheduledRestart(1));
    }

    private static RestartSettingsSnapshot CreateSettingsSnapshot(string account)
    {
        return new RestartSettingsSnapshot(
            new Settings.MainConfigHelper.MainConfig.AccountInfoConfig(account, "-"),
            "localhost",
            25565);
    }
}