You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
68 lines
1.2 KiB
68 lines
1.2 KiB
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") |
|
}) |
|
}
|
|
|