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.
62 lines
1.9 KiB
62 lines
1.9 KiB
package dao |
|
|
|
import ( |
|
"context" |
|
"testing" |
|
|
|
"github.com/smartystreets/goconvey/convey" |
|
) |
|
|
|
func TestDaoListColumnIncome(t *testing.T) { |
|
convey.Convey("ListColumnIncome", t, func(ctx convey.C) { |
|
var ( |
|
c = context.Background() |
|
mid = int64(1001) |
|
startTime = "2018-01-01" |
|
endTime = "2019-01-01" |
|
) |
|
ctx.Convey("When everything goes positive", func(ctx convey.C) { |
|
Exec(c, "INSERT INTO column_income(aid, mid, total_income, date) VALUES(1000, 1001, 100, '2018-06-01')") |
|
columns, err := d.ListColumnIncome(c, mid, startTime, endTime) |
|
ctx.Convey("Then err should be nil.columns should not be nil.", func(ctx convey.C) { |
|
ctx.So(err, convey.ShouldBeNil) |
|
ctx.So(columns, convey.ShouldNotBeNil) |
|
}) |
|
}) |
|
}) |
|
} |
|
|
|
func TestDaoListColumnIncomeByID(t *testing.T) { |
|
convey.Convey("ListColumnIncomeByID", t, func(ctx convey.C) { |
|
var ( |
|
c = context.Background() |
|
id = int64(1000) |
|
endTime = "2019-01-01" |
|
) |
|
ctx.Convey("When everything goes positive", func(ctx convey.C) { |
|
Exec(c, "INSERT INTO column_income(aid, mid, total_income, date) VALUES(1000, 1001, 100, '2018-06-01')") |
|
columns, err := d.ListColumnIncomeByID(c, id, endTime) |
|
ctx.Convey("Then err should be nil.columns should not be nil.", func(ctx convey.C) { |
|
ctx.So(err, convey.ShouldBeNil) |
|
ctx.So(columns, convey.ShouldNotBeNil) |
|
}) |
|
}) |
|
}) |
|
} |
|
|
|
func TestDaoGetColumnTitle(t *testing.T) { |
|
convey.Convey("GetColumnTitle", t, func(ctx convey.C) { |
|
var ( |
|
c = context.Background() |
|
ids = []int64{1000} |
|
) |
|
ctx.Convey("When everything goes positive", func(ctx convey.C) { |
|
Exec(c, "INSERT INTO column_income_statis(aid, title) VALUES(1000, 'test')") |
|
titles, err := d.GetColumnTitle(c, ids) |
|
ctx.Convey("Then err should be nil.titles should not be nil.", func(ctx convey.C) { |
|
ctx.So(err, convey.ShouldBeNil) |
|
ctx.So(titles, convey.ShouldNotBeNil) |
|
}) |
|
}) |
|
}) |
|
}
|
|
|