Initial commit (code only without large binaries)

EdgeCommon/pkg/serverconfigs/shared/http_header_policy_test.go

@@ -0,0 +1,57 @@
+package shared
+
+import (
+	"github.com/iwind/TeaGo/assert"
+	"testing"
+)
+
+func TestHTTPHeaderPolicy_FormatHeaders(t *testing.T) {
+	policy := &HTTPHeaderPolicy{}
+	err := policy.Init()
+	if err != nil {
+		t.Fatal(err)
+	}
+}
+
+func TestHTTPHeaderPolicy_ShouldDeleteHeader(t *testing.T) {
+	a := assert.NewAssertion(t)
+
+	{
+		policy := &HTTPHeaderPolicy{}
+		err := policy.Init()
+		if err != nil {
+			t.Fatal(err)
+		}
+		a.IsFalse(policy.ContainsDeletedHeader("Origin"))
+	}
+	{
+		policy := &HTTPHeaderPolicy{
+			DeleteHeaders: []string{"Hello", "World"},
+		}
+		err := policy.Init()
+		if err != nil {
+			t.Fatal(err)
+		}
+		a.IsFalse(policy.ContainsDeletedHeader("Origin"))
+	}
+	{
+		policy := &HTTPHeaderPolicy{
+			DeleteHeaders: []string{"origin"},
+		}
+		err := policy.Init()
+		if err != nil {
+			t.Fatal(err)
+		}
+		a.IsTrue(policy.ContainsDeletedHeader("Origin"))
+	}
+	{
+		policy := &HTTPHeaderPolicy{
+			DeleteHeaders: []string{"Origin"},
+		}
+		err := policy.Init()
+		if err != nil {
+			t.Fatal(err)
+		}
+		a.IsTrue(policy.ContainsDeletedHeader("Origin"))
+	}
+}