From ff38c0740de468be86468119033b2fb9e3d84d5b Mon Sep 17 00:00:00 2001
From: 魏曹先生 <1992414357@qq.com>
Date: Mon, 22 Sep 2025 18:40:34 +0800
Subject: Add test for `env`

---
 crates/env/env_test/Cargo.toml |  5 +++++
 crates/env/env_test/src/lib.rs | 24 ++++++++++++++++++++++++
 2 files changed, 29 insertions(+)

(limited to 'crates/env/env_test')

diff --git a/crates/env/env_test/Cargo.toml b/crates/env/env_test/Cargo.toml
index 5e63c74..e4a2d14 100644
--- a/crates/env/env_test/Cargo.toml
+++ b/crates/env/env_test/Cargo.toml
@@ -4,4 +4,9 @@ edition = "2024"
 version.workspace = true
 
 [dependencies]
+tcp_connection = { path = "../../utils/tcp_connection" }
+cfg_file = { path = "../../utils/cfg_file", features = ["default"] }
 env = { path = "../../env" }
+
+# Async & Networking
+tokio = { version = "1.46.1", features = ["full"] }
diff --git a/crates/env/env_test/src/lib.rs b/crates/env/env_test/src/lib.rs
index 8b13789..7c9fbcb 100644
--- a/crates/env/env_test/src/lib.rs
+++ b/crates/env/env_test/src/lib.rs
@@ -1 +1,25 @@
+use std::{
+    env::{current_dir, set_current_dir},
+    path::PathBuf,
+};
 
+use tokio::fs;
+
+#[cfg(test)]
+pub mod test_vault_setup_and_member_register;
+
+#[cfg(test)]
+pub mod test_virtual_file_creation_and_update;
+
+pub async fn get_and_correct_test_dir(area: &str) -> Result<PathBuf, std::io::Error> {
+    let dir = current_dir()?.join(".temp").join("test").join(area);
+    if !dir.exists() {
+        std::fs::create_dir_all(&dir)?;
+    } else {
+        // Regenerate existing directory
+        fs::remove_dir_all(&dir).await?;
+        fs::create_dir_all(&dir).await?;
+    }
+    set_current_dir(dir.clone())?;
+    Ok(dir)
+}
-- 
cgit