add output

pull/120/head
1vn 2018-10-31 12:27:15 -04:00
parent c6c49ff995
commit 3c2c9d9076
3 changed files with 14 additions and 4 deletions

View File

@ -26,8 +26,16 @@ func main() {
args := flags.Args()
if len(args) > 1 && (args[0] == "create" || args[0] == "fix") {
if err := goose.Run(args[0], nil, *dir, args[1:]...); err != nil {
if len(args) > 1 && args[0] == "create" {
if err := goose.Run("create", nil, *dir, args[1:]...); err != nil {
log.Fatalf("goose run: %v", err)
}
return
}
// TODO clean up arg/flag parsing flow
if args[0] == "fix" {
if err := goose.Run("fix", nil, *dir); err != nil {
log.Fatalf("goose run: %v", err)
}
return

2
fix.go
View File

@ -3,6 +3,7 @@ package goose
import (
"fmt"
"os"
"path/filepath"
"strings"
)
@ -37,6 +38,7 @@ func Fix(dir string) error {
return err
}
log.Printf("RENAMED %s => %s", filepath.Base(oldPath), filepath.Base(newPath))
version++
}

View File

@ -25,7 +25,7 @@ func TestFix(t *testing.T) {
fmt.Sprintf("./goose -dir=%s create add_users", dir),
fmt.Sprintf("./goose -dir=%s create add_indices", dir),
fmt.Sprintf("./goose -dir=%s create update_users", dir),
fmt.Sprintf("./goose -dir=%s sqlite3 sql.db fix", dir),
fmt.Sprintf("./goose -dir=%s fix", dir),
}
for _, cmd := range commands {
@ -54,7 +54,7 @@ func TestFix(t *testing.T) {
commands = []string{
fmt.Sprintf("./goose -dir=%s create remove_column", dir),
fmt.Sprintf("./goose -dir=%s create create_books_table", dir),
fmt.Sprintf("./goose -dir=%s sqlite3 sql.db fix", dir),
fmt.Sprintf("./goose -dir=%s fix", dir),
}
for _, cmd := range commands {