package db import ( "go_dreamfactory/lego/sys/mgo" "log" "os" "testing" ) var db *DB func TestMain(m *testing.M) { imgo, err := mgo.NewSys(mgo.SetMongodbUrl("mongodb://admin:123456@10.0.0.9:27018"), mgo.SetMongodbDatabase("dreamfactory")) if err != nil { log.Fatal(err) } db = &DB{ mgo: imgo, } defer os.Exit(m.Run()) }