Remove docker module dependency

database/cassandra/cassandra_test.go

@@ -9,8 +9,8 @@ import (
 )
 
 import (
-	"github.com/dhui/dktest"
 	"github.com/gocql/gocql"
+	"github.com/golang-migrate/migrate/v4/dktest"
 )
 
 import (

database/clickhouse/clickhouse_test.go

@@ -9,10 +9,10 @@ import (
 	"testing"
 
 	_ "github.com/ClickHouse/clickhouse-go"
-	"github.com/dhui/dktest"
 	"github.com/golang-migrate/migrate/v4"
 	"github.com/golang-migrate/migrate/v4/database/clickhouse"
 	dt "github.com/golang-migrate/migrate/v4/database/testing"
+	"github.com/golang-migrate/migrate/v4/dktest"
 	"github.com/golang-migrate/migrate/v4/dktesting"
 	_ "github.com/golang-migrate/migrate/v4/source/file"
 )

database/cockroachdb/cockroachdb_test.go

@@ -13,7 +13,7 @@ import (
 )
 
 import (
-	"github.com/dhui/dktest"
+	"github.com/golang-migrate/migrate/v4/dktest"
 	_ "github.com/lib/pq"
 )
 

database/firebird/firebird_test.go

@@ -12,7 +12,7 @@ import (
 	"strings"
 	"testing"
 
-	"github.com/dhui/dktest"
+	"github.com/golang-migrate/migrate/v4/dktest"
 
 	dt "github.com/golang-migrate/migrate/v4/database/testing"
 	"github.com/golang-migrate/migrate/v4/dktesting"

database/mongodb/mongodb_test.go

@@ -16,7 +16,7 @@ import (
 )
 
 import (
-	"github.com/dhui/dktest"
+	"github.com/golang-migrate/migrate/v4/dktest"
 	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/mongo"
 	"go.mongodb.org/mongo-driver/mongo/options"

database/mysql/mysql_test.go

@@ -17,10 +17,10 @@ import (
 	"strconv"
 	"testing"
 
-	"github.com/dhui/dktest"
 	"github.com/go-sql-driver/mysql"
 	"github.com/golang-migrate/migrate/v4"
 	dt "github.com/golang-migrate/migrate/v4/database/testing"
+	"github.com/golang-migrate/migrate/v4/dktest"
 	"github.com/golang-migrate/migrate/v4/dktesting"
 	_ "github.com/golang-migrate/migrate/v4/source/file"
 	"github.com/stretchr/testify/assert"

database/neo4j/neo4j_test.go

@@ -7,7 +7,7 @@ import (
 	"log"
 	"testing"
 
-	"github.com/dhui/dktest"
+	"github.com/golang-migrate/migrate/v4/dktest"
 	"github.com/neo4j/neo4j-go-driver/neo4j"
 
 	"github.com/golang-migrate/migrate/v4"

database/pgx/pgx_test.go

@@ -15,9 +15,9 @@ import (
 	"sync"
 	"testing"
 
-	"github.com/dhui/dktest"
 	"github.com/golang-migrate/migrate/v4"
 	"github.com/golang-migrate/migrate/v4/database"
+	"github.com/golang-migrate/migrate/v4/dktest"
 
 	dt "github.com/golang-migrate/migrate/v4/database/testing"
 	"github.com/golang-migrate/migrate/v4/dktesting"

database/pgx/v5/pgx_test.go

@@ -17,7 +17,7 @@ import (
 
 	"github.com/golang-migrate/migrate/v4"
 
-	"github.com/dhui/dktest"
+	"github.com/golang-migrate/migrate/v4/dktest"
 
 	"github.com/golang-migrate/migrate/v4/database"
 	dt "github.com/golang-migrate/migrate/v4/database/testing"

database/postgres/postgres_test.go

@@ -17,7 +17,7 @@ import (
 
 	"github.com/golang-migrate/migrate/v4"
 
-	"github.com/dhui/dktest"
+	"github.com/golang-migrate/migrate/v4/dktest"
 
 	"github.com/golang-migrate/migrate/v4/database"
 	dt "github.com/golang-migrate/migrate/v4/database/testing"

database/redshift/redshift_test.go

@@ -18,7 +18,7 @@ import (
 )
 
 import (
-	"github.com/dhui/dktest"
+	"github.com/golang-migrate/migrate/v4/dktest"
 )
 
 import (

database/rqlite/rqlite_test.go

@@ -8,7 +8,7 @@ import (
 	"net/http"
 	"testing"
 
-	"github.com/dhui/dktest"
+	"github.com/golang-migrate/migrate/v4/dktest"
 	"github.com/rqlite/gorqlite"
 	"github.com/stretchr/testify/assert"
 

database/sqlserver/sqlserver_test.go

@@ -11,8 +11,8 @@ import (
 	"testing"
 	"time"
 
-	"github.com/dhui/dktest"
 	"github.com/golang-migrate/migrate/v4"
+	"github.com/golang-migrate/migrate/v4/dktest"
 
 	dt "github.com/golang-migrate/migrate/v4/database/testing"
 	"github.com/golang-migrate/migrate/v4/dktesting"

database/yugabytedb/yugabytedb_test.go

@@ -11,8 +11,8 @@ import (
 	"testing"
 	"time"
 
-	"github.com/dhui/dktest"
 	"github.com/golang-migrate/migrate/v4"
+	"github.com/golang-migrate/migrate/v4/dktest"
 
 	_ "github.com/lib/pq"
 

dktest/container_info.go

@@ -0,0 +1,96 @@
+package dktest
+
+import (
+	"fmt"
+	"strconv"
+
+	"github.com/docker/go-connections/nat"
+)
+
+// ContainerInfo holds information about a running Docker container.
+type ContainerInfo struct {
+	ID        string
+	Name      string
+	ImageName string
+	Ports     nat.PortMap
+}
+
+func (c ContainerInfo) String() string {
+	return fmt.Sprintf("dktest.ContainerInfo{ID:%q, Name:%q, ImageName:%q}", c.ID, c.Name, c.ImageName)
+}
+
+// Port gets the specified published TCP port.
+func (c ContainerInfo) Port(containerPort uint16) (hostIP string, hostPort string, err error) {
+	port, err := nat.NewPort("tcp", strconv.Itoa(int(containerPort)))
+	if err != nil {
+		return "", "", err
+	}
+	return mapPort(c.Ports, port)
+}
+
+// UDPPort gets the specified published UDP port.
+func (c ContainerInfo) UDPPort(containerPort uint16) (hostIP string, hostPort string, err error) {
+	port, err := nat.NewPort("udp", strconv.Itoa(int(containerPort)))
+	if err != nil {
+		return "", "", err
+	}
+	return mapPort(c.Ports, port)
+}
+
+// FirstPort gets the first published TCP port.
+func (c ContainerInfo) FirstPort() (hostIP string, hostPort string, err error) {
+	return firstPort(c.Ports, "tcp")
+}
+
+// FirstUDPPort gets the first published UDP port.
+func (c ContainerInfo) FirstUDPPort() (hostIP string, hostPort string, err error) {
+	return firstPort(c.Ports, "udp")
+}
+
+func mapPort(portMap nat.PortMap, port nat.Port) (hostIP string, hostPort string, err error) {
+	if bindings, ok := portMap[port]; ok {
+		for _, binding := range bindings {
+			return mapHost(binding.HostIP), binding.HostPort, nil
+		}
+	}
+
+	portInt := port.Int()
+	proto := port.Proto()
+	for p, bindings := range portMap {
+		if p.Proto() != proto {
+			continue
+		}
+		start, end, err := p.Range()
+		if err != nil || portInt < start || portInt > end {
+			continue
+		}
+		offset := portInt - start
+		if offset >= len(bindings) {
+			continue
+		}
+		binding := bindings[offset]
+		return mapHost(binding.HostIP), binding.HostPort, nil
+	}
+	return "", "", errNoPort
+}
+
+func firstPort(portMap nat.PortMap, proto string) (hostIP string, hostPort string, err error) {
+	for port, bindings := range portMap {
+		if port.Proto() != proto {
+			continue
+		}
+		for _, binding := range bindings {
+			return mapHost(binding.HostIP), binding.HostPort, nil
+		}
+	}
+	return "", "", errNoPort
+}
+
+func mapHost(host string) string {
+	switch host {
+	case "", "0.0.0.0":
+		return "127.0.0.1"
+	default:
+		return host
+	}
+}