package test import ( "app.yhyue.com/moapp/jy_docs/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_UserDocDelete(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.DocDelete(context.Background(), &userlibclient.UserDeleteRequest{ //DocId: "8", }) 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) }