MessageCenter/rpc/test/send_test.go

package test

import (
	"context"
	"log"
	"testing"
	"time"

	"app.yhyue.com/moapp/MessageCenter/rpc/messageclient"
	"github.com/tal-tech/go-zero/core/discov"
	"github.com/tal-tech/go-zero/zrpc"
)

func Test_SendMsg(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	std := messageclient.NewMessage(zrpc.MustNewClient(zrpc.RpcClientConf{Etcd: discov.EtcdConf{Hosts: []string{"127.0.0.1:2379"}, Key: "message.rpc"}}))
	req := &messageclient.SendMsgRequest{
		Appid:         "10000",
		ReceiveUserId: "6042120adca8410f1ef2ec84",
		ReceiveName:   "rjj",
		SendUserId:    "5fa3bb6d059e75bcdf8dab6a",
		SendName:      "xzh",
		Title:         "超级订阅",
		Content:       "444",
		Link:          "",
		CiteId:        int64(3),
		MsgType:       int64(4),
	}
	res, err := std.SendUserMsg(ctx, req)
	log.Println("err ", err, res)
}

func Test_FindUserMsg(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	std := messageclient.NewMessage(zrpc.MustNewClient(zrpc.RpcClientConf{Etcd: discov.EtcdConf{Hosts: []string{"127.0.0.1:2379"}, Key: "message.rpc"}}))
	req := &messageclient.FindUserMsgReq{
		Appid:         "10000",
		ReceiveUserId: "6042120adca8410f1ef2ec84",
		UserId:        "5fa3bb6d059e75bcdf8dab6a",
		OffSet:        int64(1),
		PageSize:      int64(10),
	}
	res, err := std.FindUserMsg(ctx, req)
	log.Println("err ", err, res.Code, res.Message, res.Count, res.Data)
}

func Test_ClassUnread(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
	std := messageclient.NewMessage(zrpc.MustNewClient(zrpc.RpcClientConf{Etcd: discov.EtcdConf{Hosts: []string{"127.0.0.1:2379"}, Key: "message.rpc"}}))
	req := &messageclient.GetClassUnreadCountReq{
		Appid:   "10000",
		UserId:  "6042120adca8410f1ef2ec84",
		MsgType: 4,
	}
	res, err := std.GetClassUnreadCount(ctx, req)
	log.Println("err ", err, res)
}