shad-go/dockertest/docker_test.go

110 lines
2.2 KiB
Go
Raw Normal View History

2021-03-18 21:08:54 +00:00
package dockertest
import (
"database/sql"
"fmt"
"log"
"net"
"os"
"strings"
"testing"
"time"
"github.com/cenkalti/backoff"
"github.com/go-redis/redis"
"github.com/google/uuid"
_ "github.com/lib/pq"
"github.com/stretchr/testify/require"
tc "github.com/testcontainers/testcontainers-go"
)
var (
postgresDB *sql.DB
redisDB *redis.Client
)
func TestMain(m *testing.M) {
path := "./docker-compose.yaml"
compose, err := StartDockerCompose(path)
if err != nil {
log.Fatal(err)
}
host := os.Getenv("APP_HOST")
if err = Retry(func() error {
postgresDB, err = sql.Open("postgres", fmt.Sprintf("postgres://%s:%s@%s:%s/%s?sslmode=disable", "gopher", "pass", host, "5432", "test"))
if err != nil {
return err
}
return postgresDB.Ping()
}); err != nil {
log.Fatalf("could not connect to postgres: %s", err)
}
if err = Retry(func() error {
redisDB = redis.NewClient(&redis.Options{
Addr: net.JoinHostPort(host, "6379"),
})
return redisDB.Ping().Err()
}); err != nil {
log.Fatalf("could not connect to redis: %s", err)
}
code := m.Run()
if err := compose.Down().Error; err != nil {
log.Fatalf("could not stop containers from %v: %v", path, err)
}
os.Exit(code)
}
func StartDockerCompose(paths ...string) (*tc.LocalDockerCompose, error) {
id := strings.ToLower(uuid.New().String())
compose := tc.NewLocalDockerCompose(paths, id)
execError := compose.
WithCommand([]string{"up", "-d"}).
Invoke()
err := execError.Error
if err != nil {
return nil, fmt.Errorf("could not start containers from %v: %v", paths, err)
}
return compose, nil
}
func Retry(f func() error) error {
b := backoff.NewExponentialBackOff()
b.MaxInterval = time.Second * 5
b.MaxElapsedTime = time.Minute
return backoff.Retry(f, b)
}
func TestPostgres(t *testing.T) {
rows, err := postgresDB.Query(`SELECT table_name
FROM information_schema.tables
WHERE table_schema = 'public'
`)
require.NoError(t, err)
var table string
for rows.Next() {
err := rows.Scan(&table)
require.NoError(t, err)
t.Logf("table: %s", table)
}
require.NoError(t, rows.Err())
}
func TestRedis(t *testing.T) {
cmd := redisDB.ClientList()
require.NoError(t, cmd.Err())
t.Logf("value: %v", cmd.Val())
}