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.
121 lines
3.3 KiB
121 lines
3.3 KiB
package dao |
|
|
|
import ( |
|
"context" |
|
"testing" |
|
"time" |
|
|
|
"go-common/app/job/main/vip/model" |
|
xtime "go-common/library/time" |
|
|
|
"github.com/smartystreets/goconvey/convey" |
|
gock "gopkg.in/h2non/gock.v1" |
|
) |
|
|
|
// go test -test.v -test.run TestDaoSalaryCoupon |
|
func TestDaoSalaryCoupon(t *testing.T) { |
|
convey.Convey("TestDaoSalaryCoupon salary coupon", t, func() { |
|
var ( |
|
c = context.TODO() |
|
mid int64 = 123 |
|
ct int8 = 2 |
|
count int64 = 2 |
|
err error |
|
) |
|
err = d.SalaryCoupon(c, mid, ct, count, "cartoon_1_2018_06") |
|
convey.So(err, convey.ShouldBeNil) |
|
}) |
|
} |
|
|
|
func TestDao_SendMultipMsg(t *testing.T) { |
|
convey.Convey("send multipmsg", t, func() { |
|
defer gock.OffAll() |
|
httpMock("POST", _message).Reply(200).JSON(`{"code":0,"data":1}`) |
|
err := d.SendMultipMsg(context.TODO(), "27515256", "test", "test", "10_1_2", 4) |
|
convey.So(err, convey.ShouldBeNil) |
|
}) |
|
} |
|
|
|
func TestDaoPushData(t *testing.T) { |
|
pushData := &model.VipPushData{ |
|
Title: "TEST", |
|
PushStartTime: "15:04:05", |
|
PushEndTime: "15:04:05", |
|
} |
|
convey.Convey("PushData", t, func() { |
|
defer gock.OffAll() |
|
httpMock("POST", _pushData).Reply(200).JSON(`{"code":0,"data":1}`) |
|
rel, err := d.PushData(context.TODO(), []int64{7593623}, pushData, "2006-01-02") |
|
convey.So(err, convey.ShouldBeNil) |
|
convey.So(rel, convey.ShouldNotBeNil) |
|
}) |
|
} |
|
|
|
func TestDaoSendMedal(t *testing.T) { |
|
convey.Convey("SendMedal", t, func() { |
|
defer gock.OffAll() |
|
httpMock("GET", _sendMedal).Reply(200).JSON(`{"code":0,"data":{"status":1}}`) |
|
status := d.SendMedal(context.TODO(), 0, 0) |
|
convey.So(status, convey.ShouldNotBeNil) |
|
}) |
|
} |
|
|
|
func TestDaoSendCleanCache(t *testing.T) { |
|
hv := &model.HandlerVip{Mid: 7593623} |
|
convey.Convey("SendCleanCache", t, func() { |
|
defer gock.OffAll() |
|
httpMock("GET", _cleanCache).Reply(200).JSON(`{"code":0,"data":{"status":1}}`) |
|
err := d.SendCleanCache(context.TODO(), hv) |
|
convey.So(err, convey.ShouldBeNil) |
|
}) |
|
} |
|
|
|
func TestDaoSendBcoin(t *testing.T) { |
|
convey.Convey("SendBcoin", t, func() { |
|
defer gock.OffAll() |
|
httpMock("POST", _addBcoin).Reply(200).JSON(`{"code":0,"data":{"status":1}}`) |
|
err := d.SendBcoin(context.TODO(), []int64{7593623}, 0, xtime.Time(time.Now().Unix()), "") |
|
convey.So(err, convey.ShouldBeNil) |
|
}) |
|
} |
|
|
|
func TestDaoSendAppCleanCache(t *testing.T) { |
|
var ( |
|
hv = &model.HandlerVip{Mid: 7593623} |
|
app = &model.VipAppInfo{ |
|
PurgeURL: "http://bilibili.com/test", |
|
} |
|
) |
|
convey.Convey("SendAppCleanCache", t, func() { |
|
defer gock.OffAll() |
|
httpMock("GET", app.PurgeURL).Reply(200).JSON(`{"code":0}`) |
|
err := d.SendAppCleanCache(context.TODO(), hv, app) |
|
convey.So(err, convey.ShouldBeNil) |
|
}) |
|
} |
|
|
|
func TestDaosortParamsKey(t *testing.T) { |
|
var v map[string]string |
|
convey.Convey("sortParamsKey", t, func() { |
|
p1 := d.sortParamsKey(v) |
|
convey.So(p1, convey.ShouldNotBeNil) |
|
}) |
|
} |
|
|
|
func TestDaoPaySign(t *testing.T) { |
|
var params map[string]string |
|
convey.Convey("PaySign", t, func() { |
|
sign := d.PaySign(params, "test") |
|
convey.So(sign, convey.ShouldNotBeNil) |
|
}) |
|
} |
|
|
|
func TestDaodoNomalSend(t *testing.T) { |
|
var path = "/x/internal/vip/user/info" |
|
convey.Convey("doNomalSend", t, func() { |
|
defer gock.OffAll() |
|
httpMock("POST", path).Reply(200).JSON(`{"code":0}`) |
|
err := d.doNomalSend(context.TODO(), "http://api.bilibili.com", path, "", nil, nil, new(model.VipPushResq)) |
|
convey.So(err, convey.ShouldBeNil) |
|
}) |
|
}
|
|
|