feat: add rbac policy matrix
This commit is contained in:
@@ -0,0 +1,58 @@
|
||||
package rbac
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestCan(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
role Role
|
||||
action Action
|
||||
want bool
|
||||
}{
|
||||
{name: "owner can manage project", role: RoleOwner, action: ActionProjectManage, want: true},
|
||||
{name: "owner can read audit", role: RoleOwner, action: ActionAuditRead, want: true},
|
||||
{name: "admin can delete vm", role: RoleAdmin, action: ActionVMDelete, want: true},
|
||||
{name: "admin can manage ssh keys", role: RoleAdmin, action: ActionSSHKeyManage, want: true},
|
||||
{name: "member can create vm", role: RoleMember, action: ActionVMCreate, want: true},
|
||||
{name: "member cannot delete vm", role: RoleMember, action: ActionVMDelete, want: false},
|
||||
{name: "member cannot read audit", role: RoleMember, action: ActionAuditRead, want: false},
|
||||
{name: "viewer can read vm", role: RoleViewer, action: ActionVMRead, want: true},
|
||||
{name: "viewer can open console", role: RoleViewer, action: ActionVMConsole, want: true},
|
||||
{name: "viewer cannot power vm", role: RoleViewer, action: ActionVMPower, want: false},
|
||||
{name: "viewer cannot create vm", role: RoleViewer, action: ActionVMCreate, want: false},
|
||||
{name: "tenant roles cannot manage cluster", role: RoleOwner, action: ActionClusterManage, want: false},
|
||||
{name: "unknown role denied", role: Role("unknown"), action: ActionVMRead, want: false},
|
||||
{name: "unknown action denied", role: RoleOwner, action: Action("unknown.action"), want: false},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
if got := Can(tt.role, tt.action); got != tt.want {
|
||||
t.Fatalf("Can(%q, %q) = %v, want %v", tt.role, tt.action, got, tt.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestPolicyMatrix(t *testing.T) {
|
||||
roles := []Role{RoleOwner, RoleAdmin, RoleMember, RoleViewer}
|
||||
actions := []Action{
|
||||
ActionVMRead,
|
||||
ActionVMPower,
|
||||
ActionVMCreate,
|
||||
ActionVMDelete,
|
||||
ActionVMConsole,
|
||||
ActionProjectRead,
|
||||
ActionProjectManage,
|
||||
ActionSSHKeyRead,
|
||||
ActionSSHKeyManage,
|
||||
ActionAuditRead,
|
||||
ActionClusterManage,
|
||||
}
|
||||
|
||||
for _, role := range roles {
|
||||
for _, action := range actions {
|
||||
_ = Can(role, action)
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user