我有一系列测试。有一些测试需要访问共享资源（外部库/API/硬件设备）。如果这些测试中的任何一个并行运行，它们都会失败。
我知道我可以使用 --test-threads=1 运行所有内容，但我发现仅进行几次特殊测试就很不方便。
有什么方法可以让所有测试保持并行运行，并且有一些例外？理想情况下，我想说不要同时运行 X、Y、Z。
【问题讨论】：
Mutex 访问共享资源。
标签： rust rust-cargo
【讨论】：
作为mcarton mentions in the comments，你可以使用Mutex来防止多段代码同时运行：
use once_cell::sync::Lazy; // 1.4.0
use std::{sync::Mutex, thread::sleep, time::Duration};
static THE_RESOURCE: Lazy<Mutex<()>> = Lazy::new(Mutex::default);
type TestResult<T = (), E = Box<dyn std::error::Error>> = std::result::Result<T, E>;
#[test]
fn one() -> TestResult {
let _shared = THE_RESOURCE.lock()?;
eprintln!("Starting test one");
sleep(Duration::from_secs(1));
eprintln!("Finishing test one");
Ok(())
}
#[test]
fn two() -> TestResult {
let _shared = THE_RESOURCE.lock()?;
eprintln!("Starting test two");
sleep(Duration::from_secs(1));
eprintln!("Finishing test two");
Ok(())
}
如果您使用cargo test -- --nocapture 运行，您可以看到行为上的差异：
无锁
running 2 tests
Starting test one
Starting test two
Finishing test two
Finishing test one
test one ... ok
test two ... ok
带锁
running 2 tests
Starting test one
Finishing test one
Starting test two
test one ... ok
Finishing test two
test two ... ok
理想情况下，您应该将外部资源本身放在Mutex 中，以使代码表示它是单例的事实，并且无需记住锁定其他未使用的@987654331 @。
这确实有一个巨大的缺点，即测试中的恐慌（又名assert! 失败）会导致Mutex 中毒。这将导致后续测试无法获取锁。如果您需要避免这种情况并且您知道锁定的资源处于良好状态（() 应该没问题...）您可以处理中毒：
let _shared = THE_RESOURCE.lock().unwrap_or_else(|e| e.into_inner());
如果您需要能够并行运行一组有限的线程，您可以使用信号量。在这里，我使用Condvar 和Mutex 构建了一个糟糕的：
use std::{
sync::{Condvar, Mutex},
thread::sleep,
time::Duration,
};
#[derive(Debug)]
struct Semaphore {
mutex: Mutex<usize>,
condvar: Condvar,
}
impl Semaphore {
fn new(count: usize) -> Self {
Semaphore {
mutex: Mutex::new(count),
condvar: Condvar::new(),
}
}
fn wait(&self) -> TestResult {
let mut count = self.mutex.lock().map_err(|_| "unable to lock")?;
while *count == 0 {
count = self.condvar.wait(count).map_err(|_| "unable to lock")?;
}
*count -= 1;
Ok(())
}
fn signal(&self) -> TestResult {
let mut count = self.mutex.lock().map_err(|_| "unable to lock")?;
*count += 1;
self.condvar.notify_one();
Ok(())
}
fn guarded(&self, f: impl FnOnce() -> TestResult) -> TestResult {
// Not panic-safe!
self.wait()?;
let x = f();
self.signal()?;
x
}
}
lazy_static! {
static ref THE_COUNT: Semaphore = Semaphore::new(4);
}
THE_COUNT.guarded(|| {
eprintln!("Starting test {}", id);
sleep(Duration::from_secs(1));
eprintln!("Finishing test {}", id);
Ok(())
})
另见：
【讨论】：
您始终可以提供自己的测试工具。您可以通过将[[test]] 条目添加到Cargo.toml 来做到这一点：
[[test]]
name = "my_test"
# If your test file is not `tests/my_test.rs`, add this key:
#path = "path/to/my_test.rs"
harness = false
在这种情况下，cargo test 会将my_test.rs 编译为普通的可执行文件。这意味着您必须提供一个main 函数并自己添加所有“运行测试”逻辑。是的，这是一些工作，但至少您可以自己决定运行测试的一切。
您还可以创建两个测试文件：
tests/
- sequential.rs
- parallel.rs
然后您需要运行cargo test --test sequential -- --test-threads=1 和cargo test --test parallel。因此它不适用于单个 cargo test，但您不需要编写自己的测试工具逻辑。
【讨论】：