feat(docker-compose): allow people to disable the default wait behavior (#922)

Closes #877 by allowing people to optionally disable the `--wait` flag
usually passed in

Feel free to push any changes here or fixes / issues you have with it,
would love a new release as soon as this is merged as well if it all
possible

Author: ControlCplusControlV

testcontainers/src/compose/client.rs

@@ -28,6 +28,7 @@ pub(super) struct UpCommand {
     pub(super) env_vars: std::collections::HashMap<String, String>,
     pub(super) build: bool,
     pub(super) pull: bool,
+    pub(super) wait: bool,
 }
 
 pub(super) struct DownCommand {

testcontainers/src/compose/client/containerised.rs

@@ -73,9 +73,11 @@ impl ComposeInterface for ContainerisedComposeCli {
             cmd_parts.push("always".to_string());
         }
 
-        cmd_parts.push("--wait".to_string());
-        cmd_parts.push("--wait-timeout".to_string());
-        cmd_parts.push(command.wait_timeout.as_secs().to_string());
+        if command.wait {
+            cmd_parts.push("--wait".to_string());
+            cmd_parts.push("--wait-timeout".to_string());
+            cmd_parts.push(command.wait_timeout.as_secs().to_string());
+        }
 
         let exec = ExecCommand::new(cmd_parts)
             .with_cmd_ready_condition(CmdWaitFor::exit_code(0))

testcontainers/src/compose/client/local.rs

@@ -51,9 +51,11 @@ impl ComposeInterface for LocalComposeCli {
             cmd.arg("--pull").arg("always");
         }
 
-        cmd.arg("--wait")
-            .arg("--wait-timeout")
-            .arg(command.wait_timeout.as_secs().to_string());
+        if command.wait {
+            cmd.arg("--wait")
+                .arg("--wait-timeout")
+                .arg(command.wait_timeout.as_secs().to_string());
+        }
 
         for (key, value) in &command.env_vars {
             cmd.env(key, value);

testcontainers/src/compose/mod.rs

@@ -142,6 +142,7 @@ pub struct DockerCompose {
     remove_images: bool,
     build: bool,
     pull: bool,
+    wait: bool,
     services: HashMap<String, RawContainer>,
     env_vars: HashMap<String, String>,
     wait_strategies: HashMap<String, WaitFor>,
@@ -244,6 +245,7 @@ impl DockerCompose {
                 env_vars: self.env_vars.clone(),
                 build: self.build,
                 pull: self.pull,
+                wait: self.wait,
             })
             .await?;
 
@@ -353,6 +355,16 @@ impl DockerCompose {
         self
     }
 
+    /// Control whether docker compose waits for all services to be healthy before returning (default: true).
+    ///
+    /// Set to `false` when using one-shot containers (e.g., init or migration containers)
+    /// that exit after completing their task, as `--wait` would otherwise time out.
+    /// When disabled, use [`DockerCompose::with_wait_for_service`] for per-service readiness.
+    pub fn with_wait(mut self, wait: bool) -> Self {
+        self.wait = wait;
+        self
+    }
+
     /// Remove volumes when dropping the docker compose or not (removed by default)
     pub fn with_remove_volumes(&mut self, remove_volumes: bool) -> &mut Self {
         self.remove_volumes = remove_volumes;
@@ -376,6 +388,7 @@ impl DockerCompose {
             remove_images: false,
             build: false,
             pull: false,
+            wait: true,
             services: HashMap::new(),
             env_vars: HashMap::new(),
             wait_strategies: HashMap::new(),
@@ -747,4 +760,26 @@ mod tests {
 
         test_compose_client(compose, "containerised").await
     }
+
+    #[tokio::test]
+    async fn test_compose_with_wait_disabled() -> anyhow::Result<()> {
+        let _ = pretty_env_logger::try_init();
+
+        let path = PathBuf::from(format!(
+            "{}/tests/test-compose.yml",
+            env!("CARGO_MANIFEST_DIR")
+        ));
+
+        let mut compose = DockerCompose::with_local_client(&[path.as_path()]).with_wait(false);
+
+        compose.up().await?;
+
+        assert_eq!(compose.services().len(), 2, "Should have 2 services");
+        assert!(compose.service("hello1").is_some());
+        assert!(compose.service("hello2").is_some());
+
+        compose.down().await?;
+
+        Ok(())
+    }
 }