test/perm-test1.pl

   1 #!/usr/bin/perl -w
   2
   3 use strict;
   4 use PVE::Tools;
   5 use PVE::AccessControl;
   6 use PVE::RPCEnvironment;
   7 use Getopt::Long;
   8
   9 my $rpcenv = PVE::RPCEnvironment->init('cli');
  10
  11 my $cfgfn = "test1.cfg";
  12 $rpcenv->init_request(userconfig => $cfgfn);
  13
  14 sub check_roles {
  15     my ($user, $path, $expected_result) = @_;
  16
  17     my $roles = PVE::AccessControl::roles($rpcenv->{user_cfg}, $user, $path);
  18     my $res = join(',', sort keys %$roles);
  19
  20     die "unexpected result\nneed '${expected_result}'\ngot '$res'\n"
  21         if $res ne $expected_result;
  22
  23     print "ROLES:$path:$user:$res\n";
  24 }
  25
  26 sub check_permission {
  27     my ($user, $path, $expected_result) = @_;
  28
  29     my $perm = $rpcenv->permissions($user, $path);
  30     my $res = join(',', sort keys %$perm);
  31
  32     die "unexpected result\nneed '${expected_result}'\ngot '$res'\n"
  33         if $res ne $expected_result;
  34
  35     $perm = $rpcenv->permissions($user, $path);
  36     $res = join(',', sort keys %$perm);
  37     die "unexpected result (compiled)\nneed '${expected_result}'\ngot '$res'\n"
  38         if $res ne $expected_result;
  39
  40     print "PERM:$path:$user:$res\n";
  41 }
  42
  43 check_roles('max@pve', '/', '');
  44 check_roles('max@pve', '/vms', 'vm_admin');
  45
  46 #user permissions overrides group permissions
  47 check_roles('max@pve', '/vms/100', 'customer');
  48 check_roles('max@pve', '/vms/101', 'vm_admin');
  49
  50 check_permission('max@pve', '/', '');
  51 check_permission('max@pve', '/vms', 'Permissions.Modify,VM.Allocate,VM.Audit,VM.Console');
  52 check_permission('max@pve', '/vms/100', 'VM.Audit,VM.PowerMgmt');
  53
  54 check_permission('alex@pve', '/vms', '');
  55 check_permission('alex@pve', '/vms/100', 'VM.Audit,VM.PowerMgmt');
  56
  57
  58 check_roles('max@pve', '/vms/200', 'storage_manager');
  59 check_roles('joe@pve', '/vms/200', 'vm_admin');
  60 check_roles('sue@pve', '/vms/200', 'NoAccess');
  61
  62 print "all tests passed\n";
  63
  64 exit (0);