Add test to ProtectedModeMemoryCore for tracking malloc exhaustion

and switching to different segments.
mbbsemu · Jan 4, 2022 · 0ef2e06 · 0ef2e06
1 parent 7111b63
commit 0ef2e06
Showing 1 changed file with 19 additions and 2 deletions.
diff --git a/MBBSEmu.Tests/Memory/ProtectedModeMemoryCore_Tests.cs b/MBBSEmu.Tests/Memory/ProtectedModeMemoryCore_Tests.cs
@@ -1,4 +1,5 @@
-using MBBSEmu.DependencyInjection;
+using FluentAssertions;
+using MBBSEmu.DependencyInjection;
 using MBBSEmu.Memory;
 using NLog;
 using System.Text;
@@ -24,8 +25,24 @@ public void EndOfSegmentString()
 
             var stringFromMemory = Encoding.ASCII.GetString((memoryCore as IMemoryCore).GetString(1, testStringOffset, stripNull: false));
 
-            Assert.Equal(testString, stringFromMemory);
+            stringFromMemory.Should().Be(testString);
+        }
+
+        [Fact]
+        public void MultiSegmentAllocation()
+        {
+            var memoryCore = new ProtectedModeMemoryCore(_logger);
+            var data1 = memoryCore.Malloc(0xFF00);
+            data1.Should().NotBeNull();
+
+            var data2 = memoryCore.Malloc(0xFF00);
+            data2.Should().NotBeNull();
+            data2.Segment.Should().NotBe(data1.Segment);
 
+            var data3 = memoryCore.Malloc(0xFF00);
+            data3.Should().NotBeNull();
+            data3.Segment.Should().NotBe(data2.Segment);
+            data3.Segment.Should().NotBe(data1.Segment);
         }
     }
 }