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.
116 lines
2.6 KiB
116 lines
2.6 KiB
package service |
|
|
|
import ( |
|
"context" |
|
ht "net/http" |
|
"sync" |
|
"testing" |
|
|
|
"go-common/app/interface/main/push-archive/conf" |
|
"go-common/app/interface/main/push-archive/model" |
|
pb "go-common/app/service/main/push/api/grpc/v1" |
|
"go-common/library/ecode" |
|
bm "go-common/library/net/http/blademaster" |
|
|
|
"github.com/smartystreets/goconvey/convey" |
|
) |
|
|
|
func Test_getsetting(t *testing.T) { |
|
initd() |
|
|
|
mid := int64(11111111) |
|
s.SetSetting(context.TODO(), mid, nil) |
|
convey.Convey("获取默认的用户开关设置", t, func() { |
|
st, err := s.Setting(context.TODO(), mid) |
|
convey.So(err, convey.ShouldBeNil) |
|
convey.So(st.Type, convey.ShouldEqual, model.PushTypeSpecial) |
|
}) |
|
} |
|
|
|
func Test_setsetting(t *testing.T) { |
|
initd() |
|
|
|
mid := int64(11111111) |
|
convey.Convey("存储用户开关设置", t, func() { |
|
err := s.SetSetting(context.TODO(), mid, &model.Setting{Type: model.PushTypeAttention}) |
|
convey.So(err, convey.ShouldBeNil) |
|
}) |
|
} |
|
|
|
func Test_ps(t *testing.T) { |
|
initd() |
|
|
|
url := "http://127.0.0.1:7031/x/push-archive/setting/get?access_key=848657e31639317257ab274741c6ec7d" |
|
client := bm.NewClient(conf.Conf.HTTPClient) |
|
type _response struct { |
|
Code int `json:"code"` |
|
Data model.Setting `json:"data"` |
|
} |
|
|
|
wg := sync.WaitGroup{} |
|
all := 10 |
|
busyIn := 0 |
|
errIn := 0 |
|
normalIn := 0 |
|
mx := sync.Mutex{} |
|
for i := 0; i < all; i++ { |
|
wg.Add(1) |
|
|
|
go func(i int) { |
|
defer wg.Done() |
|
defer mx.Unlock() |
|
|
|
req, err := ht.NewRequest("GET", url, nil) |
|
if err != nil { |
|
mx.Lock() |
|
errIn++ |
|
return |
|
} |
|
|
|
r := &_response{} |
|
err = client.Do(context.TODO(), req, r) |
|
mx.Lock() |
|
if err != nil { |
|
errIn++ |
|
return |
|
} |
|
|
|
if r.Code == ecode.ServiceUnavailable.Code() { |
|
busyIn++ |
|
return |
|
} |
|
normalIn++ |
|
}(i) |
|
} |
|
|
|
wg.Wait() |
|
convey.Convey("并行访问推送开关，有频率限制", t, func() { |
|
convey.So(errIn, convey.ShouldEqual, 0) |
|
convey.So(busyIn, convey.ShouldBeGreaterThan, 0) |
|
convey.So(errIn+busyIn+normalIn, convey.ShouldEqual, all) |
|
}) |
|
t.Logf("the errin(%d), busyin(%d), normalin(%d)", errIn, busyIn, normalIn) |
|
} |
|
|
|
func Test_pushrpc(t *testing.T) { |
|
initd() |
|
|
|
convey.Convey("推送平台rpc设置推送开关", t, func() { |
|
set := &pb.SetSettingRequest{ |
|
Mid: int64(111111111), |
|
Type: 1, |
|
Value: 3, |
|
} |
|
res, err := s.pushRPC.Setting(context.TODO(), &pb.SettingRequest{Mid: set.Mid}) |
|
convey.So(err, convey.ShouldBeNil) |
|
t.Logf("setting(%+v)", res) |
|
|
|
_, err = s.pushRPC.SetSetting(context.TODO(), set) |
|
convey.So(err, convey.ShouldBeNil) |
|
|
|
res, err = s.pushRPC.Setting(context.TODO(), &pb.SettingRequest{Mid: set.Mid}) |
|
convey.So(err, convey.ShouldBeNil) |
|
t.Logf("setting(%+v)", res) |
|
}) |
|
|
|
}
|
|
|