|
| 1 | +use bevy::{prelude::*, state::app::StatesPlugin}; |
| 2 | +use bevy_replicon::{ |
| 3 | + client::UserdataReceived, prelude::*, server::ReplicationUserdata, |
| 4 | + shared::backend::channels::ServerChannel, test_app::ServerTestAppExt, |
| 5 | +}; |
| 6 | +use serde::{Deserialize, Serialize}; |
| 7 | +use test_log::test; |
| 8 | + |
| 9 | +#[test] |
| 10 | +fn update_message() { |
| 11 | + let mut server_app = App::new(); |
| 12 | + let mut client_app = App::new(); |
| 13 | + for app in [&mut server_app, &mut client_app] { |
| 14 | + app.add_plugins(( |
| 15 | + MinimalPlugins, |
| 16 | + StatesPlugin, |
| 17 | + RepliconPlugins.set(ServerPlugin::new(PostUpdate)), |
| 18 | + )) |
| 19 | + .init_resource::<ReceivedUserdata>() |
| 20 | + .add_observer(receive_userdata) |
| 21 | + .replicate::<TestComponent>() |
| 22 | + .finish(); |
| 23 | + } |
| 24 | + |
| 25 | + server_app.connect_client(&mut client_app); |
| 26 | + |
| 27 | + let mut userdata = server_app.world_mut().resource_mut::<ReplicationUserdata>(); |
| 28 | + userdata.extend_from_slice(&USERDATA.to_le_bytes()); |
| 29 | + |
| 30 | + server_app.world_mut().spawn((Replicated, TestComponent)); |
| 31 | + |
| 32 | + server_app.update(); |
| 33 | + server_app.exchange_with_client(&mut client_app); |
| 34 | + |
| 35 | + let messages = client_app.world().resource::<ClientMessages>(); |
| 36 | + assert_eq!(messages.received_count(ServerChannel::Updates), 1); |
| 37 | + assert_eq!(messages.received_count(ServerChannel::Mutations), 0); |
| 38 | + |
| 39 | + client_app.update(); |
| 40 | + |
| 41 | + let received = client_app.world().resource::<ReceivedUserdata>(); |
| 42 | + assert_eq!(received.0, USERDATA); |
| 43 | +} |
| 44 | + |
| 45 | +#[test] |
| 46 | +fn mutate_message() { |
| 47 | + let mut server_app = App::new(); |
| 48 | + let mut client_app = App::new(); |
| 49 | + for app in [&mut server_app, &mut client_app] { |
| 50 | + app.add_plugins(( |
| 51 | + MinimalPlugins, |
| 52 | + StatesPlugin, |
| 53 | + RepliconPlugins.set(ServerPlugin::new(PostUpdate)), |
| 54 | + )) |
| 55 | + .init_resource::<ReceivedUserdata>() |
| 56 | + .add_observer(receive_userdata) |
| 57 | + .replicate::<TestComponent>() |
| 58 | + .finish(); |
| 59 | + } |
| 60 | + |
| 61 | + server_app.connect_client(&mut client_app); |
| 62 | + |
| 63 | + let server_entity = server_app |
| 64 | + .world_mut() |
| 65 | + .spawn((Replicated, TestComponent)) |
| 66 | + .id(); |
| 67 | + |
| 68 | + server_app.update(); |
| 69 | + server_app.exchange_with_client(&mut client_app); |
| 70 | + client_app.update(); |
| 71 | + server_app.exchange_with_client(&mut client_app); |
| 72 | + |
| 73 | + let mut component = server_app |
| 74 | + .world_mut() |
| 75 | + .get_mut::<TestComponent>(server_entity) |
| 76 | + .unwrap(); |
| 77 | + component.set_changed(); |
| 78 | + |
| 79 | + let mut userdata = server_app.world_mut().resource_mut::<ReplicationUserdata>(); |
| 80 | + userdata.extend_from_slice(&USERDATA.to_le_bytes()); |
| 81 | + |
| 82 | + server_app.update(); |
| 83 | + server_app.exchange_with_client(&mut client_app); |
| 84 | + |
| 85 | + let messages = client_app.world().resource::<ClientMessages>(); |
| 86 | + assert_eq!(messages.received_count(ServerChannel::Updates), 0); |
| 87 | + assert_eq!(messages.received_count(ServerChannel::Mutations), 1); |
| 88 | + |
| 89 | + client_app.update(); |
| 90 | + |
| 91 | + let received = client_app.world().resource::<ReceivedUserdata>(); |
| 92 | + assert_eq!(received.0, USERDATA); |
| 93 | +} |
| 94 | + |
| 95 | +const USERDATA: u32 = 42; |
| 96 | + |
| 97 | +#[derive(Component, Deserialize, Serialize)] |
| 98 | +struct TestComponent; |
| 99 | + |
| 100 | +#[derive(Resource, Default)] |
| 101 | +struct ReceivedUserdata(u32); |
| 102 | + |
| 103 | +fn receive_userdata(received: On<UserdataReceived>, mut storage: ResMut<ReceivedUserdata>) { |
| 104 | + let bytes = received.bytes.as_ref().try_into().unwrap(); |
| 105 | + storage.0 = u32::from_le_bytes(bytes); |
| 106 | +} |
0 commit comments