wn/traefik-forward-auth
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Simplify oauth server testing

Browse Source
This commit is contained in:
Thom Seddon 2020-01-15 21:52:41 +00:00
parent c9289d6fc1
commit 391793b220
1 changed files with 34 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
internal/server_test.go
View File

@ -121,19 +121,19 @@ func TestServerAuthCallback(t *testing.T) {
assert := assert.New(t) assert := assert.New(t)
config = newDefaultConfig() config = newDefaultConfig()
// Setup token server // Setup OAuth server
tokenServerHandler := &TokenServerHandler{} server, serverURL := NewOAuthServer(t)
tokenServer := httptest.NewServer(tokenServerHandler) defer server.Close()
defer tokenServer.Close() config.Providers.Google.TokenURL = &url.URL{
tokenUrl, _ := url.Parse(tokenServer.URL) Scheme: serverURL.Scheme,
config.Providers.Google.TokenURL = tokenUrl Host: serverURL.Host,
Path: "/token",
// Setup user server }
userServerHandler := &UserServerHandler{} config.Providers.Google.UserURL = &url.URL{
userServer := httptest.NewServer(userServerHandler) Scheme: serverURL.Scheme,
defer userServer.Close() Host: serverURL.Host,
userUrl, _ := url.Parse(userServer.URL) Path: "/userinfo",
config.Providers.Google.UserURL = userUrl }
// Should pass auth response request to callback // Should pass auth response request to callback
req := newDefaultHttpRequest("/_oauth") req := newDefaultHttpRequest("/_oauth")
@ -342,21 +342,30 @@ func TestServerRouteQuery(t *testing.T) {
* Utilities * Utilities
*/ */
type TokenServerHandler struct{} type OAuthServer struct {
t *testing.T
func (t *TokenServerHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
fmt.Fprint(w, `{"access_token":"123456789"}`)
} }
type UserServerHandler struct{} func (s *OAuthServer) ServeHTTP(w http.ResponseWriter, r *http.Request) {
if r.URL.Path == "/token" {
fmt.Fprintf(w, `{"access_token":"123456789"}`)
} else if r.URL.Path == "/userinfo" {
fmt.Fprint(w, `{
"id":"1",
"email":"example@example.com",
"verified_email":true,
"hd":"example.com"
}`)
} else {
s.t.Fatal("Unrecognised request: ", r.Method, r.URL)
}
}
func (t *UserServerHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) { func NewOAuthServer(t *testing.T) (*httptest.Server, *url.URL) {
fmt.Fprint(w, `{ handler := &OAuthServer{}
"id":"1", server := httptest.NewServer(handler)
"email":"example@example.com", serverURL, _ := url.Parse(server.URL)
"verified_email":true, return server, serverURL
"hd":"example.com"
}`)
} }
func doHttpRequest(r *http.Request, c *http.Cookie) (*http.Response, string) { func doHttpRequest(r *http.Request, c *http.Cookie) (*http.Response, string) {