From 572366e07806bfb6a269c9a132c9afbd2a9191d0 Mon Sep 17 00:00:00 2001
From: antonfirsov <antonfir@gmail.com>
Date: Wed, 10 Apr 2024 23:33:34 +0200
Subject: [PATCH] test allocation limits

---
 ...niformUnmanagedPoolMemoryAllocatorTests.cs | 24 +++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/tests/ImageSharp.Tests/Memory/Allocators/UniformUnmanagedPoolMemoryAllocatorTests.cs b/tests/ImageSharp.Tests/Memory/Allocators/UniformUnmanagedPoolMemoryAllocatorTests.cs
index 6187b536ed..077d0afed8 100644
--- a/tests/ImageSharp.Tests/Memory/Allocators/UniformUnmanagedPoolMemoryAllocatorTests.cs
+++ b/tests/ImageSharp.Tests/Memory/Allocators/UniformUnmanagedPoolMemoryAllocatorTests.cs
@@ -418,4 +418,28 @@ static void RunTest()
             _ = MemoryAllocator.Create();
         }
     }
+
+    [Fact]
+    public void Allocate_OverLimit_ThrowsInvalidMemoryOperationException()
+    {
+        MemoryAllocator allocator = MemoryAllocator.Create(new MemoryAllocatorOptions()
+        {
+            AllocationLimitMegabytes = 4
+        });
+        const int oneMb = 1 << 20;
+        allocator.Allocate<byte>(4 * oneMb).Dispose(); // Should work
+        Assert.Throws<InvalidMemoryOperationException>(() => allocator.Allocate<byte>(5 * oneMb));
+    }
+
+    [Fact]
+    public void AllocateGroup_OverLimit_ThrowsInvalidMemoryOperationException()
+    {
+        MemoryAllocator allocator = MemoryAllocator.Create(new MemoryAllocatorOptions()
+        {
+            AllocationLimitMegabytes = 4
+        });
+        const int oneMb = 1 << 20;
+        allocator.AllocateGroup<byte>(4 * oneMb, 1024).Dispose(); // Should work
+        Assert.Throws<InvalidMemoryOperationException>(() => allocator.AllocateGroup<byte>(5 * oneMb, 1024));
+    }
 }