bilibili-backup/app/interface/main/creative/dao/assist/live_test.go

package assist

import (
	"context"
	"net/url"
	"strconv"
	"testing"

	"go-common/app/interface/main/creative/model/assist"

	"github.com/smartystreets/goconvey/convey"
)

func TestAssistLiveStatus(t *testing.T) {
	var (
		c   = context.TODO()
		mid = int64(0)
		ip  = ""
		res = struct {
			Code int `json:"code"`
		}{
			Code: 0,
		}
	)
	convey.Convey("LiveStatus", t, func(ctx convey.C) {
		httpMock("GET", d.liveStatusURL+"?uid="+strconv.FormatInt(mid, 10)).Reply(0).JSON(res)
		ok, err := d.LiveStatus(c, mid, ip)
		ctx.Convey("Then err should be nil.ok should not be nil.", func(ctx convey.C) {
			ctx.So(err, convey.ShouldBeNil)
			ctx.So(ok, convey.ShouldNotBeNil)
		})
	})
}

var msgRes = struct {
	Code    int    `json:"code"`
	Message string `json:"message"`
}{
	Code:    0,
	Message: "message",
}

func TestAssistLiveAddAssist(t *testing.T) {
	var (
		c         = context.TODO()
		mid       = int64(1)
		assistMid = int64(10)
		cookie    = ""
		ip        = ""
		params    = url.Values{}
	)
	params.Set("admin", strconv.FormatInt(assistMid, 10))
	params.Set("anchor_id", strconv.FormatInt(mid, 10))
	convey.Convey("LiveAddAssist", t, func(ctx convey.C) {
		httpMock("POST", d.liveAddAssistURL+"?"+params.Encode()).
			Reply(0).JSON(msgRes)
		err := d.LiveAddAssist(c, mid, assistMid, cookie, ip)
		ctx.Convey("Then err should be nil.", func(ctx convey.C) {
			ctx.So(err, convey.ShouldBeNil)
		})
	})
}

func TestAssistLiveDelAssist(t *testing.T) {
	var (
		c         = context.TODO()
		mid       = int64(0)
		assistMid = int64(0)
		cookie    = ""
		ip        = ""
		params    = url.Values{}
	)
	params.Set("admin", strconv.FormatInt(assistMid, 10))
	params.Set("anchor_id", strconv.FormatInt(mid, 10))
	convey.Convey("LiveDelAssist", t, func(ctx convey.C) {
		httpMock("POST", d.liveDelAssistURL+"?"+params.Encode()).Reply(0).JSON(msgRes)
		err := d.LiveDelAssist(c, mid, assistMid, cookie, ip)
		ctx.Convey("Then err should be nil.", func(ctx convey.C) {
			ctx.So(err, convey.ShouldBeNil)
		})
	})
}

func TestAssistLiveBannedRevoc(t *testing.T) {
	var (
		c      = context.TODO()
		mid    = int64(0)
		banID  = ""
		cookie = ""
		ip     = ""
		params = url.Values{}
	)
	params.Set("uid", strconv.FormatInt(mid, 10))
	params.Set("id", banID)
	convey.Convey("LiveBannedRevoc", t, func(ctx convey.C) {
		httpMock("POST", d.liveRevocBannedURL+"?"+params.Encode()).Reply(200).JSON(msgRes)
		err := d.LiveBannedRevoc(c, mid, banID, cookie, ip)
		ctx.Convey("Then err should be nil.", func(ctx convey.C) {
			ctx.So(err, convey.ShouldBeNil)
		})
	})
}

func TestAssistLiveAssists(t *testing.T) {
	var (
		c      = context.TODO()
		mid    = int64(0)
		ip     = ""
		params = url.Values{}
		res    = struct {
			Code int                  `json:"code"`
			Data []*assist.LiveAssist `json:"data"`
		}{
			Code: 0,
			Data: []*assist.LiveAssist{
				{
					AssistMid: 0,
					RoomID:    0,
					CTime:     0,
					Datetime:  "",
				},
			},
		}
	)
	params.Set("anchor_id", strconv.FormatInt(mid, 10))
	convey.Convey("LiveAssists", t, func(ctx convey.C) {
		httpMock("GET", d.liveAssistsURL+"?"+params.Encode()).Reply(200).JSON(res)
		assists, err := d.LiveAssists(c, mid, ip)
		ctx.Convey("Then err should be nil.assists should not be nil.", func(ctx convey.C) {
			ctx.So(err, convey.ShouldBeNil)
			ctx.So(assists, convey.ShouldNotBeNil)
		})
	})
}

func TestAssistLiveCheckAssist(t *testing.T) {
	var (
		c         = context.TODO()
		mid       = int64(0)
		assistMid = int64(0)
		ip        = ""
		params    = url.Values{}
		res       = struct {
			Code int `json:"code"`
		}{
			Code: 0,
		}
	)
	params.Set("uid", strconv.FormatInt(assistMid, 10))
	params.Set("anchor_id", strconv.FormatInt(mid, 10))
	convey.Convey("LiveCheckAssist", t, func(ctx convey.C) {
		httpMock("POST", d.liveCheckAssURL+"?"+params.Encode()).Reply(200).JSON(res)
		isAss, err := d.LiveCheckAssist(c, mid, assistMid, ip)
		ctx.Convey("Then err should be nil.isAss should not be nil.", func(ctx convey.C) {
			ctx.So(err, convey.ShouldBeNil)
			ctx.So(isAss, convey.ShouldNotBeNil)
		})
	})
}