annika/jwtauth
1
0
Fork 0
mirror of https://forgejo.merr.is/annika/jwtauth.git synced 2025-12-11 18:41:09 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Test status code in tests

Browse source
This commit is contained in:
Peter Kieltyka 2016-01-21 14:19:49 -05:00
parent 2d088c1f3a
commit b2ccd8612d
1 changed files with 15 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

30
jwtauth_test.go
View file

@ -41,22 +41,22 @@ func TestSimple(t *testing.T) {
defer ts.Close() defer ts.Close()
// sending unauthorized requests // sending unauthorized requests
if resp := testRequest(t, ts, "GET", "/", nil, nil); resp != "Unauthorized\n" { if status, resp := testRequest(t, ts, "GET", "/", nil, nil); status != 401 && resp != "Unauthorized\n" {
t.Fatalf(resp) t.Fatalf(resp)
} }
h := http.Header{} h := http.Header{}
h.Set("Authorization", "BEARER "+newJwtToken([]byte("wrong"), map[string]interface{}{})) h.Set("Authorization", "BEARER "+newJwtToken([]byte("wrong"), map[string]interface{}{}))
if resp := testRequest(t, ts, "GET", "/", h, nil); resp != "Unauthorized\n" { if status, resp := testRequest(t, ts, "GET", "/", h, nil); status != 401 && resp != "Unauthorized\n" {
t.Fatalf(resp) t.Fatalf(resp)
} }
h.Set("Authorization", "BEARER asdf") h.Set("Authorization", "BEARER asdf")
if resp := testRequest(t, ts, "GET", "/", h, nil); resp != "Unauthorized\n" { if status, resp := testRequest(t, ts, "GET", "/", h, nil); status != 401 && resp != "Unauthorized\n" {
t.Fatalf(resp) t.Fatalf(resp)
} }
// sending authorized requests // sending authorized requests
if resp := testRequest(t, ts, "GET", "/", newAuthHeader(), nil); resp != "welcome" { if status, resp := testRequest(t, ts, "GET", "/", newAuthHeader(), nil); status != 200 && resp != "welcome" {
t.Fatalf(resp) t.Fatalf(resp)
} }
} }
@ -114,32 +114,32 @@ func TestMore(t *testing.T) {
defer ts.Close() defer ts.Close()
// sending unauthorized requests // sending unauthorized requests
if resp := testRequest(t, ts, "GET", "/admin", nil, nil); resp != "Unauthorized\n" { if status, resp := testRequest(t, ts, "GET", "/admin", nil, nil); status != 401 && resp != "Unauthorized\n" {
t.Fatalf(resp) t.Fatalf(resp)
} }
h := http.Header{} h := http.Header{}
h.Set("Authorization", "BEARER "+newJwtToken([]byte("wrong"), map[string]interface{}{})) h.Set("Authorization", "BEARER "+newJwtToken([]byte("wrong"), map[string]interface{}{}))
if resp := testRequest(t, ts, "GET", "/admin", h, nil); resp != "Unauthorized\n" { if status, resp := testRequest(t, ts, "GET", "/admin", h, nil); status != 401 && resp != "Unauthorized\n" {
t.Fatalf(resp) t.Fatalf(resp)
} }
h.Set("Authorization", "BEARER asdf") h.Set("Authorization", "BEARER asdf")
if resp := testRequest(t, ts, "GET", "/admin", h, nil); resp != "Unauthorized\n" { if status, resp := testRequest(t, ts, "GET", "/admin", h, nil); status != 401 && resp != "Unauthorized\n" {
t.Fatalf(resp) t.Fatalf(resp)
} }
h = newAuthHeader((jwtauth.Claims{}).Set("exp", jwtauth.EpochNow()-1000)) h = newAuthHeader((jwtauth.Claims{}).Set("exp", jwtauth.EpochNow()-1000))
if resp := testRequest(t, ts, "GET", "/admin", h, nil); resp != "expired\n" { if status, resp := testRequest(t, ts, "GET", "/admin", h, nil); status != 401 && resp != "expired\n" {
t.Fatalf(resp) t.Fatalf(resp)
} }
// sending authorized requests // sending authorized requests
if resp := testRequest(t, ts, "GET", "/", nil, nil); resp != "welcome" { if status, resp := testRequest(t, ts, "GET", "/", nil, nil); status != 200 && resp != "welcome" {
t.Fatalf(resp) t.Fatalf(resp)
} }
h = newAuthHeader((jwtauth.Claims{}).SetExpiryIn(5 * time.Minute)) h = newAuthHeader((jwtauth.Claims{}).SetExpiryIn(5 * time.Minute))
if resp := testRequest(t, ts, "GET", "/admin", h, nil); resp != "protected" { if status, resp := testRequest(t, ts, "GET", "/admin", h, nil); status != 200 && resp != "protected" {
t.Fatalf(resp) t.Fatalf(resp)
} }
} }
@ -148,11 +148,11 @@ func TestMore(t *testing.T) {
// Test helper functions // Test helper functions
// //
func testRequest(t *testing.T, ts *httptest.Server, method, path string, header http.Header, body io.Reader) string { func testRequest(t *testing.T, ts *httptest.Server, method, path string, header http.Header, body io.Reader) (int, string) {
req, err := http.NewRequest(method, ts.URL+path, body) req, err := http.NewRequest(method, ts.URL+path, body)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
return "" return 0, ""
} }
if header != nil { if header != nil {
@ -164,17 +164,17 @@ func testRequest(t *testing.T, ts *httptest.Server, method, path string, header
resp, err := http.DefaultClient.Do(req) resp, err := http.DefaultClient.Do(req)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
return "" return 0, ""
} }
respBody, err := ioutil.ReadAll(resp.Body) respBody, err := ioutil.ReadAll(resp.Body)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
return "" return 0, ""
} }
defer resp.Body.Close() defer resp.Body.Close()
return string(respBody) return resp.StatusCode, string(respBody)
} }
func newJwtToken(secret []byte, claims ...jwtauth.Claims) string { func newJwtToken(secret []byte, claims ...jwtauth.Claims) string {