59 lines
1.8 KiB
Go
59 lines
1.8 KiB
Go
//go:build pgint
|
|
|
|
package pgint
|
|
|
|
import (
|
|
"context"
|
|
"net/url"
|
|
"os"
|
|
"testing"
|
|
|
|
"felis.lolicon.best/internal/api"
|
|
"felis.lolicon.best/internal/store"
|
|
)
|
|
|
|
// ---- user admin ----------------------------------------------------------------
|
|
|
|
// A user's detail whose linked accounts cannot be read is an error. It used to come
|
|
// back with an empty list, which tells the admin the user has linked nothing.
|
|
func TestUserDetailLinkReadFailure(t *testing.T) {
|
|
ctx := context.Background()
|
|
u := newUser(t, "user", "udl")
|
|
mc := testUUID(t)
|
|
mustExec(t, `INSERT INTO account_links (user_id, mc_uuid) VALUES ($1, $2)`, u.ID, mc)
|
|
|
|
// A role that reads users and servers and nothing else.
|
|
const role = "pgint_nolinks"
|
|
drop := func() {
|
|
for _, stmt := range []string{"DROP OWNED BY " + role, "DROP ROLE " + role} {
|
|
if _, err := db.ExecContext(ctx, stmt); err != nil {
|
|
t.Errorf("%s: %v", stmt, err)
|
|
}
|
|
}
|
|
}
|
|
mustExec(t, "CREATE ROLE "+role+" LOGIN")
|
|
t.Cleanup(drop)
|
|
mustExec(t, "GRANT USAGE ON SCHEMA public TO "+role)
|
|
mustExec(t, "GRANT SELECT ON users, servers TO "+role)
|
|
dsn, err := url.Parse(os.Getenv("FELIS_TEST_PG_URL"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
dsn.User = url.User(role)
|
|
drv, err := store.Open(ctx, dsn.String())
|
|
if err != nil {
|
|
t.Fatalf("open as %s: %v", role, err)
|
|
}
|
|
defer drv.Close()
|
|
limited := api.NewPGRepo(drv.DB())
|
|
|
|
if d, err := limited.UserDetail(ctx, u.ID); err == nil {
|
|
t.Fatalf("UserDetail with account_links unreadable = %+v, nil; want its error", d)
|
|
}
|
|
mustExec(t, "GRANT SELECT ON account_links TO "+role)
|
|
d, err := limited.UserDetail(ctx, u.ID)
|
|
if err != nil || d.Username != u.Username || len(d.LinkedAccounts) != 1 || d.LinkedAccounts[0].MCUUID != mc {
|
|
t.Fatalf("UserDetail with account_links readable = %+v, %v; want %s with %s linked", d, err, u.Username, mc)
|
|
}
|
|
}
|