WAF策略中增加验证码相关定制设置

internal/waf/waf_manager_test.go

@@ -0,0 +1,45 @@
+package waf_test
+
+import (
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/firewallconfigs"
+	"github.com/TeaOSLab/EdgeNode/internal/waf"
+	"github.com/iwind/TeaGo/logs"
+	"testing"
+)
+
+func TestWAFManager_convert(t *testing.T) {
+	p := &firewallconfigs.HTTPFirewallPolicy{
+		Id:   1,
+		IsOn: true,
+		Inbound: &firewallconfigs.HTTPFirewallInboundConfig{
+			IsOn: true,
+			Groups: []*firewallconfigs.HTTPFirewallRuleGroup{
+				{
+					Id: 1,
+					Sets: []*firewallconfigs.HTTPFirewallRuleSet{
+						{
+							Id: 1,
+						},
+						{
+							Id: 2,
+							Rules: []*firewallconfigs.HTTPFirewallRule{
+								{
+									Id: 1,
+								},
+								{
+									Id: 2,
+								},
+							},
+						},
+					},
+				},
+			},
+		},
+	}
+	w, err := waf.SharedWAFManager.ConvertWAF(p)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	logs.PrintAsJSON(w, t)
+}