persist files in sim memory io for integrity check

simulator/runner/io.rs

@@ -79,6 +79,11 @@ impl SimIO for SimulatorIO {
     fn close_files(&self) {
         self.files.borrow_mut().clear()
     }
+
+    fn persist_files(&self) -> anyhow::Result<()> {
+        // Files are persisted automatically
+        Ok(())
+    }
 }
 
 impl Clock for SimulatorIO {

simulator/runner/memory/io.rs

@@ -190,6 +190,17 @@ impl SimIO for MemorySimIO {
             file.closed.set(true);
         }
     }
+
+    fn persist_files(&self) -> anyhow::Result<()> {
+        let files = self.files.borrow();
+        for (file_path, file) in files.iter() {
+            if file_path.ends_with(".db") || file_path.ends_with("wal") || file_path.ends_with("lg")
+            {
+                std::fs::write(file_path, &*file.buffer.borrow())?;
+            }
+        }
+        Ok(())
+    }
 }
 
 impl Clock for MemorySimIO {

simulator/runner/mod.rs

@@ -20,4 +20,6 @@ pub trait SimIO: turso_core::IO {
     fn syncing(&self) -> bool;
 
     fn close_files(&self);
+
+    fn persist_files(&self) -> anyhow::Result<()>;
 }