82 lines
3 KiB
Go
82 lines
3 KiB
Go
package application
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"goauthentik.io/api/v3"
|
|
"goauthentik.io/internal/outpost/proxyv2/constants"
|
|
)
|
|
|
|
func TestRedirectToStart_Proxy(t *testing.T) {
|
|
a := newTestApplication()
|
|
a.proxyConfig.Mode = *api.NewNullableProxyMode(api.PROXYMODE_PROXY.Ptr())
|
|
a.proxyConfig.ExternalHost = "https://test.goauthentik.io"
|
|
req, _ := http.NewRequest("GET", "/foo/bar/baz", nil)
|
|
|
|
rr := httptest.NewRecorder()
|
|
a.redirectToStart(rr, req)
|
|
|
|
assert.Equal(t, http.StatusFound, rr.Code)
|
|
loc, _ := rr.Result().Location()
|
|
assert.Equal(t, "https://test.goauthentik.io/outpost.goauthentik.io/start?rd=https%3A%2F%2Ftest.goauthentik.io%2Ffoo%2Fbar%2Fbaz", loc.String())
|
|
|
|
s, _ := a.sessions.Get(req, a.SessionName())
|
|
assert.Equal(t, "https://test.goauthentik.io/foo/bar/baz", s.Values[constants.SessionRedirect])
|
|
}
|
|
|
|
func TestRedirectToStart_Forward(t *testing.T) {
|
|
a := newTestApplication()
|
|
a.proxyConfig.Mode = *api.NewNullableProxyMode(api.PROXYMODE_FORWARD_SINGLE.Ptr())
|
|
a.proxyConfig.ExternalHost = "https://test.goauthentik.io"
|
|
req, _ := http.NewRequest("GET", "/foo/bar/baz", nil)
|
|
|
|
rr := httptest.NewRecorder()
|
|
a.redirectToStart(rr, req)
|
|
|
|
assert.Equal(t, http.StatusFound, rr.Code)
|
|
loc, _ := rr.Result().Location()
|
|
assert.Equal(t, "https://test.goauthentik.io/outpost.goauthentik.io/start?rd=https%3A%2F%2Ftest.goauthentik.io%2Ffoo%2Fbar%2Fbaz", loc.String())
|
|
|
|
s, _ := a.sessions.Get(req, a.SessionName())
|
|
assert.Equal(t, "https://test.goauthentik.io/foo/bar/baz", s.Values[constants.SessionRedirect])
|
|
}
|
|
|
|
func TestRedirectToStart_Forward_Domain_Invalid(t *testing.T) {
|
|
a := newTestApplication()
|
|
a.proxyConfig.CookieDomain = api.PtrString("foo")
|
|
a.proxyConfig.Mode = *api.NewNullableProxyMode(api.PROXYMODE_FORWARD_DOMAIN.Ptr())
|
|
a.proxyConfig.ExternalHost = "https://test.goauthentik.io"
|
|
req, _ := http.NewRequest("GET", "/foo/bar/baz", nil)
|
|
|
|
rr := httptest.NewRecorder()
|
|
a.redirectToStart(rr, req)
|
|
|
|
assert.Equal(t, http.StatusFound, rr.Code)
|
|
loc, _ := rr.Result().Location()
|
|
assert.Equal(t, "https://test.goauthentik.io/outpost.goauthentik.io/start?rd=https%3A%2F%2Ftest.goauthentik.io", loc.String())
|
|
|
|
s, _ := a.sessions.Get(req, a.SessionName())
|
|
assert.Equal(t, "https://test.goauthentik.io", s.Values[constants.SessionRedirect])
|
|
}
|
|
|
|
func TestRedirectToStart_Forward_Domain(t *testing.T) {
|
|
a := newTestApplication()
|
|
a.proxyConfig.CookieDomain = api.PtrString("goauthentik.io")
|
|
a.proxyConfig.Mode = *api.NewNullableProxyMode(api.PROXYMODE_FORWARD_DOMAIN.Ptr())
|
|
a.proxyConfig.ExternalHost = "https://test.goauthentik.io"
|
|
req, _ := http.NewRequest("GET", "/foo/bar/baz", nil)
|
|
|
|
rr := httptest.NewRecorder()
|
|
a.redirectToStart(rr, req)
|
|
|
|
assert.Equal(t, http.StatusFound, rr.Code)
|
|
loc, _ := rr.Result().Location()
|
|
assert.Equal(t, "https://test.goauthentik.io/outpost.goauthentik.io/start?rd=https%3A%2F%2Ftest.goauthentik.io", loc.String())
|
|
|
|
s, _ := a.sessions.Get(req, a.SessionName())
|
|
assert.Equal(t, "https://test.goauthentik.io", s.Values[constants.SessionRedirect])
|
|
}
|