Use 'count_n_players' with Query

richelbilderbeek · Jul 14, 2024 · a08db23 · a08db23
1 parent 3a45972
commit a08db23
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/src/app.rs b/src/app.rs
@@ -46,14 +46,9 @@ fn respond_to_mouse_wheel_press(
 }
 
 #[cfg(test)]
-pub fn count_n_players(app: &App) -> usize {
-    let mut n = 0;
-    for c in app.world().components().iter() {
-        if c.name().contains("::Player") {
-            n += 1;
-        }
-    }
-    n
+fn count_n_players(app: &mut App) -> usize {
+    let mut query = app.world_mut().query::<&Player>();
+    return query.iter(app.world_mut()).len();
 }
 
 #[cfg(test)]