technofab/dbmate
1
0
Fork 0
mirror of https://github.com/TECHNOFAB11/dbmate.git synced 2025-12-12 16:10:03 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Abstract command tests to support multiple databases

Browse source
This commit is contained in:
Adrian Macneil 2015-11-30 22:06:12 -08:00
parent 456bee585c
commit 8cb676158e
1 changed files with 60 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

100
commands_test.go
View file

@ -15,7 +15,7 @@ import (
var stubsDir string
func testContext(t *testing.T) *cli.Context {
func testContext(t *testing.T, u *url.URL) *cli.Context {
var err error
if stubsDir == "" {
stubsDir, err = filepath.Abs("./stubs")
@ -25,7 +25,6 @@ func testContext(t *testing.T) *cli.Context {
err = os.Chdir(stubsDir)
require.Nil(t, err)
u := testURL(t)
err = os.Setenv("DATABASE_URL", u.String())
require.Nil(t, err)
@ -38,7 +37,7 @@ func testContext(t *testing.T) *cli.Context {
return cli.NewContext(app, flagset, nil)
}
func testURL(t *testing.T) *url.URL {
func postgresTestURL(t *testing.T) *url.URL {
str := os.Getenv("POSTGRES_PORT")
require.NotEmpty(t, str, "missing POSTGRES_PORT environment variable")
@ -53,6 +52,13 @@ func testURL(t *testing.T) *url.URL {
return u
}
func testURLs(t *testing.T) []*url.URL {
return []*url.URL{
postgresTestURL(t),
}
}
func mustClose(c io.Closer) {
if err := c.Close(); err != nil {
panic(err)
@ -60,21 +66,20 @@ func mustClose(c io.Closer) {
}
func TestGetDatabaseUrl(t *testing.T) {
ctx := testContext(t)
err := os.Setenv("DATABASE_URL", "postgres://example.org/db")
envURL, err := url.Parse("foo://example.org/db")
require.Nil(t, err)
ctx := testContext(t, envURL)
u, err := main.GetDatabaseURL(ctx)
require.Nil(t, err)
require.Equal(t, "postgres", u.Scheme)
require.Equal(t, "foo", u.Scheme)
require.Equal(t, "example.org", u.Host)
require.Equal(t, "/db", u.Path)
}
func TestMigrateCommand(t *testing.T) {
ctx := testContext(t)
func testMigrateCommandURL(t *testing.T, u *url.URL) {
ctx := testContext(t, u)
// drop and recreate database
err := main.DropCommand(ctx)
@ -87,8 +92,40 @@ func TestMigrateCommand(t *testing.T) {
require.Nil(t, err)
// verify results
u := testURL(t)
db, err := sql.Open("postgres", u.String())
db, err := sql.Open(u.Scheme, u.String())
require.Nil(t, err)
defer mustClose(db)
count := 0
err = db.QueryRow(`select count(*) from schema_migrations
where version = '20151129054053'`).Scan(&count)
require.Nil(t, err)
require.Equal(t, 1, count)
err = db.QueryRow("select count(*) from users").Scan(&count)
require.Nil(t, err)
require.Equal(t, 1, count)
}
func TestMigrateCommand(t *testing.T) {
for _, u := range testURLs(t) {
testMigrateCommandURL(t, u)
}
}
func testUpCommandURL(t *testing.T, u *url.URL) {
ctx := testContext(t, u)
// drop database
err := main.DropCommand(ctx)
require.Nil(t, err)
// create and migrate
err = main.UpCommand(ctx)
require.Nil(t, err)
// verify results
db, err := sql.Open(u.Scheme, u.String())
require.Nil(t, err)
defer mustClose(db)
@ -104,35 +141,13 @@ func TestMigrateCommand(t *testing.T) {
}
func TestUpCommand(t *testing.T) {
ctx := testContext(t)
// drop database
err := main.DropCommand(ctx)
require.Nil(t, err)
// create and migrate
err = main.UpCommand(ctx)
require.Nil(t, err)
// verify results
u := testURL(t)
db, err := sql.Open("postgres", u.String())
require.Nil(t, err)
defer mustClose(db)
count := 0
err = db.QueryRow(`select count(*) from schema_migrations
where version = '20151129054053'`).Scan(&count)
require.Nil(t, err)
require.Equal(t, 1, count)
err = db.QueryRow("select count(*) from users").Scan(&count)
require.Nil(t, err)
require.Equal(t, 1, count)
for _, u := range testURLs(t) {
testUpCommandURL(t, u)
}
}
func TestRollbackCommand(t *testing.T) {
ctx := testContext(t)
func testRollbackCommandURL(t *testing.T, u *url.URL) {
ctx := testContext(t, u)
// drop, recreate, and migrate database
err := main.DropCommand(ctx)
@ -143,8 +158,7 @@ func TestRollbackCommand(t *testing.T) {
require.Nil(t, err)
// verify migration
u := testURL(t)
db, err := sql.Open("postgres", u.String())
db, err := sql.Open(u.Scheme, u.String())
require.Nil(t, err)
defer mustClose(db)
@ -166,3 +180,9 @@ func TestRollbackCommand(t *testing.T) {
err = db.QueryRow("select count(*) from users").Scan(&count)
require.Equal(t, "pq: relation \"users\" does not exist", err.Error())
}
func TestRollbackCommand(t *testing.T) {
for _, u := range testURLs(t) {
testRollbackCommandURL(t, u)
}
}