81 lines
2.5 KiB
Go
81 lines
2.5 KiB
Go
|
package dockertest
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
"net/url"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
_ "github.com/lib/pq"
|
||
|
|
)
|
||
|
|
|
||
|
|
// SetupPostgreSQLContainer sets up a real PostgreSQL instance for testing purposes,
|
||
|
|
// using a Docker container. It returns the container ID and its IP address,
|
||
|
|
// or makes the test fail on error.
|
||
|
|
func SetupPostgreSQLContainer() (c ContainerID, ip string, port int, err error) {
|
||
|
|
port = RandomPort()
|
||
|
|
forward := fmt.Sprintf("%d:%d", port, 5432)
|
||
|
|
if BindDockerToLocalhost != "" {
|
||
|
|
forward = "127.0.0.1:" + forward
|
||
|
|
}
|
||
|
|
c, ip, err = SetupContainer(PostgresImageName, port, 15*time.Second, func() (string, error) {
|
||
|
|
return run("--name", GenerateContainerID(), "-d", "-p", forward, "-e", fmt.Sprintf("POSTGRES_PASSWORD=%s", PostgresPassword), PostgresImageName)
|
||
|
|
})
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// ConnectToPostgreSQL starts a PostgreSQL image and passes the database url to the connector callback.
|
||
|
|
func ConnectToPostgreSQL(tries int, delay time.Duration, connector func(url string) bool) (c ContainerID, err error) {
|
||
|
|
c, ip, port, err := SetupPostgreSQLContainer()
|
||
|
|
if err != nil {
|
||
|
|
return c, fmt.Errorf("Could not set up PostgreSQL container: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
for try := 0; try <= tries; try++ {
|
||
|
|
time.Sleep(delay)
|
||
|
|
url := fmt.Sprintf("postgres://%s:%s@%s:%d/postgres?sslmode=disable", PostgresUsername, PostgresPassword, ip, port)
|
||
|
|
if connector(url) {
|
||
|
|
return c, nil
|
||
|
|
}
|
||
|
|
log.Printf("Try %d failed. Retrying.", try)
|
||
|
|
}
|
||
|
|
return c, errors.New("Could not set up PostgreSQL container.")
|
||
|
|
}
|
||
|
|
|
||
|
|
// SetUpPostgreDatabase connects postgre container with given $connectURL and also creates a new database named $databaseName
|
||
|
|
// A modified url used to connect the created database will be returned
|
||
|
|
func SetUpPostgreDatabase(databaseName, connectURL string) (modifiedURL string, err error) {
|
||
|
|
db, err := sql.Open("postgres", connectURL)
|
||
|
|
if err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
defer db.Close()
|
||
|
|
|
||
|
|
count := 0
|
||
|
|
err = db.QueryRow(
|
||
|
|
fmt.Sprintf("SELECT COUNT(*) FROM pg_catalog.pg_database WHERE datname = '%s' ;", databaseName)).
|
||
|
|
Scan(&count)
|
||
|
|
if err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
if count == 0 {
|
||
|
|
// not found for $databaseName, create it
|
||
|
|
_, err = db.Exec(fmt.Sprintf("CREATE DATABASE %s", databaseName))
|
||
|
|
if err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// replace dbname in url
|
||
|
|
// from: postgres://postgres:docker@192.168.99.100:9071/postgres?sslmode=disable
|
||
|
|
// to: postgres://postgres:docker@192.168.99.100:9071/$databaseName?sslmode=disable
|
||
|
|
u, err := url.Parse(connectURL)
|
||
|
|
if err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
u.Path = fmt.Sprintf("/%s", databaseName)
|
||
|
|
return u.String(), nil
|
||
|
|
}
|