Files
EdgeCommon/pkg/serverconfigs/shared/http_header_config_test.go
2020-09-13 19:27:47 +08:00

32 lines
511 B
Go

package shared
import (
"github.com/iwind/TeaGo/assert"
"testing"
)
func TestHeaderConfig_Match(t *testing.T) {
a := assert.NewAssertion(t)
h := NewHeaderConfig()
err := h.Init()
if err != nil {
t.Fatal(err)
}
a.IsFalse(h.Match(200))
a.IsFalse(h.Match(400))
h.Status = &HTTPStatusConfig{
Always: false,
Codes: []int{200, 301, 302, 400},
}
err = h.Init()
if err != nil {
t.Fatal(err)
}
a.IsTrue(h.Match(400))
a.IsFalse(h.Match(500))
h.Status.Always = true
a.IsTrue(h.Match(500))
}