Initial commit

driver/postgres/postgres_test.go

@@ -0,0 +1,63 @@
+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())
+	}()
+}