goStatic/customHeaders_test.go
Alex Tan d0e5eca1f4 Don't panic when regex doesn't compile
Instead, print a warning and discard the invalid rule.
2021-07-11 16:09:55 -05:00

39 lines
762 B
Go

package main
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestHeaderConfigWithValidRegex(t *testing.T) {
assert := assert.New(t)
config := HeaderConfig{Regex: "/$"}
ok := config.Init()
assert.True(ok)
assert.NotNil(config.CompiledRegex)
assert.True(config.UsesRegex())
}
func TestHeaderConfigWithInvalidRegex(t *testing.T) {
assert := assert.New(t)
config := HeaderConfig{Regex: "["}
ok := config.Init()
assert.False(ok)
assert.Nil(config.CompiledRegex)
}
func TestHeaderConfigWithoutRegex(t *testing.T) {
assert := assert.New(t)
config := HeaderConfig{
Path: "/page-data",
FileExtension: "json",
}
ok := config.Init()
assert.True(ok)
assert.Nil(config.CompiledRegex)
assert.False(config.UsesRegex())
}