openbilibili/app/admin/main/dm/dao/mysql_test.go

package dao

import (
	"context"
	"testing"

	"go-common/app/admin/main/dm/model"

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

func TestUpSubjectAttr(t *testing.T) {
	Convey("test update subject attr", t, func() {
		_, err := testDao.UpSubjectAttr(context.TODO(), 1, 1221, 16)
		So(err, ShouldBeNil)
	})
}

func TestUpSubjectCount(t *testing.T) {
	Convey("update count in subject", t, func() {
		_, err := testDao.UpSubjectCount(context.TODO(), 1, 1221, 1, 1)
		So(err, ShouldBeNil)
	})
}

func TestIncrSubjectCount(t *testing.T) {
	Convey("update count in subject", t, func() {
		_, err := testDao.IncrSubjectCount(context.TODO(), 1, 1221, 1)
		So(err, ShouldBeNil)
	})
}

func TestIncrSubMoveCount(t *testing.T) {
	Convey("update move count in subject", t, func() {
		_, err := testDao.IncrSubMoveCount(context.TODO(), 1, 1221, 1)
		So(err, ShouldBeNil)
	})
}

func TestIncrSubCount(t *testing.T) {
	Convey("update count in subject", t, func() {
		_, err := testDao.IncrSubjectCount(context.TODO(), 1, 1221, 1)
		So(err, ShouldBeNil)
	})
}

func TestUpSubjectPool(t *testing.T) {
	Convey("update childpool in subject", t, func() {
		_, err := testDao.UpSubjectPool(context.TODO(), 1, 1221, 1)
		So(err, ShouldBeNil)
	})
}

func TestUpSubjectState(t *testing.T) {
	Convey("update state in subject", t, func() {
		_, err := testDao.UpSubjectState(context.TODO(), 1, 1221, 1)
		So(err, ShouldBeNil)
	})
}

func TestUpSubjectMaxlimit(t *testing.T) {
	Convey("update maxlimit in subject", t, func() {
		_, err := testDao.UpSubjectMaxlimit(context.TODO(), 1, 1221, 8000)
		So(err, ShouldBeNil)
	})
}

func TestSubject(t *testing.T) {
	var (
		tp  int32 = 1
		oid int64 = 1508
		c         = context.TODO()
	)
	Convey("subject test", t, func() {
		_, err := testDao.Subject(c, tp, oid)
		So(err, ShouldBeNil)
	})
}

func TestSubjects(t *testing.T) {
	oids := []int64{1221, 1321}
	Convey("subject test", t, func() {
		res, err := testDao.Subjects(context.TODO(), 1, oids)
		So(err, ShouldBeNil)
		So(res, ShouldNotBeEmpty)
	})
}

func TestChangeReportStat(t *testing.T) {
	var (
		c           = context.TODO()
		cid   int64 = 1
		dmids       = []int64{1, 2, 3, 4, 5}
		state       = model.StatFirstDelete
	)
	Convey("update dm report state and err shoule be nil", t, func() {
		err := testDao.ChangeReportStat(c, cid, dmids, state)
		So(err, ShouldBeNil)
	})
}

func TestIgnoreReport(t *testing.T) {
	var (
		cid   int64 = 1
		c           = context.TODO()
		dmids       = []int64{1, 2, 3, 4, 5}
		state int8  = 4
	)
	Convey("ignore dm report state and err shoule be nil", t, func() {
		err := testDao.IgnoreReport(c, cid, dmids, state)
		So(err, ShouldBeNil)
	})
}

func TestReports(t *testing.T) {
	var (
		cid   int64 = 10109027
		c           = context.TODO()
		dmids       = []int64{719218595}
	)
	Convey("", t, func() {
		res, err := testDao.Reports(c, cid, dmids)
		So(err, ShouldBeNil)
		So(res, ShouldNotBeEmpty)
		for _, r := range res {
			t.Logf("===:%+v", r)
		}
	})
}

func TestReportUsers(t *testing.T) {
	var (
		tableID int64 = 1
		c             = context.TODO()
		state         = model.NoticeUnsend
		dmids         = []int64{1, 2, 3, 4, 5}
	)
	Convey("", t, func() {
		_, err := testDao.ReportUsers(c, tableID, dmids, state)
		So(err, ShouldBeNil)
	})
}

func TestUpReportUserState(t *testing.T) {
	var (
		tableID int64 = 1
		c             = context.TODO()
		state         = model.NoticeSend
		dmids         = []int64{1, 2, 3, 4, 5}
	)
	Convey("", t, func() {
		_, err := testDao.UpReportUserState(c, tableID, dmids, state)
		So(err, ShouldBeNil)
	})
}

func TestAddReportLog(t *testing.T) {
	var (
		tableID int64 = 1
		c             = context.TODO()
		lg            = &model.ReportLog{
			ID:      1,
			Did:     1234,
			AdminID: 1,
		}
	)
	Convey("", t, func() {
		err := testDao.AddReportLog(c, tableID, []*model.ReportLog{lg})
		So(err, ShouldBeNil)
	})
}

func TestReportLog(t *testing.T) {
	var (
		dmid int64 = 719918888
		c          = context.TODO()
	)
	Convey("", t, func() {
		res, err := testDao.ReportLog(c, dmid)
		So(err, ShouldBeNil)
		So(res, ShouldNotBeEmpty)
	})
}