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:       "1",
		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:        "5",
	})
	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:2379"},
			Key:   "jydocs.userlib.rpc",
		},
	})
	userLib := userlibclient.NewUserLib(client)
	resp, err := userLib.DocDelete(context.Background(), &userlibclient.UserDeleteRequest{
		DocId: "1",
	})
	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",
		Cost:100,
	})
	if err != nil {
		log.Println("UserDoc Collect call error:", err)
	}
	log.Print("UserDoc Collect call response:", resp)
}