messageCenter/rpc/messagecenter/test/message_test.go

package test

import (
	"context"
	"flag"
	"github.com/zeromicro/go-zero/core/conf"
	"github.com/zeromicro/go-zero/zrpc"
	"log"
	"messagecenter/rpc/messagecenter/internal/config"
	"messagecenter/rpc/messagecenter/messagecenter"
	"testing"
	"time"
)

var configFile = flag.String("f", "../etc/messagecenter.yaml", "the config file")
var c config.Config

func init() {
	conf.MustLoad(*configFile, &c)
}

//数量查询
func Test_Count(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
	TestSystem := messagecenter.NewMessageCenter(zrpc.MustNewClient(c.TestConf))
	req := &messagecenter.CountReq{
		MsgType:   5,
		UserType:  2,
		SendId:    "",
		EntId:     10000,
		ReceiveId: "2",
	}
	res, err := TestSystem.Count(ctx, req)
	log.Println("res ", res)
	log.Println("err ", err)
}

//用户数量查询
func Test_UserList(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
	TestSystem := messagecenter.NewMessageCenter(zrpc.MustNewClient(c.TestConf))
	req := &messagecenter.UserReq{
		Phone:             "",
		UserType:          1,
		StartTime:         "",
		CustomerServiceId: "3",
		EndTime:           "",
	}
	res, err := TestSystem.UserList(ctx, req)
	log.Println("res ", res)
	log.Println("err ", err)
}

//聊天信息查询
func Test_FindMessage(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
	TestSystem := messagecenter.NewMessageCenter(zrpc.MustNewClient(c.TestConf))
	req := &messagecenter.MessageReq{
		MsgType:   5,
		UserType:  1,
		SendId:    "2",
		OwnId:     "3",
		PageSize:  1,
		PageIndex: 2,
	}
	res, err := TestSystem.FindMessage(ctx, req)
	log.Println("res ", res)
	log.Println("err ", err)
}

//聊天保存
func Test_SaveMessage(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
	TestSystem := messagecenter.NewMessageCenter(zrpc.MustNewClient(c.TestConf))
	req := &messagecenter.MessageEntity{
		Title:     "11",
		Content:   "请问在吗?",
		Item:      8,
		ItemType:  4,
		Link:      "1111",
		Appid:     "10000",
		ReceiveId: "2",
		SendId:    "3",
		OwnType:   int64(2),
		Type:      1,
	}
	res, err := TestSystem.SaveMessage(ctx, req)
	log.Println("res ", res)
	log.Println("err ", err)
}