userCenter/service/ent_test.go

package service

import (
	"context"

	// "flag"
	"log"
	"testing"
	"time"

	resourcepb "bp.jydev.jianyu360.cn/BaseService/resourceCenter/rpc/pb"
	"bp.jydev.jianyu360.cn/BaseService/resourceCenter/rpc/resource"
	"bp.jydev.jianyu360.cn/BaseService/userCenter/rpc/pb"

	"github.com/zeromicro/go-zero/core/discov"
	"github.com/zeromicro/go-zero/zrpc"
)

//type Config struct {
//	zrpc.RpcServerConf
//	Node           int    // 节点
//	CalleeId       string // 服务名字
//	FileSystemConf zrpc.RpcClientConf
//}
//
//var c Config
//
//func init() {
//	conf.MustLoad("usercenter.yaml", &c)
//}

// go test -v -run Test_GetStatusByCode
func Test_GetStatusByCode(t *testing.T) {
	//ctx, _ := context.WithTimeout(context.Background(), 2*time.Second)
	FileSystem := AutoInit() //usercenterclient.NewUserCenter(zrpc.MustNewClient(c.FileSystemConf))
	req := &pb.GetStatusByCodeReq{
		Code:  "91230881MA18Y5NM11",
		Phone: "15225181827",
	}
	res, err := FileSystem.GetStatusByCode(ctx, req)
	log.Println("err ", err)
	log.Println("res:", res)
}

// go test -v -run Test_ExamineInfo
func Test_ExamineInfo(t *testing.T) {
	//ctx, _ := context.WithTimeout(context.Background(), 2*time.Second)
	FileSystem := AutoInit() //usercenterclient.NewUserCenter(zrpc.MustNewClient(c.FileSystemConf))
	req := &pb.CheckExamineReq{
		ExamineId: 18,
	}
	res, err := FileSystem.ExamineInfo(ctx, req)
	log.Println("err ", err)
	log.Println("res:", res)
}

// go test -v -run Test_UpdateEnt
func Test_UpdateEnt(t *testing.T) {
	//ctx, _ := context.WithTimeout(context.Background(), 2*time.Second)
	FileSystem := AutoInit() //usercenterclient.NewUserCenter(zrpc.MustNewClient(c.FileSystemConf))
	req := &pb.EntUpdateReq{
		EntId:      14823,
		UpdateType: 2,
	}
	res, err := FileSystem.EntUpdate(ctx, req)
	log.Println("err ", err)
	log.Println("res:", res)
}

// go test -v -run Test_EntInfo
func Test_EntInfo(t *testing.T) {
	//ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	FileSystem := AutoInit() //usercenterclient.NewUserCenter(zrpc.MustNewClient(c.FileSystemConf))
	req := &pb.CheckEntReq{
		EntId: 14783,
	}
	res, err := FileSystem.EntInfo(ctx, req)
	log.Println("err ", err)
	log.Println("res:", res)
}

// go test -v -run Test_Resource
func Test_Resource(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	client := zrpc.MustNewClient(zrpc.RpcClientConf{
		Etcd: discov.EtcdConf{
			Hosts: []string{"192.168.3.240:2379"},
			Key:   "resource.rpc",
		},
	})
	fileSystem := resource.NewResource(client)
	req := &resourcepb.PowerReq{
		Appid:     "10000",
		GoodsCode: "xxfb_gyxx",
		EntId:     1,
		//StartTime: time.Now().Format("2006-01-02 15:04:05"),
		//EndTime:   time.Now().AddDate(1, 0, 0).Format("2006-01-02 15:04:05"),
		Type: 1,
	}
	resp, err := fileSystem.PowerHandle(ctx, req)
	log.Println("err:", err)
	log.Println("resp:", resp)
}

// go test -v -run Test_CheckEnt
func Test_CheckEnt(t *testing.T) {
	//ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	FileSystem := AutoInit() //usercenterclient.NewUserCenter(zrpc.MustNewClient(c.FileSystemConf))
	req := &pb.CheckEntReq{
		EntId: 1,
	}
	res, err := FileSystem.CheckEnt(ctx, req)
	log.Println("err ", err)
	log.Println("res:", res)
}

//
func Test_EntAuth(t *testing.T) {
	//ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	FileSystem := AutoInit() //usercenterclient.NewUserCenter(zrpc.MustNewClient(c.FileSystemConf))
	req := &pb.EntAuthReq{
		Name:             "张鑫磊企业2",
		AuthPhone:        "15538267985",
		CreditCode:       "asdqweqweasd",
		Phone:            "15538267985",
		ComPanyType:      "1",
		AreaNumber:       "1c00100010001",
		Business:         "https://www.baidu.com",
		OfficialLetter:   "https://www.jianyu360.cn",
		AuthName:         "郭靖2",
		OrganizationType: "2",
		EntId:            21,
	}

	res, err := FileSystem.EntAuth(ctx, req)
	log.Println("err ", err)
	log.Println("req ", res)
}

// go test -v -run Test_EntExamine
func Test_EntExamine(t *testing.T) {
	//ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	FileSystem := AutoInit() //usercenterclient.NewUserCenter(zrpc.MustNewClient(c.FileSystemConf))
	req := &pb.ExamineReq{
		EntId:     14823,
		AuthType:  "2",
		Reason:    "",
		AuditUser: "少爷",
		ExamineId: "18",
	}

	res, err := FileSystem.EntExamine(ctx, req)
	log.Println("err ", err)
	log.Println("req ", res)
}

// go test -v -run Test_EntList
func Test_EntList(t *testing.T) {
	//ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	FileSystem := AutoInit() //usercenterclient.NewUserCenter(zrpc.MustNewClient(c.FileSystemConf))
	req := &pb.EntListReq{
		Name:       "企业",
		AuthStatus: "1",
		PageNum:    "1",
		PageSize:   "10",
		// CreditCode:      data.CreditCode,
		// CreateStartTime: data.CreateStartTime,
		// CreateEndTime:   data.CreateEndTime,
		// ValidStartTime:  data.ValidStartTime,
		// ValidEndTime:    data.ValidEndTime,
		// FrozenStatus:    data.FrozenStatus,
		// RegPhone:        data.RegPhone,
		// AuthPhone:       data.AuthPhone,
	}

	res, err := FileSystem.EntList(ctx, req)
	log.Println("err ", err)
	log.Println("res:", res)
}

// go test -v -run Test_ExamineList
func Test_ExamineList(t *testing.T) {
	//ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	FileSystem := AutoInit() //usercenterclient.NewUserCenter(zrpc.MustNewClient(c.FileSystemConf))
	req := &pb.ExamineListReq{
		Name:          "中凯",
		AuthPhone:     "",
		RegPhone:      "",
		AuthType:      "",
		AuthStartTime: "",
		AuthEndTime:   "",
		PageNum:       "",
		PageSize:      "",
	}

	res, err := FileSystem.ExamineList(ctx, req)
	log.Println("err ", err)
	log.Println("res:", res)
}

// go test -v -run Test_UserInfo
func Test_UserInfo(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	FileSystem := AutoInit() //usercenterclient.NewUserCenter(zrpc.MustNewClient(c.FileSystemConf))
	req := &pb.UserReq{
		UserId: 57685,
		Types:  "f",
		// EntId:     14904,
		// EntUserId: 4254,
	}

	res, err := FileSystem.GetUserInfo(ctx, req)
	log.Println("err ", err)
	log.Println("res:", res)
}

// go test -v -run Test_EntUserInfo
//获取用户企业信息
func Test_EntUserInfo(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	FileSystem := AutoInit() //usercenterclient.NewUserCenter(zrpc.MustNewClient(c.FileSystemConf))
	req := &pb.EntUserReq{
		EntUserId: 4254,
		AppId:     "10000",
		EntId:     14904,
	}

	res, err := FileSystem.GetEntUserInfo(ctx, req)

	log.Println("res:", res)
	log.Println("err ", err)
}

// go test -v -run Test_EntUserList
//获取企业列表/员工列表
func Test_EntUserList(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	FileSystem := AutoInit() //usercenterclient.NewUserCenter(zrpc.MustNewClient(c.FileSystemConf))
	req := &pb.EntUserListReq{
		EntId: 149041,
		AppId: "10000",
		// DeptId: 3821,  //非必填
		// Name: "后端",//非必填
	}

	res, err := FileSystem.GetEntUserList(ctx, req)

	log.Println("res:", res)
	log.Println("err ", err)
}

// go test -v -run Test_CheckIsEntAdmin
//查看用户是否是管理员 1-企业管理员 2-部门管理员 3-员工
func Test_CheckIsEntAdmin(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	FileSystem := AutoInit() //usercenterclient.NewUserCenter(zrpc.MustNewClient(c.FileSystemConf))
	req := &pb.EntUserReq{
		EntUserId: 4254,
		AppId:     "10000",
		EntId:     14904,
	}

	res, err := FileSystem.CheckIsEntAdmin(ctx, req)

	log.Println("res:", res)
	log.Println("err ", err)
}