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.
118 lines
3.6 KiB
118 lines
3.6 KiB
package dao |
|
|
|
import ( |
|
"context" |
|
"fmt" |
|
"testing" |
|
"time" |
|
|
|
"github.com/smartystreets/goconvey/convey" |
|
) |
|
|
|
func TestDaoGetDayExpenseCount(t *testing.T) { |
|
convey.Convey("GetDayExpenseCount", t, func(ctx convey.C) { |
|
var ( |
|
c = context.Background() |
|
beginDate = time.Now() |
|
ctype = int(0) |
|
) |
|
ctx.Convey("When everything goes positive", func(ctx convey.C) { |
|
total, err := d.GetDayExpenseCount(c, beginDate, ctype) |
|
ctx.Convey("Then err should be nil.total should not be nil.", func(ctx convey.C) { |
|
ctx.So(err, convey.ShouldBeNil) |
|
ctx.So(total, convey.ShouldNotBeNil) |
|
}) |
|
}) |
|
}) |
|
} |
|
|
|
func TestDaoGetAllDayExpenseInfo(t *testing.T) { |
|
convey.Convey("GetAllDayExpenseInfo", t, func(ctx convey.C) { |
|
var ( |
|
c = context.Background() |
|
beginDate, _ = time.ParseInLocation("2018-06-01", "2018-01-01", time.Local) |
|
ctype = int(0) |
|
from = int(0) |
|
limit = int(10) |
|
) |
|
ctx.Convey("When everything goes positive", func(ctx convey.C) { |
|
Exec(c, "INSERT INTO expense_daily_info(up_acount, date) VALUES(100, '2018-01-01')") |
|
_, err := d.GetAllDayExpenseInfo(c, beginDate, ctype, from, limit) |
|
ctx.Convey("Then err should be nil.infos should not be nil.", func(ctx convey.C) { |
|
ctx.So(err, convey.ShouldBeNil) |
|
}) |
|
}) |
|
}) |
|
} |
|
|
|
func TestDaoGetDayTotalExpenseInfo(t *testing.T) { |
|
convey.Convey("GetDayTotalExpenseInfo", t, func(ctx convey.C) { |
|
var ( |
|
c = context.Background() |
|
date = time.Date(2018, 9, 1, 0, 0, 0, 0, time.Local) |
|
ctype = int(0) |
|
) |
|
ctx.Convey("When everything goes positive", func(ctx convey.C) { |
|
Exec(c, fmt.Sprintf("INSERT INTO expense_daily_info(date, total_expense) VALUES('%s', 100)", date.Format("2006-01-02"))) |
|
_, err := d.GetDayTotalExpenseInfo(c, date, ctype) |
|
ctx.Convey("Then err should be nil.totalExpense should not be nil.", func(ctx convey.C) { |
|
ctx.So(err, convey.ShouldBeNil) |
|
}) |
|
}) |
|
}) |
|
} |
|
|
|
func TestDaoGetMonthExpenseCount(t *testing.T) { |
|
convey.Convey("GetMonthExpenseCount", t, func(ctx convey.C) { |
|
var ( |
|
c = context.Background() |
|
month = "2019-01-01" |
|
beginMonth = "2018-01-01" |
|
ctype = int(0) |
|
) |
|
ctx.Convey("When everything goes positive", func(ctx convey.C) { |
|
total, err := d.GetMonthExpenseCount(c, month, beginMonth, ctype) |
|
ctx.Convey("Then err should be nil.total should not be nil.", func(ctx convey.C) { |
|
ctx.So(err, convey.ShouldBeNil) |
|
ctx.So(total, convey.ShouldNotBeNil) |
|
}) |
|
}) |
|
}) |
|
} |
|
|
|
func TestDaoGetAllMonthExpenseInfo(t *testing.T) { |
|
convey.Convey("GetAllMonthExpenseInfo", t, func(ctx convey.C) { |
|
var ( |
|
c = context.Background() |
|
month = "2019-01" |
|
beginMonth = "2018-01" |
|
ctype = int(0) |
|
from = int(0) |
|
limit = int(10) |
|
) |
|
ctx.Convey("When everything goes positive", func(ctx convey.C) { |
|
Exec(c, fmt.Sprintf("INSERT INTO expense_monthly_info(date, total_expense) VALUES('2018-02', 100)")) |
|
_, err := d.GetAllMonthExpenseInfo(c, month, beginMonth, ctype, from, limit) |
|
ctx.Convey("Then err should be nil.infos should not be nil.", func(ctx convey.C) { |
|
ctx.So(err, convey.ShouldBeNil) |
|
}) |
|
}) |
|
}) |
|
} |
|
|
|
func TestDaoGetLatelyExpenseDate(t *testing.T) { |
|
convey.Convey("GetLatelyExpenseDate", t, func(ctx convey.C) { |
|
var ( |
|
c = context.Background() |
|
table = "daily" |
|
ctype = int(0) |
|
) |
|
ctx.Convey("When everything goes positive", func(ctx convey.C) { |
|
date, err := d.GetLatelyExpenseDate(c, table, ctype) |
|
ctx.Convey("Then err should be nil.date should not be nil.", func(ctx convey.C) { |
|
ctx.So(err, convey.ShouldBeNil) |
|
ctx.So(date, convey.ShouldNotBeNil) |
|
}) |
|
}) |
|
}) |
|
}
|
|
|