RestApiv2/internal/bd/testhelper.go

27 lines
468 B
Go

package bd
import (
"fmt"
"strings"
"testing"
)
func Testdb(t *testing.T, databaseurl string) (*Bd, func(...string)) {
t.Helper()
config := NewConfig()
config.BaseUrlBd = databaseurl
s := New(config)
if err := s.Open(); err != nil {
t.Fatal(err)
}
return s, func(tables ...string) {
if len(tables) > 0 {
if _, err := s.db.Exec(fmt.Sprintf("TRUNCATE %s CASCADE", strings.Join(tables, ", "))); err != nil {
t.Fatal(err)
}
}
s.Close()
}
}