You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
677 B
41 lines
677 B
5 years ago
|
package dao
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"flag"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/bilibili/kratos/pkg/conf/paladin"
|
||
|
|
"github.com/bilibili/kratos/pkg/testing/lich"
|
||
|
|
)
|
||
|
|
|
||
|
|
var d *dao
|
||
|
|
var ctx = context.Background()
|
||
|
|
|
||
|
|
func TestMain(m *testing.M) {
|
||
|
|
flag.Set("conf", "../../test")
|
||
|
|
flag.Set("f", "../../test/docker-compose.yaml")
|
||
|
|
flag.Parse()
|
||
|
|
disableLich := os.Getenv("DISABLE_LICH") != ""
|
||
|
|
if !disableLich {
|
||
|
|
if err := lich.Setup(); err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
var err error
|
||
|
|
if err = paladin.Init(); err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
var cf func()
|
||
|
|
if d, cf, err = newTestDao();err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
ret := m.Run()
|
||
|
|
cf()
|
||
|
|
if !disableLich {
|
||
|
|
_ = lich.Teardown()
|
||
|
|
}
|
||
|
|
os.Exit(ret)
|
||
|
|
}
|