143 lines
4.1 KiB
C#
143 lines
4.1 KiB
C#
using System;
|
|
using NUnit.Framework;
|
|
using Unity.Burst;
|
|
using Unity.Collections;
|
|
using Unity.Collections.LowLevel.Unsafe;
|
|
using Unity.Jobs;
|
|
|
|
internal class UnsafeRingQueueTests
|
|
{
|
|
[Test]
|
|
public void UnsafeRingQueue_Enqueue_Dequeue()
|
|
{
|
|
var test = new UnsafeRingQueue<int>(16, Allocator.Persistent);
|
|
|
|
Assert.AreEqual(0, test.Length);
|
|
|
|
int item;
|
|
Assert.False(test.TryDequeue(out item));
|
|
|
|
test.Enqueue(123);
|
|
Assert.AreEqual(1, test.Length);
|
|
|
|
Assert.True(test.TryEnqueue(456));
|
|
Assert.AreEqual(2, test.Length);
|
|
|
|
Assert.True(test.TryDequeue(out item));
|
|
Assert.AreEqual(123, item);
|
|
Assert.AreEqual(1, test.Length);
|
|
|
|
Assert.AreEqual(456, test.Dequeue());
|
|
Assert.AreEqual(0, test.Length);
|
|
|
|
test.Dispose();
|
|
}
|
|
|
|
[Test]
|
|
public unsafe void UnsafeRingQueue_Enqueue_Dequeue_View()
|
|
{
|
|
var list = new UnsafeList<int>(16, Allocator.Persistent);
|
|
list.Length = 16;
|
|
var test = new UnsafeRingQueue<int>(list.Ptr, list.Length);
|
|
|
|
Assert.AreEqual(0, test.Length);
|
|
|
|
int item;
|
|
Assert.False(test.TryDequeue(out item));
|
|
|
|
test.Enqueue(123);
|
|
Assert.AreEqual(1, test.Length);
|
|
|
|
Assert.True(test.TryEnqueue(456));
|
|
Assert.AreEqual(2, test.Length);
|
|
|
|
Assert.True(test.TryDequeue(out item));
|
|
Assert.AreEqual(123, item);
|
|
Assert.AreEqual(1, test.Length);
|
|
|
|
Assert.AreEqual(456, test.Dequeue());
|
|
Assert.AreEqual(0, test.Length);
|
|
|
|
test.Dispose();
|
|
list.Dispose();
|
|
}
|
|
|
|
[Test]
|
|
public void UnsafeRingQueue_Throws()
|
|
{
|
|
using (var test = new UnsafeRingQueue<int>(1, Allocator.Persistent))
|
|
{
|
|
Assert.Throws<InvalidOperationException>(() => { test.Dequeue(); });
|
|
|
|
Assert.DoesNotThrow(() => { test.Enqueue(123); });
|
|
Assert.Throws<InvalidOperationException>(() => { test.Enqueue(456); });
|
|
|
|
int item = 0;
|
|
Assert.DoesNotThrow(() => { item = test.Dequeue(); });
|
|
Assert.AreEqual(123, item);
|
|
|
|
Assert.DoesNotThrow(() => { test.Enqueue(456); });
|
|
Assert.DoesNotThrow(() => { item = test.Dequeue(); });
|
|
Assert.AreEqual(456, item);
|
|
|
|
Assert.Throws<InvalidOperationException>(() => { test.Dequeue(); });
|
|
}
|
|
}
|
|
|
|
[Test]
|
|
public void UnsafeRingQueue_CustomAllocatorTest()
|
|
{
|
|
AllocatorManager.Initialize();
|
|
var allocatorHelper = new AllocatorHelper<CustomAllocatorTests.CountingAllocator>(AllocatorManager.Persistent);
|
|
ref var allocator = ref allocatorHelper.Allocator;
|
|
allocator.Initialize();
|
|
|
|
using (var container = new UnsafeRingQueue<int>(1, allocator.Handle))
|
|
{
|
|
}
|
|
|
|
Assert.IsTrue(allocator.WasUsed);
|
|
allocator.Dispose();
|
|
allocatorHelper.Dispose();
|
|
AllocatorManager.Shutdown();
|
|
}
|
|
|
|
[BurstCompile]
|
|
struct BurstedCustomAllocatorJob : IJob
|
|
{
|
|
[NativeDisableUnsafePtrRestriction]
|
|
public unsafe CustomAllocatorTests.CountingAllocator* Allocator;
|
|
|
|
public void Execute()
|
|
{
|
|
unsafe
|
|
{
|
|
using (var container = new UnsafeRingQueue<int>(1, Allocator->Handle))
|
|
{
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
[Test]
|
|
public unsafe void UnsafeRingQueue_BurstedCustomAllocatorTest()
|
|
{
|
|
AllocatorManager.Initialize();
|
|
var allocatorHelper = new AllocatorHelper<CustomAllocatorTests.CountingAllocator>(AllocatorManager.Persistent);
|
|
ref var allocator = ref allocatorHelper.Allocator;
|
|
allocator.Initialize();
|
|
|
|
var allocatorPtr = (CustomAllocatorTests.CountingAllocator*)UnsafeUtility.AddressOf<CustomAllocatorTests.CountingAllocator>(ref allocator);
|
|
unsafe
|
|
{
|
|
var handle = new BurstedCustomAllocatorJob {Allocator = allocatorPtr}.Schedule();
|
|
handle.Complete();
|
|
}
|
|
|
|
Assert.IsTrue(allocator.WasUsed);
|
|
allocator.Dispose();
|
|
allocatorHelper.Dispose();
|
|
AllocatorManager.Shutdown();
|
|
}
|
|
}
|