mirror of https://github.com/pressly/goose.git
82 lines
1.8 KiB
Go
82 lines
1.8 KiB
Go
package goose
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestFix(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "tmptest")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
defer os.RemoveAll(dir) // clean up
|
|
defer os.Remove("goose") // clean up
|
|
|
|
commands := []string{
|
|
"go build -i -o goose ./cmd/goose",
|
|
fmt.Sprintf("./goose -dir=%s create yolo", dir),
|
|
fmt.Sprintf("./goose -dir=%s create yolo", dir),
|
|
fmt.Sprintf("./goose -dir=%s create yolo", dir),
|
|
fmt.Sprintf("./goose -dir=%s create yolo", dir),
|
|
fmt.Sprintf("./goose -dir=%s sqlite3 sql.db fix", dir),
|
|
}
|
|
|
|
for _, cmd := range commands {
|
|
args := strings.Split(cmd, " ")
|
|
time.Sleep(1 * time.Second)
|
|
out, err := exec.Command(args[0], args[1:]...).CombinedOutput()
|
|
if err != nil {
|
|
t.Fatalf("%s:\n%v\n\n%s", err, cmd, out)
|
|
}
|
|
}
|
|
|
|
files, err := ioutil.ReadDir(dir)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// check that the files are in order
|
|
for i, f := range files {
|
|
expected := fmt.Sprintf("%05v", i+1)
|
|
if !strings.HasPrefix(f.Name(), expected) {
|
|
t.Errorf("failed to find %s prefix in %s", expected, f.Name())
|
|
}
|
|
}
|
|
|
|
// add more migrations and then fix it
|
|
commands = []string{
|
|
fmt.Sprintf("./goose -dir=%s create yolo", dir),
|
|
fmt.Sprintf("./goose -dir=%s create yolo", dir),
|
|
fmt.Sprintf("./goose -dir=%s sqlite3 sql.db fix", dir),
|
|
}
|
|
|
|
for _, cmd := range commands {
|
|
args := strings.Split(cmd, " ")
|
|
time.Sleep(1 * time.Second)
|
|
out, err := exec.Command(args[0], args[1:]...).CombinedOutput()
|
|
if err != nil {
|
|
t.Fatalf("%s:\n%v\n\n%s", err, cmd, out)
|
|
}
|
|
}
|
|
|
|
files, err = ioutil.ReadDir(dir)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// check that the files still in order
|
|
for i, f := range files {
|
|
expected := fmt.Sprintf("%05v", i+1)
|
|
if !strings.HasPrefix(f.Name(), expected) {
|
|
t.Errorf("failed to find %s prefix in %s", expected, f.Name())
|
|
}
|
|
}
|
|
}
|