更新单元测试

rpc/stdlib/test/doc_test.go

@@ -1,14 +1,15 @@
 package test
 
 import (
-	"app.yhyue.com/moapp/jy_docs/rpc/stdlib/stdlib"
-	"app.yhyue.com/moapp/jy_docs/rpc/stdlib/stdlibclient"
 	"context"
-	"github.com/tal-tech/go-zero/core/discov"
-	"github.com/tal-tech/go-zero/zrpc"
 	"log"
 	"testing"
 	"time"
+
+	"app.yhyue.com/moapp/jy_docs/rpc/stdlib/stdlib"
+	"app.yhyue.com/moapp/jy_docs/rpc/stdlib/stdlibclient"
+	"github.com/tal-tech/go-zero/core/discov"
+	"github.com/tal-tech/go-zero/zrpc"
 )
 
 func Test_DocOn(t *testing.T) {
@@ -37,3 +38,39 @@ func Test_DocOff(t *testing.T) {
 	//log.Println("err ", err)
 	//log.Println("req ", res)
 }
+
+func Test_DocActivity(t *testing.T) {
+	ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
+	std := stdlibclient.NewStdlib(zrpc.MustNewClient(zrpc.RpcClientConf{Etcd: discov.EtcdConf{Hosts: []string{"192.168.3.240:2379"}, Key: "jydocs.stdlib.rpc"}}))
+	req := &stdlib.DocActivityReq{PageNum: 0, PageSize: 10}
+	res, err := std.DocActivity(ctx, req)
+	log.Println("err ", err)
+	log.Println("req ", res)
+}
+
+func Test_DocStatistics(t *testing.T) {
+	ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
+	std := stdlibclient.NewStdlib(zrpc.MustNewClient(zrpc.RpcClientConf{Etcd: discov.EtcdConf{Hosts: []string{"192.168.3.240:2379"}, Key: "jydocs.stdlib.rpc"}}))
+	req := &stdlib.DocStatisticsReq{DocId: "111", DocStatisticsType: 1}
+	res, err := std.DocStatistics(ctx, req)
+	log.Println("err ", err)
+	log.Println("req ", res)
+}
+
+func Test_DocGetCheck(t *testing.T) {
+	ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
+	std := stdlibclient.NewStdlib(zrpc.MustNewClient(zrpc.RpcClientConf{Etcd: discov.EtcdConf{Hosts: []string{"192.168.3.240:2379"}, Key: "jydocs.stdlib.rpc"}}))
+	req := &stdlib.DocGetCheckReq{DocId: "111", UserId: "123"}
+	res, err := std.DocGetCheck(ctx, req)
+	log.Println("err ", err)
+	log.Println("req ", res)
+}
+
+func Test_DocGetContent(t *testing.T) {
+	ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
+	std := stdlibclient.NewStdlib(zrpc.MustNewClient(zrpc.RpcClientConf{Etcd: discov.EtcdConf{Hosts: []string{"192.168.3.240:2379"}, Key: "jydocs.stdlib.rpc"}}))
+	req := &stdlib.DocGetCheckReq{DocId: "111", UserId: "123"}
+	res, err := std.DocGetContent(ctx, req)
+	log.Println("err ", err)
+	log.Println("req ", res)
+}