98 lines
2.5 KiB
Go
98 lines
2.5 KiB
Go
package ruleset_v2
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
yaml "gopkg.in/yaml.v3"
|
|
"testing"
|
|
)
|
|
|
|
func TestRuleUnmarshalJSON(t *testing.T) {
|
|
ruleJSON := `{
|
|
"domains": [
|
|
"example.com",
|
|
"www.example.com"
|
|
],
|
|
"request_modifications": [
|
|
"SpoofUserAgent(\"googlebot\")"
|
|
],
|
|
"response_modifications": [
|
|
"APIContent()",
|
|
"SetContentSecurityPolicy(\"foobar\")",
|
|
"SetIncomingCookie(\"authorization-bearer\", \"hunter2\")"
|
|
]
|
|
}`
|
|
|
|
//fmt.Println(ruleJSON)
|
|
rule := &Rule{}
|
|
err := json.Unmarshal([]byte(ruleJSON), rule)
|
|
if err != nil {
|
|
t.Errorf("expected no error in Unmarshal, got '%s'", err)
|
|
return
|
|
}
|
|
|
|
if len(rule.Domains) != 2 {
|
|
t.Errorf("expected number of domains to be 2")
|
|
return
|
|
}
|
|
if !(rule.Domains[0] == "example.com" || rule.Domains[1] == "example.com") {
|
|
t.Errorf("expected domain to be example.com")
|
|
return
|
|
}
|
|
if len(rule.ResponseModifications) != 3 {
|
|
t.Errorf("expected number of ResponseModifications to be 3, got %d", len(rule.ResponseModifications))
|
|
}
|
|
if len(rule.RequestModifications) != 1 {
|
|
t.Errorf("expected number of RequestModifications to be 1, got %d", len(rule.RequestModifications))
|
|
}
|
|
|
|
// test marshal
|
|
jsonRule, err := json.Marshal(rule)
|
|
if err != nil {
|
|
t.Errorf("expected no error marshalling rule to json, got '%s'", err.Error())
|
|
}
|
|
fmt.Println(string(jsonRule))
|
|
}
|
|
|
|
func TestRuleUnmarshalYAML(t *testing.T) {
|
|
ruleYAML := `
|
|
domains:
|
|
- example.com
|
|
- www.example.com
|
|
request_modifications:
|
|
- SpoofUserAgent("googlebot")
|
|
response_modifications:
|
|
- APIContent()
|
|
- SetContentSecurityPolicy("foobar")
|
|
- SetIncomingCookie("authorization-bearer", "hunter2")
|
|
`
|
|
|
|
rule := &Rule{}
|
|
err := yaml.Unmarshal([]byte(ruleYAML), rule)
|
|
if err != nil {
|
|
t.Errorf("expected no error in Unmarshal, got '%s'", err)
|
|
return
|
|
}
|
|
|
|
if len(rule.Domains) != 2 {
|
|
t.Errorf("expected number of domains to be 2, got %d", len(rule.Domains))
|
|
return
|
|
}
|
|
if !(rule.Domains[0] == "example.com" || rule.Domains[1] == "example.com") {
|
|
t.Errorf("expected domain to be example.com")
|
|
return
|
|
}
|
|
if len(rule.ResponseModifications) != 3 {
|
|
t.Errorf("expected number of ResponseModifications to be 3, got %d", len(rule.ResponseModifications))
|
|
}
|
|
if len(rule.RequestModifications) != 1 {
|
|
t.Errorf("expected number of RequestModifications to be 1, got %d", len(rule.RequestModifications))
|
|
}
|
|
|
|
yamlRule, err := yaml.Marshal(rule)
|
|
if err != nil {
|
|
t.Errorf("expected no error marshalling rule to yaml, got '%s'", err.Error())
|
|
}
|
|
fmt.Println(string(yamlRule))
|
|
}
|