dex/functional/repo/testutil.go
2015-08-18 11:26:57 -07:00

25 lines
513 B
Go

package repo
import (
"fmt"
"github.com/coopernurse/gorp"
"github.com/coreos/dex/db"
)
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 = c.CreateTablesIfNotExists(); err != nil {
panic(fmt.Sprintf("Unable to create database tables: %v", err))
}
return c
}