Blame view

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