forked from olliecoleman/alloy
-
Notifications
You must be signed in to change notification settings - Fork 0
/
db.go
39 lines (33 loc) · 917 Bytes
/
db.go
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
package testutils
import (
"fmt"
"log"
"github.com/gobuffalo/envy"
"github.com/olliecoleman/alloy/app/services"
"github.com/pressly/goose"
)
func SetupDB() {
td := envy.Get("TEST_DATABASE_URL", "postgres://postgres:@localhost/alloy_test?sslmode=disable")
envy.Set("DATABASE_URL", td)
migrationsDir, err := envy.MustGet("MIGRATIONS_DIR")
if err != nil {
log.Fatal("MIGRATIONS_DIR variable not set")
}
services.InitDB()
if err := goose.Run("up", services.DB.DB, migrationsDir); err != nil {
log.Fatal("ERR", err)
}
}
func DropDB() {
migrationsDir, err := envy.MustGet("MIGRATIONS_DIR")
if err != nil {
log.Fatal("MIGRATIONS_DIR variable not set")
}
if err := goose.Run("down", services.DB.DB, migrationsDir); err != nil {
log.Fatal("ERR", err)
}
}
func ResetTable(tablename string) {
log.Printf("DELETE from %s", tablename)
services.DB.Exec(fmt.Sprintf(`DELETE FROM %s`, tablename))
}