chore: handle new error on return from internalCheck()

testcontainers · Sep 13, 2024 · d3f2a75 · d3f2a75
1 parent b6949e4
commit d3f2a75
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/wait/host_port.go b/wait/host_port.go
@@ -157,13 +157,17 @@ func (hp *HostPortStrategy) WaitUntilReady(ctx context.Context, target StrategyT
 	}
 
 	if err = internalCheck(ctx, internalPort, target); err != nil {
-		if errors.Is(errShellNotExecutable, err) {
+		switch {
+		case errors.Is(err, errShellNotExecutable):
 			log.Println("Shell not executable in container, only external port validated")
 			return nil
+		case errors.Is(err, errShellNotFound):
+			log.Println("Shell not found in container")
+			return nil
+		default:
+			return fmt.Errorf("internal check: %w", err)
 		}
-
-		return fmt.Errorf("internal check: %w", err)
-	}
+	}	
 
 	return nil
 }