Blame view

models/role_test.go 896 Bytes
ee23102d   zhangqijia   支持mongo, grpc接服务器
1
2
3
  package models
  
  import (
1584eb4b   zhangqijia   修复创建唯一索引的bug
4
5
6
  	"fmt"
  	"pro2d/components/db"
  	"pro2d/conf"
eb417b0b   zhangqijia   reactor mongo
7
  	"pro2d/protos/pb"
1584eb4b   zhangqijia   修复创建唯一索引的bug
8
  	"pro2d/utils"
ee23102d   zhangqijia   支持mongo, grpc接服务器
9
  	"testing"
ee23102d   zhangqijia   支持mongo, grpc接服务器
10
11
12
  )
  
  func TestNewRole(t *testing.T) {
1584eb4b   zhangqijia   修复创建唯一索引的bug
13
  	db.MongoDatabase = db.MongoClient.Database("game")
ee23102d   zhangqijia   支持mongo, grpc接服务器
14
  
1584eb4b   zhangqijia   修复创建唯一索引的bug
15
16
17
18
19
  	var uid = conf.SnowFlack.NextValStr()
  	ok, role := RoleExistByUid(uid)
  	if ok {
  		//uid存在 , 更新角色
  		role.Role.Device = "222222"
eb417b0b   zhangqijia   reactor mongo
20
21
22
23
24
25
26
27
28
  		role.AddHero(&pb.Hero{
  			Id:         1,
  			RoleId:     role.Role.Id,
  			Type:       0,
  			Level:      0,
  			ReinCount:  0,
  			ReinPoint:  0,
  			Equipments: "",
  		})
ee23102d   zhangqijia   支持mongo, grpc接服务器
29
30
  		role.Save()
  	}else {
1584eb4b   zhangqijia   修复创建唯一索引的bug
31
32
33
34
  		//uid不存在,创建角色
  		role = NewRole(1)
  		role.Role.Uid = uid
  		role.Role.Device = "111111"
ee23102d   zhangqijia   支持mongo, grpc接服务器
35
  		role.Role.Level = 0
1584eb4b   zhangqijia   修复创建唯一索引的bug
36
37
  		i, err := role.Create()
  		fmt.Println(i, err)
ee23102d   zhangqijia   支持mongo, grpc接服务器
38
39
  	}
  	print(role)
1584eb4b   zhangqijia   修复创建唯一索引的bug
40
41
42
43
44
45
46
  }
  
  func TestRoleIndex(t *testing.T) {
  	coll, keys := utils.FindIndex(pb.Role{})
  	for  _, index := range keys {
  		utils.Sugar.Debugf("coll: %s, key: %s", coll, index)
  	}
ee23102d   zhangqijia   支持mongo, grpc接服务器
47
  }