package models
  
  import (

  	"pro2d/protos/pb"

  	"testing"

  )
  
  func TestNewRole(t *testing.T) {

  	//db.MongoDBClient.Database(conf.GlobalConf.AccountConf.DBName).Drop(context.Background())
  	//db.MongoDBClient.Database(conf.GlobalConf.GameConf.DBName).Drop(context.Background())

  	var uid int64 = 1
  	var role *RoleModel

  	if ok, role := RoleExistByUid(uid); ok {

  		role.Role.Device = "111111"

  		role.AddHero(&pb.Hero{
  			Id:         1,
  			RoleId:     role.Role.Id,
  			Type:       0,
  			Level:      0,
  			ReinCount:  0,
  			ReinPoint:  0,
  			Equipments: "",
  		})

  		role.Save()
  	}else {

  		role = NewRole(&pb.Role{Uid: uid})
  		role.Role.Id = 1

  		role.Role.Device = "222222"
  		role.Role.Level = 0
  		role.Create()
  		role.Index("uid")
  	}
  	print(role)
  }