use citadel_io::tokio;
use citadel_sdk::prefabs::client::DefaultServerConnectionSettingsBuilder;
use citadel_sdk::prelude::*;
use std::sync::atomic::{AtomicBool, Ordering};
#[tokio::main]
async fn main() {
citadel_logging::setup_log();
let addr = get_env("CITADEL_SERVER_ADDR");
let stun0 = get_env("STUN_0_ADDR");
let stun1 = get_env("STUN_1_ADDR");
let stun2 = get_env("STUN_2_ADDR");
let finished = &AtomicBool::new(false);
let server_connection_settings =
DefaultServerConnectionSettingsBuilder::credentialed_registration(
addr,
"test-username",
"Test user",
"notsecurepassword",
)
.with_udp_mode(UdpMode::Enabled)
.build()
.unwrap();
let client =
citadel_sdk::prefabs::client::single_connection::SingleClientServerConnectionKernel::new(
server_connection_settings,
|mut connection| async move {
let chan = connection.udp_channel_rx.take();
citadel_io::tokio::task::spawn(citadel_sdk::test_common::udp_mode_assertions(
UdpMode::Enabled,
chan,
))
.await
.map_err(|err| NetworkError::Generic(err.to_string()))?;
finished.store(true, Ordering::SeqCst);
connection.shutdown_kernel().await?;
Ok(())
},
);
let _ = DefaultNodeBuilder::default()
.with_node_type(NodeType::Peer)
.with_stun_servers([stun0, stun1, stun2])
.build(client)
.unwrap()
.await
.unwrap();
assert!(finished.load(Ordering::SeqCst));
}
fn get_env(key: &'static str) -> String {
if let Ok(env) = std::env::var(key) {
env
} else {
panic!("Expected the env_var {key} set")
}
}