61 lines
2.1 KiB
Go
61 lines
2.1 KiB
Go
package upstream
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"regexp"
|
|
|
|
"github.com/oauth2-proxy/oauth2-proxy/v7/pkg/app/pagewriter"
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/ginkgo/extensions/table"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("Rewrite", func() {
|
|
type rewritePathTableInput struct {
|
|
rewriteRegex *regexp.Regexp
|
|
rewriteTarget string
|
|
requestTarget string
|
|
expectedRequestURI string
|
|
}
|
|
|
|
DescribeTable("should rewrite the request path",
|
|
func(in rewritePathTableInput) {
|
|
req := httptest.NewRequest("", in.requestTarget, nil)
|
|
rw := httptest.NewRecorder()
|
|
|
|
var gotRequestURI string
|
|
handler := newRewritePath(in.rewriteRegex, in.rewriteTarget, &pagewriter.WriterFuncs{})(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
gotRequestURI = r.RequestURI
|
|
}))
|
|
handler.ServeHTTP(rw, req)
|
|
|
|
Expect(gotRequestURI).To(Equal(in.expectedRequestURI))
|
|
},
|
|
Entry("when the path matches the regexp", rewritePathTableInput{
|
|
rewriteRegex: regexp.MustCompile("^/http/(.*)"),
|
|
rewriteTarget: "/$1",
|
|
requestTarget: "http://example.com/http/foo/bar",
|
|
expectedRequestURI: "http://example.com/foo/bar",
|
|
}),
|
|
Entry("when the path does not match the regexp", rewritePathTableInput{
|
|
rewriteRegex: regexp.MustCompile("^/http/(.*)"),
|
|
rewriteTarget: "/$1",
|
|
requestTarget: "https://example.com/https/foo/bar",
|
|
expectedRequestURI: "https://example.com/https/foo/bar",
|
|
}),
|
|
Entry("when the regexp is not anchored", rewritePathTableInput{
|
|
rewriteRegex: regexp.MustCompile("/http/(.*)"),
|
|
rewriteTarget: "/$1",
|
|
requestTarget: "http://example.com/bar/http/foo/bar",
|
|
expectedRequestURI: "http://example.com/bar/foo/bar",
|
|
}),
|
|
Entry("when the regexp is rewriting to a query", rewritePathTableInput{
|
|
rewriteRegex: regexp.MustCompile(`/articles/([a-z0-9\-]*)`),
|
|
rewriteTarget: "/article?id=$1",
|
|
requestTarget: "http://example.com/articles/blog-2021-01-01",
|
|
expectedRequestURI: "http://example.com/article?id=blog-2021-01-01",
|
|
}),
|
|
)
|
|
})
|