openbilibili/app/interface/main/mcn/dao/msg/msg_test.go

package msg

import (
	"context"
	"testing"

	"github.com/smartystreets/goconvey/convey"

	"gopkg.in/h2non/gock.v1"
)

func TestMsgMutliSendSysMsg(t *testing.T) {
	convey.Convey("MutliSendSysMsg", t, func(ctx convey.C) {
		var (
			c       = context.Background()
			allUids = []int64{}
			mc      = ""
			title   = ""
			context = ""
			ip      = ""
		)
		ctx.Convey("When everything goes positive", func(ctx convey.C) {
			err := d.MutliSendSysMsg(c, allUids, mc, title, context, ip)
			ctx.Convey("Then err should be nil.", func(ctx convey.C) {
				ctx.So(err, convey.ShouldBeNil)
			})
		})
	})
}

func TestMsgSendSysMsg(t *testing.T) {
	convey.Convey("SendSysMsg", t, func(ctx convey.C) {
		var (
			c       = context.Background()
			uids    = []int64{}
			mc      = ""
			title   = ""
			context = ""
			ip      = ""
		)
		defer gock.OffAll()
		httpMock("POST", "http://message.bilibili.co/api/notify/send.user.notify.do").Reply(200).JSON(`{"code": 0}`)

		ctx.Convey("When everything goes positive", func(ctx convey.C) {
			err := d.SendSysMsg(c, uids, mc, title, context, ip)
			ctx.Convey("Then err should be nil.", func(ctx convey.C) {
				ctx.So(err, convey.ShouldBeNil)
			})
		})
	})
}