add mount table and mount tests

2 files changed, 29 insertions, 1 deletions

diff --git a/kernel/src/filesystem/mount.tests.cpp b/kernel/src/filesystem/mount.tests.cpp
index 40b7cbb..c9ff82e 100644
--- a/kernel/src/filesystem/mount.tests.cpp
+++ b/kernel/src/filesystem/mount.tests.cpp
@@ -34,9 +34,10 @@ SCENARIO("Mount construction", "[filesystem][mount]")
         REQUIRE(mount.is_ready_to_unmount());
       }
 
-      THEN("the mount has no parent mount")
+      THEN("the mount has no parent mount and no source mount")
       {
         REQUIRE(mount.parent_mount() == nullptr);
+        REQUIRE(mount.source_mount() == nullptr);
       }
     }
 
diff --git a/kernel/src/filesystem/mount_table.tests.cpp b/kernel/src/filesystem/mount_table.tests.cpp
index 8118e19..19b47b2 100644
--- a/kernel/src/filesystem/mount_table.tests.cpp
+++ b/kernel/src/filesystem/mount_table.tests.cpp
@@ -155,3 +155,30 @@ SCENARIO("Adding, finding and removing mounts in the mount table", "[filesystem]
     }
   }
 }
+
+SCENARIO("Mount reference counting", "[filesystem][mount_table]")
+{
+  kernel::filesystem::mount_table table;
+
+  GIVEN("a filesystem and a root dentry")
+  {
+    auto fs = kstd::make_shared<kernel::tests::filesystem::filesystem>();
+    auto root_inode = kstd::make_shared<kernel::tests::filesystem::inode>();
+    auto root_dentry = kstd::make_shared<kernel::filesystem::dentry>(nullptr, root_inode, "/");
+
+    auto source_mount = kstd::make_shared<kernel::filesystem::mount>(root_dentry, root_dentry, fs, nullptr, nullptr);
+    auto mount = kstd::make_shared<kernel::filesystem::mount>(root_dentry, root_dentry, fs, nullptr, source_mount);
+
+    THEN("reference count of source mount is incremented when a mount is added to the mount table and decremented when "
+         "the mount is removed")
+    {
+      REQUIRE(source_mount->ref_count() == 0);
+
+      table.add_mount(mount);
+      REQUIRE(source_mount->ref_count() == 1);
+
+      REQUIRE(table.remove_mount("/") == kernel::filesystem::mount_table::operation_result::removed);
+      REQUIRE(source_mount->ref_count() == 0);
+    }
+  }
+}