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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
##############################################################################
#
# Copyright (c) 2001, 2002 Zope Foundation and Contributors.
# All Rights Reserved.
#
# This software is subject to the provisions of the Zope Public License,
# Version 2.1 (ZPL). A copy of the ZPL should accompany this distribution.
# THIS SOFTWARE IS PROVIDED "AS IS" AND ANY AND ALL EXPRESS OR IMPLIED
# WARRANTIES ARE DISCLAIMED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF TITLE, MERCHANTABILITY, AGAINST INFRINGEMENT, AND FITNESS
# FOR A PARTICULAR PURPOSE.
#
##############################################################################
"""Test handler for 'protectClass' directive
"""
import unittest
from zope.security.interfaces import PUBLIC_PERMISSION_NAME as zope_Public
class Test_protectName(unittest.TestCase):
def setUp(self):
from zope.security.checker import _clear
_clear()
def tearDown(self):
from zope.security.checker import _clear
_clear()
def _callFUT(self, class_, name, permission):
from zope.security.protectclass import protectName
return protectName(class_, name, permission)
def test_wo_existing_checker_w_zope_Public(self):
from zope.security.checker import CheckerPublic
from zope.security.checker import _checkers
self._callFUT(Foo, 'bar', zope_Public)
self.assertTrue(_checkers[Foo].get_permissions['bar'] is CheckerPublic)
def test_w_existing_checker(self):
from zope.security.checker import Checker
from zope.security.checker import _checkers
checker = _checkers[Foo] = Checker({})
permission = object()
self._callFUT(Foo, 'bar', permission)
self.assertTrue(_checkers[Foo] is checker)
self.assertTrue(checker.get_permissions['bar'] is permission)
class Test_protectSetAttribute(unittest.TestCase):
def setUp(self):
from zope.security.checker import _clear
_clear()
def tearDown(self):
from zope.security.checker import _clear
_clear()
def _callFUT(self, class_, name, permission):
from zope.security.protectclass import protectSetAttribute
return protectSetAttribute(class_, name, permission)
def test_wo_existing_checker_w_zope_Public(self):
from zope.security.checker import CheckerPublic
from zope.security.checker import _checkers
self._callFUT(Foo, 'bar', zope_Public)
self.assertTrue(_checkers[Foo].set_permissions['bar'] is CheckerPublic)
def test_w_existing_checker(self):
from zope.security.checker import Checker
from zope.security.checker import _checkers
checker = _checkers[Foo] = Checker({})
permission = object()
self._callFUT(Foo, 'bar', permission)
self.assertTrue(_checkers[Foo] is checker)
self.assertTrue(checker.set_permissions['bar'] is permission)
class Test_protectLikeUnto(unittest.TestCase):
def setUp(self):
from zope.security.checker import _clear
_clear()
def tearDown(self):
from zope.security.checker import _clear
_clear()
def _callFUT(self, class_, like_unto):
from zope.security.protectclass import protectLikeUnto
return protectLikeUnto(class_, like_unto)
def test_wo_existing_like_unto_checker(self):
from zope.security.checker import _checkers
self.assertFalse(Foo in _checkers)
self._callFUT(Bar, Foo)
self.assertFalse(Foo in _checkers)
self.assertFalse(Bar in _checkers)
def test_w_existing_like_unto_checker_wo_existing_checker(self):
from zope.security.checker import Checker
from zope.security.checker import CheckerPublic
from zope.security.checker import defineChecker
from zope.security.checker import _checkers
permission = object()
foo_checker = Checker({'bar': CheckerPublic}, {'bar': permission})
defineChecker(Foo, foo_checker)
self._callFUT(Bar, Foo)
bar_checker = _checkers[Bar]
self.assertEqual(bar_checker.get_permissions,
foo_checker.get_permissions)
self.assertEqual(bar_checker.set_permissions,
foo_checker.set_permissions)
def test_w_existing_like_unto_checker_w_existing_checker(self):
from zope.security.checker import Checker
from zope.security.checker import CheckerPublic
from zope.security.checker import defineChecker
from zope.security.checker import _checkers
permission1, permission2 = object(), object()
foo_checker = Checker({'bar': CheckerPublic}, {'bar': permission2})
defineChecker(Foo, foo_checker)
bar_checker = Checker({'bar': permission1, 'baz': CheckerPublic}, {})
defineChecker(Bar, bar_checker)
self._callFUT(Bar, Foo)
bar_checker = _checkers[Bar]
self.assertEqual(bar_checker.get_permissions,
{'bar': CheckerPublic, 'baz': CheckerPublic})
self.assertEqual(bar_checker.set_permissions,
foo_checker.set_permissions)
class Foo(object):
bar = 'Bar'
class Bar(Foo):
baz = 'Baz'
def test_suite():
return unittest.TestSuite((
unittest.defaultTestLoader.loadTestsFromTestCase(Test_protectName),
unittest.defaultTestLoader.loadTestsFromTestCase(Test_protectSetAttribute),
unittest.defaultTestLoader.loadTestsFromTestCase(Test_protectLikeUnto),
))
|