181 lines
8.4 KiB
Go
181 lines
8.4 KiB
Go
// Vikunja is a to-do list application to facilitate your life.
|
|
// Copyright 2018-present Vikunja and contributors. All rights reserved.
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package webtests
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"code.vikunja.io/api/pkg/user"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// Bot owner from pkg/db/fixtures/users.yml: user21 owns bot 23. No api_tokens
|
|
// fixture is owned by a bot, so listing a caller-owned bot's tokens returns an
|
|
// authorized (but empty) list.
|
|
var testuser21 = user.User{ID: 21, Username: "user_bot_owner_a", Issuer: "local"}
|
|
|
|
// TestHumaAPIToken ports the v1 model-level permission matrix
|
|
// (pkg/models/api_tokens_test.go) onto the v2 HTTP surface so /api/v2/tokens
|
|
// independently proves the full owner/bot-owner contract once v1's routes are
|
|
// gone. /tokens supports list/create/delete only — no ReadOne, no Update.
|
|
//
|
|
// Every request runs against one shared echo.Echo: setupTestEnv re-randomizes
|
|
// the global JWT secret on each call, so a second env would invalidate the
|
|
// first's tokens. One env, one secret, all callers via humaTokenFor.
|
|
//
|
|
// Fixture topology (pkg/db/fixtures/api_tokens.yml):
|
|
// - tokens #1, #2: owned by user1.
|
|
// - token #3: owned by user2 — never visible to user1, never deletable by user1.
|
|
// - tokens #4, #5: owned by disabled/locked users 17/18.
|
|
// - tokens #6, #7: owned by user15.
|
|
// - token #8: owned by user13.
|
|
func TestHumaAPIToken(t *testing.T) {
|
|
e, err := setupTestEnv()
|
|
require.NoError(t, err)
|
|
user1Token := humaTokenFor(t, &testuser1)
|
|
user2Token := humaTokenFor(t, &testuser2)
|
|
botOwnerToken := humaTokenFor(t, &testuser21)
|
|
|
|
t.Run("ReadAll", func(t *testing.T) {
|
|
t.Run("Normal - exact owned set for user1", func(t *testing.T) {
|
|
rec := humaRequest(t, e, http.MethodGet, "/api/v2/tokens", "", user1Token, "")
|
|
require.Equal(t, http.StatusOK, rec.Code, "body: %s", rec.Body.String())
|
|
ids := apiTokenIDsFromReadAll(t, rec.Body.Bytes())
|
|
// user1 owns exactly tokens #1 and #2; cardinality is pinned.
|
|
assert.ElementsMatch(t, []int64{1, 2}, ids,
|
|
"ReadAll must return exactly {1,2}; body: %s", rec.Body.String())
|
|
assert.Equal(t, int64(2), apiTokenTotalFromReadAll(t, rec.Body.Bytes()))
|
|
assert.NotContains(t, ids, int64(3), "token #3 (owned by user2) must be hidden")
|
|
})
|
|
t.Run("Isolation - user2 sees only its own token", func(t *testing.T) {
|
|
rec := humaRequest(t, e, http.MethodGet, "/api/v2/tokens", "", user2Token, "")
|
|
require.Equal(t, http.StatusOK, rec.Code, "body: %s", rec.Body.String())
|
|
ids := apiTokenIDsFromReadAll(t, rec.Body.Bytes())
|
|
assert.ElementsMatch(t, []int64{3}, ids,
|
|
"user2 must see only token #3; body: %s", rec.Body.String())
|
|
assert.NotContains(t, ids, int64(1), "token #1 (owned by user1) must be hidden from user2")
|
|
assert.NotContains(t, ids, int64(2), "token #2 (owned by user1) must be hidden from user2")
|
|
})
|
|
t.Run("Search by title", func(t *testing.T) {
|
|
rec := humaRequest(t, e, http.MethodGet, "/api/v2/tokens?q=test+token+1", "", user1Token, "")
|
|
require.Equal(t, http.StatusOK, rec.Code, "body: %s", rec.Body.String())
|
|
ids := apiTokenIDsFromReadAll(t, rec.Body.Bytes())
|
|
assert.ElementsMatch(t, []int64{1}, ids,
|
|
"search must narrow to token #1; body: %s", rec.Body.String())
|
|
})
|
|
t.Run("owner_id - caller's own bot returns an authorized (empty) list", func(t *testing.T) {
|
|
rec := humaRequest(t, e, http.MethodGet, "/api/v2/tokens?owner_id=23", "", botOwnerToken, "")
|
|
require.Equal(t, http.StatusOK, rec.Code, "body: %s", rec.Body.String())
|
|
ids := apiTokenIDsFromReadAll(t, rec.Body.Bytes())
|
|
assert.Empty(t, ids, "bot 23 has no tokens; list must be empty but authorized; body: %s", rec.Body.String())
|
|
})
|
|
t.Run("owner_id - forbidden when not the bot's owner", func(t *testing.T) {
|
|
// user1 is not the owner of bot 23 (owned by user21).
|
|
rec := humaRequest(t, e, http.MethodGet, "/api/v2/tokens?owner_id=23", "", user1Token, "")
|
|
assert.Equal(t, http.StatusForbidden, rec.Code, "body: %s", rec.Body.String())
|
|
})
|
|
t.Run("owner_id - forbidden when the target is not a bot", func(t *testing.T) {
|
|
// user2 is a real user, not a bot owned by user1.
|
|
rec := humaRequest(t, e, http.MethodGet, "/api/v2/tokens?owner_id=2", "", user1Token, "")
|
|
assert.Equal(t, http.StatusForbidden, rec.Code, "body: %s", rec.Body.String())
|
|
})
|
|
})
|
|
|
|
t.Run("Create", func(t *testing.T) {
|
|
t.Run("Normal - returns cleartext token once and sets owner", func(t *testing.T) {
|
|
rec := humaRequest(t, e, http.MethodPost, "/api/v2/tokens",
|
|
`{"title":"new token","permissions":{"tasks":["read_all"]},"expires_at":"2099-01-01T00:00:00Z"}`,
|
|
user1Token, "")
|
|
require.Equal(t, http.StatusCreated, rec.Code, "body: %s", rec.Body.String())
|
|
|
|
var created struct {
|
|
ID int64 `json:"id"`
|
|
Token string `json:"token"`
|
|
Title string `json:"title"`
|
|
OwnerID int64 `json:"owner_id"`
|
|
}
|
|
require.NoError(t, json.Unmarshal(rec.Body.Bytes(), &created))
|
|
assert.Equal(t, "new token", created.Title)
|
|
assert.NotEmpty(t, created.Token, "create response must include the cleartext token exactly once")
|
|
require.Greater(t, len(created.Token), 3)
|
|
assert.Equal(t, "tk_", created.Token[:3], "cleartext token must carry the tk_ prefix; got %q", created.Token)
|
|
assert.Equal(t, int64(1), created.OwnerID, "owner must default to the authenticated user")
|
|
})
|
|
t.Run("Empty title", func(t *testing.T) {
|
|
// v2 enforces required title at the schema level → 422.
|
|
rec := humaRequest(t, e, http.MethodPost, "/api/v2/tokens",
|
|
`{"title":"","permissions":{"tasks":["read_all"]},"expires_at":"2099-01-01T00:00:00Z"}`,
|
|
user1Token, "")
|
|
assert.Equal(t, http.StatusUnprocessableEntity, rec.Code, "body: %s", rec.Body.String())
|
|
})
|
|
t.Run("owner_id - forbidden when the target is not a caller-owned bot", func(t *testing.T) {
|
|
// user1 cannot mint a token for user2 (not a bot they own).
|
|
rec := humaRequest(t, e, http.MethodPost, "/api/v2/tokens",
|
|
`{"title":"sneaky","owner_id":2,"permissions":{"tasks":["read_all"]},"expires_at":"2099-01-01T00:00:00Z"}`,
|
|
user1Token, "")
|
|
assert.Equal(t, http.StatusForbidden, rec.Code, "body: %s", rec.Body.String())
|
|
})
|
|
})
|
|
|
|
t.Run("Delete", func(t *testing.T) {
|
|
t.Run("Forbidden - token of another user", func(t *testing.T) {
|
|
// Token #3 belongs to user2; user1 must not be able to delete it.
|
|
rec := humaRequest(t, e, http.MethodDelete, "/api/v2/tokens/3", "", user1Token, "")
|
|
assert.Equal(t, http.StatusForbidden, rec.Code, "body: %s", rec.Body.String())
|
|
})
|
|
t.Run("Nonexisting", func(t *testing.T) {
|
|
// CanDelete returns (false, nil) for a missing token → generic forbidden.
|
|
rec := humaRequest(t, e, http.MethodDelete, "/api/v2/tokens/9999", "", user1Token, "")
|
|
assert.Equal(t, http.StatusForbidden, rec.Code, "body: %s", rec.Body.String())
|
|
})
|
|
t.Run("Normal - own token", func(t *testing.T) {
|
|
// Runs last so deleting #1 doesn't disturb the ReadAll cardinality assertions.
|
|
rec := humaRequest(t, e, http.MethodDelete, "/api/v2/tokens/1", "", user1Token, "")
|
|
require.Equal(t, http.StatusNoContent, rec.Code, "body: %s", rec.Body.String())
|
|
assert.Empty(t, rec.Body.String())
|
|
})
|
|
})
|
|
}
|
|
|
|
func apiTokenIDsFromReadAll(t *testing.T, body []byte) []int64 {
|
|
t.Helper()
|
|
var resp struct {
|
|
Items []struct {
|
|
ID int64 `json:"id"`
|
|
} `json:"items"`
|
|
}
|
|
require.NoError(t, json.Unmarshal(body, &resp), "ReadAll body must be a paginated envelope: %s", string(body))
|
|
ids := make([]int64, 0, len(resp.Items))
|
|
for _, it := range resp.Items {
|
|
ids = append(ids, it.ID)
|
|
}
|
|
return ids
|
|
}
|
|
|
|
func apiTokenTotalFromReadAll(t *testing.T, body []byte) int64 {
|
|
t.Helper()
|
|
var resp struct {
|
|
Total int64 `json:"total"`
|
|
}
|
|
require.NoError(t, json.Unmarshal(body, &resp), "ReadAll body must be a paginated envelope: %s", string(body))
|
|
return resp.Total
|
|
}
|