EdgeAPI/internal/rpc/services/service_server_test.go

package services

import (
	"testing"

	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"
)

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")
}