wait until scripts loaded fully in benchmarks

Author: makspll

crates/bevy_mod_scripting_core/src/pipeline/machines.rs

@@ -178,6 +178,11 @@ impl<P: IntoScriptPluginParams> ActiveMachines<P> {
             internal_state: MachineExecutionState::Initialized(Box::new(state)),
         });
     }
+
+    /// Returns the amount of active machines
+    pub fn active_machines(&self) -> usize {
+        self.machines.len()
+    }
 }
 
 /// A machine, which combines the inputs to its states with the state of the world and generates state transitions,

crates/bevy_mod_scripting_core/src/pipeline/mod.rs

@@ -317,14 +317,39 @@ impl<P> Default for RequestProcessingPipelineRun<P> {
     }
 }
 
-/// A system which runs [`RunProcessingPipelineOnce`] command for the plugin only if [`RequestProcessingPipelineRun`] resource had [`RequestProcessingPipelineRun::request_run`] called on it
+/// A system which runs [`RunProcessingPipelineOnce`] command for the plugin only if [`RequestProcessingPipelineRun`] resource had [`RequestProcessingPipelineRun::request_run`] called on it,
+/// and or if there are active machines
 pub fn automatic_pipeline_runner<P: IntoScriptPluginParams>(world: &mut World) {
     let mut res = world.get_resource_or_init::<RequestProcessingPipelineRun<P>>();
-    if res.get_and_unset() {
+    if res.get_and_unset()
+        || world
+            .get_resource::<ActiveMachines<P>>()
+            .is_some_and(|machines| machines.active_machines() > 0)
+    {
         RunProcessingPipelineOnce::<P>::new().apply(world);
     }
 }
 
+/// A helper trait for App implementations
+pub trait PipelineRun {
+    /// Runs update's until all scripts are processed and no more machines are running
+    fn update_until_all_scripts_processed<P: IntoScriptPluginParams>(&mut self);
+}
+
+impl PipelineRun for App {
+    fn update_until_all_scripts_processed<P: IntoScriptPluginParams>(&mut self) {
+        loop {
+            let world = self.world_mut();
+            let machines = world.get_resource::<ActiveMachines<P>>();
+            let has_active = machines.is_some_and(|machines| machines.active_machines() > 0);
+            if !has_active {
+                break;
+            }
+            self.update();
+        }
+    }
+}
+
 #[cfg(test)]
 mod test {
     use bevy_asset::{AssetApp, AssetId, AssetPlugin};

crates/testing_crates/script_integration_test_harness/src/lib.rs

@@ -32,6 +32,7 @@ use bevy_mod_scripting_core::{
     BMSScriptingInfrastructurePlugin, IntoScriptPluginParams,
     commands::AttachScript,
     error::ScriptError,
+    pipeline::PipelineRun,
     script::{DisplayProxy, ScriptAttachment, ScriptComponent, ScriptContext},
 };
 use bevy_mod_scripting_functions::ScriptFunctionsPlugin;
@@ -309,7 +310,7 @@ where
         }
     }
 
-    app.update();
+    app.update_until_all_scripts_processed::<P>();
 
     let script_contexts = app
         .world_mut()