dbmate/driver/postgres/postgres_test.go

package postgres_test

import (
	"database/sql"
	"github.com/adrianmacneil/dbmate/driver/postgres"
	"github.com/stretchr/testify/require"
	"net/url"
	"os"
	"testing"
)

func testURL(t *testing.T) *url.URL {
	str := os.Getenv("POSTGRES_PORT")
	require.NotEmpty(t, str)

	u, err := url.Parse(str)
	require.Nil(t, err)

	u.Scheme = "postgres"
	u.User = url.User("postgres")
	u.Path = "/dbmate"
	u.RawQuery = "sslmode=disable"

	return u
}

func TestCreateDropDatabase(t *testing.T) {
	d := postgres.Driver{}
	u := testURL(t)

	// drop any existing database
	err := d.DropDatabase(u)
	require.Nil(t, err)

	// create database
	err = d.CreateDatabase(u)
	require.Nil(t, err)

	// check that database exists and we can connect to it
	func() {
		db, err := sql.Open("postgres", u.String())
		require.Nil(t, err)
		defer db.Close()

		err = db.Ping()
		require.Nil(t, err)
	}()

	// drop the database
	err = d.DropDatabase(u)
	require.Nil(t, err)

	// check that database no longer exists
	func() {
		db, err := sql.Open("postgres", u.String())
		require.Nil(t, err)
		defer db.Close()

		err = db.Ping()
		require.NotNil(t, err)
		require.Equal(t, "pq: database \"dbmate\" does not exist", err.Error())
	}()
}