运行货物测试时，如何抑制多个线程的输出

Question

如果我用cargo test执行以下测试用例，one_thread_test的输出将被抑制，如文档中所述。

但是，multi_thread_test的输出将显示在stdout上。有可能匹配单线程和多线程测试用例的行为吗？

#[test]
fn one_thread_test() {
println!("A");
println!("B");
}
#[test]
fn multi_thread_test() {
use std::thread;
let mut threads = vec![];
for _ in 0..100 {
let t = thread::spawn(move || {
println!("from thread");
});
threads.push(t);
}
for thread in threads {
thread.join().unwrap();
}
}

Answer 1

这里有一个快速而肮脏的解决方法。

它的工作方式是将消息发送到主线程中结构所拥有的接收器。接收器在丢弃时打印所有累积的消息——这一点很重要，这样断言失败引起的恐慌就不会阻止打印。

use std::sync::mpsc::{channel, Sender, Receiver};
struct TestPrinter {
receiver: Receiver<String>,
sender: Sender<String>,
}
impl TestPrinter {
fn new() -> TestPrinter {
let (sender, receiver) = channel();
TestPrinter { receiver, sender }
}
fn sender(&self) -> Sender<String> {
self.sender.clone()
}
}
impl Drop for TestPrinter {
fn drop(&mut self) {
while let Some(v) = self.receiver.try_recv().ok() {
println!("later: {}", v);
}
}
}

和一个方便的宏，所以感觉就像调用println!:

macro_rules! myprint {
($send: expr, $($arg:tt)*) => { 
(*&$send).send(format!($($arg)*));
};
}  

为了发送要打印的消息，每个线程都有一个sender：

#[test]
fn multi_thread_test() {
use std::thread;
let mut threads = vec![];
let printer = TestPrinter::new();
for _ in 0..100 {
let sender = printer.sender();
let t = thread::spawn(move || {
myprint!(sender, "from thread");
});
threads.push(t);
}
for thread in threads {
thread.join().unwrap();
}
}

实际打印发生在printer超出范围时。它在主线程中，因此在成功测试期间不会打印，除非指定了--nocapture。