75 lines
2.4 KiB
Rust
75 lines
2.4 KiB
Rust
use super::*;
|
|
|
|
#[test]
|
|
fn recv_once_returns_empty_on_timeout() {
|
|
let params = default_params();
|
|
let server = open_success_server(params);
|
|
let config = default_test_config();
|
|
let mut client = Client::connect(server.addr.to_string(), config).unwrap();
|
|
client
|
|
.set_recv_timeout(Some(Duration::from_millis(50)))
|
|
.unwrap();
|
|
assert_open_started(client.open().unwrap());
|
|
let events = client.recv_once().unwrap();
|
|
assert!(events.is_empty());
|
|
server.join();
|
|
}
|
|
|
|
#[test]
|
|
fn recv_once_decodes_only_received_bytes_after_longer_datagram() {
|
|
let params = default_params();
|
|
let server_params = params.clone();
|
|
let server = start_fake_server(move |socket, tx| {
|
|
let (_, peer) = recv_request(&socket, &tx);
|
|
let reply = open_reply(FLAG_OPEN | FLAG_REPLY, TOKEN, &server_params, None);
|
|
socket.send_to(&reply, peer).unwrap();
|
|
|
|
let mut seqs = Vec::new();
|
|
for _ in 0..2 {
|
|
let (packet, _) = recv_request(&socket, &tx);
|
|
seqs.push(u32::from_le_bytes(packet[12..16].try_into().unwrap()));
|
|
}
|
|
|
|
let mut longer = echo_reply_packet(
|
|
TOKEN,
|
|
seqs[0],
|
|
&server_params,
|
|
&TimestampFields::default(),
|
|
None,
|
|
);
|
|
longer.extend_from_slice(b"trailing bytes");
|
|
socket.send_to(&longer, peer).unwrap();
|
|
|
|
let shorter = echo_reply_packet(
|
|
TOKEN,
|
|
seqs[1],
|
|
&server_params,
|
|
&TimestampFields::default(),
|
|
None,
|
|
);
|
|
socket.send_to(&shorter, 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 _ = client.recv_once().unwrap();
|
|
let events = client.recv_once().unwrap();
|
|
assert_eq!(events.len(), 1);
|
|
match &events[0] {
|
|
ClientEvent::EchoReply { seq, bytes, .. } => {
|
|
assert_eq!(*seq, 1);
|
|
assert_eq!(*bytes, echo_packet_len(false, ¶ms));
|
|
}
|
|
other => panic!("expected EchoReply after shorter datagram, got {other:?}"),
|
|
}
|
|
server.join();
|
|
}
|