vendor dependencies with dep

vendor/github.com/go-pg/migrations/db_test.go

@@ -0,0 +1,161 @@
+package migrations_test
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/go-pg/migrations"
+
+	"github.com/go-pg/pg"
+)
+
+func connectDB() *pg.DB {
+	db := pg.Connect(&pg.Options{
+		User: "postgres",
+	})
+
+	_, err := db.Exec("DROP TABLE IF EXISTS gopg_migrations")
+	if err != nil {
+		panic(err)
+	}
+
+	return db
+}
+
+func TestVersion(t *testing.T) {
+	db := connectDB()
+
+	_, _, err := migrations.Run(db, "init")
+	if err != nil {
+		t.Fatalf("init failed: %s", err)
+	}
+
+	version, err := migrations.Version(db)
+	if err != nil {
+		t.Fatalf("Version failed: %s", err)
+	}
+	if version != 0 {
+		t.Fatalf("got version %d, wanted 0", version)
+	}
+
+	if err := migrations.SetVersion(db, 999); err != nil {
+		t.Fatalf("SetVersion failed: %s", err)
+	}
+
+	version, err = migrations.Version(db)
+	if err != nil {
+		t.Fatalf("Version failed: %s", err)
+	}
+	if version != 999 {
+		t.Fatalf("got version %d, wanted %d", version)
+	}
+}
+
+func TestUpDown(t *testing.T) {
+	db := connectDB()
+
+	_, _, err := migrations.Run(db, "init")
+	if err != nil {
+		t.Fatalf("init failed: %s", err)
+	}
+
+	migrations.Set([]migrations.Migration{
+		{Version: 2, Up: doNothing, Down: doNothing},
+		{Version: 1, Up: doNothing, Down: doNothing},
+		{Version: 3, Up: doNothing, Down: doNothing},
+	})
+	oldVersion, newVersion, err := migrations.Run(db, "up")
+	if err != nil {
+		t.Fatal(err)
+	}
+	if oldVersion != 0 {
+		t.Fatalf("got %d, wanted 0", oldVersion)
+	}
+	if newVersion != 3 {
+		t.Fatalf("got %d, wanted 3", newVersion)
+	}
+
+	version, err := migrations.Version(db)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if version != 3 {
+		t.Fatalf("got version %d, wanted 3", version)
+	}
+
+	for i := 2; i >= -5; i-- {
+		wantOldVersion := int64(i + 1)
+		wantNewVersion := int64(i)
+		if wantNewVersion < 0 {
+			wantOldVersion = 0
+			wantNewVersion = 0
+		}
+
+		oldVersion, newVersion, err = migrations.Run(db, "down")
+		if err != nil {
+			t.Fatal(err)
+		}
+		if oldVersion != wantOldVersion {
+			t.Fatalf("got %d, wanted %d", oldVersion, wantOldVersion)
+		}
+		if newVersion != wantNewVersion {
+			t.Fatalf("got %d, wanted %d", newVersion, wantNewVersion)
+		}
+
+		version, err = migrations.Version(db)
+		if err != nil {
+			t.Fatal(err)
+		}
+		if version != wantNewVersion {
+			t.Fatalf("got version %d, wanted %d", version, wantNewVersion)
+		}
+	}
+}
+
+func TestSetVersion(t *testing.T) {
+	db := connectDB()
+
+	_, _, err := migrations.Run(db, "init")
+	if err != nil {
+		t.Fatalf("init failed: %s", err)
+	}
+
+	migrations.Set([]migrations.Migration{
+		{Version: 1, Up: doPanic, Down: doPanic},
+		{Version: 2, Up: doPanic, Down: doPanic},
+		{Version: 3, Up: doPanic, Down: doPanic},
+	})
+
+	for i := 0; i < 5; i++ {
+		wantOldVersion := int64(i)
+		wantNewVersion := int64(i + 1)
+
+		oldVersion, newVersion, err := migrations.Run(
+			db, "set_version", fmt.Sprint(wantNewVersion))
+		if err != nil {
+			t.Fatal(err)
+		}
+		if oldVersion != wantOldVersion {
+			t.Fatalf("got %d, wanted %d", oldVersion, wantOldVersion)
+		}
+		if newVersion != wantNewVersion {
+			t.Fatalf("got %d, wanted %d", newVersion, wantNewVersion)
+		}
+
+		version, err := migrations.Version(db)
+		if err != nil {
+			t.Fatal(err)
+		}
+		if version != wantNewVersion {
+			t.Fatalf("got version %d, wanted %d", version, wantNewVersion)
+		}
+	}
+}
+
+func doNothing(db migrations.DB) error {
+	return nil
+}
+
+func doPanic(db migrations.DB) error {
+	panic("this migration should not be run")
+}