150 lines
4 KiB
Go
150 lines
4 KiB
Go
package replacepathregex
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/containous/traefik/v2/pkg/config/dynamic"
|
|
"github.com/containous/traefik/v2/pkg/middlewares/replacepath"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestReplacePathRegex(t *testing.T) {
|
|
testCases := []struct {
|
|
desc string
|
|
path string
|
|
config dynamic.ReplacePathRegex
|
|
expectedPath string
|
|
expectedRawPath string
|
|
expectedHeader string
|
|
expectsError bool
|
|
}{
|
|
{
|
|
desc: "simple regex",
|
|
path: "/whoami/and/whoami",
|
|
config: dynamic.ReplacePathRegex{
|
|
Replacement: "/who-am-i/$1",
|
|
Regex: `^/whoami/(.*)`,
|
|
},
|
|
expectedPath: "/who-am-i/and/whoami",
|
|
expectedRawPath: "/who-am-i/and/whoami",
|
|
expectedHeader: "/whoami/and/whoami",
|
|
},
|
|
{
|
|
desc: "simple replace (no regex)",
|
|
path: "/whoami/and/whoami",
|
|
config: dynamic.ReplacePathRegex{
|
|
Replacement: "/who-am-i",
|
|
Regex: `/whoami`,
|
|
},
|
|
expectedPath: "/who-am-i/and/who-am-i",
|
|
expectedRawPath: "/who-am-i/and/who-am-i",
|
|
expectedHeader: "/whoami/and/whoami",
|
|
},
|
|
{
|
|
desc: "no match",
|
|
path: "/whoami/and/whoami",
|
|
config: dynamic.ReplacePathRegex{
|
|
Replacement: "/whoami",
|
|
Regex: `/no-match`,
|
|
},
|
|
expectedPath: "/whoami/and/whoami",
|
|
},
|
|
{
|
|
desc: "multiple replacement",
|
|
path: "/downloads/src/source.go",
|
|
config: dynamic.ReplacePathRegex{
|
|
Replacement: "/downloads/$1-$2",
|
|
Regex: `^(?i)/downloads/([^/]+)/([^/]+)$`,
|
|
},
|
|
expectedPath: "/downloads/src-source.go",
|
|
expectedRawPath: "/downloads/src-source.go",
|
|
expectedHeader: "/downloads/src/source.go",
|
|
},
|
|
{
|
|
desc: "invalid regular expression",
|
|
path: "/invalid/regexp/test",
|
|
config: dynamic.ReplacePathRegex{
|
|
Replacement: "/valid/regexp/$1",
|
|
Regex: `^(?err)/invalid/regexp/([^/]+)$`,
|
|
},
|
|
expectedPath: "/invalid/regexp/test",
|
|
expectsError: true,
|
|
},
|
|
{
|
|
desc: "replacement with escaped char",
|
|
path: "/aaa/bbb",
|
|
config: dynamic.ReplacePathRegex{
|
|
Replacement: "/foo%2Fbar",
|
|
Regex: `/aaa/bbb`,
|
|
},
|
|
expectedPath: "/foo/bar",
|
|
expectedRawPath: "/foo%2Fbar",
|
|
expectedHeader: "/aaa/bbb",
|
|
},
|
|
{
|
|
desc: "path and regex with escaped char",
|
|
path: "/aaa%2Fbbb",
|
|
config: dynamic.ReplacePathRegex{
|
|
Replacement: "/foo/bar",
|
|
Regex: `/aaa%2Fbbb`,
|
|
},
|
|
expectedPath: "/foo/bar",
|
|
expectedRawPath: "/foo/bar",
|
|
expectedHeader: "/aaa%2Fbbb",
|
|
},
|
|
{
|
|
desc: "path with escaped char (no match)",
|
|
path: "/aaa%2Fbbb",
|
|
config: dynamic.ReplacePathRegex{
|
|
Replacement: "/foo/bar",
|
|
Regex: `/aaa/bbb`,
|
|
},
|
|
expectedPath: "/aaa/bbb",
|
|
expectedRawPath: "/aaa%2Fbbb",
|
|
},
|
|
}
|
|
|
|
for _, test := range testCases {
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
var actualPath, actualRawPath, actualHeader, requestURI string
|
|
next := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
actualPath = r.URL.Path
|
|
actualRawPath = r.URL.RawPath
|
|
actualHeader = r.Header.Get(replacepath.ReplacedPathHeader)
|
|
requestURI = r.RequestURI
|
|
})
|
|
|
|
handler, err := New(context.Background(), next, test.config, "foo-replace-path-regexp")
|
|
if test.expectsError {
|
|
require.Error(t, err)
|
|
return
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
|
|
server := httptest.NewServer(handler)
|
|
defer server.Close()
|
|
|
|
resp, err := http.Get(server.URL + test.path)
|
|
require.NoError(t, err, "Unexpected error while making test request")
|
|
require.Equal(t, http.StatusOK, resp.StatusCode)
|
|
|
|
assert.Equal(t, test.expectedPath, actualPath, "Unexpected path.")
|
|
assert.Equal(t, test.expectedRawPath, actualRawPath, "Unexpected raw path.")
|
|
|
|
if actualRawPath == "" {
|
|
assert.Equal(t, actualPath, requestURI, "Unexpected request URI.")
|
|
} else {
|
|
assert.Equal(t, actualRawPath, requestURI, "Unexpected request URI.")
|
|
}
|
|
|
|
if test.expectedHeader != "" {
|
|
assert.Equal(t, test.expectedHeader, actualHeader, "Unexpected '%s' header.", replacepath.ReplacedPathHeader)
|
|
}
|
|
})
|
|
}
|
|
}
|