use kubi_udp::{ server::{Server, ServerConfig, ServerEvent}, client::{Client, ClientConfig, ClientEvent}, }; use std::{thread, time::Duration}; const TEST_ADDR: &str = "127.0.0.1:22342"; type CtsMessage = u32; type StcMessage = u64; const CTS_MSG: CtsMessage = 0xbeef_face; const STC_MSG: StcMessage = 0xdead_beef_cafe_face; #[test] fn test_connection() { //Init logging kubi_logging::init(); //Create server and client let mut server: Server = Server::bind( TEST_ADDR.parse().expect("Invalid TEST_ADDR"), ServerConfig::default() ).expect("Failed to create server"); let mut client: Client = Client::new( TEST_ADDR.parse().unwrap(), ClientConfig::default() ).expect("Failed to create client"); //Start server update thread let server_handle = thread::spawn(move || { let mut message_received = false; loop { server.update().unwrap(); let events: Vec<_> = server.process_events().collect(); for event in events { match event { ServerEvent::Connected(id) => { assert_eq!(id.get(), 1, "Unexpected client id"); server.send_message(id, STC_MSG).unwrap(); }, ServerEvent::Disconnected(id) => { assert!(message_received, "Client {id} disconnected from the server before sending the message"); return; }, ServerEvent::MessageReceived { from, message } => { log::info!("server received message"); assert_eq!(message, CTS_MSG, "Received message not equal"); message_received = true; }, _ => () } } } }); //Wait a bit thread::sleep(Duration::from_secs(1)); //Connect client client.connect().expect("Client connect failed"); //Start updating the client let client_handle = thread::spawn(move || { let mut message_received = false; loop { client.update().unwrap(); let events: Vec<_> = client.process_events().collect(); for event in events { match event { ClientEvent::Connected(id) => { assert_eq!(id.get(), 1, "Unexpected client id"); client.send_message(CTS_MSG).unwrap(); }, ClientEvent::Disconnected(reason) => { assert!(message_received, "Client lost connection to the server before sending the message with reason: {reason:?}"); return; }, ClientEvent::MessageReceived(data) => { log::info!("client received message"); assert_eq!(data, STC_MSG, "Received message not equal"); message_received = true; client.disconnect(); }, _ => () } } } }); server_handle.join().unwrap(); client_handle.join().unwrap(); }