1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
#!/usr/bin/env perl
use strict;
use warnings;
# Disable IPv6, epoll and kqueue
BEGIN { $ENV{MOJO_NO_IPV6} = $ENV{MOJO_POLL} = 1 }
use Test::More;
plan tests => 42;
# testing code starts here
use Mojolicious::Lite;
use Test::Mojo;
my %roles = (role1=>{priv1=>1},
role2=>{priv1=>1,priv2=>1});
plugin 'authorization', {
has_priv => sub {
my $self = shift;
my ($priv, $extradata) = @_;
return 0
unless($self->session('role'));
my $role = $self->session('role');
my $privs = $roles{$role};
return 1
if exists($privs->{$priv});
return 0;
},
is_role => sub {
my $self = shift;
my ($role, $extradata) = @_;
return 0
unless($self->session('role'));
return 1
if ($self->session('role') eq $role);
return 0;
},
user_privs => sub {
my $self = shift;
my ($extradata) = @_;
return []
unless($self->session('role'));
my $role = $self->session('role');
my $privs = $roles{$role};
return sort keys(%{$privs});
},
user_role => sub {
my $self = shift;
my ($extradata) = @_;
return $self->session('role');
},
};
get '/' => sub {
my $self = shift;
$self->session('role'=>'role1');
$self->render(text => 'index page');
};
get '/priv1' => sub {
my $self = shift;
$self->render(text=> $self->has_priv('priv1') ? 'Priv 1' : 'fail');
};
get '/priv2' => sub {
my $self = shift;
$self->render(text=> $self->has_priv('priv2') ? 'Priv 2' : 'fail');
};
get '/privilege1' => sub {
my $self = shift;
$self->render(text=> $self->has_privilege('priv1') ? 'Priv 1' : 'fail');
};
get '/privilege2' => sub {
my $self = shift;
$self->render(text=> $self->has_privilege('priv2') ? 'Priv 2' : 'fail');
};
get '/role1' => sub {
my $self = shift;
$self->render(text=> $self->role('role1') ? 'Role 1' : 'fail');
};
get '/role2' => sub {
my $self = shift;
$self->render(text=> $self->role('role2') ? 'Role 2' : 'fail');
};
get '/change/:role' => sub {
my $self = shift;
my $role = $self->param('role');
$self->session('role'=>$role);
my $new_role = $self->role;
$self->render(text=>$new_role);
};
get '/myrole' => sub {
my $self = shift;
my $new_role = $self->role;
$self->render(text=>$new_role);
};
get '/myprivs' => sub {
my $self = shift;
my @privs = $self->privileges();
my $priv = join(':',@privs);
$self->render(text=>$priv);
};
my $t = Test::Mojo->new;
$t->get_ok('/')->status_is(200)->content_is('index page');
$t->get_ok('/priv1')->status_is(200)->content_is('Priv 1');
$t->get_ok('/priv2')->status_is(200)->content_is('fail');
$t->get_ok('/privilege1')->status_is(200)->content_is('Priv 1');
$t->get_ok('/privilege2')->status_is(200)->content_is('fail');
$t->get_ok('/myrole')->status_is(200)->content_is('role1');
$t->get_ok('/myprivs')->status_is(200)->content_is('priv1');
$t->get_ok('/change/role2')->status_is(200)->content_is('role2');
$t->get_ok('/priv1')->status_is(200)->content_is('Priv 1');
$t->get_ok('/priv2')->status_is(200)->content_is('Priv 2');
$t->get_ok('/privilege1')->status_is(200)->content_is('Priv 1');
$t->get_ok('/privilege2')->status_is(200)->content_is('Priv 2');
$t->get_ok('/myrole')->status_is(200)->content_is('role2');
$t->get_ok('/myprivs')->status_is(200)->content_is('priv1:priv2');
|
