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.
92 lines
2.1 KiB
92 lines
2.1 KiB
package gorpc |
|
|
|
import ( |
|
"context" |
|
"testing" |
|
|
|
pushsrv "go-common/app/service/main/push/api/gorpc" |
|
"go-common/app/service/main/push/model" |
|
|
|
. "github.com/smartystreets/goconvey/convey" |
|
) |
|
|
|
var ( |
|
// _noArg = &struct{}{} |
|
// _noRes = &struct{}{} |
|
ctx = context.TODO() |
|
) |
|
|
|
func WithRPC(f func(client *pushsrv.Service)) func() { |
|
return func() { |
|
client := pushsrv.New(nil) |
|
f(client) |
|
} |
|
} |
|
|
|
func Test_AddReport(t *testing.T) { |
|
Convey("AddReport", t, WithRPC(func(client *pushsrv.Service) { |
|
arg := &model.ArgReport{ |
|
APPID: 1, |
|
PlatformID: 1, |
|
Mid: 1, |
|
Buvid: "b", |
|
DeviceToken: "d", |
|
Build: 8080, |
|
TimeZone: 8, |
|
NotifySwitch: 1, |
|
} |
|
err := client.AddReport(ctx, arg) |
|
So(err, ShouldBeNil) |
|
})) |
|
} |
|
|
|
func Test_Setting(t *testing.T) { |
|
Convey("get setting", t, WithRPC(func(client *pushsrv.Service) { |
|
arg := &model.ArgMid{Mid: 88888888} |
|
res, err := client.Setting(ctx, arg) |
|
So(err, ShouldBeNil) |
|
t.Logf("setting(%v)", res) |
|
})) |
|
|
|
Convey("set setting", t, WithRPC(func(client *pushsrv.Service) { |
|
arg := &model.ArgSetting{Mid: 999999999, Type: model.UserSettingArchive, Value: model.SwitchOff} |
|
err := client.SetSetting(ctx, arg) |
|
So(err, ShouldBeNil) |
|
|
|
argMid := &model.ArgMid{Mid: 999999999} |
|
res, err := client.Setting(ctx, argMid) |
|
So(err, ShouldBeNil) |
|
t.Logf("setting(%v)", res) |
|
})) |
|
} |
|
|
|
func TestAddUserReportCache(t *testing.T) { |
|
Convey("AddUserReportCache", t, WithRPC(func(client *pushsrv.Service) { |
|
arg := &model.ArgUserReports{Mid: 123456, Reports: []*model.Report{{ |
|
APPID: 1, |
|
PlatformID: 1, |
|
Mid: 123456, |
|
DeviceToken: "dtrpc", |
|
}}} |
|
err := client.AddUserReportCache(context.Background(), arg) |
|
So(err, ShouldBeNil) |
|
})) |
|
} |
|
|
|
func TestAddTokensCache(t *testing.T) { |
|
Convey("AddTokensCache", t, WithRPC(func(client *pushsrv.Service) { |
|
arg := &model.ArgReports{Reports: []*model.Report{{ |
|
APPID: 1, |
|
PlatformID: 1, |
|
Mid: 123456, |
|
DeviceToken: "dtrpc", |
|
}, { |
|
APPID: 1, |
|
PlatformID: 1, |
|
Mid: 123456, |
|
DeviceToken: "dtrpc2", |
|
}}} |
|
err := client.AddTokensCache(context.Background(), arg) |
|
So(err, ShouldBeNil) |
|
})) |
|
}
|
|
|