1026 lines
30 KiB
Go
1026 lines
30 KiB
Go
package http
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/traefik/traefik/v3/pkg/middlewares/requestdecorator"
|
|
)
|
|
|
|
func TestClientIPMatcher(t *testing.T) {
|
|
testCases := []struct {
|
|
desc string
|
|
rule string
|
|
expected map[string]int
|
|
expectedError bool
|
|
}{
|
|
{
|
|
desc: "invalid ClientIP matcher",
|
|
rule: "ClientIP(`1`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid ClientIP matcher (no parameter)",
|
|
rule: "ClientIP()",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid ClientIP matcher (empty parameter)",
|
|
rule: "ClientIP(``)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid ClientIP matcher (too many parameters)",
|
|
rule: "ClientIP(`127.0.0.1`, `192.168.1.0/24`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "valid ClientIP matcher",
|
|
rule: "ClientIP(`127.0.0.1`)",
|
|
expected: map[string]int{
|
|
"127.0.0.1": http.StatusOK,
|
|
"192.168.1.1": http.StatusNotFound,
|
|
},
|
|
},
|
|
{
|
|
desc: "valid ClientIP matcher but invalid remote address",
|
|
rule: "ClientIP(`127.0.0.1`)",
|
|
expected: map[string]int{
|
|
"1": http.StatusNotFound,
|
|
},
|
|
},
|
|
{
|
|
desc: "valid ClientIP matcher using CIDR",
|
|
rule: "ClientIP(`192.168.1.0/24`)",
|
|
expected: map[string]int{
|
|
"192.168.1.1": http.StatusOK,
|
|
"192.168.1.100": http.StatusOK,
|
|
"192.168.2.1": http.StatusNotFound,
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, test := range testCases {
|
|
test := test
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {})
|
|
muxer, err := NewMuxer()
|
|
require.NoError(t, err)
|
|
|
|
err = muxer.AddRoute(test.rule, 0, handler)
|
|
if test.expectedError {
|
|
require.Error(t, err)
|
|
return
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
|
|
results := make(map[string]int)
|
|
for remoteAddr := range test.expected {
|
|
w := httptest.NewRecorder()
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "https://example.com", http.NoBody)
|
|
req.RemoteAddr = remoteAddr
|
|
|
|
muxer.ServeHTTP(w, req)
|
|
results[remoteAddr] = w.Code
|
|
}
|
|
assert.Equal(t, test.expected, results)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestMethodMatcher(t *testing.T) {
|
|
testCases := []struct {
|
|
desc string
|
|
rule string
|
|
expected map[string]int
|
|
expectedError bool
|
|
}{
|
|
{
|
|
desc: "invalid Method matcher (no parameter)",
|
|
rule: "Method()",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid Method matcher (empty parameter)",
|
|
rule: "Method(``)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid Method matcher (too many parameters)",
|
|
rule: "Method(`GET`, `POST`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "valid Method matcher",
|
|
rule: "Method(`GET`)",
|
|
expected: map[string]int{
|
|
http.MethodGet: http.StatusOK,
|
|
http.MethodPost: http.StatusNotFound,
|
|
strings.ToLower(http.MethodGet): http.StatusNotFound,
|
|
},
|
|
},
|
|
{
|
|
desc: "valid Method matcher (lower case)",
|
|
rule: "Method(`get`)",
|
|
expected: map[string]int{
|
|
http.MethodGet: http.StatusOK,
|
|
http.MethodPost: http.StatusNotFound,
|
|
strings.ToLower(http.MethodGet): http.StatusNotFound,
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, test := range testCases {
|
|
test := test
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {})
|
|
muxer, err := NewMuxer()
|
|
require.NoError(t, err)
|
|
|
|
err = muxer.AddRoute(test.rule, 0, handler)
|
|
if test.expectedError {
|
|
require.Error(t, err)
|
|
return
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
|
|
results := make(map[string]int)
|
|
for method := range test.expected {
|
|
w := httptest.NewRecorder()
|
|
|
|
req := httptest.NewRequest(method, "https://example.com", http.NoBody)
|
|
|
|
muxer.ServeHTTP(w, req)
|
|
results[method] = w.Code
|
|
}
|
|
assert.Equal(t, test.expected, results)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestHostMatcher(t *testing.T) {
|
|
testCases := []struct {
|
|
desc string
|
|
rule string
|
|
expected map[string]int
|
|
expectedError bool
|
|
}{
|
|
{
|
|
desc: "invalid Host matcher (no parameter)",
|
|
rule: "Host()",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid Host matcher (empty parameter)",
|
|
rule: "Host(``)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid Host matcher (non-ASCII)",
|
|
rule: "Host(`🦭.com`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid Host matcher (too many parameters)",
|
|
rule: "Host(`example.com`, `example.org`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "valid Host matcher",
|
|
rule: "Host(`example.com`)",
|
|
expected: map[string]int{
|
|
"https://example.com": http.StatusOK,
|
|
"https://example.com:8080": http.StatusOK,
|
|
"https://example.com/path": http.StatusOK,
|
|
"https://EXAMPLE.COM/path": http.StatusOK,
|
|
"https://example.org": http.StatusNotFound,
|
|
"https://example.org/path": http.StatusNotFound,
|
|
},
|
|
},
|
|
{
|
|
desc: "valid Host matcher - matcher ending with a dot",
|
|
rule: "Host(`example.com.`)",
|
|
expected: map[string]int{
|
|
"https://example.com": http.StatusOK,
|
|
"https://example.com/path": http.StatusOK,
|
|
"https://example.org": http.StatusNotFound,
|
|
"https://example.org/path": http.StatusNotFound,
|
|
"https://example.com.": http.StatusOK,
|
|
"https://example.com./path": http.StatusOK,
|
|
"https://example.org.": http.StatusNotFound,
|
|
"https://example.org./path": http.StatusNotFound,
|
|
},
|
|
},
|
|
{
|
|
desc: "valid Host matcher - URL ending with a dot",
|
|
rule: "Host(`example.com`)",
|
|
expected: map[string]int{
|
|
"https://example.com.": http.StatusOK,
|
|
"https://example.com./path": http.StatusOK,
|
|
"https://example.org.": http.StatusNotFound,
|
|
"https://example.org./path": http.StatusNotFound,
|
|
},
|
|
},
|
|
{
|
|
desc: "valid Host matcher - matcher with UPPER case",
|
|
rule: "Host(`EXAMPLE.COM`)",
|
|
expected: map[string]int{
|
|
"https://example.com": http.StatusOK,
|
|
"https://example.com/path": http.StatusOK,
|
|
"https://example.org": http.StatusNotFound,
|
|
"https://example.org/path": http.StatusNotFound,
|
|
},
|
|
},
|
|
{
|
|
desc: "valid Host matcher - puny-coded emoji",
|
|
rule: "Host(`xn--9t9h.com`)",
|
|
expected: map[string]int{
|
|
"https://xn--9t9h.com": http.StatusOK,
|
|
"https://xn--9t9h.com/path": http.StatusOK,
|
|
"https://example.com": http.StatusNotFound,
|
|
"https://example.com/path": http.StatusNotFound,
|
|
// The request's sender must use puny-code.
|
|
"https://🦭.com": http.StatusNotFound,
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, test := range testCases {
|
|
test := test
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {})
|
|
muxer, err := NewMuxer()
|
|
require.NoError(t, err)
|
|
|
|
err = muxer.AddRoute(test.rule, 0, handler)
|
|
if test.expectedError {
|
|
require.Error(t, err)
|
|
return
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
|
|
// RequestDecorator is necessary for the Host matcher
|
|
reqHost := requestdecorator.New(nil)
|
|
|
|
results := make(map[string]int)
|
|
for calledURL := range test.expected {
|
|
w := httptest.NewRecorder()
|
|
|
|
req := httptest.NewRequest(http.MethodGet, calledURL, http.NoBody)
|
|
|
|
reqHost.ServeHTTP(w, req, muxer.ServeHTTP)
|
|
results[calledURL] = w.Code
|
|
}
|
|
assert.Equal(t, test.expected, results)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestHostRegexpMatcher(t *testing.T) {
|
|
testCases := []struct {
|
|
desc string
|
|
rule string
|
|
expected map[string]int
|
|
expectedError bool
|
|
}{
|
|
{
|
|
desc: "invalid HostRegexp matcher (no parameter)",
|
|
rule: "HostRegexp()",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid HostRegexp matcher (empty parameter)",
|
|
rule: "HostRegexp(``)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid HostRegexp matcher (non-ASCII)",
|
|
rule: "HostRegexp(`🦭.com`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid HostRegexp matcher (invalid regexp)",
|
|
rule: "HostRegexp(`(example.com`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid HostRegexp matcher (too many parameters)",
|
|
rule: "HostRegexp(`example.com`, `example.org`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "valid HostRegexp matcher",
|
|
rule: "HostRegexp(`^[a-zA-Z-]+\\.com$`)",
|
|
expected: map[string]int{
|
|
"https://example.com": http.StatusOK,
|
|
"https://example.com:8080": http.StatusOK,
|
|
"https://example.com/path": http.StatusOK,
|
|
"https://example.org": http.StatusNotFound,
|
|
"https://example.org/path": http.StatusNotFound,
|
|
},
|
|
},
|
|
{
|
|
desc: "valid HostRegexp matcher with case sensitive regexp",
|
|
rule: "HostRegexp(`^[A-Z]+\\.com$`)",
|
|
expected: map[string]int{
|
|
"https://example.com": http.StatusNotFound,
|
|
"https://EXAMPLE.com": http.StatusNotFound,
|
|
"https://example.com/path": http.StatusNotFound,
|
|
"https://example.org": http.StatusNotFound,
|
|
"https://example.org/path": http.StatusNotFound,
|
|
},
|
|
},
|
|
{
|
|
desc: "valid HostRegexp matcher with Traefik v2 syntax",
|
|
rule: "HostRegexp(`{domain:[a-zA-Z-]+\\.com}`)",
|
|
expected: map[string]int{
|
|
"https://example.com": http.StatusNotFound,
|
|
"https://example.com/path": http.StatusNotFound,
|
|
"https://example.org": http.StatusNotFound,
|
|
"https://example.org/path": http.StatusNotFound,
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, test := range testCases {
|
|
test := test
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {})
|
|
muxer, err := NewMuxer()
|
|
require.NoError(t, err)
|
|
|
|
err = muxer.AddRoute(test.rule, 0, handler)
|
|
if test.expectedError {
|
|
require.Error(t, err)
|
|
return
|
|
}
|
|
require.NoError(t, err)
|
|
|
|
// RequestDecorator is necessary for the HostRegexp matcher
|
|
reqHost := requestdecorator.New(nil)
|
|
|
|
results := make(map[string]int)
|
|
for calledURL := range test.expected {
|
|
w := httptest.NewRecorder()
|
|
|
|
req := httptest.NewRequest(http.MethodGet, calledURL, http.NoBody)
|
|
|
|
reqHost.ServeHTTP(w, req, muxer.ServeHTTP)
|
|
results[calledURL] = w.Code
|
|
}
|
|
assert.Equal(t, test.expected, results)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPathMatcher(t *testing.T) {
|
|
testCases := []struct {
|
|
desc string
|
|
rule string
|
|
expected map[string]int
|
|
expectedError bool
|
|
}{
|
|
{
|
|
desc: "invalid Path matcher (no parameter)",
|
|
rule: "Path()",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid Path matcher (empty parameter)",
|
|
rule: "Path(``)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid Path matcher (no leading /)",
|
|
rule: "Path(`css`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid Path matcher (too many parameters)",
|
|
rule: "Path(`/css`, `/js`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "valid Path matcher",
|
|
rule: "Path(`/css`)",
|
|
expected: map[string]int{
|
|
"https://example.com": http.StatusNotFound,
|
|
"https://example.com/html": http.StatusNotFound,
|
|
"https://example.org/css": http.StatusOK,
|
|
"https://example.com/css": http.StatusOK,
|
|
"https://example.com/css/": http.StatusNotFound,
|
|
"https://example.com/css/main.css": http.StatusNotFound,
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, test := range testCases {
|
|
test := test
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {})
|
|
muxer, err := NewMuxer()
|
|
require.NoError(t, err)
|
|
|
|
err = muxer.AddRoute(test.rule, 0, handler)
|
|
if test.expectedError {
|
|
require.Error(t, err)
|
|
return
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
|
|
results := make(map[string]int)
|
|
for calledURL := range test.expected {
|
|
w := httptest.NewRecorder()
|
|
|
|
req := httptest.NewRequest(http.MethodGet, calledURL, http.NoBody)
|
|
|
|
muxer.ServeHTTP(w, req)
|
|
results[calledURL] = w.Code
|
|
}
|
|
assert.Equal(t, test.expected, results)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPathRegexpMatcher(t *testing.T) {
|
|
testCases := []struct {
|
|
desc string
|
|
rule string
|
|
expected map[string]int
|
|
expectedError bool
|
|
}{
|
|
{
|
|
desc: "invalid PathRegexp matcher (no parameter)",
|
|
rule: "PathRegexp()",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid PathRegexp matcher (empty parameter)",
|
|
rule: "PathRegexp(``)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid PathRegexp matcher (invalid regexp)",
|
|
rule: "PathRegexp(`/(css`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid PathRegexp matcher (too many parameters)",
|
|
rule: "PathRegexp(`/css`, `/js`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "valid PathRegexp matcher",
|
|
rule: "PathRegexp(`^/(css|js)`)",
|
|
expected: map[string]int{
|
|
"https://example.com": http.StatusNotFound,
|
|
"https://example.com/html": http.StatusNotFound,
|
|
"https://example.org/css": http.StatusOK,
|
|
"https://example.com/CSS": http.StatusNotFound,
|
|
"https://example.com/css": http.StatusOK,
|
|
"https://example.com/css/": http.StatusOK,
|
|
"https://example.com/css/main.css": http.StatusOK,
|
|
"https://example.com/js": http.StatusOK,
|
|
"https://example.com/js/": http.StatusOK,
|
|
"https://example.com/js/main.js": http.StatusOK,
|
|
},
|
|
},
|
|
{
|
|
desc: "valid PathRegexp matcher with Traefik v2 syntax",
|
|
rule: `PathRegexp("/{path:(css|js)}")`,
|
|
expected: map[string]int{
|
|
"https://example.com": http.StatusNotFound,
|
|
"https://example.com/html": http.StatusNotFound,
|
|
"https://example.org/css": http.StatusNotFound,
|
|
"https://example.com/{path:css}": http.StatusOK,
|
|
"https://example.com/{path:css}/": http.StatusOK,
|
|
"https://example.com/%7Bpath:css%7D": http.StatusOK,
|
|
"https://example.com/%7Bpath:css%7D/": http.StatusOK,
|
|
"https://example.com/{path:js}": http.StatusOK,
|
|
"https://example.com/{path:js}/": http.StatusOK,
|
|
"https://example.com/%7Bpath:js%7D": http.StatusOK,
|
|
"https://example.com/%7Bpath:js%7D/": http.StatusOK,
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, test := range testCases {
|
|
test := test
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {})
|
|
muxer, err := NewMuxer()
|
|
require.NoError(t, err)
|
|
|
|
err = muxer.AddRoute(test.rule, 0, handler)
|
|
if test.expectedError {
|
|
require.Error(t, err)
|
|
return
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
|
|
results := make(map[string]int)
|
|
for calledURL := range test.expected {
|
|
w := httptest.NewRecorder()
|
|
|
|
req := httptest.NewRequest(http.MethodGet, calledURL, http.NoBody)
|
|
|
|
muxer.ServeHTTP(w, req)
|
|
results[calledURL] = w.Code
|
|
}
|
|
assert.Equal(t, test.expected, results)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPathPrefixMatcher(t *testing.T) {
|
|
testCases := []struct {
|
|
desc string
|
|
rule string
|
|
expected map[string]int
|
|
expectedError bool
|
|
}{
|
|
{
|
|
desc: "invalid PathPrefix matcher (no parameter)",
|
|
rule: "PathPrefix()",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid PathPrefix matcher (empty parameter)",
|
|
rule: "PathPrefix(``)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid PathPrefix matcher (no leading /)",
|
|
rule: "PathPrefix(`css`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid PathPrefix matcher (too many parameters)",
|
|
rule: "PathPrefix(`/css`, `/js`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "valid PathPrefix matcher",
|
|
rule: `PathPrefix("/css")`,
|
|
expected: map[string]int{
|
|
"https://example.com": http.StatusNotFound,
|
|
"https://example.com/html": http.StatusNotFound,
|
|
"https://example.org/css": http.StatusOK,
|
|
"https://example.com/css": http.StatusOK,
|
|
"https://example.com/css/": http.StatusOK,
|
|
"https://example.com/css/main.css": http.StatusOK,
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, test := range testCases {
|
|
test := test
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {})
|
|
muxer, err := NewMuxer()
|
|
require.NoError(t, err)
|
|
|
|
err = muxer.AddRoute(test.rule, 0, handler)
|
|
if test.expectedError {
|
|
require.Error(t, err)
|
|
return
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
|
|
results := make(map[string]int)
|
|
for calledURL := range test.expected {
|
|
w := httptest.NewRecorder()
|
|
|
|
req := httptest.NewRequest(http.MethodGet, calledURL, http.NoBody)
|
|
|
|
muxer.ServeHTTP(w, req)
|
|
results[calledURL] = w.Code
|
|
}
|
|
assert.Equal(t, test.expected, results)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestHeaderMatcher(t *testing.T) {
|
|
testCases := []struct {
|
|
desc string
|
|
rule string
|
|
expected map[*http.Header]int
|
|
expectedError bool
|
|
}{
|
|
{
|
|
desc: "invalid Header matcher (no parameter)",
|
|
rule: "Header()",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid Header matcher (missing value parameter)",
|
|
rule: "Header(`X-Forwarded-Host`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid Header matcher (missing value parameter)",
|
|
rule: "Header(`X-Forwarded-Host`, ``)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid Header matcher (missing key parameter)",
|
|
rule: "Header(``, `example.com`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid Header matcher (too many parameters)",
|
|
rule: "Header(`X-Forwarded-Host`, `example.com`, `example.org`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "valid Header matcher",
|
|
rule: "Header(`X-Forwarded-Proto`, `https`)",
|
|
expected: map[*http.Header]int{
|
|
{"X-Forwarded-Proto": []string{"https"}}: http.StatusOK,
|
|
{"x-forwarded-proto": []string{"https"}}: http.StatusNotFound,
|
|
{"X-Forwarded-Proto": []string{"http", "https"}}: http.StatusOK,
|
|
{"X-Forwarded-Proto": []string{"https", "http"}}: http.StatusOK,
|
|
{"X-Forwarded-Host": []string{"example.com"}}: http.StatusNotFound,
|
|
},
|
|
},
|
|
{
|
|
desc: "valid Header matcher (non-canonical form)",
|
|
rule: "Header(`x-forwarded-proto`, `https`)",
|
|
expected: map[*http.Header]int{
|
|
{"X-Forwarded-Proto": []string{"https"}}: http.StatusOK,
|
|
{"x-forwarded-proto": []string{"https"}}: http.StatusNotFound,
|
|
{"X-Forwarded-Proto": []string{"http", "https"}}: http.StatusOK,
|
|
{"X-Forwarded-Proto": []string{"https", "http"}}: http.StatusOK,
|
|
{"X-Forwarded-Host": []string{"example.com"}}: http.StatusNotFound,
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, test := range testCases {
|
|
test := test
|
|
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {})
|
|
muxer, err := NewMuxer()
|
|
require.NoError(t, err)
|
|
|
|
err = muxer.AddRoute(test.rule, 0, handler)
|
|
if test.expectedError {
|
|
require.Error(t, err)
|
|
return
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
|
|
for headers := range test.expected {
|
|
w := httptest.NewRecorder()
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "https://example.com", http.NoBody)
|
|
req.Header = *headers
|
|
|
|
muxer.ServeHTTP(w, req)
|
|
assert.Equal(t, test.expected[headers], w.Code, headers)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestHeaderRegexpMatcher(t *testing.T) {
|
|
testCases := []struct {
|
|
desc string
|
|
rule string
|
|
expected map[*http.Header]int
|
|
expectedError bool
|
|
}{
|
|
{
|
|
desc: "invalid HeaderRegexp matcher (no parameter)",
|
|
rule: "HeaderRegexp()",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid HeaderRegexp matcher (missing value parameter)",
|
|
rule: "HeaderRegexp(`X-Forwarded-Host`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid HeaderRegexp matcher (missing value parameter)",
|
|
rule: "HeaderRegexp(`X-Forwarded-Host`, ``)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid HeaderRegexp matcher (missing key parameter)",
|
|
rule: "HeaderRegexp(``, `example.com`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid HeaderRegexp matcher (invalid regexp)",
|
|
rule: "HeaderRegexp(`X-Forwarded-Host`,`(example.com`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid HeaderRegexp matcher (too many parameters)",
|
|
rule: "HeaderRegexp(`X-Forwarded-Host`, `example.com`, `example.org`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "valid HeaderRegexp matcher",
|
|
rule: "HeaderRegexp(`X-Forwarded-Proto`, `^https?$`)",
|
|
expected: map[*http.Header]int{
|
|
{"X-Forwarded-Proto": []string{"http"}}: http.StatusOK,
|
|
{"x-forwarded-proto": []string{"http"}}: http.StatusNotFound,
|
|
{"X-Forwarded-Proto": []string{"https"}}: http.StatusOK,
|
|
{"X-Forwarded-Proto": []string{"HTTPS"}}: http.StatusNotFound,
|
|
{"X-Forwarded-Proto": []string{"ws", "https"}}: http.StatusOK,
|
|
{"X-Forwarded-Host": []string{"example.com"}}: http.StatusNotFound,
|
|
},
|
|
},
|
|
{
|
|
desc: "valid HeaderRegexp matcher (non-canonical form)",
|
|
rule: "HeaderRegexp(`x-forwarded-proto`, `^https?$`)",
|
|
expected: map[*http.Header]int{
|
|
{"X-Forwarded-Proto": []string{"http"}}: http.StatusOK,
|
|
{"x-forwarded-proto": []string{"http"}}: http.StatusNotFound,
|
|
{"X-Forwarded-Proto": []string{"https"}}: http.StatusOK,
|
|
{"X-Forwarded-Proto": []string{"HTTPS"}}: http.StatusNotFound,
|
|
{"X-Forwarded-Proto": []string{"ws", "https"}}: http.StatusOK,
|
|
{"X-Forwarded-Host": []string{"example.com"}}: http.StatusNotFound,
|
|
},
|
|
},
|
|
{
|
|
desc: "valid HeaderRegexp matcher with Traefik v2 syntax",
|
|
rule: "HeaderRegexp(`X-Forwarded-Proto`, `http{secure:s?}`)",
|
|
expected: map[*http.Header]int{
|
|
{"X-Forwarded-Proto": []string{"http"}}: http.StatusNotFound,
|
|
{"X-Forwarded-Proto": []string{"https"}}: http.StatusNotFound,
|
|
{"X-Forwarded-Proto": []string{"http{secure:}"}}: http.StatusOK,
|
|
{"X-Forwarded-Proto": []string{"HTTP{secure:}"}}: http.StatusNotFound,
|
|
{"X-Forwarded-Proto": []string{"http{secure:s}"}}: http.StatusOK,
|
|
{"X-Forwarded-Proto": []string{"http{secure:S}"}}: http.StatusNotFound,
|
|
{"X-Forwarded-Proto": []string{"HTTPS"}}: http.StatusNotFound,
|
|
{"X-Forwarded-Proto": []string{"ws", "http{secure:s}"}}: http.StatusOK,
|
|
{"X-Forwarded-Host": []string{"example.com"}}: http.StatusNotFound,
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, test := range testCases {
|
|
test := test
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {})
|
|
muxer, err := NewMuxer()
|
|
require.NoError(t, err)
|
|
|
|
err = muxer.AddRoute(test.rule, 0, handler)
|
|
if test.expectedError {
|
|
require.Error(t, err)
|
|
return
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
|
|
for headers := range test.expected {
|
|
w := httptest.NewRecorder()
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "https://example.com", http.NoBody)
|
|
req.Header = *headers
|
|
|
|
muxer.ServeHTTP(w, req)
|
|
assert.Equal(t, test.expected[headers], w.Code, *headers)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestQueryMatcher(t *testing.T) {
|
|
testCases := []struct {
|
|
desc string
|
|
rule string
|
|
expected map[string]int
|
|
expectedError bool
|
|
}{
|
|
{
|
|
desc: "invalid Query matcher (no parameter)",
|
|
rule: "Query()",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid Query matcher (empty key, one parameter)",
|
|
rule: "Query(``)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid Query matcher (empty key)",
|
|
rule: "Query(``, `traefik`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid Query matcher (empty value)",
|
|
rule: "Query(`q`, ``)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid Query matcher (too many parameters)",
|
|
rule: "Query(`q`, `traefik`, `proxy`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "valid Query matcher",
|
|
rule: "Query(`q`, `traefik`)",
|
|
expected: map[string]int{
|
|
"https://example.com": http.StatusNotFound,
|
|
"https://example.com?q=traefik": http.StatusOK,
|
|
"https://example.com?rel=ddg&q=traefik": http.StatusOK,
|
|
"https://example.com?q=traefik&q=proxy": http.StatusOK,
|
|
"https://example.com?q=awesome&q=traefik": http.StatusOK,
|
|
"https://example.com?q=nginx": http.StatusNotFound,
|
|
"https://example.com?rel=ddg": http.StatusNotFound,
|
|
"https://example.com?q=TRAEFIK": http.StatusNotFound,
|
|
"https://example.com?Q=traefik": http.StatusNotFound,
|
|
"https://example.com?rel=traefik": http.StatusNotFound,
|
|
},
|
|
},
|
|
{
|
|
desc: "valid Query matcher with empty value",
|
|
rule: "Query(`mobile`)",
|
|
expected: map[string]int{
|
|
"https://example.com": http.StatusNotFound,
|
|
"https://example.com?mobile": http.StatusOK,
|
|
"https://example.com?mobile=true": http.StatusNotFound,
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, test := range testCases {
|
|
test := test
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {})
|
|
muxer, err := NewMuxer()
|
|
require.NoError(t, err)
|
|
|
|
err = muxer.AddRoute(test.rule, 0, handler)
|
|
if test.expectedError {
|
|
require.Error(t, err)
|
|
return
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
|
|
results := make(map[string]int)
|
|
for calledURL := range test.expected {
|
|
w := httptest.NewRecorder()
|
|
|
|
req := httptest.NewRequest(http.MethodGet, calledURL, http.NoBody)
|
|
|
|
muxer.ServeHTTP(w, req)
|
|
results[calledURL] = w.Code
|
|
}
|
|
assert.Equal(t, test.expected, results)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestQueryRegexpMatcher(t *testing.T) {
|
|
testCases := []struct {
|
|
desc string
|
|
rule string
|
|
expected map[string]int
|
|
expectedError bool
|
|
}{
|
|
{
|
|
desc: "invalid QueryRegexp matcher (no parameter)",
|
|
rule: "QueryRegexp()",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid QueryRegexp matcher (empty parameter)",
|
|
rule: "QueryRegexp(``)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid QueryRegexp matcher (invalid regexp)",
|
|
rule: "QueryRegexp(`q`, `(traefik`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "invalid QueryRegexp matcher (too many parameters)",
|
|
rule: "QueryRegexp(`q`, `traefik`, `proxy`)",
|
|
expectedError: true,
|
|
},
|
|
{
|
|
desc: "valid QueryRegexp matcher",
|
|
rule: "QueryRegexp(`q`, `^(traefik|nginx)$`)",
|
|
expected: map[string]int{
|
|
"https://example.com": http.StatusNotFound,
|
|
"https://example.com?q=traefik": http.StatusOK,
|
|
"https://example.com?rel=ddg&q=traefik": http.StatusOK,
|
|
"https://example.com?q=traefik&q=proxy": http.StatusOK,
|
|
"https://example.com?q=awesome&q=traefik": http.StatusOK,
|
|
"https://example.com?q=TRAEFIK": http.StatusNotFound,
|
|
"https://example.com?Q=traefik": http.StatusNotFound,
|
|
"https://example.com?rel=traefik": http.StatusNotFound,
|
|
"https://example.com?q=nginx": http.StatusOK,
|
|
"https://example.com?rel=ddg&q=nginx": http.StatusOK,
|
|
"https://example.com?q=nginx&q=proxy": http.StatusOK,
|
|
"https://example.com?q=awesome&q=nginx": http.StatusOK,
|
|
"https://example.com?q=NGINX": http.StatusNotFound,
|
|
"https://example.com?Q=nginx": http.StatusNotFound,
|
|
"https://example.com?rel=nginx": http.StatusNotFound,
|
|
"https://example.com?q=haproxy": http.StatusNotFound,
|
|
"https://example.com?rel=ddg": http.StatusNotFound,
|
|
},
|
|
},
|
|
{
|
|
desc: "valid QueryRegexp matcher",
|
|
rule: "QueryRegexp(`q`, `^.*$`)",
|
|
expected: map[string]int{
|
|
"https://example.com": http.StatusNotFound,
|
|
"https://example.com?q=traefik": http.StatusOK,
|
|
"https://example.com?rel=ddg&q=traefik": http.StatusOK,
|
|
"https://example.com?q=traefik&q=proxy": http.StatusOK,
|
|
"https://example.com?q=awesome&q=traefik": http.StatusOK,
|
|
"https://example.com?q=TRAEFIK": http.StatusOK,
|
|
"https://example.com?Q=traefik": http.StatusNotFound,
|
|
"https://example.com?rel=traefik": http.StatusNotFound,
|
|
"https://example.com?q=nginx": http.StatusOK,
|
|
"https://example.com?rel=ddg&q=nginx": http.StatusOK,
|
|
"https://example.com?q=nginx&q=proxy": http.StatusOK,
|
|
"https://example.com?q=awesome&q=nginx": http.StatusOK,
|
|
"https://example.com?q=NGINX": http.StatusOK,
|
|
"https://example.com?Q=nginx": http.StatusNotFound,
|
|
"https://example.com?rel=nginx": http.StatusNotFound,
|
|
"https://example.com?q=haproxy": http.StatusOK,
|
|
"https://example.com?rel=ddg": http.StatusNotFound,
|
|
},
|
|
},
|
|
{
|
|
desc: "valid QueryRegexp matcher with Traefik v2 syntax",
|
|
rule: "QueryRegexp(`q`, `{value:(traefik|nginx)}`)",
|
|
expected: map[string]int{
|
|
"https://example.com?q=traefik": http.StatusNotFound,
|
|
"https://example.com?q={value:traefik}": http.StatusOK,
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, test := range testCases {
|
|
test := test
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {})
|
|
muxer, err := NewMuxer()
|
|
require.NoError(t, err)
|
|
|
|
err = muxer.AddRoute(test.rule, 0, handler)
|
|
if test.expectedError {
|
|
require.Error(t, err)
|
|
return
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
|
|
results := make(map[string]int)
|
|
for calledURL := range test.expected {
|
|
w := httptest.NewRecorder()
|
|
|
|
req := httptest.NewRequest(http.MethodGet, calledURL, http.NoBody)
|
|
|
|
muxer.ServeHTTP(w, req)
|
|
results[calledURL] = w.Code
|
|
}
|
|
assert.Equal(t, test.expected, results)
|
|
})
|
|
}
|
|
}
|