MessageCenter/rpc/test/message_test.go

package test

import (
	"app.yhyue.com/moapp/MessageCenter/rpc/messageclient"
	"context"
	"github.com/tal-tech/go-zero/core/discov"
	"github.com/tal-tech/go-zero/zrpc"
	"log"
	"testing"
	"time"
)

func  Test_ChangeReadStatus(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"}}))
	//std := stdlibclient.NewStdlib(zrpc.MustNewClient(zrpc.RpcClientConf{Etcd: discov.EtcdConf{Hosts: []string{"127.0.0.1:2379"}, Key: "jydocs.stdlib.rpc"}}))
	req := &messageclient.ChangeReadStatusRequest{Id: 1,ReadStatus: 1,Appid: "10000"}
	res, err := std.ChangeReadStatus(ctx, req)
	log.Println("err ", err,res)
	//if res.State == true {
	//	log.Println("req true")
	//} else {
	//	log.Println("req false", res)
	//}
}
func  Test_DeleteSingleMessage(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"}}))
	//std := stdlibclient.NewStdlib(zrpc.MustNewClient(zrpc.RpcClientConf{Etcd: discov.EtcdConf{Hosts: []string{"127.0.0.1:2379"}, Key: "jydocs.stdlib.rpc"}}))
	req := &messageclient.DeleteSingleMessageRequest{Id: 3,Appid: "10000"}
	res, err := std.DeleteSingleMessage(ctx, req)
	log.Println("err ", err,res)
	//if res.State == true {
	//	log.Println("req true")
	//} else {
	//	log.Println("req false", res)
	//}
}
func  Test_DeleteMultipleMessage(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"}}))
	//std := stdlibclient.NewStdlib(zrpc.MustNewClient(zrpc.RpcClientConf{Etcd: discov.EtcdConf{Hosts: []string{"127.0.0.1:2379"}, Key: "jydocs.stdlib.rpc"}}))
	req := &messageclient.DeleteMultipleMessageRequest{Id: "4,5",Appid: "10000"}
	res, err := std.DeleteMultipleMessage(ctx, req)
	log.Println("err ", err,res)
	//if res.State == true {
	//	log.Println("req true")
	//} else {
	//	log.Println("req false", res)
	//}
}
func  Test_LastMessage(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"}}))
	//std := stdlibclient.NewStdlib(zrpc.MustNewClient(zrpc.RpcClientConf{Etcd: discov.EtcdConf{Hosts: []string{"127.0.0.1:2379"}, Key: "jydocs.stdlib.rpc"}}))
	req := &messageclient.GetLastMessageReq{UserId: "6042120adca8410f1ef2ec84",Appid: "10000",MsgType: 4}
	res, err := std.GetLastMessage(ctx, req)
	log.Println("err ", err,res)
	//if res.State == true {
	//	log.Println("req true")
	//} else {
	//	log.Println("req false", res)
	//}
}
func  Test_CountUnreadMessage(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"}}))
	//std := stdlibclient.NewStdlib(zrpc.MustNewClient(zrpc.RpcClientConf{Etcd: discov.EtcdConf{Hosts: []string{"127.0.0.1:2379"}, Key: "jydocs.stdlib.rpc"}}))
	req := &messageclient.GetUnreadCountRequest{UserId: "6042120adca8410f1ef2ec84",Appid: "10000"}
	res, err := std.GetUnreadCount(ctx, req)
	log.Println("err ", err,res)
	//if res.State == true {
	//	log.Println("req true")
	//} else {
	//	log.Println("req false", res)
	//}
}