Null checks in ContainerState.getMappedPort

Guard against `getNetworkSettings()` returning null
testcontainers · Mar 26, 2021 · 18bd092 · 18bd092
1 parent a0e3e47
commit 18bd092
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/core/src/main/java/org/testcontainers/containers/ContainerState.java b/core/src/main/java/org/testcontainers/containers/ContainerState.java
@@ -32,6 +32,7 @@
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
+import java.util.Optional;
 import java.util.Objects;
 import java.util.stream.Collectors;
 
@@ -142,9 +143,11 @@ default Integer getMappedPort(int originalPort) {
 
         Ports.Binding[] binding = new Ports.Binding[0];
         final InspectContainerResponse containerInfo = this.getContainerInfo();
-        if (containerInfo != null) {
-            binding = containerInfo.getNetworkSettings().getPorts().getBindings().get(new ExposedPort(originalPort));
-        }
+        binding = Optional.ofNullable(containerInfo)
+            .flatMap(it -> Optional.ofNullable(it.getNetworkSettings()))
+            .flatMap(it -> Optional.ofNullable(it.getPorts()))
+            .flatMap(it -> Optional.ofNullable(it.getBindings()))
+            .flatMap(it -> Optional.ofNullable(it.get(new ExposedPort(originalPort)))).orElse(null);
 
         if (binding != null && binding.length > 0 && binding[0] != null) {
             return Integer.valueOf(binding[0].getHostPortSpec());