2
0
Fork 0
mirror of https://github.com/bevyengine/bevy synced 2024-12-30 06:53:13 +00:00
bevy/tests/how_to_test_systems.rs
Clar Fon efda7f3f9c
Simpler lint fixes: makes ci lints work but disables a lint for now ()
Takes the first two commits from  and adds suggestions from this
comment:
https://github.com/bevyengine/bevy/pull/15375#issuecomment-2366968300

See  for more reasoning/motivation.

## Rebasing (rerunning)

```rust
git switch simpler-lint-fixes
git reset --hard main
cargo fmt --all -- --unstable-features --config normalize_comments=true,imports_granularity=Crate
cargo fmt --all
git add --update
git commit --message "rustfmt"
cargo clippy --workspace --all-targets --all-features --fix
cargo fmt --all -- --unstable-features --config normalize_comments=true,imports_granularity=Crate
cargo fmt --all
git add --update
git commit --message "clippy"
git cherry-pick e6c0b94f6795222310fb812fa5c4512661fc7887
```
2024-09-24 11:42:59 +00:00

174 lines
4 KiB
Rust

#![allow(missing_docs)]
use bevy::prelude::*;
#[derive(Component, Default)]
struct Enemy {
hit_points: u32,
score_value: u32,
}
#[derive(Event)]
struct EnemyDied(u32);
#[derive(Resource)]
struct Score(u32);
fn update_score(mut dead_enemies: EventReader<EnemyDied>, mut score: ResMut<Score>) {
for value in dead_enemies.read() {
score.0 += value.0;
}
}
fn despawn_dead_enemies(
mut commands: Commands,
mut dead_enemies: EventWriter<EnemyDied>,
enemies: Query<(Entity, &Enemy)>,
) {
for (entity, enemy) in &enemies {
if enemy.hit_points == 0 {
commands.entity(entity).despawn_recursive();
dead_enemies.send(EnemyDied(enemy.score_value));
}
}
}
fn hurt_enemies(mut enemies: Query<&mut Enemy>) {
for mut enemy in &mut enemies {
enemy.hit_points -= 1;
}
}
fn spawn_enemy(mut commands: Commands, keyboard_input: Res<ButtonInput<KeyCode>>) {
if keyboard_input.just_pressed(KeyCode::Space) {
commands.spawn(Enemy {
hit_points: 5,
score_value: 3,
});
}
}
#[test]
fn did_hurt_enemy() {
// Setup app
let mut app = App::new();
// Add Score resource
app.insert_resource(Score(0));
// Add `EnemyDied` event
app.add_event::<EnemyDied>();
// Add our two systems
app.add_systems(Update, (hurt_enemies, despawn_dead_enemies).chain());
// Setup test entities
let enemy_id = app
.world_mut()
.spawn(Enemy {
hit_points: 5,
score_value: 3,
})
.id();
// Run systems
app.update();
// Check resulting changes
assert!(app.world().get::<Enemy>(enemy_id).is_some());
assert_eq!(app.world().get::<Enemy>(enemy_id).unwrap().hit_points, 4);
}
#[test]
fn did_despawn_enemy() {
// Setup app
let mut app = App::new();
// Add Score resource
app.insert_resource(Score(0));
// Add `EnemyDied` event
app.add_event::<EnemyDied>();
// Add our two systems
app.add_systems(Update, (hurt_enemies, despawn_dead_enemies).chain());
// Setup test entities
let enemy_id = app
.world_mut()
.spawn(Enemy {
hit_points: 1,
score_value: 1,
})
.id();
// Run systems
app.update();
// Check enemy was despawned
assert!(app.world().get::<Enemy>(enemy_id).is_none());
// Get `EnemyDied` event reader
let enemy_died_events = app.world().resource::<Events<EnemyDied>>();
let mut enemy_died_reader = enemy_died_events.get_cursor();
let enemy_died = enemy_died_reader.read(enemy_died_events).next().unwrap();
// Check the event has been sent
assert_eq!(enemy_died.0, 1);
}
#[test]
fn spawn_enemy_using_input_resource() {
// Setup app
let mut app = App::new();
// Add our systems
app.add_systems(Update, spawn_enemy);
// Setup test resource
let mut input = ButtonInput::<KeyCode>::default();
input.press(KeyCode::Space);
app.insert_resource(input);
// Run systems
app.update();
// Check resulting changes, one entity has been spawned with `Enemy` component
assert_eq!(app.world_mut().query::<&Enemy>().iter(app.world()).len(), 1);
// Clear the `just_pressed` status for all `KeyCode`s
app.world_mut()
.resource_mut::<ButtonInput<KeyCode>>()
.clear();
// Run systems
app.update();
// Check resulting changes, no new entity has been spawned
assert_eq!(app.world_mut().query::<&Enemy>().iter(app.world()).len(), 1);
}
#[test]
fn update_score_on_event() {
// Setup app
let mut app = App::new();
// Add Score resource
app.insert_resource(Score(0));
// Add `EnemyDied` event
app.add_event::<EnemyDied>();
// Add our systems
app.add_systems(Update, update_score);
// Send an `EnemyDied` event
app.world_mut()
.resource_mut::<Events<EnemyDied>>()
.send(EnemyDied(3));
// Run systems
app.update();
// Check resulting changes
assert_eq!(app.world().resource::<Score>().0, 3);
}