mirror of
https://github.com/portainer/portainer.git
synced 2025-07-24 15:59:41 +02:00
- added new method to get api key by ID - added tests
This commit is contained in:
parent
7cc28b10a0
commit
5839f96787
5 changed files with 56 additions and 0 deletions
|
@ -56,6 +56,15 @@ func (handler *Handler) userRemoveAccessToken(w http.ResponseWriter, r *http.Req
|
|||
return &httperror.HandlerError{http.StatusInternalServerError, "Unable to find a user with the specified identifier inside the database", err}
|
||||
}
|
||||
|
||||
// check if the key exists and the key belongs to the user
|
||||
apiKey, err := handler.apiKeyService.GetAPIKey(portainer.APIKeyID(apiKeyID))
|
||||
if err != nil {
|
||||
return &httperror.HandlerError{http.StatusInternalServerError, "API Key not found", err}
|
||||
}
|
||||
if apiKey.UserID != portainer.UserID(userID) {
|
||||
return &httperror.HandlerError{http.StatusForbidden, "Permission denied to remove api-key", httperrors.ErrUnauthorized}
|
||||
}
|
||||
|
||||
err = handler.apiKeyService.DeleteAPIKey(portainer.APIKeyID(apiKeyID))
|
||||
if err != nil {
|
||||
if err == apikey.ErrInvalidAPIKey {
|
||||
|
|
|
@ -97,4 +97,25 @@ func Test_userRemoveAccessToken(t *testing.T) {
|
|||
|
||||
is.Equal(0, len(keys))
|
||||
})
|
||||
|
||||
t.Run("user cannot delete another users API Keys using api-key auth", func(t *testing.T) {
|
||||
_, adminAPIKey, err := apiKeyService.GenerateApiKey(*adminUser, "admin-key")
|
||||
is.NoError(err)
|
||||
|
||||
rawAPIKey, _, err := apiKeyService.GenerateApiKey(*user, "user-key")
|
||||
is.NoError(err)
|
||||
|
||||
req := httptest.NewRequest(http.MethodDelete, fmt.Sprintf("/users/%d/tokens/%d", user.ID, adminAPIKey.ID), nil)
|
||||
req.Header.Add("x-api-key", rawAPIKey)
|
||||
|
||||
rr := httptest.NewRecorder()
|
||||
h.ServeHTTP(rr, req)
|
||||
|
||||
is.Equal(http.StatusForbidden, rr.Code)
|
||||
|
||||
adminKeyGot, err := apiKeyService.GetAPIKey(adminAPIKey.ID)
|
||||
is.NoError(err)
|
||||
|
||||
is.Equal(adminAPIKey, adminKeyGot)
|
||||
})
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue