openbilibili/app/service/bbq/user/internal/dao/user_black_test.go

package dao

import (
	"context"
	"github.com/smartystreets/goconvey/convey"
	"go-common/app/service/bbq/user/internal/model"
	xtime "go-common/library/time"
	"testing"
	"time"
)

func TestDaoTxAddUserBlack(t *testing.T) {
	convey.Convey("TxAddUserBlack", t, func(ctx convey.C) {
		var (
			c     = context.Background()
			mid   = int64(88895104)
			upMid = int64(88895105)
		)
		tx, _ := d.BeginTran(c)
		defer func() {
			tx.Rollback()
		}()
		ctx.Convey("When everything goes positive", func(ctx convey.C) {
			num, err := d.TxAddUserBlack(c, tx, mid, upMid)
			ctx.Convey("Then err should be nil.num should not be nil.", func(ctx convey.C) {
				ctx.So(err, convey.ShouldBeNil)
				ctx.So(num, convey.ShouldNotBeNil)
			})
		})
	})
}

func TestDaoTxCancelUserBlack(t *testing.T) {
	convey.Convey("TxCancelUserBlack", t, func(ctx convey.C) {
		var (
			c     = context.Background()
			mid   = int64(88895104)
			upMid = int64(88895105)
		)
		tx, _ := d.BeginTran(c)
		defer func() {
			tx.Rollback()
		}()
		ctx.Convey("When everything goes positive", func(ctx convey.C) {
			num, err := d.TxCancelUserBlack(c, tx, mid, upMid)
			ctx.Convey("Then err should be nil.num should not be nil.", func(ctx convey.C) {
				ctx.So(err, convey.ShouldBeNil)
				ctx.So(num, convey.ShouldNotBeNil)
			})
		})
	})
}

func TestDaoFetchBlackList(t *testing.T) {
	convey.Convey("FetchBlackList", t, func(ctx convey.C) {
		var (
			c   = context.Background()
			mid = int64(88895104)
		)
		ctx.Convey("When everything goes positive", func(ctx convey.C) {
			upMid, err := d.FetchBlackList(c, mid)
			ctx.Convey("Then err should be nil.upMid should not be nil.", func(ctx convey.C) {
				ctx.So(err, convey.ShouldBeNil)
				ctx.So(upMid, convey.ShouldNotBeNil)
			})
		})
	})
}

func TestDaoFetchPartBlackList(t *testing.T) {
	convey.Convey("FetchPartBlackList", t, func(ctx convey.C) {
		var (
			c      = context.Background()
			mid    = int64(88895104)
			cursor model.CursorValue
			size   = int(10)
		)
		cursor.CursorID = model.MaxInt64
		cursor.CursorTime = xtime.Time(time.Now().Unix())
		ctx.Convey("When everything goes positive", func(ctx convey.C) {
			MID2IDMap, blackMIDs, err := d.FetchPartBlackList(c, mid, cursor, size)
			ctx.Convey("Then err should be nil.MID2IDMap,blackMIDs should not be nil.", func(ctx convey.C) {
				ctx.So(err, convey.ShouldBeNil)
				ctx.So(blackMIDs, convey.ShouldNotBeNil)
				ctx.So(MID2IDMap, convey.ShouldNotBeNil)
			})
		})
	})
}

func TestDaoIsBlack(t *testing.T) {
	convey.Convey("IsBlack", t, func(ctx convey.C) {
		var (
			c             = context.Background()
			mid           = int64(88895104)
			candidateMIDs = []int64{88895105}
		)
		ctx.Convey("When everything goes positive", func(ctx convey.C) {
			MIDMap := d.IsBlack(c, mid, candidateMIDs)
			ctx.Convey("Then MIDMap should not be nil.", func(ctx convey.C) {
				ctx.So(MIDMap, convey.ShouldNotBeNil)
			})
		})
	})
}