Add Rollback component to integration test players

gschup · Oct 19, 2023 · 062dcce · 062dcce
1 parent a5defd2
commit 062dcce
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/tests/integration.rs b/tests/integration.rs
@@ -6,8 +6,8 @@ use bevy::{
     MinimalPlugins,
 };
 use bevy_ggrs::{
-    GgrsPlugin, GgrsSchedule, LocalInputs, LocalPlayers, PlayerInputs, ReadInputs, Rollback,
-    Session,
+    AddRollbackCommandExtension, GgrsPlugin, GgrsSchedule, LocalInputs, LocalPlayers, PlayerInputs,
+    ReadInputs, Rollback, Session,
 };
 use bytemuck::{Pod, Zeroable};
 use ggrs::{Config, P2PSession, PlayerHandle, PlayerType, SessionBuilder, UdpNonBlockingSocket};
@@ -213,10 +213,12 @@ pub fn spawn_players(mut commands: Commands, session: Res<Session<GgrsConfig>>)
     };
 
     for handle in 0..num_players {
-        commands.spawn((
-            PlayerComponent { handle },
-            Velocity::default(),
-            Transform::default(),
-        ));
+        commands
+            .spawn((
+                PlayerComponent { handle },
+                Velocity::default(),
+                Transform::default(),
+            ))
+            .add_rollback();
     }
 }