检查 Rust 是否正在运行测试构建 [重复]

Question

我可以使用cfg!(debug_assertions) 检查 Rust 项目是否在开发模式下运行，但我想知道如何检查测试是否正在运行。我可以在if 语句中使用类似的测试标志吗？

原因是在运行集成测试时防止数据库写入。

Answer 1

你可以使用cfg!(test):

if cfg!(test) {
    // do test stuff
} else {
    // do non-test stuff
}

但是，这可能会导致代码脆弱。例如，您可能不小心在生产环境中执行了某些代码，即使测试都通过了。在项目级别切换代码通常更健壮，如果你弄错了会导致编译错误：

#[cfg(test)]
impl Write for DB {
    fn write(&mut self, buf: &[u8]) -> Result<usize> {
        // test stuff
    }

    fn flush(&mut self) -> Result<()> {
        // test stuff
    }
}

#[cfg(not(test))]
impl Write for DB {
    fn write(&mut self, buf: &[u8]) -> Result<usize> {
        // actually write
    }

    fn flush(&mut self) -> Result<()> {
        // actually flush
    }
}

如果忘记实现not(test)版本，会出现编译错误。