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
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
|
"""helpers for bps unittests"""
#=========================================================
#imports
#=========================================================
#core
import os
import unittest
import logging; log = logging.getLogger(__name__)
#pkg
#local
__all__ = [
'TestCase',
'Param',
'enable_test',
]
#=========================================================
#helper for assertFunctionResults() method
#=========================================================
class Params(object):
"helper to represent params for function call"
@classmethod
def norm(cls, value):
if isinstance(value, cls):
return value
if isinstance(value, (list,tuple)):
return cls(*value)
return cls(**value)
def __init__(self, *args, **kwds):
self.args = args
self.kwds = kwds
def render(self, offset=0):
"""render parenthesized parameters"""
txt = ''
for a in self.args[offset:]:
txt += "%r, " % (a,)
kwds = self.kwds
for k in sorted(kwds):
txt += "%s=%r, " % (k, kwds[k])
if txt.endswith(", "):
txt = txt[:-2]
return txt
#=========================================================
#custom test base
#=========================================================
class TestCase(unittest.TestCase):
"""passlib-specific test case class
this class mainly overriddes many of the common assert methods
so to give a default message which includes the values
as well as the class-specific case_prefix string.
this latter bit makes the output of various test cases
easier to distinguish from eachother.
"""
case_prefix = None
def __init__(self, *a, **k):
#set the doc strings for all test messages to begin w/ case_prefix
#yes, this is incredibly hacked.
prefix = self.case_prefix
if prefix:
if callable(prefix):
prefix = prefix()
for attr in dir(self):
if not attr.startswith("test"):
continue
v = getattr(self, attr)
if not hasattr(v, "im_func"):
continue
d = v.im_func.__doc__ or v.im_func.__name__
idx = d.find(": ")
if idx > -1:
d = d[idx+1:]
v.im_func.__doc__ = d = "%s: %s" % (prefix, d.lstrip())
assert v.__doc__ == d
unittest.TestCase.__init__(self, *a, **k)
def assertEquals(self, real, correct, msg=None):
#NOTE: overriding this to get msg formatting capability
msg = self._format_msg(msg, "got %r, expected would equal %r", real, correct)
return self.assert_(real == correct, msg)
def assertEqual(self, *a, **k):
return self.assertEquals(*a, **k)
def assertNotEquals(self, real, correct, msg=None):
#NOTE: overriding this to get msg formatting capability
msg = self._format_msg(msg, "got %r, expected would equal %r", real, correct)
return self.assert_(real != correct, msg)
def assertNotEqual(self, *a, **k):
return self.assertNotEquals(*a, **k)
def assertIs(self, real, correct, msg=None):
msg = self._format_msg(msg, "got %r, expected would be %r", real, correct)
return self.assert_(real is correct, msg)
def assertIsNot(self, real, correct, msg=None):
msg = self._format_msg(msg, "expected would not be %r", real)
return self.assert_(real is not correct, msg)
def assertIsInstance(self, obj, klass, msg=None):
msg = self._format_msg(msg, "got %r, expected instance of %r", obj, klass)
return self.assert_(isinstance(obj, klass), msg)
def assertRaises(self, type, func, *args, **kwds):
msg = kwds.pop("__msg__", None)
err = None
try:
result = func(*args, **kwds)
except Exception, err:
pass
if err is None:
msg = self._format_msg(msg, "function returned %r, expected it to raise %r", result, type)
raise AssertionError(msg)
elif not isinstance(err, type):
msg = self._format_msg(msg, "function raised %r, expected %r", err, type)
raise AssertionError(msg)
def assertFunctionResults(self, func, cases):
"""helper for running through function calls.
func should be the function to call.
cases should be list of Param instances,
where first position argument is expected return value,
and remaining args and kwds are passed to function.
"""
for elem in cases:
elem = Params.norm(elem)
correct = elem.args[0]
result = func(*elem.args[1:], **elem.kwds)
self.assertEqual(result, correct,
"error for case %s: got %r, expected would equal %r" % (elem.render(1), result, correct)
)
def _format_msg(self, msg, template, *args, **kwds):
"helper for generating default message"
if msg and not msg.endswith(":"):
return msg
if args:
template %= args
if kwds:
template %= kwds
if msg:
return msg + " " + template
return template
#=========================================================
#helper funcs
#=========================================================
DEFAULT_TESTS = "backends"
tests = [
v.strip()
for v
in os.environ.get("PASSLIB_TESTS", DEFAULT_TESTS).lower().split(",")
]
def enable_test(*names):
"""check if a given test should be included based on the env var.
test flags:
all run ALL tests
backends test active backends
fallback-backends test inactive backends
slow required to enable really slow tests (eg builtin bcrypt backend)
"""
return 'all' in tests or any(name in tests for name in names)
#=========================================================
#EOF
#=========================================================
|
