1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
package migrations
import (
"database/sql"
"fmt"
"io/ioutil"
"math/rand"
"os"
"os/user"
"path"
"path/filepath"
"reflect"
"testing"
"github.com/go-sql-driver/mysql"
_ "github.com/mattn/go-sqlite3"
migrate "github.com/rubenv/sql-migrate"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestSQLiteMigrations(t *testing.T) {
subdir := "sqlite3"
db, err := sql.Open(subdir, ":memory:")
require.NoError(t, err, "Unable to open sqlite connection")
runMigrations(t, db, subdir)
db.Close()
}
func TestMySQLMigrations(t *testing.T) {
if os.Getenv("MYSQL_TEST") == "" {
t.Skip("No MYSQL_TEST environment variable")
}
r := require.New(t)
subdir := "mysql"
dsn := mysql.NewConfig()
dsn.Net = "tcp"
dsn.ParseTime = true
dsn.Addr = os.Getenv("MYSQL_TEST_HOST")
dsn.Passwd = os.Getenv("MYSQL_TEST_PASS")
u, _ := user.Current()
if testUser, ok := os.LookupEnv("MYSQL_TEST_USER"); ok {
dsn.User = testUser
} else {
dsn.User = u.Username
}
db, err := sql.Open(subdir, dsn.FormatDSN())
r.NoError(err, "Unable to open mysql connection")
rnd := make([]byte, 4)
rand.Read(rnd)
suffix := fmt.Sprintf("_%x", string(rnd))
_, err = db.Exec("CREATE DATABASE migrations_test" + suffix)
r.NoError(err)
_, err = db.Exec("USE migrations_test" + suffix)
r.NoError(err)
runMigrations(t, db, subdir)
db.Exec("DROP DATABASE IF EXISTS migrations_test" + suffix)
db.Close()
}
func runMigrations(t *testing.T, db *sql.DB, directory string) {
a := assert.New(t)
r := require.New(t)
m := &migrate.FileMigrationSource{
Dir: directory,
}
files, err := filepath.Glob(path.Join(directory, "*.sql"))
// Verify that there is at least one migration to run
r.NoError(err, "No migrations to run")
r.NotEmpty(files)
// Verify that migrating up works
n, err := migrate.Exec(db, directory, m, migrate.Up)
if a.NoError(err) {
a.Len(files, n)
}
// Verify that a subsequent run has no migrations
n, err = migrate.Exec(db, directory, m, migrate.Up)
if a.NoError(err) {
a.Equal(0, n)
}
// Verify that reversing migrations works
n, err = migrate.Exec(db, directory, m, migrate.Down)
if a.NoError(err) {
a.Len(files, n)
}
}
// Test that all migration directories contain the same set of migrations files.
func TestMigationDirectoryContents(t *testing.T) {
names := map[string][]string{}
contents, err := ioutil.ReadDir(".")
assert.NoError(t, err)
for _, i := range contents {
if i.IsDir() {
dir := path.Join(i.Name(), "*.sql")
files, _ := filepath.Glob(dir)
trimmed := []string{}
for _, f := range files {
trimmed = append(trimmed, filepath.Base(f))
}
names[i.Name()] = trimmed
}
}
// Use one entry from the `names` map as a reference for all the others.
first := names[reflect.ValueOf(names).MapKeys()[0].String()]
for _, v := range names {
assert.EqualValues(t, first, v)
}
}
|
