RobustToolbox/Robust.Shared.Tests/Collections/OverflowQueue_Test.cs

using NUnit.Framework;
using Robust.Shared.Collections;

namespace Robust.Shared.Tests.Collections;

[Parallelizable(ParallelScope.All | ParallelScope.Fixtures)]
[TestFixture, TestOf(typeof(OverflowQueue<>))]
internal sealed class OverflowQueue_Test
{
    private static IEnumerable<(int size, int iterations)> TestParams => new[]
    {
        (10, 17),
        (10, 4)
    };

    [Test]
    public void TestQueue([ValueSource(nameof(TestParams))] (int size, int iterations) test)
    {
        var queue = new OverflowQueue<int>(test.size);

        Assert.That(queue.Contains(0), Is.False);

        for (int i = 0; i < test.iterations; i++)
        {
            queue.Enqueue(i);
        }

        var overlap = Math.Max(test.iterations - test.size, 0);
        for (int i = overlap; i < test.iterations; i++)
        {
            Assert.That(queue.Contains(i));
        }

        Assert.That(queue.Contains(test.iterations-1));
        Assert.That(queue.Contains(-1), Is.False);

        Assert.That(queue.Peek(), Is.EqualTo(overlap));
        var array = queue.ToArray();
        Assert.That(array.Length, Is.EqualTo(Math.Min(test.size, test.iterations)));
        for (var i = 0; i < array.Length; i++)
        {
            Assert.That(array[i], Is.EqualTo(i + overlap));
        }

        for (int i = overlap; i < test.iterations; i++)
        {
            Assert.That(queue.TryDequeue(out var item));
            Assert.That(item, Is.EqualTo(i));
        }

        Assert.Throws<InvalidOperationException>(() => queue.Dequeue());
        Assert.That(queue.TryDequeue(out _), Is.False);
    }
}