even-perfect-numbers-scanner/PerfectNumbers.Core.Tests/GpuWorkLimiterTests.cs

using System.Diagnostics;
using FluentAssertions;
using PerfectNumbers.Core.Gpu;
using Xunit;

namespace PerfectNumbers.Core.Tests;

[Collection("GpuNtt")]
public class GpuWorkLimiterTests
{
    [Fact]
    [Trait("Category", "Fast")]
    public async Task Acquire_blocks_when_limit_reached()
    {
        GpuWorkLimiter.SetLimit(1);

        var startedSecond = new TaskCompletionSource<bool>();
        var sw = new Stopwatch();

        var first = GpuWorkLimiter.Acquire();
        sw.Start();

        var secondTask = Task.Run(() =>
        {
            var second = GpuWorkLimiter.Acquire();
            try
            {
                startedSecond.TrySetResult(true);
            }
            finally
            {
                second.Dispose();
            }
        });

        try
        {
            await Task.Delay(50);
            startedSecond.Task.IsCompleted.Should().BeFalse();
        }
        finally
        {
            first.Dispose();
        }

        await secondTask.WaitAsync(TimeSpan.FromSeconds(1));
        sw.Stop();

        sw.ElapsedMilliseconds.Should().BeGreaterThanOrEqualTo(50);
    }
}