-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathaccess_test.go
111 lines (98 loc) · 2.45 KB
/
access_test.go
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
111
package buzza
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestAccessMerge(t *testing.T) {
assert := assert.New(t)
assert.Equal(AccessAllowed, AccessUndefined.merge(AccessAllowed))
assert.Equal(AccessAllowed, AccessAllowed.merge(AccessAllowed))
assert.Equal(AccessAllowed, AccessAllowed.merge(AccessUndefined))
assert.Equal(AccessAllowed, AccessForbidden.merge(AccessAllowed))
assert.Equal(AccessUndefined, AccessUndefined.merge(AccessUndefined))
assert.Equal(AccessForbidden, AccessUndefined.merge(AccessForbidden))
assert.Equal(AccessForbidden, AccessForbidden.merge(AccessForbidden))
assert.Equal(AccessForbidden, AccessForbidden.merge(AccessUndefined))
assert.Equal(AccessForbidden, AccessAllowed.merge(AccessForbidden))
rolePermitted := Role{Permissions: map[PermissionName]bool{PermissionDownloadPro: true}}
roleUndefined := Role{Permissions: map[PermissionName]bool{}}
roleForbidden := Role{Permissions: map[PermissionName]bool{PermissionDownloadPro: false}}
allowedCases := []Roles{
{
rolePermitted,
},
{
roleUndefined,
rolePermitted,
rolePermitted,
},
{
rolePermitted,
roleUndefined,
rolePermitted,
},
}
for _, equalCase := range allowedCases {
assert.Equal(AccessAllowed, equalCase.Access(PermissionDownloadPro))
}
forbiddenCases := []Roles{
{
roleForbidden,
},
{
roleUndefined,
rolePermitted,
roleForbidden,
},
{
rolePermitted,
roleUndefined,
roleForbidden,
},
{
roleForbidden,
roleUndefined,
roleForbidden,
},
{
roleForbidden,
roleForbidden,
},
{
roleForbidden,
rolePermitted,
roleForbidden,
},
}
for i, equalCase := range forbiddenCases {
assert.Equal(AccessForbidden, equalCase.Access(PermissionDownloadPro), "access index: %d", i)
}
}
func TestMapRolesById(t *testing.T) {
assert := assert.New(t)
roleAdmin := Role{
Id: RoleIdAdmin,
Permissions: map[PermissionName]bool{
PermissionDownloadPro: true,
PermissionAdminDashboard: true,
},
}
rolePro := Role{
Id: RoleIdPro,
Permissions: map[PermissionName]bool{
PermissionDownloadPro: true,
},
}
rolesMapped := mapRolesById(roleAdmin, rolePro)
assert.Equal(roleAdmin, rolesMapped[RoleIdAdmin])
assert.Equal(rolePro, rolesMapped[RoleIdPro])
assert.Panics(func() {
mapRolesById(roleAdmin, roleAdmin)
})
assert.Panics(func() {
mapRolesById(roleAdmin, rolePro, roleAdmin)
})
assert.Panics(func() {
mapRolesById(rolePro, rolePro, roleAdmin)
})
}