sq/magefile_sakila.go
Neil O'Toole 23022f2b7f
Go1.19 compatibility (#104)
* upgraded mattn/go-isatty version because it was importing a incompatible version of golang.org/x/sys/unix

* changed parsing of SQL Server DSN due to changes to url.ParseQuery no longer accepting semicolon

* updated xo/dburl pkg to v0.11.0

* removed dead underscore assignment

* updated GH workflow go action to 1.19
2022-08-08 13:14:42 -06:00

150 lines
4.6 KiB
Go

//go:build mage
// This magefile contains targets for starting and stopping
// the Sakila test containers locally.
package main
import (
"context"
"fmt"
"sync"
"time"
"github.com/magefile/mage/mg"
"github.com/testcontainers/testcontainers-go/wait"
"golang.org/x/sync/errgroup"
"github.com/testcontainers/testcontainers-go"
)
// Sakila is the mage namespace for sakila targets.
type Sakila mg.Namespace
const startupTimeout = time.Minute * 5
// containers is the set of database server containers required for
// integration testing.
//
// NOTE: The wait.ForLog mechanism is probably not fully correct for any of the
//
// containers (e.g. the string may appear multiple times in the log output),
// thus the containers may not have started fully when WaitingFor completes.
var containers = []testcontainers.ContainerRequest{
{
Image: "sakiladb/sqlserver:2017-CU19",
Name: "sakiladb-sqlserver-2017",
ExposedPorts: []string{"14337:1433"},
WaitingFor: wait.ForLog("Changed database context to 'sakila'.").WithStartupTimeout(startupTimeout),
},
{
Image: "sakiladb/postgres:9",
Name: "sakiladb-postgres-9",
ExposedPorts: []string{"54329:5432"},
WaitingFor: wait.ForLog("PostgreSQL init process complete; ready for start up.").WithStartupTimeout(startupTimeout),
},
{
Image: "sakiladb/postgres:10",
Name: "sakiladb-postgres-10",
ExposedPorts: []string{"54330:5432"},
WaitingFor: wait.ForLog("PostgreSQL init process complete; ready for start up.").WithStartupTimeout(startupTimeout),
},
{
Image: "sakiladb/postgres:11",
Name: "sakiladb-postgres-11",
ExposedPorts: []string{"54331:5432"},
WaitingFor: wait.ForLog("PostgreSQL init process complete; ready for start up.").WithStartupTimeout(startupTimeout),
},
{
Image: "sakiladb/postgres:12",
Name: "sakiladb-postgres-12",
ExposedPorts: []string{"54332:5432"},
WaitingFor: wait.ForLog("PostgreSQL init process complete; ready for start up.").WithStartupTimeout(startupTimeout),
},
{
Image: "sakiladb/mysql:5.6",
Name: "sakiladb-mysql-5.6",
ExposedPorts: []string{"33066:3306"},
WaitingFor: wait.ForLog("[Note] mysqld: ready for connections.").WithStartupTimeout(startupTimeout),
},
{
Image: "sakiladb/mysql:5.7",
Name: "sakiladb-mysql-5.7",
ExposedPorts: []string{"33067:3306"},
WaitingFor: wait.ForLog("[Note] mysqld: ready for connections.").WithStartupTimeout(startupTimeout),
},
{
Image: "sakiladb/mysql:8",
Name: "sakiladb-mysql-8",
ExposedPorts: []string{"33068:3306"},
WaitingFor: wait.ForLog("[Server] /usr/sbin/mysqld: ready for connections.").WithStartupTimeout(startupTimeout),
},
}
// StartAll starts all the sakila database server containers locally.
// Use RemoveAll to stop & remove the containers.
func (Sakila) StartAll() error {
const envarInfo = `export SQ_TEST_SRC__SAKILA_MY56=localhost:33066
export SQ_TEST_SRC__SAKILA_MY57=localhost:33067
export SQ_TEST_SRC__SAKILA_MY8=localhost:33068
export SQ_TEST_SRC__SAKILA_PG9=localhost:54329
export SQ_TEST_SRC__SAKILA_PG10=localhost:54330
export SQ_TEST_SRC__SAKILA_PG11=localhost:54331
export SQ_TEST_SRC__SAKILA_PG12=localhost:54332
export SQ_TEST_SRC__SAKILA_MS17=localhost:14337`
fmt.Println("Starting all containers...")
fmt.Printf("'src' the following:\n===\n%s\n===\n\n", envarInfo)
errGroup, ctx := errgroup.WithContext(context.Background())
for _, container := range containers {
container := container
// The user invokes mage sakila:RemoveAll to remove all the containers.
container.SkipReaper = true
errGroup.Go(func() error {
fmt.Printf("Starting container %s for %s\n", container.Name, container.Image)
_, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
ContainerRequest: container,
Started: true,
})
if err != nil {
return fmt.Errorf("image %s: %w", container.Image, err)
}
fmt.Printf("SUCCESS: started container %s for %s\n", container.Name, container.Image)
return nil
})
}
err := errGroup.Wait()
if err != nil {
fmt.Println("ERROR: problem reported trying to start all containers (some or all containers may actually have started)")
return err
}
fmt.Println("SUCCESS: All database containers started")
return nil
}
// RemoveAll removes all the Sakila docker containers.
func (Sakila) RemoveAll() {
wg := &sync.WaitGroup{}
wg.Add(len(containers))
for _, container := range containers {
container := container
go func() {
defer wg.Done()
// We don't care if there's an error
_ = execDocker("rm", "-f", container.Name)
}()
}
wg.Wait()
}