init_test.go 894 Bytes
package models

import (
	"context"
	"pro2d/components/db"
	"pro2d/conf"
	_ "pro2d/conf"
	"pro2d/utils"
	"testing"
)


func TestInitModels(t *testing.T) {
	err := db.Connect(conf.GlobalConf.MongoConf.User, conf.GlobalConf.MongoConf.Password, conf.GlobalConf.MongoConf.Host, conf.GlobalConf.MongoConf.Port, conf.GlobalConf.MongoConf.MaxNum, conf.GlobalConf.MongoConf.TimeOut,"account")
	if err != nil {
		utils.Sugar.Errorf("mongodb init error: %v", err)
	}

	InitAccountServerModels()
	db.MongoClient.Disconnect(context.TODO())
	err = db.Connect(conf.GlobalConf.MongoConf.User, conf.GlobalConf.MongoConf.Password, conf.GlobalConf.MongoConf.Host, conf.GlobalConf.MongoConf.Port, conf.GlobalConf.MongoConf.MaxNum, conf.GlobalConf.MongoConf.TimeOut,"game")
	if err != nil {
		utils.Sugar.Errorf("mongodb init error: %v", err)
	}
	InitGameServerModels()
	db.MongoClient.Disconnect(context.TODO())
}