Allow overriding default compose wait strategy

Allow the default wait strategy to be overriden for docker compose
environments. The default "listening ports" strategy doesn't work for
distroless images and in complex setups, having to explicitly override
the wait strategy for each service is error-prone.

Author: jalaziz

packages/testcontainers/src/docker-compose-environment/docker-compose-environment.test.ts

@@ -98,6 +98,16 @@ describe("DockerComposeEnvironment", { timeout: 180_000 }, () => {
     await startedEnvironment.down();
   });
 
+  it("should support default wait strategy", async () => {
+    const startedEnvironment = await new DockerComposeEnvironment(fixtures, "docker-compose-with-healthcheck.yml")
+      .withDefaultWaitStrategy(Wait.forHealthCheck())
+      .up();
+
+    await checkEnvironmentContainerIsHealthy(startedEnvironment, await composeContainerName("container"));
+
+    await startedEnvironment.down();
+  });
+
   it("should support log message wait strategy", async () => {
     const startedEnvironment = await new DockerComposeEnvironment(fixtures, "docker-compose.yml")
       .withWaitStrategy(await composeContainerName("container"), Wait.forLogMessage("Listening on port 8080"))

packages/testcontainers/src/docker-compose-environment/docker-compose-environment.ts

@@ -23,6 +23,7 @@ export class DockerComposeEnvironment {
   private profiles: string[] = [];
   private environment: Environment = {};
   private pullPolicy: ImagePullPolicy = PullPolicy.defaultPolicy();
+  private defaultWaitStrategy: WaitStrategy = Wait.forListeningPorts();
   private waitStrategy: { [containerName: string]: WaitStrategy } = {};
   private startupTimeout?: number;
 
@@ -63,6 +64,11 @@ export class DockerComposeEnvironment {
     return this;
   }
 
+  public withDefaultWaitStrategy(waitStrategy: WaitStrategy): this {
+    this.defaultWaitStrategy = waitStrategy;
+    return this;
+  }
+
   public withWaitStrategy(containerName: string, waitStrategy: WaitStrategy): this {
     this.waitStrategy[containerName] = waitStrategy;
     return this;
@@ -140,7 +146,7 @@ export class DockerComposeEnvironment {
           const boundPorts = BoundPorts.fromInspectResult(client.info.containerRuntime.hostIps, mappedInspectResult);
           const waitStrategy = this.waitStrategy[containerName]
             ? this.waitStrategy[containerName]
-            : Wait.forListeningPorts();
+            : this.defaultWaitStrategy;
           if (this.startupTimeout !== undefined) {
             waitStrategy.withStartupTimeout(this.startupTimeout);
           }