mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-04 15:51:54 +08:00
120 lines
4.0 KiB
Go
120 lines
4.0 KiB
Go
// Copyright 2022 GoEdge goedge.cdn@gmail.com. All rights reserved.
|
|
|
|
package stats_test
|
|
|
|
import (
|
|
"github.com/TeaOSLab/EdgeNode/internal/stats"
|
|
"github.com/TeaOSLab/EdgeNode/internal/utils/testutils"
|
|
"github.com/iwind/TeaGo/assert"
|
|
"github.com/iwind/TeaGo/rands"
|
|
"github.com/iwind/TeaGo/types"
|
|
"runtime"
|
|
"runtime/debug"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestUserAgentParser_Parse(t *testing.T) {
|
|
var parser = stats.NewUserAgentParser()
|
|
for i := 0; i < 4; i++ {
|
|
t.Log(parser.Parse("Mozilla/5.0 (Windows NT 10.0; WOW64; rv:49.0) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/88.0.4324.96 Safari/537.36 Test/1"))
|
|
t.Log(parser.Parse("Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.110 Safari/537.36"))
|
|
}
|
|
}
|
|
|
|
func TestUserAgentParser_Parse_Unknown(t *testing.T) {
|
|
var parser = stats.NewUserAgentParser()
|
|
t.Log(parser.Parse("Mozilla/5.0 (Wind 10.0; WOW64; rv:49.0) Apple/537.36 (KHTML, like Gecko) Chr/88.0.4324.96 Sa/537.36 Test/1"))
|
|
t.Log(parser.Parse(""))
|
|
}
|
|
|
|
func TestUserAgentParser_Memory(t *testing.T) {
|
|
var stat1 = &runtime.MemStats{}
|
|
runtime.ReadMemStats(stat1)
|
|
|
|
var parser = stats.NewUserAgentParser()
|
|
|
|
for i := 0; i < 1_000_000; i++ {
|
|
parser.Parse("Mozilla/5.0 (Windows NT 10.0; WOW64; rv:49.0) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/88.0.4324.96 Safari/537.36 Test/" + types.String(rands.Int(0, 1_000_000)))
|
|
}
|
|
|
|
runtime.GC()
|
|
debug.FreeOSMemory()
|
|
|
|
var stat2 = &runtime.MemStats{}
|
|
runtime.ReadMemStats(stat2)
|
|
|
|
t.Log("max cache items:", parser.MaxCacheItems())
|
|
t.Log("cache:", parser.Len(), "usage:", (stat2.HeapInuse-stat1.HeapInuse)>>20, "MB")
|
|
}
|
|
|
|
func TestNewUserAgentParser_GC(t *testing.T) {
|
|
if !testutils.IsSingleTesting() {
|
|
return
|
|
}
|
|
|
|
var parser = stats.NewUserAgentParser()
|
|
|
|
for i := 0; i < 1_000_000; i++ {
|
|
parser.Parse("Mozilla/5.0 (Windows NT 10.0; WOW64; rv:49.0) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/88.0.4324.96 Safari/537.36 Test/" + types.String(rands.Int(0, 100_000)))
|
|
}
|
|
|
|
time.Sleep(60 * time.Second) // wait to gc
|
|
t.Log(parser.Len(), "cache items")
|
|
}
|
|
|
|
func TestNewUserAgentParser_Mobile(t *testing.T) {
|
|
var a = assert.NewAssertion(t)
|
|
var parser = stats.NewUserAgentParser()
|
|
for _, userAgent := range []string{
|
|
"Mozilla/5.0 (iPhone; CPU iPhone OS 12_2 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Mobile/15E148",
|
|
"Mozilla/5.0 (Linux; U; Android 2.2.1; en-us; Nexus One Build/FRG83) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1",
|
|
} {
|
|
a.IsTrue(parser.Parse(userAgent).IsMobile)
|
|
}
|
|
}
|
|
|
|
func BenchmarkUserAgentParser_Parse_Many_LimitCPU(b *testing.B) {
|
|
runtime.GOMAXPROCS(4)
|
|
|
|
var parser = stats.NewUserAgentParser()
|
|
b.RunParallel(func(pb *testing.PB) {
|
|
for pb.Next() {
|
|
parser.Parse("Mozilla/5.0 (Windows NT 10.0; WOW64; rv:49.0) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/88.0.4324.96 Safari/537.36 Test/" + types.String(rands.Int(0, 1_000_000)))
|
|
}
|
|
})
|
|
b.Log(parser.Len())
|
|
}
|
|
|
|
func BenchmarkUserAgentParser_Parse_Many(b *testing.B) {
|
|
var parser = stats.NewUserAgentParser()
|
|
b.RunParallel(func(pb *testing.PB) {
|
|
for pb.Next() {
|
|
parser.Parse("Mozilla/5.0 (Windows NT 10.0; WOW64; rv:49.0) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/88.0.4324.96 Safari/537.36 Test/" + types.String(rands.Int(0, 1_000_000)))
|
|
}
|
|
})
|
|
b.Log(parser.Len())
|
|
}
|
|
|
|
func BenchmarkUserAgentParser_Parse_Few_LimitCPU(b *testing.B) {
|
|
runtime.GOMAXPROCS(4)
|
|
|
|
var parser = stats.NewUserAgentParser()
|
|
b.RunParallel(func(pb *testing.PB) {
|
|
for pb.Next() {
|
|
parser.Parse("Mozilla/5.0 (Windows NT 10.0; WOW64; rv:49.0) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/88.0.4324.96 Safari/537.36 Test/" + types.String(rands.Int(0, 100_000)))
|
|
}
|
|
})
|
|
b.Log(parser.Len())
|
|
}
|
|
|
|
func BenchmarkUserAgentParser_Parse_Few(b *testing.B) {
|
|
var parser = stats.NewUserAgentParser()
|
|
b.RunParallel(func(pb *testing.PB) {
|
|
for pb.Next() {
|
|
parser.Parse("Mozilla/5.0 (Windows NT 10.0; WOW64; rv:49.0) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/88.0.4324.96 Safari/537.36 Test/" + types.String(rands.Int(0, 100_000)))
|
|
}
|
|
})
|
|
b.Log(parser.Len())
|
|
}
|