use std::net::{IpAddr, Ipv4Addr, SocketAddr};
use tokio::time::Duration;
use quilkin::{codec::qcmp::Protocol, test::TestHelper};
#[tokio::test]
#[cfg_attr(target_os = "macos", ignore)]
async fn proxy_ping() {
let mut t = TestHelper::default();
let qcmp = quilkin::net::raw_socket_with_reuse(0).unwrap();
let qcmp_port = quilkin::net::socket_port(&qcmp);
let server_proxy = quilkin::components::proxy::Proxy {
qcmp,
to: vec![(Ipv4Addr::UNSPECIFIED, 0).into()],
..<_>::default()
};
let server_config = TestHelper::new_config();
t.run_server(server_config, Some(server_proxy), None).await;
ping(qcmp_port).await;
}
async fn ping(port: u16) {
tokio::time::sleep(std::time::Duration::from_millis(250)).await;
let socket = tokio::net::UdpSocket::bind((Ipv4Addr::UNSPECIFIED, 0))
.await
.unwrap();
let local_addr = SocketAddr::new(IpAddr::V4(Ipv4Addr::LOCALHOST), port);
let ping = Protocol::ping();
let mut ping_packet = quilkin::codec::qcmp::QcmpPacket::default();
socket
.send_to(ping.encode(&mut ping_packet), &local_addr)
.await
.unwrap();
let mut buf = [0; quilkin::codec::qcmp::MAX_QCMP_PACKET_LEN];
let (size, _) = tokio::time::timeout(Duration::from_secs(1), socket.recv_from(&mut buf))
.await
.unwrap()
.unwrap();
let recv_time = quilkin::time::UtcTimestamp::now();
let reply = Protocol::parse(&buf[..size]).unwrap().unwrap();
assert_eq!(ping.nonce(), reply.nonce());
const MAX: std::time::Duration = std::time::Duration::from_millis(50);
let delay = reply.round_trip_delay(recv_time).unwrap();
assert!(
MAX > delay.duration(),
"Delay {:?} greater than {MAX:?}",
delay.duration(),
);
}