fix(psqldocker): fix port bug

purposeinplay · Aug 21, 2024 · ec5dce8 · ec5dce8
1 parent 247f3a6
commit ec5dce8
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/psqldocker/psql_container.go b/psqldocker/psql_container.go
@@ -11,6 +11,7 @@ import (
 	_ "github.com/lib/pq"
 	"github.com/ory/dockertest/v3"
 	"github.com/ory/dockertest/v3/docker"
+	"strings"
 )
 
 // ensure Container implements the io.Closer interface.
@@ -102,6 +103,12 @@ func NewContainer(
 		opts[i].apply(&options)
 	}
 
+	exposedPSQLPort := options.dbPort
+
+	if !strings.Contains(exposedPSQLPort, "/tcp") {
+		exposedPSQLPort = exposedPSQLPort + "/tcp"
+	}
+
 	return &Container{
 		user:     user,
 		password: password,
@@ -113,7 +120,7 @@ func NewContainer(
 			Cmd:          []string{"-p " + options.dbPort},
 			Repository:   "postgres",
 			Tag:          options.imageTag,
-			ExposedPorts: []string{options.dbPort},
+			ExposedPorts: []string{exposedPSQLPort},
 			Env:          envVars(user, password, dbName),
 		},
 		expiration:       options.expirationSeconds,