76 lines
2.0 KiB
Go
76 lines
2.0 KiB
Go
package service
|
||
|
||
import (
|
||
"go-common/app/interface/main/push-archive/dao"
|
||
"go-common/app/interface/main/push-archive/model"
|
||
"testing"
|
||
|
||
"github.com/smartystreets/goconvey/convey"
|
||
)
|
||
|
||
func Test_groupparam(t *testing.T) {
|
||
initd()
|
||
|
||
expect := map[string]string{
|
||
"1#ai:pushlist_follow_recent": "follow",
|
||
"1#ai:pushlist_play_recent": "play",
|
||
"1#ai:pushlist_offline_up": "offline",
|
||
"2#special": "special",
|
||
}
|
||
convey.Convey("推送的group参数", t, func() {
|
||
for k, g := range s.dao.FanGroups {
|
||
group := s.getGroupParam(g)
|
||
convey.So(group, convey.ShouldEqual, expect[k])
|
||
}
|
||
})
|
||
}
|
||
|
||
func Test_usersettingfilter(t *testing.T) {
|
||
initd()
|
||
|
||
mid := int64(11111111)
|
||
s.userSettings[mid] = &model.Setting{Type: model.PushTypeForbid}
|
||
allow := s.filterUserSetting(mid, model.RelationSpecial)
|
||
convey.Convey("usersettings filter关闭开关，则排除", t, func() {
|
||
convey.So(allow, convey.ShouldEqual, false)
|
||
})
|
||
|
||
s.userSettings[mid] = nil
|
||
allow = s.filterUserSetting(mid, model.RelationSpecial)
|
||
convey.Convey("usersettings filter未设置开关，则允许", t, func() {
|
||
convey.So(allow, convey.ShouldEqual, true)
|
||
})
|
||
|
||
s.userSettings[mid] = &model.Setting{Type: model.PushTypeAttention}
|
||
allow = s.filterUserSetting(mid, model.RelationSpecial)
|
||
convey.Convey("usersettings filter设置未所有关注，则允许", t, func() {
|
||
convey.So(allow, convey.ShouldEqual, true)
|
||
})
|
||
}
|
||
|
||
func Test_ispgc(t *testing.T) {
|
||
arc := new(model.Archive)
|
||
convey.Convey("pgc稿件判断", t, func() {
|
||
arc.Attribute = int32(110336)
|
||
convey.So(s.isPGC(arc), convey.ShouldEqual, true)
|
||
|
||
arc.Attribute = int32(16512)
|
||
convey.So(s.isPGC(arc), convey.ShouldEqual, false)
|
||
})
|
||
}
|
||
|
||
func TestServicefansByAbtest(t *testing.T) {
|
||
initd()
|
||
group := &dao.FanGroup{
|
||
Hitby: "ab_test",
|
||
HBaseTable: "push_archive_ab_test",
|
||
HBaseFamily: []string{"cf"},
|
||
}
|
||
fans := []int64{1, 2, 3, 4, 5, 6}
|
||
convey.Convey("fansByAbtest", t, func() {
|
||
exists, notExists := s.fansByAbtest(group, fans)
|
||
t.Logf("exists(%v)", exists)
|
||
t.Logf("notExists(%v)", notExists)
|
||
})
|
||
}
|