Files
ChatMask-Server/tests/registry.rs
T

35 lines
1.3 KiB
Rust
Raw Normal View History

use super::*;
#[test]
fn authenticated_observed_endpoints_expire_and_capacity_and_retry_rates_are_bounded() {
let (a, b) = (PeerId::random(), PeerId::random());
let mut registry = Registry::new(2, 60);
let ep_a = "127.0.0.1:9000".parse().unwrap();
let ep_b = "127.0.0.1:9001".parse().unwrap();
assert!(registry.connect(a, b).is_err());
assert!(
matches!(registry.register(a,ep_a),Response::Registered{endpoint,..} if endpoint==ep_a)
);
registry.register(b, ep_b);
assert!(matches!(
registry.register(PeerId::random(), ep_a),
Response::Error { .. }
));
let (actual_a, actual_b, _) = registry.connect(a, b).unwrap();
assert_eq!((actual_a, actual_b), (ep_a, ep_b));
registry.register(a, ep_a);
assert!(registry.connect(a, b).is_err());
assert!(registry.connect(a, a).is_err());
registry.clients.get_mut(&b).unwrap().seen -= Duration::from_secs(61);
registry.expire();
assert!(registry.connect(a, b).is_err());
registry.remove(a);
assert!(registry.clients.is_empty());
}
#[test]
fn only_quic_addresses_produce_udp_candidates() {
assert!(socket_address(&"/ip4/127.0.0.1/tcp/9000".parse().unwrap()).is_none());
assert_eq!(
socket_address(&"/ip4/127.0.0.1/udp/9000/quic-v1".parse().unwrap()).unwrap(),
"127.0.0.1:9000".parse().unwrap()
);
}