使用StdoutLock输出的内容无法被cargo test捕获，如何避免测试时打印？

你遇到的问题是：使用std::io::stdout().lock()保证多线程日志原子性时，输出会绕过cargo test的捕获机制直接打印到终端，而println!的输出则能被正常捕获。

原因 #

cargo test会重定向标准输出流来捕获测试中的输出，但直接调用lock()获取的是操作系统级别的原始stdout句柄，这个句柄不受测试框架的重定向影响，因此会直接输出到终端。

方法一：条件编译，测试时丢弃输出 #

通过Rust的条件编译属性，让生产环境使用stdout lock保证原子性，测试环境将输出导向sink（一个会丢弃所有写入内容的Writer）。

修改后的代码：

use std::io::{self, sink, Write as _};

pub fn print() {
    println!("Hello, world! (print)");
}

pub fn write() {
    // 生产环境使用stdout lock保证原子性
    #[cfg(not(test))]
    let mut handle = io::stdout().lock();
    // 测试环境丢弃所有输出
    #[cfg(test)]
    let mut handle = sink();

    writeln!(&mut handle, "Hello, world! (write)").unwrap();
    handle.flush().unwrap();
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_print() {
        print();
    }

    #[test]
    fn test_write() {
        write();
    }
}

运行cargo test --lib后，write()的输出会被完全丢弃，不会显示在终端中。

方法二：抽象日志输出目标（适合扩展） #

如果后续需要支持更多输出方式（比如文件日志），可以抽象一个日志写入器的Trait，分别实现生产和测试环境的版本：

use std::io::{self, sink, Write as _};

trait LogWriter: Write {
    fn lock(&mut self) -> Box<dyn Write + '_>;
}

// 生产环境的stdout写入器
struct StdoutWriter;

impl Write for StdoutWriter {
    fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
        io::stdout().write(buf)
    }

    fn flush(&mut self) -> io::Result<()> {
        io::stdout().flush()
    }
}

impl LogWriter for StdoutWriter {
    fn lock(&mut self) -> Box<dyn Write + '_> {
        Box::new(io::stdout().lock())
    }
}

// 测试环境的丢弃写入器
struct TestWriter;

impl Write for TestWriter {
    fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
        sink().write(buf)
    }

    fn flush(&mut self) -> io::Result<()> {
        sink().flush()
    }
}

impl LogWriter for TestWriter {
    fn lock(&mut self) -> Box<dyn Write + '_> {
        Box::new(sink())
    }
}

// 根据环境返回对应写入器
fn get_log_writer() -> Box<dyn LogWriter> {
    #[cfg(not(test))]
    Box::new(StdoutWriter)
    #[cfg(test)]
    Box::new(TestWriter)
}

pub fn write() {
    let mut writer = get_log_writer();
    let mut handle = writer.lock();
    writeln!(&mut handle, "Hello, world! (write)").unwrap();
    handle.flush().unwrap();
}

pub fn print() {
    println!("Hello, world! (print)");
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_print() {
        print();
    }

    #[test]
    fn test_write() {
        write();
    }
}

这种方式可以灵活扩展日志的输出目标，同时保持测试环境的输出隔离。

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