dex/functional/repo/testutil.go

29 lines
608 B
Go
Raw Normal View History

2015-08-18 05:57:27 +05:30
package repo
import (
"fmt"
"github.com/coreos/dex/db"
"github.com/go-gorp/gorp"
2015-08-18 05:57:27 +05:30
)
func initDB(dsn string) *gorp.DbMap {
c, err := db.NewConnection(db.Config{DSN: dsn})
if err != nil {
panic(fmt.Sprintf("Unable to connect to database: %v", err))
}
if err = c.DropTablesIfExists(); err != nil {
panic(fmt.Sprintf("Unable to drop database tables: %v", err))
}
if err = db.DropMigrationsTable(c); err != nil {
panic(fmt.Sprintf("Unable to drop migration table: %v", err))
2015-08-18 05:57:27 +05:30
}
if _, err = db.MigrateToLatest(c); err != nil {
panic(fmt.Sprintf("Unable to migrate: %v", err))
}
2015-08-18 05:57:27 +05:30
return c
}