TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-02 22:10:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化测试用例

Browse Source
This commit is contained in:
GoEdgeLab
2023-12-10 08:54:39 +08:00
parent 3e2c5e876d
commit 198c742c04
1 changed files with 45 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
internal/waf/template_test.go
View File

@@ -13,6 +13,7 @@ import (
"mime/multipart"
"net/http"
"net/url"
"runtime"
"strings"
"testing"
"time"
@@ -66,6 +67,8 @@ func Test_Template2(t *testing.T) {
}
func BenchmarkTemplate(b *testing.B) {
runtime.GOMAXPROCS(4)
wafInstance, err := waf.Template()
if err != nil {
b.Fatal(err)
@@ -73,15 +76,17 @@ func BenchmarkTemplate(b *testing.B) {
b.ResetTimer()
for i := 0; i < b.N; i++ {
req, err := http.NewRequest(http.MethodGet, "https://example.com/index.php?id=123", nil)
if err != nil {
b.Fatal(err)
}
req.Header.Set("User-Agent", testUserAgent)
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
req, err := http.NewRequest(http.MethodGet, "https://example.com/index.php?id=123"+types.String(rand.Int()%10000), nil)
if err != nil {
b.Fatal(err)
}
req.Header.Set("User-Agent", testUserAgent)
_, _, _, _, _ = wafInstance.MatchRequest(requests.NewTestRequest(req), nil, firewallconfigs.ServerCaptchaTypeNone)
}
_, _, _, _, _ = wafInstance.MatchRequest(requests.NewTestRequest(req), nil, firewallconfigs.ServerCaptchaTypeNone)
}
})
}
func testTemplate1010(a *assert.Assertion, t *testing.T, template *waf.WAF) {
@@ -430,3 +435,35 @@ func testTemplate20001(a *assert.Assertion, t *testing.T, template *waf.WAF) {
}
}
}
func BenchmarkTemplatePathTraversal(b *testing.B) {
runtime.GOMAXPROCS(4)
template, err := waf.Template()
if err != nil {
b.Fatal(err)
}
var group = template.FindRuleGroupWithCode("pathTraversal")
if group == nil {
b.Fatal("group not found")
return
}
b.ResetTimer()
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
req, err := http.NewRequest(http.MethodPost, "https://example.com/?id=1234"+types.String(rand.Int()%10000)+"&name=lily&time=12345678910", nil)
if err != nil {
b.Fatal(err)
}
req.Header.Set("User-Agent", testUserAgent)
_, _, result, err := group.MatchRequest(requests.NewTestRequest(req))
if err != nil {
b.Fatal(err)
}
_ = result
}
})
}