jy_docs/rpc/stdlib/test/doc_test.go

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"
)

func Test_DocOn(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.DocChangeReq{Id: "123", Reason: 1}
	res, err := std.DocOn(ctx, req)
	log.Println("err ", err)
	if res.State == true {
		log.Println("req true")
	} else {
		log.Println("req false", res)
	}
}
func Test_DocOff(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 120*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.DocChangeReq{Id: "123", Reason: 10}
	res, err := std.DocOff(ctx, req)
	log.Println("err ", err)
	if res.State == true {
		log.Println("req true")
	} else {
		log.Println("req false", res)
	}
	//log.Println("err ", err)
	//log.Println("req ", res)
}