userCenter/rpc/test/ent_test.go

package test

import (
	"context"
	// "flag"
	"log"
	"testing"
	"time"

	"bp.jydev.jianyu360.cn/BaseService/userCenter/rpc/pb"
	usercenterclient "bp.jydev.jianyu360.cn/BaseService/userCenter/rpc/usercenter"

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

	"github.com/zeromicro/go-zero/core/conf"
	"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_ExamineInfo
func Test_ExamineInfo(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 2*time.Second)
	FileSystem := 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 := 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 := usercenterclient.NewUserCenter(zrpc.MustNewClient(c.FileSystemConf))
	req := &pb.CheckEntReq{
		EntId: 14823,
	}
	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 := 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 := 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 := 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 := 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 := 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)
}