Retry failed calls to MappedPort to workaround inspect problem

testcontainers · Apr 10, 2024 · b397286 · b397286
1 parent a3ff7aa
commit b397286
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/docker.go b/docker.go
@@ -157,7 +157,16 @@ func (c *DockerContainer) Host(ctx context.Context) (string, error) {
 }
 
 // MappedPort gets externally mapped port for a container port
-func (c *DockerContainer) MappedPort(ctx context.Context, port nat.Port) (nat.Port, error) {
+func (c *DockerContainer) MappedPort(ctx context.Context, port nat.Port) (mappedPort nat.Port, err error) {
+	err = backoff.Retry(func() (err error) {
+		mappedPort, err = c.mappedPort(ctx, port)
+		return
+	}, backoff.WithContext(backoff.NewExponentialBackOff(), ctx))
+
+	return
+}
+
+func (c *DockerContainer) mappedPort(ctx context.Context, port nat.Port) (nat.Port, error) {
 	inspect, err := c.inspectContainer(ctx)
 	if err != nil {
 		return "", err