1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
//! Test utilities for allocating an address for the server and then spawning clients to connect to
//! it.
#![allow(dead_code, unused_macros)]
#[macro_use]
mod eyre;
#[macro_use]
mod namegen;
mod choke;
mod drive;
mod wdt;
mod xorshift;
#[allow(unused_imports)]
pub use {drive::*, eyre::*, namegen::*, xorshift::*};
#[cfg(feature = "tokio")]
pub mod tokio;
use {
color_eyre::eyre::WrapErr,
std::{
fmt::{Arguments, Debug},
io,
sync::Arc,
},
};
fn intvar(nam: &str) -> Option<u32> {
let val = std::env::var(nam).ok()?;
val.trim().parse().ok()
}
pub fn num_clients() -> u32 {
intvar("INTERPROCESS_TEST_NUM_CLIENTS").filter(|n| *n > 0).unwrap_or(80)
}
pub fn num_concurrent_clients() -> u32 {
intvar("INTERPROCESS_TEST_NUM_CONCURRENT_CLIENTS").filter(|n| *n > 0).unwrap_or(6)
}
pub fn test_wrapper(f: impl (FnOnce() -> TestResult) + Send + 'static) -> TestResult {
eyre::install();
self::wdt::run_under_wachdog(f)
}
pub fn message(msg: Option<Arguments<'_>>, server: bool, terminator: Option<char>) -> Box<str> {
let msg = msg.unwrap_or_else(|| format_args!("Message"));
let sc = if server { "server" } else { "client" };
let mut msg = format!("{msg} from {sc}!");
if let Some(t) = terminator {
msg.push(t);
}
msg.into()
}
pub fn listen_and_pick_name<L: Debug, N: Debug + ?Sized, F: FnMut(u32) -> NameResult<N>>(
namegen: &mut NameGen<N, F>,
mut bindfn: impl FnMut(&N) -> io::Result<L>,
) -> TestResult<(Arc<N>, L)> {
use std::io::ErrorKind::*;
let listener = namegen
.find_map(|nm| {
eprintln!("Trying name {nm:?}...");
let nm = match nm {
Ok(ok) => ok,
Err(e) => return Some(Err(e)),
};
let l = match bindfn(&nm) {
Ok(l) => l,
Err(e) if matches!(e.kind(), AddrInUse | PermissionDenied) => {
eprintln!("\"{}\", skipping", e.kind());
return None;
}
Err(e) => return Some(Err(e)),
};
Some(Ok((nm, l)))
})
.unwrap() // Infinite iterator
.context("listener bind failed")?;
eprintln!("Listener successfully created: {listener:#?}");
Ok(listener)
}
|
