unit-testing - 我应该在 Rust 的什么地方放置测试实用程序函数？

Question

我有以下代码定义了可以放置生成的文件的路径:

fn gen_test_dir() -> tempdir::TempDir {                                        
    tempdir::TempDir::new_in(Path::new("/tmp"), "filesyncer-tests").unwrap()   
} 

此函数在 tests/lib.rs 中定义，用于该文件中的测试，我也想在位于 src/lib.rs 的单元测试中使用它。

这是否可以在不将实用程序函数编译为非测试二进制文件且不复制代码的情况下实现？

Answer 1

您可以从您的 #[cfg(test)] 模块导入其他 #[cfg(test)] 模块，例如，在 main.rs 或在其他模块中，您可以执行以下操作:

#[cfg(test)]
pub mod test_util {
    pub fn return_two() -> usize { 2 }
}

然后从您项目中的任何其他地方:

#[cfg(test)]
mod test {
    use crate::test_util::return_two;

    #[test]
    fn test_return_two() {
        assert_eq!(return_two(), 2);
    }
}