Function TestRuleSet_MatchRequest_Allow

internal/waf/rule_set_test.go:78–122

(t *testing.T)

}

func TestRuleSet_MatchRequest_Allow(t *testing.T) {
	var a = assert.NewAssertion(t)

	var set = waf.NewRuleSet()
	set.Connector = waf.RuleConnectorOr

	set.Rules = []*waf.Rule{
		{
			Param:    "${requestPath}",
			Operator: waf.RuleOperatorMatch,
			Value:    "hello",
		},
	}

	set.Actions = []*waf.ActionConfig{
		{
			Code: "allow",
			Options: maps.Map{
				"scope": waf.AllowScopeGroup,
			},
		},
	}

	var wafInstance = waf.NewWAF()

	err := set.Init(wafInstance)
	if err != nil {
		t.Fatal(err)
	}

	rawReq, err := http.NewRequest(http.MethodGet, "http://teaos.cn/hello?name=lu&age=20", nil)
	if err != nil {
		t.Fatal(err)
	}
	var req = requests.NewTestRequest(rawReq)
	b, _, err := set.MatchRequest(req)
	if err != nil {
		t.Fatal(err)
	}
	a.IsTrue(b)

	var result = set.PerformActions(wafInstance, &waf.RuleGroup{}, req, nil)
	a.IsTrue(result.IsAllowed)
	t.Log("scope:", result.AllowScope)
}

func BenchmarkRuleSet_MatchRequest(b *testing.B) {
	runtime.GOMAXPROCS(1)