jy_docs/rpc/userlib/test/userLib_test.go

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{"127.0.0.1:2379"},
			Key:   "jydocs.userlib.rpc",
		},
	})
	userLib := userlibclient.NewUserLib(client)
	resp, err := userLib.DocCollect(context.Background(), &userlibclient.UserCollectRequest{
		DocId:  "1",
		UserId: "2",
		AppId:  10000,
		Cost:   0,
	})
	if err != nil {
		log.Println("UserDoc Collect call error:", err)
	}
	log.Print("UserDoc Collect call response:", resp)
}

//用户取消收藏单元测试
func Test_UserCancelDocCollect(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.DocCancelCollect(context.Background(), &userlibclient.UserDeleteRequest{
		DocId: "3",
	})
	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{"127.0.0.1:2380"},
			Key:   "jydocs.userlib.rpc",
		},
	})
	userLib := userlibclient.NewUserLib(client)
	resp, err := userLib.DocDelete(context.Background(), &userlibclient.UserRecoveryRequest{
		UserDocId: int32(10),
		AppId:     int64(10000),
	})
	if err != nil {
		log.Println("UserDoc Collect call error:", err)
	}
	log.Print("UserDoc Collect call response:", resp)
}

//删除找回
func Test_UserDocRestore(t *testing.T) {
	client := zrpc.MustNewClient(zrpc.RpcClientConf{
		Etcd: discov.EtcdConf{
			Hosts: []string{"127.0.0.1:2380"},
			Key:   "jydocs.userlib.rpc",
		},
	})
	userLib := userlibclient.NewUserLib(client)
	resp, err := userLib.DocRestore(context.Background(), &userlibclient.UserRecoveryRequest{
		UserDocId: int32(10),
		AppId:     int64(10000),
	})
	if err != nil {
		log.Println("UserDoc Collect call error:", err)
	}
	log.Print("UserDoc Collect call response:", resp)
}

//永久删除
func Test_UserDocPermanentDelete(t *testing.T) {
	client := zrpc.MustNewClient(zrpc.RpcClientConf{
		Etcd: discov.EtcdConf{
			Hosts: []string{"127.0.0.1:2380"},
			Key:   "jydocs.userlib.rpc",
		},
	})
	userLib := userlibclient.NewUserLib(client)
	resp, err := userLib.DocPermanentDelete(context.Background(), &userlibclient.UserRecoveryRequest{
		UserDocId: int32(10),
		AppId:     int64(10000),
	})
	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:2380"},
			Key:   "jydocs.userlib.rpc",
		},
	})
	userLib := userlibclient.NewUserLib(client)
	resp, err := userLib.DocDownload(context.Background(), &userlibclient.UserCollectRequest{
		DocId:  "123",
		UserId: "wh5",
		Cost:   1000,
		AppId:  int64(10000),
	})
	if err != nil {
		log.Println("UserDoc Collect call error:", err)
	}
	log.Print("UserDoc Collect call response:", resp.Message, resp.Code)
}