diff options
Diffstat (limited to '')
| -rw-r--r-- | internal/authz/middleware_test.go | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/internal/authz/middleware_test.go b/internal/authz/middleware_test.go index cc3f6d1..9ed9081 100644 --- a/internal/authz/middleware_test.go +++ b/internal/authz/middleware_test.go @@ -40,11 +40,10 @@ func TestAuthentication(t *testing.T) { description: "Good Login", handler: func(rw http.ResponseWriter, req *http.Request) { ctx := req.Context() - uid := ctx.Value("urn") + uid := ctx.Value(AuthzUrnKey) if uid != fmt.Sprintf("uid:%s", okUserName) { t.Fatal("Context UID not set") } - }, }, { @@ -72,6 +71,7 @@ func TestAuthentication(t *testing.T) { recorder := httptest.NewRecorder() authHandler.ServeHTTP(recorder, req) result := recorder.Result() + defer result.Body.Close() if result.StatusCode != tc.statusCode { t.Fatalf("Test Case %s failed Expected: %d Found: %d", tc.description, tc.statusCode, result.StatusCode) @@ -94,13 +94,13 @@ func TestAuthorization(t *testing.T) { url: fmt.Sprintf("%s/%s", baseURL, "repo/url"), user: "uid:jack", expectedStatus: 200, - description: "an autorized action should yield a 200", + description: "an authorized action should yield a 200", }, { url: fmt.Sprintf("%s/%s", baseURL, "repo/url/bar"), user: "uid:chumba", expectedStatus: 403, - description: "an unautorized action should yield a 403", + description: "an unauthorized action should yield a 403", }, } svcr := admin.NewService( @@ -115,12 +115,13 @@ func TestAuthorization(t *testing.T) { recorder := httptest.NewRecorder() req := httptest.NewRequest(http.MethodGet, tc.url, nil) ctx := req.Context() - ctx = context.WithValue(ctx, "urn", tc.user) + ctx = context.WithValue(ctx, AuthzUrnKey, tc.user) req = req.WithContext(ctx) authHandler.ServeHTTP(recorder, req) result := recorder.Result() + defer result.Body.Close() if result.StatusCode != tc.expectedStatus { - t.Fatalf("Test Case failed Expected: %d Found: %d", tc.expectedStatus, result.StatusCode) + t.Fatalf("Test Case %s failed Expected: %d Found: %d", tc.description, tc.expectedStatus, result.StatusCode) } } } |