jy_docs/rpc/userlib/test/userLib_test.go

package test

import (
	"app.yhyue.com/moapp/jydocs/rpc/userlib/userlib"
	"app.yhyue.com/moapp/jydocs/rpc/userlib/userlibclient"
	"context"
	"github.com/tal-tech/go-zero/core/discov"
	"github.com/tal-tech/go-zero/zrpc"
	"log"
	"testing"
)

//用户收藏单元测试
func Test_UserDocCollect(t *testing.T) {
	client := zrpc.MustNewClient(zrpc.RpcClientConf{
		Etcd: discov.EtcdConf{
			Hosts: []string{"192.168.3.240:2379"},
			Key:   "jydocs.userlib.rpc",
		},
	})
	userLib := userlibclient.NewUserLib(client)
	resp, err := userLib.DocCollect(context.Background(), &userlibclient.UserCollectRequest{
		DocId:        "1",
		UserId:       "1",
		SourceUserId: "2",
		DocName:      "2020招标文件白皮书.docx",
		DocFileType:  userlib.UserDocFileType_Word,
		DocFileSize:  1002,
		DocPageSize:  20,
		DocCategory:  userlib.UserDocCategory_Collect,
		DocFileSuffix:"docx",
		Cost:0,
	})
	if err != nil {
		log.Println("UserDoc Collect call error:", err)
	}
	log.Print("UserDoc Collect call response:", resp)
}
//用户转存单元测试
func Test_UserDocDwnload(t *testing.T) {
	client := zrpc.MustNewClient(zrpc.RpcClientConf{
		Etcd: discov.EtcdConf{
			Hosts: []string{"127.0.0.1:2379"},
			Key:   "jydocs.userlib.rpc",
		},
	})
	userLib := userlibclient.NewUserLib(client)
	resp, err := userLib.DocDownload(context.Background(), &userlibclient.UserCollectRequest{
		DocId:        "1",
		UserId:       "wh5",
		SourceUserId: "2",
		DocName:      "2020招标文件白皮书.docx",
		DocFileType:  userlib.UserDocFileType_Word,
		DocFileSize:  1002,
		DocPageSize:  20,
		DocCategory:  userlib.UserDocCategory_Collect,
		DocFileSuffix:"docx",
		Cost:100,
	})
	if err != nil {
		log.Println("UserDoc Collect call error:", err)
	}
	log.Print("UserDoc Collect call response:", resp)
}