openbilibili/app/job/main/block/dao/dao_test.go

package dao

import (
	"context"
	"flag"
	"testing"

	"go-common/app/job/main/block/conf"
	"go-common/app/job/main/block/model"

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

var (
	dao *Dao
	ctx = context.TODO()
)

func TestMain(m *testing.M) {
	flag.Set("conf", "../cmd/block-service-test.toml")
	var err error
	if err = conf.Init(); err != nil {
		panic(err)
	}
	dao = New()
	defer dao.Close()
	m.Run()
}

func TestDB(t *testing.T) {
	Convey("db", t, func() {
		tx, err := dao.BeginTX(ctx)
		So(err, ShouldBeNil)

		var (
			mid int64 = 46333
		)
		err = dao.UpsertAddBlockCount(ctx, mid)
		So(err, ShouldBeNil)

		err = dao.TxUpsertUser(ctx, tx, mid, model.BlockStatusFalse)
		So(err, ShouldBeNil)

		var (
			history = &model.DBHistory{
				MID:     mid,
				Source:  model.BlockSourceRemove,
				Comment: "ut test",
				Action:  model.BlockActionAdminRemove,
			}
		)
		err = dao.TxInsertHistory(ctx, tx, history)
		So(err, ShouldBeNil)

		err = tx.Rollback()
		So(err, ShouldBeNil)
	})
}

func TestTool(t *testing.T) {
	Convey("tool", t, func() {
		var (
			mids = []int64{1, 2, 3, 46333, 35858}
		)
		str := midsToParam(mids)
		So(str, ShouldEqual, "1,2,3,46333,35858")
	})
}