86 lines
1.6 KiB
Go
86 lines
1.6 KiB
Go
package pgfixture
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"gitlab.com/manytask/itmo-go/private/tools/testtool"
|
|
)
|
|
|
|
func lookPath(t *testing.T, name string) string {
|
|
t.Helper()
|
|
|
|
path, err := exec.LookPath(name)
|
|
if err == nil {
|
|
return path
|
|
}
|
|
|
|
const ubuntuPostgres = "/usr/lib/postgresql"
|
|
|
|
if dirs, err := ioutil.ReadDir(ubuntuPostgres); err == nil {
|
|
for _, d := range dirs {
|
|
path := filepath.Join(ubuntuPostgres, d.Name(), "bin", name)
|
|
|
|
if _, err := os.Stat(path); err == nil {
|
|
return path
|
|
}
|
|
}
|
|
}
|
|
|
|
t.Fatalf("%s binary not found; is postgres installed?", name)
|
|
return ""
|
|
}
|
|
|
|
func Start(t *testing.T) string {
|
|
pgconn, ok := os.LookupEnv("PGCONN")
|
|
if ok {
|
|
t.Logf("using external database: PGCONN=%s", pgconn)
|
|
return pgconn
|
|
}
|
|
|
|
pgdata := t.TempDir()
|
|
|
|
cmd := exec.Command(lookPath(t, "initdb"), "-N", "-D", pgdata)
|
|
cmd.Stderr = os.Stderr
|
|
cmd.Stdout = os.Stdout
|
|
if err := cmd.Run(); err != nil {
|
|
t.Fatalf("initdb failed: %v", err)
|
|
}
|
|
|
|
port, err := testtool.GetFreePort()
|
|
require.NoError(t, err)
|
|
|
|
pgrun := t.TempDir()
|
|
|
|
cmd = exec.Command(lookPath(t, "postgres"), "-D", pgdata, "-k", pgrun, "-F", "-p", port)
|
|
cmd.Stderr = os.Stderr
|
|
cmd.Stdout = os.Stdout
|
|
if err := cmd.Start(); err != nil {
|
|
t.Fatalf("postgres failed: %v", err)
|
|
}
|
|
|
|
finished := make(chan error, 1)
|
|
go func() {
|
|
finished <- cmd.Wait()
|
|
}()
|
|
|
|
select {
|
|
case err := <-finished:
|
|
t.Fatalf("postgres server terminated: %v", err)
|
|
|
|
case <-time.After(time.Second / 2):
|
|
}
|
|
|
|
t.Cleanup(func() {
|
|
_ = cmd.Process.Kill()
|
|
})
|
|
|
|
return fmt.Sprintf("host=localhost port=%s database=postgres", port)
|
|
}
|