use std::collections::HashSet;
use bevy::prelude::*;
use bevy_assert::AppTest;
#[test]
fn test_query_count_one() {
let mut t = AppTest::new();
let start = t.count_all();
t.world_mut().spawn(());
let end = t.count_all();
assert_eq!(end - start, 1);
}
#[test]
fn test_query_count_one_filtered() {
let mut t = AppTest::new();
let start = t.count::<With<Transform>>();
t.world_mut().spawn(());
t.world_mut().spawn(Transform::default());
let end = t.count::<With<Transform>>();
assert_eq!(end - start, 1);
}
#[test]
fn test_all_entities() {
let mut t = AppTest::new();
let mut added = t.all_entities();
added.insert(t.world_mut().spawn(()).id());
added.insert(t.world_mut().spawn(Transform::default()).id());
let total = t.all_entities();
assert_eq!(added, total);
}
#[test]
fn test_entities_filtered() {
let mut t = AppTest::new();
let mut added = HashSet::<Entity>::new();
t.world_mut().spawn(());
added.insert(t.world_mut().spawn(Transform::default()).id());
let total = t.entities::<With<Transform>>();
assert_eq!(added, total);
}