diff options
Diffstat (limited to 'tests/channels.rs')
| -rw-r--r-- | tests/channels.rs | 217 |
1 files changed, 217 insertions, 0 deletions
diff --git a/tests/channels.rs b/tests/channels.rs new file mode 100644 index 0000000..ebba9df --- /dev/null +++ b/tests/channels.rs @@ -0,0 +1,217 @@ +use bevy::prelude::App; + +use bevy_quinnet::{ + client::Client, + server::Server, + shared::{ + channel::{ChannelId, ChannelType}, + QuinnetError, + }, +}; + +// https://github.com/rust-lang/rust/issues/46379 +pub use utils::*; + +mod utils; + +/////////////////////////////////////////////////////////// +/// /// +/// Test /// +/// /// +/////////////////////////////////////////////////////////// + +#[test] +fn default_channel() { + let port = 6001; // TODO Use port 0 and retrieve the port used by the server. + let mut server_app: App = start_simple_server_app(port); + let mut client_app: App = start_simple_client_app(port); + + let client_default_channel = get_default_client_channel(&client_app); + let server_default_channel = get_default_server_channel(&server_app); + + for channel in vec![client_default_channel, server_default_channel] { + assert!( + matches!(channel, ChannelId::OrderedReliable(_)), + "Default channel should be an OrderedReliable channel" + ); + } + + close_client_channel(client_default_channel, &mut client_app); + close_server_channel(server_default_channel, &mut server_app); + + { + let mut server = server_app.world.resource_mut::<Server>(); + assert_eq!( + server.endpoint().get_default_channel(), + None, + "Default server channel should be None" + ); + + assert!( + matches!( + server + .endpoint_mut() + .broadcast_message(SharedMessage::TestMessage("".to_string())), + Err(QuinnetError::NoDefaultChannel) + ), + "Should not be able to send on default channel" + ); + } + { + let mut client = client_app.world.resource_mut::<Client>(); + assert_eq!( + client.connection().get_default_channel(), + None, + "Default client channel should be None" + ); + + assert!( + matches!( + client + .connection_mut() + .send_message(SharedMessage::TestMessage("".to_string())), + Err(QuinnetError::NoDefaultChannel) + ), + "Should not be able to send on default channel" + ); + } + + let client_channel = open_client_channel(ChannelType::OrderedReliable, &mut client_app); + let server_channel = open_server_channel(ChannelType::OrderedReliable, &mut server_app); + + let client_id = wait_for_client_connected(&mut client_app, &mut server_app); + + let mut msg_counter = 0; + send_and_test_client_message( + client_id, + client_channel, + &mut client_app, + &mut server_app, + &mut msg_counter, + ); + send_and_test_server_message( + client_id, + server_channel, + &mut server_app, + &mut client_app, + &mut msg_counter, + ); +} + +/////////////////////////////////////////////////////////// +/// /// +/// Test /// +/// /// +/////////////////////////////////////////////////////////// + +#[test] +fn single_instance_channels() { + let port = 6002; // TODO Use port 0 and retrieve the port used by the server. + let mut server_app: App = start_simple_server_app(port); + let mut client_app: App = start_simple_client_app(port); + + let client_id = wait_for_client_connected(&mut client_app, &mut server_app); + + let mut msg_counter = 0; + for (channel_type, channel_id) in vec![ + (ChannelType::UnorderedReliable, ChannelId::UnorderedReliable), + (ChannelType::Unreliable, ChannelId::Unreliable), + ] { + send_and_test_client_message( + client_id, + channel_id, + &mut client_app, + &mut server_app, + &mut msg_counter, + ); + close_client_channel(channel_id, &mut client_app); + open_client_channel(channel_type, &mut client_app); + send_and_test_client_message( + client_id, + channel_id, + &mut client_app, + &mut server_app, + &mut msg_counter, + ); + + send_and_test_server_message( + client_id, + channel_id, + &mut server_app, + &mut client_app, + &mut msg_counter, + ); + close_server_channel(channel_id, &mut server_app); + open_server_channel(channel_type, &mut server_app); + send_and_test_server_message( + client_id, + channel_id, + &mut server_app, + &mut client_app, + &mut msg_counter, + ); + } +} + +/////////////////////////////////////////////////////////// +/// /// +/// Test /// +/// /// +/////////////////////////////////////////////////////////// + +#[test] +fn multi_instance_channels() { + let port = 6003; // TODO Use port 0 and retrieve the port used by the server. + let mut server_app: App = start_simple_server_app(port); + let mut client_app_1: App = start_simple_client_app(port); + + let client_id_1 = wait_for_client_connected(&mut client_app_1, &mut server_app); + + let client_1_channel_1 = get_default_client_channel(&client_app_1); + let client_1_channel_2 = open_client_channel(ChannelType::OrderedReliable, &mut client_app_1); + + let server_channel_1 = get_default_server_channel(&server_app); + let server_channel_2 = open_server_channel(ChannelType::OrderedReliable, &mut server_app); + + let mut msg_counter = 0; + for channel in vec![client_1_channel_1, client_1_channel_2] { + send_and_test_client_message( + client_id_1, + channel, + &mut client_app_1, + &mut server_app, + &mut msg_counter, + ); + } + for channel in vec![server_channel_1, server_channel_2] { + send_and_test_server_message( + client_id_1, + channel, + &mut server_app, + &mut client_app_1, + &mut msg_counter, + ); + } + + let mut client_app_2 = start_simple_client_app(port); + let client_id_2 = wait_for_client_connected(&mut client_app_2, &mut server_app); + + for (client_id, mut client_app) in + vec![(client_id_1, client_app_1), (client_id_2, client_app_2)] + { + send_and_test_server_message( + client_id, + server_channel_1, + &mut server_app, + &mut client_app, + &mut msg_counter, + ); + send_and_test_server_message( + client_id, + server_channel_2, + &mut server_app, + &mut client_app, + &mut msg_counter, + ); + } +} |