some more tests

Author: wasm-forge

Cargo.toml

@@ -19,3 +19,4 @@ ciborium = "0.2.2"
 [dev-dependencies]
 candid = "0.10.14"
 pocket-ic = "9.0.2"
+

src/fs_tests.rs

@@ -930,10 +930,10 @@ mod tests {
 
             fs.unmount_memory_file(file_name).unwrap();
 
+            // init new memory into a file
             fs.mount_memory_file(file_name, Box::new(memory2.clone()))
                 .unwrap();
 
-            // init new memory into a file
             fs.init_memory_file(file_name).unwrap();
 
             let mut buf1 = [0u8; 10];
@@ -948,6 +948,66 @@ mod tests {
         }
     }
 
+    #[test]
+    fn test_mounts() {
+        let dir_fd = 3;
+
+        for mut fs in test_fs_setups("") {
+            let memory = new_vector_memory();
+            let file_name = "file.txt";
+            let hello_message = "Hello host".to_string();
+            let hello_message2 = "1234Hello from regular file".to_string();
+
+            fs.mount_memory_file(file_name, Box::new(memory.clone()))
+                .unwrap();
+
+            // write something into a host memory file
+            write_text_file(&mut fs, dir_fd, file_name, &hello_message, 1).unwrap();
+
+            // the memory should contain the file now
+            let v: Vec<u8> = memory.borrow().clone();
+            assert_eq!(&v[0..hello_message.len()], hello_message.as_bytes());
+
+            let str = read_text_file(&mut fs, dir_fd, file_name, 0, 1000);
+            assert_eq!(str, hello_message);
+
+            // unmount file, the file.txt should become empty
+            fs.unmount_memory_file(file_name).unwrap();
+            let str = read_text_file(&mut fs, dir_fd, file_name, 0, 1000);
+            assert_eq!(str, "".to_string());
+
+            // mount again, the old content should recover
+            fs.mount_memory_file(file_name, Box::new(memory.clone()))
+                .unwrap();
+            let str = read_text_file(&mut fs, dir_fd, file_name, 0, 1000);
+            assert_eq!(str, hello_message);
+
+            // store mounted contents into the host file, check the host file content is renewed
+            fs.store_memory_file(file_name).unwrap();
+            fs.unmount_memory_file(file_name).unwrap();
+            let str = read_text_file(&mut fs, dir_fd, file_name, 0, 1000);
+            assert_eq!(str, hello_message);
+
+            // write some other content message,
+            // check there is a new content now
+            write_text_file(&mut fs, dir_fd, file_name, &hello_message2, 1).unwrap();
+
+            let str = read_text_file(&mut fs, dir_fd, file_name, 0, 1000);
+            assert_eq!(str, hello_message2);
+
+            // after mounting, we still have the old content
+            fs.mount_memory_file(file_name, Box::new(memory.clone()))
+                .unwrap();
+            let str = read_text_file(&mut fs, dir_fd, file_name, 0, 1000);
+            assert_eq!(str, hello_message);
+
+            // initializing should recover the data from the host file to the mounted memory
+            fs.init_memory_file(file_name).unwrap();
+            let str = read_text_file(&mut fs, dir_fd, file_name, 0, 1000);
+            assert_eq!(str, hello_message2);
+        }
+    }
+
     #[test]
     fn writing_from_different_file_descriptors() {
         for mut fs in test_fs_setups("f1/f2/text.txt") {

src/runtime.rs

@@ -2,4 +2,5 @@ pub(crate) mod dir;
 pub(crate) mod fd;
 pub(crate) mod file;
 pub(crate) mod structure_helpers;
+
 pub mod types;

src/test_utils.rs

@@ -93,7 +93,9 @@ pub fn write_text_file(
 
     let file_fd = fs.open(parent_fd, path, FdStat::default(), OpenFlags::CREATE, 0)?;
 
-    write_text_fd(fs, file_fd, content, times)
+    write_text_fd(fs, file_fd, content, times)?;
+
+    fs.close(file_fd)
 }
 
 #[cfg(test)]