kernel/tests: don't rely on vector for fake memory

1 files changed, 12 insertions, 3 deletions

diff --git a/kernel/include/kernel/test_support/simulated_memory.hpp b/kernel/include/kernel/test_support/simulated_memory.hpp
index fed9f43..9a391d8 100644
--- a/kernel/include/kernel/test_support/simulated_memory.hpp
+++ b/kernel/include/kernel/test_support/simulated_memory.hpp
@@ -1,10 +1,11 @@
 #ifndef TEACHOS_KERNEL_TEST_SUPPORT_SIMULATED_MEMORY_HPP
 #define TEACHOS_KERNEL_TEST_SUPPORT_SIMULATED_MEMORY_HPP
 
+#include "kapi/memory.hpp"
+
 #include <kstd/units>
 
 #include <cstddef>
-#include <vector>
 
 namespace kernel::tests
 {
@@ -13,13 +14,21 @@ namespace kernel::tests
   {
     explicit simulated_memory(kstd::units::bytes size);
 
+    ~simulated_memory();
+
     auto clear() -> void;
 
-    auto ram_base() noexcept -> std::byte *;
+    [[nodiscard]] auto ram_base() noexcept -> std::byte *;
     [[nodiscard]] auto ram_base() const noexcept -> std::byte const *;
+    [[nodiscard]] auto heap_base() const noexcept -> kapi::memory::linear_address;
+    [[nodiscard]] auto heap_size() const noexcept -> kstd::units::bytes;
+    [[nodiscard]] auto memory_descriptor() const noexcept -> int;
 
   private:
-    std::vector<std::byte> m_memory;
+    int m_memory_descriptor{};
+    kstd::units::bytes m_size{0};
+    std::byte * m_physical_base{nullptr};
+    std::byte * m_virtual_base{nullptr};
   };
 
 }  // namespace kernel::tests