bitt 0.5.0

Bevy integration testing toolkit
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

use std::time::Duration;

use bevy::{app::AppExit, prelude::*};

use crate::TestWrangler;

#[derive(Debug, Resource)]
struct Timeout(Duration);

/// A plugin that will add an Asserter and panic if it runs for longer than the given duration.
///
/// Useful for cases when you want to test a combo of some systems in relative isolation.
#[derive(Debug)]
pub struct TimeoutAsserterPlugin(pub Duration);

impl Plugin for TimeoutAsserterPlugin {
    fn build(&self, app: &mut App) {
        app.init_resource::<TestWrangler>()
            .insert_resource(Timeout(self.0))
            .add_systems(Update, (timeout, exit_on_asserter_result).chain());
    }
}

fn timeout(mut asserter: ResMut<TestWrangler>, time: Res<Time<Real>>, timeout: Res<Timeout>) {
    if time.elapsed() >= timeout.0 {
        dbg!("Test timed out");
        asserter.fail();
    }
}

fn exit_on_asserter_result(asserter: Res<TestWrangler>, mut exit: EventWriter<AppExit>) {
    if let Some(result) = asserter.outcome {
        if result {
            exit.send(AppExit);
        } else {
            panic!("Test failed");
        }
    }
}