mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-04 16:00:24 +08:00
86 lines
1.9 KiB
Go
86 lines
1.9 KiB
Go
package services
|
|
|
|
import (
|
|
rpcutils "github.com/TeaOSLab/EdgeAPI/internal/rpc/utils"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
|
_ "github.com/iwind/TeaGo/bootstrap"
|
|
"github.com/iwind/TeaGo/dbs"
|
|
"github.com/iwind/TeaGo/logs"
|
|
timeutil "github.com/iwind/TeaGo/utils/time"
|
|
"testing"
|
|
)
|
|
|
|
func TestServerService_UploadServerHTTPRequestStat(t *testing.T) {
|
|
dbs.NotifyReady()
|
|
|
|
service := new(ServerService)
|
|
_, err := service.UploadServerHTTPRequestStat(rpcutils.NewMockNodeContext(1), &pb.UploadServerHTTPRequestStatRequest{
|
|
Month: timeutil.Format("Ym"),
|
|
RegionCities: []*pb.UploadServerHTTPRequestStatRequest_RegionCity{
|
|
{
|
|
ServerId: 1,
|
|
CountryId: 1, // 中国
|
|
ProvinceId: 12, // 安徽省
|
|
CityId: 108, // 阜阳市
|
|
CountRequests: 1,
|
|
},
|
|
},
|
|
RegionProviders: []*pb.UploadServerHTTPRequestStatRequest_RegionProvider{
|
|
{
|
|
ServerId: 1,
|
|
ProviderId: 1, // 电信
|
|
Count: 1,
|
|
},
|
|
},
|
|
Systems: []*pb.UploadServerHTTPRequestStatRequest_System{
|
|
{
|
|
ServerId: 1,
|
|
Name: "Mac OS X",
|
|
Count: 1,
|
|
Version: "20",
|
|
},
|
|
},
|
|
Browsers: []*pb.UploadServerHTTPRequestStatRequest_Browser{
|
|
{
|
|
ServerId: 1,
|
|
Name: "Chrome",
|
|
Count: 1,
|
|
Version: "70",
|
|
},
|
|
{
|
|
ServerId: 1,
|
|
Name: "Firefox",
|
|
Count: 1,
|
|
Version: "30",
|
|
},
|
|
},
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
t.Log("===countries===")
|
|
logs.PrintAsJSON(serverHTTPCountryStatMap, t)
|
|
|
|
t.Log("===provinces===")
|
|
logs.PrintAsJSON(serverHTTPProvinceStatMap, t)
|
|
|
|
t.Log("===cities===")
|
|
logs.PrintAsJSON(serverHTTPCityStatMap, t)
|
|
|
|
t.Log("===providers===")
|
|
logs.PrintAsJSON(serverHTTPProviderStatMap, t)
|
|
|
|
t.Log("===systems===")
|
|
logs.PrintAsJSON(serverHTTPSystemStatMap, t)
|
|
|
|
t.Log("===browsers===")
|
|
logs.PrintAsJSON(serverHTTPBrowserStatMap, t)
|
|
|
|
err = service.dumpServerHTTPStats()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Log("ok")
|
|
}
|