如何让Rust的fs::remove_dir_all在测试执行时删除指定文件？

问题根源 #

1. 路径不存在触发测试失败：std::fs::remove_dir_all在目标路径不存在时会直接返回NotFound错误，cargo test会将这种错误判定为测试失败（退出码101）。VSCode调试时该目录刚好存在，所以能成功执行。
2. 相对路径的不确定性：你使用的"relative path/.dblog"是相对路径，若cargo test的工作目录与预期不符（比如复杂项目结构或特殊配置），会导致路径解析错误。

解决方案 #

1. 先检查路径存在性，避免无意义报错 #

修改测试函数，先判断目录是否存在，仅在存在时执行删除操作：

use std::fs;
use std::path::Path;

#[test]
pub fn it_deletes() -> std::io::Result<()> {
    let db_log_path = Path::new("relative path/.dblog"); // 替换为你的实际路径
    if db_log_path.exists() {
        fs::remove_dir_all(db_log_path)?;
    }
    Ok(())
}

2. 使用 crate 根目录确保路径正确性 #

用env!("CARGO_MANIFEST_DIR")获取当前 crate 的根目录，拼接出绝对路径，彻底避免相对路径的工作目录问题：

use std::fs;
use std::path::Path;

#[test]
pub fn it_deletes() -> std::io::Result<()> {
    let crate_root = env!("CARGO_MANIFEST_DIR");
    let db_log_path = Path::new(crate_root).join("relative path/.dblog"); // 替换为相对于crate根的路径
    if db_log_path.exists() {
        fs::remove_dir_all(db_log_path)?;
    }
    Ok(())
}

3. 测试前自动执行清理（推荐） #

如果需要在所有测试运行前执行清理，而非单独写测试函数，可以封装清理逻辑，在每个测试开头调用：

use std::fs;
use std::path::Path;

fn cleanup_db_log() -> std::io::Result<()> {
    let crate_root = env!("CARGO_MANIFEST_DIR");
    let db_log_path = Path::new(crate_root).join(".dblog"); // 假设日志目录在crate根目录下
    if db_log_path.exists() {
        fs::remove_dir_all(db_log_path)?;
    }
    Ok(())
}

#[test]
pub fn test_rocksdb_operation() -> std::io::Result<()> {
    cleanup_db_log()?;
    // 此处编写你的测试逻辑
    Ok(())
}

额外排查技巧 #

若不确定路径是否正确，可以在测试中打印当前工作目录，辅助排查：

println!("当前工作目录: {:?}", std::env::current_dir()?);

内容的提问来源于stack exchange，提问作者andrewvious