meutel/oauth2_proxy
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Make proper PassAccessTokenTest methods

Browse Source
This commit is contained in:
Mike Bland 2015-04-06 21:35:58 -04:00
parent 9887ac3be5
commit 83ad43a571
1 changed files with 17 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
oauthproxy_test.go
View File

@ -152,24 +152,25 @@ func NewPassAccessTokenTest(opts PassAccessTokenTestOptions) *PassAccessTokenTes
return t return t
} }
func Close(t *PassAccessTokenTest) { func (pat_test *PassAccessTokenTest) Close() {
t.provider_server.Close() pat_test.provider_server.Close()
} }
func getCallbackEndpoint(pac_test *PassAccessTokenTest) (http_code int, cookie string) { func (pat_test *PassAccessTokenTest) getCallbackEndpoint() (http_code int,
cookie string) {
rw := httptest.NewRecorder() rw := httptest.NewRecorder()
req, err := http.NewRequest("GET", "/oauth2/callback?code=callback_code", req, err := http.NewRequest("GET", "/oauth2/callback?code=callback_code",
strings.NewReader("")) strings.NewReader(""))
if err != nil { if err != nil {
return 0, "" return 0, ""
} }
pac_test.proxy.ServeHTTP(rw, req) pat_test.proxy.ServeHTTP(rw, req)
return rw.Code, rw.HeaderMap["Set-Cookie"][0] return rw.Code, rw.HeaderMap["Set-Cookie"][0]
} }
func getRootEndpoint(pac_test *PassAccessTokenTest, cookie string) (http_code int, func (pat_test *PassAccessTokenTest) getRootEndpoint(
access_token string) { cookie string) (http_code int, access_token string) {
cookie_key := pac_test.proxy.CookieKey cookie_key := pat_test.proxy.CookieKey
var value string var value string
key_prefix := cookie_key + "=" key_prefix := cookie_key + "="
@ -198,43 +199,43 @@ func getRootEndpoint(pac_test *PassAccessTokenTest, cookie string) (http_code in
}) })
rw := httptest.NewRecorder() rw := httptest.NewRecorder()
pac_test.proxy.ServeHTTP(rw, req) pat_test.proxy.ServeHTTP(rw, req)
return rw.Code, rw.Body.String() return rw.Code, rw.Body.String()
} }
func TestForwardAccessTokenUpstream(t *testing.T) { func TestForwardAccessTokenUpstream(t *testing.T) {
pac_test := NewPassAccessTokenTest(PassAccessTokenTestOptions{ pat_test := NewPassAccessTokenTest(PassAccessTokenTestOptions{
PassAccessToken: true, PassAccessToken: true,
}) })
defer Close(pac_test) defer pat_test.Close()
// A successful validation will redirect and set the auth cookie. // A successful validation will redirect and set the auth cookie.
code, cookie := getCallbackEndpoint(pac_test) code, cookie := pat_test.getCallbackEndpoint()
assert.Equal(t, 302, code) assert.Equal(t, 302, code)
assert.NotEqual(t, nil, cookie) assert.NotEqual(t, nil, cookie)
// Now we make a regular request; the access_token from the cookie is // Now we make a regular request; the access_token from the cookie is
// forwarded as the "X-Forwarded-Access-Token" header. The token is // forwarded as the "X-Forwarded-Access-Token" header. The token is
// read by the test provider server and written in the response body. // read by the test provider server and written in the response body.
code, payload := getRootEndpoint(pac_test, cookie) code, payload := pat_test.getRootEndpoint(cookie)
assert.Equal(t, 200, code) assert.Equal(t, 200, code)
assert.Equal(t, "my_auth_token", payload) assert.Equal(t, "my_auth_token", payload)
} }
func TestDoNotForwardAccessTokenUpstream(t *testing.T) { func TestDoNotForwardAccessTokenUpstream(t *testing.T) {
pac_test := NewPassAccessTokenTest(PassAccessTokenTestOptions{ pat_test := NewPassAccessTokenTest(PassAccessTokenTestOptions{
PassAccessToken: false, PassAccessToken: false,
}) })
defer Close(pac_test) defer pat_test.Close()
// A successful validation will redirect and set the auth cookie. // A successful validation will redirect and set the auth cookie.
code, cookie := getCallbackEndpoint(pac_test) code, cookie := pat_test.getCallbackEndpoint()
assert.Equal(t, 302, code) assert.Equal(t, 302, code)
assert.NotEqual(t, nil, cookie) assert.NotEqual(t, nil, cookie)
// Now we make a regular request, but the access token header should // Now we make a regular request, but the access token header should
// not be present. // not be present.
code, payload := getRootEndpoint(pac_test, cookie) code, payload := pat_test.getRootEndpoint(cookie)
assert.Equal(t, 200, code) assert.Equal(t, 200, code)
assert.Equal(t, "No access token found.", payload) assert.Equal(t, "No access token found.", payload)
} }