193 lines
8.1 KiB
Go
193 lines
8.1 KiB
Go
package api
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
// The owner row is the one identity the panel may never demote, delete, or
|
|
// disable (migration 0011) — only the local break-glass console resets it.
|
|
// These guards became load-bearing the moment provisioning started actually
|
|
// writing role='owner'; before that the owner tier was simply unreachable, so
|
|
// nothing could reach them.
|
|
func TestOwnerAccountProtectedFromPanelMutations(t *testing.T) {
|
|
owner := &Principal{UserID: "usr-root", Role: "owner", ViaAdminAccess: true}
|
|
repo := newFakeRepo()
|
|
repo.seedUser(UserView{ID: "usr-root", Username: "root", Role: "owner"})
|
|
repo.seedUser(UserView{ID: "usr-owner2", Username: "spare-owner", Role: "owner"})
|
|
repo.seedUser(UserView{ID: "u2", Username: "alice", Role: "user"})
|
|
api := newTestAPI(repo, newFakeCluster())
|
|
api.External = staticExternal{p: owner}
|
|
eh := api.ExternalHandler()
|
|
|
|
t.Run("role change refused", func(t *testing.T) {
|
|
w := do(eh, "PATCH", "/api/v1/users/usr-owner2", `{"role":"admin"}`, jsonHeader)
|
|
if w.Code != http.StatusForbidden {
|
|
t.Fatalf("demote owner: code = %d body %s, want 403", w.Code, w.Body.String())
|
|
}
|
|
if got := decodeErr(t, w); got != "owner_protected" {
|
|
t.Fatalf("demote owner: error code = %q, want owner_protected", got)
|
|
}
|
|
})
|
|
t.Run("delete refused", func(t *testing.T) {
|
|
w := do(eh, "DELETE", "/api/v1/users/usr-owner2", "", nil)
|
|
if w.Code != http.StatusForbidden {
|
|
t.Fatalf("delete owner: code = %d body %s, want 403", w.Code, w.Body.String())
|
|
}
|
|
if got := decodeErr(t, w); got != "owner_protected" {
|
|
t.Fatalf("delete owner: error code = %q, want owner_protected", got)
|
|
}
|
|
})
|
|
t.Run("disable refused", func(t *testing.T) {
|
|
w := do(eh, "POST", "/api/v1/users/usr-owner2/disable", `{"disabled":true}`, jsonHeader)
|
|
if w.Code != http.StatusForbidden {
|
|
t.Fatalf("disable owner: code = %d body %s, want 403", w.Code, w.Body.String())
|
|
}
|
|
if got := decodeErr(t, w); got != "owner_protected" {
|
|
t.Fatalf("disable owner: error code = %q, want owner_protected", got)
|
|
}
|
|
})
|
|
// The caller's own row is refused as self_protected even though it is also
|
|
// an owner: that is the reason the admin can act on.
|
|
for _, tc := range []struct{ name, method, path, body string }{
|
|
{"own role", "PATCH", "/api/v1/users/usr-root", `{"role":"admin"}`},
|
|
{"own delete", "DELETE", "/api/v1/users/usr-root", ""},
|
|
{"own disable", "POST", "/api/v1/users/usr-root/disable", `{"disabled":true}`},
|
|
} {
|
|
t.Run(tc.name+" refused as self", func(t *testing.T) {
|
|
var h map[string]string
|
|
if tc.body != "" {
|
|
h = jsonHeader
|
|
}
|
|
w := do(eh, tc.method, tc.path, tc.body, h)
|
|
if w.Code != http.StatusForbidden {
|
|
t.Fatalf("code = %d body %s, want 403", w.Code, w.Body.String())
|
|
}
|
|
if got := decodeErr(t, w); got != "self_protected" {
|
|
t.Fatalf("error code = %q, want self_protected", got)
|
|
}
|
|
})
|
|
}
|
|
t.Run("email edits on an owner stay allowed", func(t *testing.T) {
|
|
w := do(eh, "PATCH", "/api/v1/users/usr-owner2", `{"email":"[email protected]"}`, jsonHeader)
|
|
if w.Code != http.StatusOK {
|
|
t.Fatalf("edit owner email: code = %d body %s, want 200", w.Code, w.Body.String())
|
|
}
|
|
})
|
|
t.Run("control: a normal user can still be promoted", func(t *testing.T) {
|
|
w := do(eh, "PATCH", "/api/v1/users/u2", `{"role":"admin"}`, jsonHeader)
|
|
if w.Code != http.StatusOK {
|
|
t.Fatalf("promote user: code = %d body %s, want 200", w.Code, w.Body.String())
|
|
}
|
|
})
|
|
}
|
|
|
|
// The user-scoped admin sub-resources (quotas, account links) must answer 404
|
|
// for an unknown user id. Before the requireLiveUser guard the quota upsert and
|
|
// the link insert reached the users(id) foreign key and surfaced as an opaque
|
|
// 500 (found live against the drill cluster, audit #30), and the quotas read
|
|
// answered a zero-value "unlimited" view as if the id existed.
|
|
func TestAdminSubresourcesRequireLiveUser(t *testing.T) {
|
|
owner := &Principal{UserID: "usr-root", Role: "owner", ViaAdminAccess: true}
|
|
repo := newFakeRepo()
|
|
repo.seedUser(UserView{ID: "usr-root", Username: "root", Role: "owner"})
|
|
repo.seedUser(UserView{ID: "u2", Username: "alice", Role: "user"})
|
|
api := newTestAPI(repo, newFakeCluster())
|
|
api.External = staticExternal{p: owner}
|
|
eh := api.ExternalHandler()
|
|
|
|
t.Run("quotas read of an unknown user", func(t *testing.T) {
|
|
w := do(eh, "GET", "/api/v1/users/usr-nope/quotas", "", nil)
|
|
if w.Code != http.StatusNotFound {
|
|
t.Fatalf("code = %d body %s, want 404", w.Code, w.Body.String())
|
|
}
|
|
})
|
|
t.Run("quotas write of an unknown user", func(t *testing.T) {
|
|
w := do(eh, "PUT", "/api/v1/users/usr-nope/quotas", `{"max_servers":1}`, jsonHeader)
|
|
if w.Code != http.StatusNotFound {
|
|
t.Fatalf("code = %d body %s, want 404", w.Code, w.Body.String())
|
|
}
|
|
})
|
|
t.Run("account link of an unknown user", func(t *testing.T) {
|
|
w := do(eh, "POST", "/api/v1/users/usr-nope/links",
|
|
`{"mc_uuid":"22222222-3333-4444-5555-666666666666"}`, jsonHeader)
|
|
if w.Code != http.StatusNotFound {
|
|
t.Fatalf("code = %d body %s, want 404", w.Code, w.Body.String())
|
|
}
|
|
})
|
|
t.Run("control: a live user still accepts both writes", func(t *testing.T) {
|
|
w := do(eh, "PUT", "/api/v1/users/u2/quotas", `{"max_servers":2}`, jsonHeader)
|
|
if w.Code != http.StatusOK {
|
|
t.Fatalf("set quotas: code = %d body %s, want 200", w.Code, w.Body.String())
|
|
}
|
|
w = do(eh, "POST", "/api/v1/users/u2/links",
|
|
`{"mc_uuid":"22222222-3333-4444-5555-666666666677"}`, jsonHeader)
|
|
if w.Code != http.StatusOK {
|
|
t.Fatalf("link: code = %d body %s, want 200", w.Code, w.Body.String())
|
|
}
|
|
})
|
|
}
|
|
|
|
// The quotas form replaces all four caps at once, so an owner can lift a cap they
|
|
// set: a missing or null field is unlimited, 0 grants none of it. Before, a null
|
|
// field meant "leave it", the panel sent null for every emptied box, and a cap once
|
|
// set could only be moved, never removed; a negative one was written as is.
|
|
func TestSetQuotasReplacesAllCaps(t *testing.T) {
|
|
owner := &Principal{UserID: "usr-root", Role: "owner", ViaAdminAccess: true}
|
|
repo := newFakeRepo()
|
|
repo.seedUser(UserView{ID: "usr-root", Username: "root", Role: "owner"})
|
|
repo.seedUser(UserView{ID: "u2", Username: "alice", Role: "user"})
|
|
api := newTestAPI(repo, newFakeCluster())
|
|
api.External = staticExternal{p: owner}
|
|
eh := api.ExternalHandler()
|
|
|
|
caps := func() string {
|
|
t.Helper()
|
|
w := do(eh, "GET", "/api/v1/users/u2/quotas", "", nil)
|
|
if w.Code != http.StatusOK {
|
|
t.Fatalf("get quotas: code = %d body %s", w.Code, w.Body.String())
|
|
}
|
|
return strings.TrimSpace(w.Body.String())
|
|
}
|
|
put := func(body string, want int) {
|
|
t.Helper()
|
|
w := do(eh, "PUT", "/api/v1/users/u2/quotas", body, jsonHeader)
|
|
if w.Code != want {
|
|
t.Fatalf("PUT %s: code = %d body %s, want %d", body, w.Code, w.Body.String(), want)
|
|
}
|
|
}
|
|
|
|
put(`{"max_servers":0,"max_cpu_milli":2000,"max_memory_mb":4096,"max_storage_gb":20}`, http.StatusOK)
|
|
if got, want := caps(), `{"user_id":"u2","max_servers":0,"max_cpu_milli":2000,"max_memory_mb":4096,"max_storage_gb":20}`; got != want {
|
|
t.Fatalf("after setting every cap: %s, want %s", got, want)
|
|
}
|
|
// What the panel sends after the owner empties two boxes.
|
|
put(`{"max_servers":null,"max_cpu_milli":1000,"max_memory_mb":null,"max_storage_gb":20}`, http.StatusOK)
|
|
if got, want := caps(), `{"user_id":"u2","max_cpu_milli":1000,"max_storage_gb":20}`; got != want {
|
|
t.Fatalf("after emptying two boxes: %s, want %s", got, want)
|
|
}
|
|
put(`{}`, http.StatusOK)
|
|
if got, want := caps(), `{"user_id":"u2"}`; got != want {
|
|
t.Fatalf("after lifting every cap: %s, want %s", got, want)
|
|
}
|
|
|
|
put(`{"max_servers":3}`, http.StatusOK)
|
|
for _, body := range []string{
|
|
`{"max_servers":-1}`,
|
|
`{"max_servers":1,"max_storage_gb":-5}`,
|
|
`{"max_memory_mb":2147483648}`,
|
|
`{"max_cpu_milli":1.5}`,
|
|
} {
|
|
put(body, http.StatusBadRequest)
|
|
}
|
|
if got, want := caps(), `{"user_id":"u2","max_servers":3}`; got != want {
|
|
t.Fatalf("a refused write changed the caps: %s, want %s", got, want)
|
|
}
|
|
w := do(eh, "PUT", "/api/v1/users/u2/quotas", `{"max_storage_gb":-5}`, jsonHeader)
|
|
if !strings.Contains(w.Body.String(), `"invalid_quota"`) || !strings.Contains(w.Body.String(), "max_storage_gb") {
|
|
t.Fatalf("a negative cap should name the field: %s", w.Body.String())
|
|
}
|
|
put(`{"max_memory_mb":2147483647}`, http.StatusOK)
|
|
}
|