85 lines
2.4 KiB
Go
85 lines
2.4 KiB
Go
|
package dao
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"encoding/json"
|
|||
|
|
"testing"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"go-common/app/interface/main/push-archive/model"
|
|||
|
|
|
|||
|
|
"github.com/smartystreets/goconvey/convey"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func Test_upperlimit(t *testing.T) {
|
|||
|
|
upper := int64(998)
|
|||
|
|
d.UpperLimitExpire = 1 // 1s
|
|||
|
|
exist, err := d.ExistUpperLimitCache(context.TODO(), upper)
|
|||
|
|
convey.Convey("upper主推送频率限制，没存储过", t, func() {
|
|||
|
|
convey.So(err, convey.ShouldBeNil)
|
|||
|
|
convey.So(exist, convey.ShouldEqual, false)
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
err = d.AddUpperLimitCache(context.TODO(), upper)
|
|||
|
|
convey.Convey("upper主推送频率限制,添加推送1次，再次获取已存在, 失效后不存在", t, func() {
|
|||
|
|
convey.So(err, convey.ShouldBeNil)
|
|||
|
|
exist, err = d.ExistUpperLimitCache(context.TODO(), upper)
|
|||
|
|
convey.So(err, convey.ShouldBeNil)
|
|||
|
|
convey.So(exist, convey.ShouldEqual, true)
|
|||
|
|
time.Sleep(2 * time.Second)
|
|||
|
|
exist, err = d.ExistUpperLimitCache(context.TODO(), upper)
|
|||
|
|
convey.So(err, convey.ShouldBeNil)
|
|||
|
|
convey.So(exist, convey.ShouldEqual, false)
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func Test_statisticscache(t *testing.T) {
|
|||
|
|
ps, err := d.GetStatisticsCache(context.TODO())
|
|||
|
|
convey.Convey("从redis获取统计数据, 没有数据", t, func() {
|
|||
|
|
convey.So(err, convey.ShouldBeNil)
|
|||
|
|
convey.So(ps, convey.ShouldBeNil)
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
per := int64(1000)
|
|||
|
|
start := int64(1000000)
|
|||
|
|
var mids []int64
|
|||
|
|
for i := start; i < start+per; i++ {
|
|||
|
|
mids = append(mids, i)
|
|||
|
|
}
|
|||
|
|
midscount := len(mids)
|
|||
|
|
midsstr, _ := json.Marshal(mids)
|
|||
|
|
ps = &model.PushStatistic{
|
|||
|
|
Aid: int64(121321),
|
|||
|
|
Group: "ai:pushlist_offline_up",
|
|||
|
|
Type: 1,
|
|||
|
|
Mids: string(midsstr),
|
|||
|
|
MidsCounter: midscount,
|
|||
|
|
CTime: time.Now(),
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
err = d.AddStatisticsCache(context.TODO(), ps)
|
|||
|
|
convey.Convey("添加统计数据到redis", t, func() {
|
|||
|
|
convey.So(err, convey.ShouldBeNil)
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func Test_perupperlimit(t *testing.T) {
|
|||
|
|
upper := int64(10)
|
|||
|
|
fan := int64(20)
|
|||
|
|
err := d.AddPerUpperLimitCache(context.TODO(), fan, upper, 1, 1)
|
|||
|
|
convey.Convey("添加推送次数限制", t, func() {
|
|||
|
|
convey.So(err, convey.ShouldEqual, nil)
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
total, err := d.GetPerUpperLimitCache(context.TODO(), fan, upper)
|
|||
|
|
convey.Convey("获取推送次数限制, 失效后不存在", t, func() {
|
|||
|
|
convey.So(err, convey.ShouldEqual, nil)
|
|||
|
|
convey.So(total, convey.ShouldEqual, 1)
|
|||
|
|
time.Sleep(time.Second * 2)
|
|||
|
|
total, err = d.GetPerUpperLimitCache(context.TODO(), fan, upper)
|
|||
|
|
convey.So(err, convey.ShouldEqual, nil)
|
|||
|
|
convey.So(total, convey.ShouldEqual, 0)
|
|||
|
|
|
|||
|
|
})
|
|||
|
|
}
|