test: Fix BootMem page alignment issues.

Signed-off-by: TunaCici <[email protected]>
TunaCici · Jan 14, 2024 · 23c1cc6 · 23c1cc6
1 parent f263225
commit 23c1cc6
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/Tests/MemoryBootMem.cpp b/Tests/MemoryBootMem.cpp
@@ -10,7 +10,7 @@ extern "C" {
 
 TEST(MemoryBootMem, init)
 {
-        uint8_t *playground = new uint8_t[BM_ARENA_SIZE * PAGE_SIZE];
+        uint8_t *playground = new uint8_t[(BM_ARENA_SIZE + 1) * PAGE_SIZE];
         for (auto i = 0; i < BM_ARENA_SIZE * PAGE_SIZE; i++) {
                 playground[i] = 0;
         }
@@ -21,11 +21,13 @@ TEST(MemoryBootMem, init)
 
 TEST(MemoryBootMem, alloc)
 {
-        uint8_t *playground = new uint8_t[BM_ARENA_SIZE * PAGE_SIZE];
+        uint8_t *playground = new uint8_t[(BM_ARENA_SIZE + 1) * PAGE_SIZE];
         for (auto i = 0; i < BM_ARENA_SIZE * PAGE_SIZE; i++) {
                 playground[i] = 0;
         }
 
+        playground = (uint8_t*) PALIGN(playground);
+
         uint32_t pageCount = bootmem_init((void*) playground);
         EXPECT_EQ(pageCount, BM_ARENA_SIZE);
 
@@ -54,4 +56,10 @@ TEST(MemoryBootMem, alloc)
         /* no more space;( */
         tmp = bootmem_alloc(1);
         EXPECT_EQ(tmp, (void*) 0);
+
+        /* addresses are valid? */
+        for (auto i = 0; i < BM_ARENA_SIZE * PAGE_SIZE; i++) {
+                playground[i] = 42;
+                EXPECT_EQ(playground[i], 42);
+        }
 }