108 lines
3.7 KiB
Rust
108 lines
3.7 KiB
Rust
use super::*;
|
|
|
|
#[test]
|
|
fn duplicate_reply_emits_duplicate_event() {
|
|
let params = default_params();
|
|
let server = start_fake_server(move |socket, tx| {
|
|
let (_, peer) = recv_request(&socket, &tx);
|
|
let reply = open_reply(FLAG_OPEN | FLAG_REPLY, TOKEN, ¶ms, None);
|
|
socket.send_to(&reply, peer).unwrap();
|
|
|
|
socket
|
|
.set_read_timeout(Some(Duration::from_secs(2)))
|
|
.unwrap();
|
|
let mut buf = [0_u8; 2048];
|
|
if let Ok((size, _)) = socket.recv_from(&mut buf) {
|
|
tx.send(buf[..size].to_vec()).unwrap();
|
|
let seq = u32::from_le_bytes(buf[12..16].try_into().unwrap());
|
|
let ts = TimestampFields::default();
|
|
let reply_packet = echo_reply_packet(TOKEN, seq, ¶ms, &ts, None);
|
|
socket.send_to(&reply_packet, peer).unwrap();
|
|
thread::sleep(Duration::from_millis(10));
|
|
socket.send_to(&reply_packet, peer).unwrap();
|
|
}
|
|
});
|
|
let config = default_test_config();
|
|
let mut client = Client::connect(server.addr.to_string(), config).unwrap();
|
|
client
|
|
.set_recv_timeout(Some(Duration::from_millis(200)))
|
|
.unwrap();
|
|
assert_open_started(client.open().unwrap());
|
|
client.send_probe().unwrap();
|
|
thread::sleep(Duration::from_millis(50));
|
|
|
|
let events1 = client.recv_once().unwrap();
|
|
assert_eq!(events1.len(), 1);
|
|
assert!(matches!(&events1[0], ClientEvent::EchoReply { .. }));
|
|
|
|
thread::sleep(Duration::from_millis(30));
|
|
let events2 = client.recv_once().unwrap();
|
|
assert_eq!(events2.len(), 1);
|
|
assert!(matches!(
|
|
&events2[0],
|
|
ClientEvent::DuplicateReply { seq: 0, .. }
|
|
));
|
|
server.join();
|
|
}
|
|
|
|
#[test]
|
|
fn out_of_order_reply_emits_late_event() {
|
|
let params = default_params();
|
|
let server = start_fake_server(move |socket, tx| {
|
|
let (_, peer) = recv_request(&socket, &tx);
|
|
let reply = open_reply(FLAG_OPEN | FLAG_REPLY, TOKEN, ¶ms, None);
|
|
socket.send_to(&reply, peer).unwrap();
|
|
|
|
socket
|
|
.set_read_timeout(Some(Duration::from_secs(2)))
|
|
.unwrap();
|
|
|
|
let mut seqs = Vec::new();
|
|
for _ in 0..2 {
|
|
let mut buf = [0_u8; 2048];
|
|
if let Ok((size, _)) = socket.recv_from(&mut buf) {
|
|
tx.send(buf[..size].to_vec()).unwrap();
|
|
let seq = u32::from_le_bytes(buf[12..16].try_into().unwrap());
|
|
seqs.push(seq);
|
|
}
|
|
}
|
|
let ts = TimestampFields::default();
|
|
let reply1 = echo_reply_packet(TOKEN, seqs[1], ¶ms, &ts, None);
|
|
socket.send_to(&reply1, peer).unwrap();
|
|
thread::sleep(Duration::from_millis(10));
|
|
let reply0 = echo_reply_packet(TOKEN, seqs[0], ¶ms, &ts, None);
|
|
socket.send_to(&reply0, peer).unwrap();
|
|
});
|
|
let config = default_test_config();
|
|
let mut client = Client::connect(server.addr.to_string(), config).unwrap();
|
|
client
|
|
.set_recv_timeout(Some(Duration::from_millis(200)))
|
|
.unwrap();
|
|
assert_open_started(client.open().unwrap());
|
|
client.send_probe().unwrap();
|
|
client.send_probe().unwrap();
|
|
thread::sleep(Duration::from_millis(50));
|
|
|
|
let ev1 = client.recv_once().unwrap();
|
|
assert_eq!(ev1.len(), 1);
|
|
assert!(matches!(&ev1[0], ClientEvent::EchoReply { seq: 1, .. }));
|
|
|
|
thread::sleep(Duration::from_millis(30));
|
|
let ev2 = client.recv_once().unwrap();
|
|
assert_eq!(ev2.len(), 1);
|
|
match &ev2[0] {
|
|
ClientEvent::LateReply {
|
|
seq,
|
|
highest_seen,
|
|
rtt,
|
|
..
|
|
} => {
|
|
assert_eq!(*seq, 0);
|
|
assert_eq!(*highest_seen, 1);
|
|
assert!(rtt.is_some());
|
|
}
|
|
other => panic!("expected LateReply, got {other:?}"),
|
|
}
|
|
server.join();
|
|
}
|