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
#![allow(non_camel_case_types)]
#[macro_use]
extern crate bitflags;
extern crate libc;
mod integrations;
mod types;
mod utils;
pub use integrations::*;
pub use types::*;
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn result_is_ok_for_any_flags() {
let af_flags_combs = (0..AddressFamilyFlags::all().bits() + 1)
.filter_map(|x| AddressFamilyFlags::from_bits(x))
.collect::<Vec<AddressFamilyFlags>>();
let proto_flags_combs = (0..ProtocolFlags::all().bits() + 1)
.filter_map(|x| ProtocolFlags::from_bits(x))
.collect::<Vec<ProtocolFlags>>();
for af_flags in af_flags_combs.iter() {
for proto_flags in proto_flags_combs.iter() {
assert!(get_sockets_info(*af_flags, *proto_flags).is_ok());
}
}
}
#[test]
fn result_is_empty_for_empty_flags() {
let sockets_info =
get_sockets_info(AddressFamilyFlags::empty(), ProtocolFlags::empty()).unwrap();
assert!(sockets_info.len() == 0);
}
#[test]
fn api_is_consistent() {
let af_flags = AddressFamilyFlags::all();
let proto_flags = ProtocolFlags::all();
assert_eq!(
Result_(get_sockets_info(af_flags, proto_flags)),
Result_(flatten_result(
iterate_sockets_info(af_flags, proto_flags).map(|r| r.collect())
))
);
#[derive(Debug)]
struct Result_(Result<Vec<SocketInfo>, Error>);
impl PartialEq<Result_> for Result_ {
fn eq(&self, other: &Result_) -> bool {
match (&self.0, &other.0) {
(Ok(a), Ok(b)) => a.len() == b.len(),
_ => false,
}
}
}
fn flatten_result<T, E>(r: Result<Result<T, E>, E>) -> Result<T, E> {
match r {
Ok(Ok(x)) => Ok(x),
Ok(Err(e)) => Err(e),
Err(e) => Err(e),
}
}
}
}