gitee.com/woood2/luca@v1.0.4/internal/repository/main_test.go (about)

     1  package repository
     2  
     3  import (
     4  	"gitee.com/woood2/luca/internal/conf"
     5  	"gitee.com/woood2/luca/internal/db"
     6  	"gitee.com/woood2/luca/internal/trace"
     7  	"gitee.com/woood2/luca/internal/util"
     8  	"go.uber.org/zap"
     9  	"testing"
    10  )
    11  
    12  var appRepo AppRepo
    13  
    14  func TestMain(m *testing.M) {
    15  	if !util.CI() {
    16  		attr := conf.Load("application.yml", "configs/application.yml")
    17  		//gorm
    18  		gormDB := db.NewGormDB(attr.Env, attr.Mysql, zap.NewExample())
    19  		appRepo = NewAppRepoImpl(zap.NewExample(), gormDB)
    20  		//mongo
    21  		mongoDB := db.NewMongoDB(attr.Mongo)
    22  		db.SetMongoDB(mongoDB)
    23  		//zipkin
    24  		trace.Open(attr.Zipkin)
    25  		defer trace.Close()
    26  	}
    27  	m.Run()
    28  }