mirror of
https://github.com/lukaszraczylo/traefikoidc.git
synced 2026-10-17 04:37:48 +00:00
feat(lib): add TrustForwardedURI to honor X-Forwarded-Uri for post-login redirect target
This commit is contained in:
@@ -0,0 +1,33 @@
|
||||
package traefikoidc
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestOriginalRequestURI_DefaultOff(t *testing.T) {
|
||||
tr := &TraefikOidc{trustForwardedURI: false}
|
||||
req := httptest.NewRequest(http.MethodGet, "/protected?x=1", nil)
|
||||
req.Header.Set("X-Forwarded-Uri", "/spoofed")
|
||||
if got := tr.originalRequestURI(req); got != "/protected?x=1" {
|
||||
t.Fatalf("default-off: want /protected?x=1, got %q", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestOriginalRequestURI_TrustEnabled(t *testing.T) {
|
||||
tr := &TraefikOidc{trustForwardedURI: true}
|
||||
req := httptest.NewRequest(http.MethodGet, "/protected?x=1", nil)
|
||||
req.Header.Set("X-Forwarded-Uri", "/real?y=2")
|
||||
if got := tr.originalRequestURI(req); got != "/real?y=2" {
|
||||
t.Fatalf("trust-on with header: want /real?y=2, got %q", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestOriginalRequestURI_TrustEnabledNoHeader(t *testing.T) {
|
||||
tr := &TraefikOidc{trustForwardedURI: true}
|
||||
req := httptest.NewRequest(http.MethodGet, "/protected", nil)
|
||||
if got := tr.originalRequestURI(req); got != "/protected" {
|
||||
t.Fatalf("trust-on no header: want /protected, got %q", got)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user