mirror of https://github.com/VinGarcia/ksql.git
93 lines
2.4 KiB
Go
93 lines
2.4 KiB
Go
package kpostgres
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
"github.com/jackc/pgx/v4/pgxpool"
|
|
_ "github.com/lib/pq"
|
|
"github.com/ory/dockertest"
|
|
"github.com/ory/dockertest/docker"
|
|
"github.com/vingarcia/ksql"
|
|
"github.com/vingarcia/ksql/sqldialect"
|
|
"io"
|
|
"log"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestSQLAdapter(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
postgresURL, closePostgres := startPostgresDB(ctx, "ksql")
|
|
defer closePostgres()
|
|
|
|
ksql.RunTestsForAdapter(t, "kpgx", sqldialect.PostgresDialect{}, postgresURL, func(t *testing.T) (ksql.DBAdapter, io.Closer) {
|
|
sqldb, err := sql.Open("postgres", postgresURL)
|
|
if err != nil {
|
|
t.Fatal(err.Error())
|
|
}
|
|
|
|
return SQLAdapter{sqldb}, sqldb
|
|
})
|
|
}
|
|
|
|
func startPostgresDB(ctx context.Context, dbName string) (databaseURL string, closer func()) {
|
|
// uses a sensible default on windows (tcp/http) and linux/osx (socket)
|
|
dockerPool, err := dockertest.NewPool("")
|
|
if err != nil {
|
|
log.Fatalf("Could not connect to docker: %s", err)
|
|
}
|
|
|
|
// pulls an image, creates a container based on it and runs it
|
|
resource, err := dockerPool.RunWithOptions(
|
|
&dockertest.RunOptions{
|
|
Repository: "postgres",
|
|
Tag: "14.0",
|
|
Env: []string{
|
|
"POSTGRES_PASSWORD=postgres",
|
|
"POSTGRES_USER=postgres",
|
|
"POSTGRES_DB=" + dbName,
|
|
"listen_addresses = '*'",
|
|
},
|
|
},
|
|
func(config *docker.HostConfig) {
|
|
// set AutoRemove to true so that stopped container goes away by itself
|
|
config.AutoRemove = true
|
|
config.RestartPolicy = docker.RestartPolicy{Name: "no"}
|
|
},
|
|
)
|
|
if err != nil {
|
|
log.Fatalf("Could not start resource: %s", err)
|
|
}
|
|
|
|
hostAndPort := resource.GetHostPort("5432/tcp")
|
|
databaseUrl := fmt.Sprintf("postgres://postgres:postgres@%s/%s?sslmode=disable", hostAndPort, dbName)
|
|
|
|
fmt.Println("Connecting to postgres on url: ", databaseUrl)
|
|
|
|
resource.Expire(40) // Tell docker to hard kill the container in 40 seconds
|
|
|
|
var sqlDB *pgxpool.Pool
|
|
// exponential backoff-retry, because the application in the container might not be ready to accept connections yet
|
|
dockerPool.MaxWait = 10 * time.Second
|
|
dockerPool.Retry(func() error {
|
|
sqlDB, err = pgxpool.Connect(ctx, databaseUrl)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return sqlDB.Ping(ctx)
|
|
})
|
|
if err != nil {
|
|
log.Fatalf("Could not connect to docker: %s", err)
|
|
}
|
|
sqlDB.Close()
|
|
|
|
return databaseUrl, func() {
|
|
if err := dockerPool.Purge(resource); err != nil {
|
|
log.Fatalf("Could not purge resource: %s", err)
|
|
}
|
|
}
|
|
}
|