Blame view

models/role_test.go 994 Bytes
ee23102d   zhangqijia   支持mongo, grpc接服务器
1
2
3
  package models
  
  import (
1584eb4b   zhangqijia   修复创建唯一索引的bug
4
  	"fmt"
765431a4   zhangqijia   增加schema接口, 抽象 mo...
5
  	"pro2d/common"
436e0af4   zhangqijia   reactor: dir; Ac...
6
  	"pro2d/common/db/mongoproxy"
765431a4   zhangqijia   增加schema接口, 抽象 mo...
7
  	"pro2d/common/logger"
7f269318   zhangqijia   add pb.go; 添加关闭连接...
8
  	"pro2d/pb"
0e5d52de   zhangqijia   reactor: 重构底层框架1.0
9
  	"pro2d/utils"
ee23102d   zhangqijia   支持mongo, grpc接服务器
10
  	"testing"
ee23102d   zhangqijia   支持mongo, grpc接服务器
11
12
13
  )
  
  func TestNewRole(t *testing.T) {
436e0af4   zhangqijia   reactor: dir; Ac...
14
  	err := mongoproxy.ConnectMongo(common.GlobalConf.MongoConf)
765431a4   zhangqijia   增加schema接口, 抽象 mo...
15
16
17
18
  	if err != nil {
  		logger.Error(err)
  		return
  	}
ee23102d   zhangqijia   支持mongo, grpc接服务器
19
  
33ea26ab   zhangqijia   使用schema封装mongo
20
  	var uid = "141815055745814528"
cad2b7f3   zhangqijia   reactor: 重构目录, 重构...
21
22
  	role := RoleExistByUid(uid)
  	if role != nil {
1584eb4b   zhangqijia   修复创建唯一索引的bug
23
  		//uid存在 , 更新角色
33ea26ab   zhangqijia   使用schema封装mongo
24
25
26
27
28
29
30
31
32
  		//role.AddHero(&pb.Hero{
  		//	Id:         1,
  		//	RoleId:     role.Role.Id,
  		//	Type:       0,
  		//	Level:      0,
  		//	ReinCount:  0,
  		//	ReinPoint:  0,
  		//	Equipments: "",
  		//})
7f269318   zhangqijia   add pb.go; 添加关闭连接...
33
  		role.SetProperty("Device", "1111")
33ea26ab   zhangqijia   使用schema封装mongo
34
  		//role.Save()
ee23102d   zhangqijia   支持mongo, grpc接服务器
35
  	}else {
1584eb4b   zhangqijia   修复创建唯一索引的bug
36
  		//uid不存在,创建角色
33ea26ab   zhangqijia   使用schema封装mongo
37
  		role = NewRole("1")
1584eb4b   zhangqijia   修复创建唯一索引的bug
38
39
  		role.Role.Uid = uid
  		role.Role.Device = "111111"
ee23102d   zhangqijia   支持mongo, grpc接服务器
40
  		role.Role.Level = 0
33ea26ab   zhangqijia   使用schema封装mongo
41
42
  		err := role.Create()
  		fmt.Println(err)
ee23102d   zhangqijia   支持mongo, grpc接服务器
43
44
  	}
  	print(role)
1584eb4b   zhangqijia   修复创建唯一索引的bug
45
46
47
48
49
  }
  
  func TestRoleIndex(t *testing.T) {
  	coll, keys := utils.FindIndex(pb.Role{})
  	for  _, index := range keys {
cad2b7f3   zhangqijia   reactor: 重构目录, 重构...
50
  		logger.Debug("coll: %s, key: %s", coll, index)
1584eb4b   zhangqijia   修复创建唯一索引的bug
51
  	}
ee23102d   zhangqijia   支持mongo, grpc接服务器
52
  }