root / snf-django-lib / snf_django / utils / testing.py @ f019f93d
History | View | Annotate | Download (10.4 kB)
1 |
# Copyright 2011-2013 GRNET S.A. All rights reserved.
|
---|---|
2 |
#
|
3 |
# Redistribution and use in source and binary forms, with or
|
4 |
# without modification, are permitted provided that the following
|
5 |
# conditions are met:
|
6 |
#
|
7 |
# 1. Redistributions of source code must retain the above
|
8 |
# copyright notice, this list of conditions and the following
|
9 |
# disclaimer.
|
10 |
#
|
11 |
# 2. Redistributions in binary form must reproduce the above
|
12 |
# copyright notice, this list of conditions and the following
|
13 |
# disclaimer in the documentation and/or other materials
|
14 |
# provided with the distribution.
|
15 |
#
|
16 |
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
|
17 |
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
18 |
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
19 |
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
|
20 |
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
21 |
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
22 |
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
|
23 |
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
|
24 |
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
25 |
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
|
26 |
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
27 |
# POSSIBILITY OF SUCH DAMAGE.
|
28 |
#
|
29 |
# The views and conclusions contained in the software and
|
30 |
# documentation are those of the authors and should not be
|
31 |
# interpreted as representing official policies, either expressed
|
32 |
# or implied, of GRNET S.A.
|
33 |
|
34 |
|
35 |
from contextlib import contextmanager |
36 |
from django.test import TestCase |
37 |
from django.utils import simplejson as json |
38 |
from synnefo.util import text |
39 |
from mock import patch |
40 |
|
41 |
|
42 |
@contextmanager
|
43 |
def override_settings(settings, **kwargs): |
44 |
"""
|
45 |
Helper context manager to override django settings within the provided
|
46 |
context.
|
47 |
|
48 |
All keyword arguments provided are set to the django settings object and
|
49 |
get reverted/removed when the manager exits.
|
50 |
|
51 |
>>> from synnefo.util.testing import override_settings
|
52 |
>>> from django.conf import settings
|
53 |
>>> with override_settings(settings, DEBUG=True):
|
54 |
... assert settings.DEBUG == True
|
55 |
|
56 |
The special arguemnt ``prefix`` can be set to prefix all setting keys with
|
57 |
the provided value.
|
58 |
|
59 |
>>> from django.conf import settings
|
60 |
>>> from django.core import mail
|
61 |
>>> with override_settings(settings, CONTACT_EMAILS=['kpap@grnet.gr'],
|
62 |
... prefix='MYAPP_'):
|
63 |
... from django.core.mail import send_mail
|
64 |
... send_mail("hello", "I love you kpap", settings.DEFAULT_FROM_EMAIL,
|
65 |
... settings.MYAPP_CONTACT_EMAILS)
|
66 |
... assert 'kpap@grnet.gr' in mail.mailbox[0].recipients()
|
67 |
|
68 |
If you plan to reuse it
|
69 |
|
70 |
>>> import functools
|
71 |
>>> from synnefo.util.testing import override_settings
|
72 |
>>> from django.conf import settings
|
73 |
>>> myapp_settings = functools.partial(override_settings, prefix='MYAPP_')
|
74 |
>>> with myapp_settings(CONTACT_EMAILS=['kpap@grnet.gr']):
|
75 |
... assert settings.MYAPP_CONTACT_EMAILS == ['kpap@grnet.gr']
|
76 |
|
77 |
"""
|
78 |
|
79 |
_prefix = kwargs.get('prefix', '') |
80 |
prefix = lambda key: '%s%s' % (_prefix, key) |
81 |
|
82 |
oldkeys = [k for k in dir(settings) if k.upper() == k] |
83 |
oldsettings = dict([(k, getattr(settings, k)) for k in oldkeys]) |
84 |
|
85 |
toremove = [] |
86 |
for key, value in kwargs.iteritems(): |
87 |
key = prefix(key) |
88 |
if not hasattr(settings, key): |
89 |
toremove.append(key) |
90 |
setattr(settings, key, value)
|
91 |
|
92 |
yield
|
93 |
|
94 |
# Remove keys that didn't exist
|
95 |
for key in toremove: |
96 |
delattr(settings, key)
|
97 |
|
98 |
# Remove keys that added during the execution of the context
|
99 |
if kwargs.get('reset_changes', True): |
100 |
newkeys = [k for k in dir(settings) if k.upper() == k] |
101 |
for key in newkeys: |
102 |
if not key in oldkeys: |
103 |
delattr(settings, key)
|
104 |
|
105 |
# Revert old keys
|
106 |
for key in oldkeys: |
107 |
if key == key.upper():
|
108 |
setattr(settings, key, oldsettings.get(key))
|
109 |
|
110 |
|
111 |
def with_settings(settings, prefix='', **override): |
112 |
def wrapper(func): |
113 |
def inner(*args, **kwargs): |
114 |
with override_settings(settings, prefix=prefix, **override):
|
115 |
ret = func(*args, **kwargs) |
116 |
return ret
|
117 |
return inner
|
118 |
return wrapper
|
119 |
|
120 |
serial = 0
|
121 |
|
122 |
|
123 |
@contextmanager
|
124 |
def astakos_user(user): |
125 |
"""
|
126 |
Context manager to mock astakos response.
|
127 |
|
128 |
usage:
|
129 |
with astakos_user("user@user.com"):
|
130 |
.... make api calls ....
|
131 |
|
132 |
"""
|
133 |
with patch("snf_django.lib.api.get_token") as get_token: |
134 |
get_token.return_value = "DummyToken"
|
135 |
with patch('astakosclient.AstakosClient.authenticate') as m2: |
136 |
m2.return_value = {"access": {
|
137 |
"token": {
|
138 |
"expires": "2013-06-19T15:23:59.975572+00:00", |
139 |
"id": "DummyToken", |
140 |
"tenant": {
|
141 |
"id": text.udec(user, 'utf8'), |
142 |
"name": "Firstname Lastname" |
143 |
} |
144 |
}, |
145 |
"serviceCatalog": [],
|
146 |
"user": {
|
147 |
"roles_links": [],
|
148 |
"id": text.udec(user, 'utf8'), |
149 |
"roles": [{"id": 1, "name": "default"}], |
150 |
"name": "Firstname Lastname"}} |
151 |
} |
152 |
|
153 |
with patch('astakosclient.AstakosClient.get_quotas') as m3: |
154 |
m3.return_value = { |
155 |
"system": {
|
156 |
"pithos.diskspace": {
|
157 |
"usage": 0, |
158 |
"limit": 1073741824, # 1GB |
159 |
"pending": 0 |
160 |
} |
161 |
} |
162 |
} |
163 |
issue_fun = \ |
164 |
"astakosclient.AstakosClient.issue_one_commission"
|
165 |
with patch(issue_fun) as m3: |
166 |
serials = [] |
167 |
append = serials.append |
168 |
|
169 |
def get_serial(*args, **kwargs): |
170 |
global serial
|
171 |
serial += 1
|
172 |
append(serial) |
173 |
return serial
|
174 |
|
175 |
m3.side_effect = get_serial |
176 |
resolv_fun = \ |
177 |
'astakosclient.AstakosClient.resolve_commissions'
|
178 |
with patch(resolv_fun) as m4: |
179 |
m4.return_value = {'accepted': serials,
|
180 |
'rejected': [],
|
181 |
'failed': []}
|
182 |
users_fun = \ |
183 |
'astakosclient.AstakosClient.get_usernames'
|
184 |
with patch(users_fun) as m5: |
185 |
|
186 |
def get_usernames(*args, **kwargs): |
187 |
uuids = args[-1]
|
188 |
return dict((uuid, uuid) for uuid in uuids) |
189 |
|
190 |
m5.side_effect = get_usernames |
191 |
yield
|
192 |
|
193 |
|
194 |
serial = 0
|
195 |
|
196 |
|
197 |
@contextmanager
|
198 |
def mocked_quotaholder(success=True): |
199 |
with patch("synnefo.quotas.Quotaholder.get") as astakos: |
200 |
global serial
|
201 |
serial += 10
|
202 |
|
203 |
def foo(*args, **kwargs): |
204 |
return (len(astakos.return_value.issue_one_commission.mock_calls) + |
205 |
serial) |
206 |
astakos.return_value.issue_one_commission.side_effect = foo |
207 |
astakos.return_value.resolve_commissions.return_value = {"failed": []}
|
208 |
yield astakos.return_value
|
209 |
|
210 |
|
211 |
class BaseAPITest(TestCase): |
212 |
def get(self, url, user='user', *args, **kwargs): |
213 |
with astakos_user(user):
|
214 |
with mocked_quotaholder():
|
215 |
response = self.client.get(url, *args, **kwargs)
|
216 |
return response
|
217 |
|
218 |
def delete(self, url, user='user'): |
219 |
with astakos_user(user):
|
220 |
with mocked_quotaholder():
|
221 |
response = self.client.delete(url)
|
222 |
return response
|
223 |
|
224 |
def post(self, url, user='user', params={}, ctype='json', *args, **kwargs): |
225 |
if ctype == 'json': |
226 |
content_type = 'application/json'
|
227 |
with astakos_user(user):
|
228 |
with mocked_quotaholder():
|
229 |
response = self.client.post(url, params,
|
230 |
content_type=content_type, |
231 |
*args, **kwargs) |
232 |
return response
|
233 |
|
234 |
def put(self, url, user='user', params={}, ctype='json', *args, **kwargs): |
235 |
if ctype == 'json': |
236 |
content_type = 'application/json'
|
237 |
with astakos_user(user):
|
238 |
with mocked_quotaholder():
|
239 |
response = self.client.put(url, params,
|
240 |
content_type=content_type, |
241 |
*args, **kwargs) |
242 |
return response
|
243 |
|
244 |
def assertSuccess(self, response): |
245 |
self.assertTrue(response.status_code in [200, 202, 203, 204]) |
246 |
|
247 |
def assertFault(self, response, status_code, name): |
248 |
self.assertEqual(response.status_code, status_code)
|
249 |
fault = json.loads(response.content) |
250 |
self.assertEqual(fault.keys(), [name])
|
251 |
|
252 |
def assertBadRequest(self, response): |
253 |
self.assertFault(response, 400, 'badRequest') |
254 |
|
255 |
def assertItemNotFound(self, response): |
256 |
self.assertFault(response, 404, 'itemNotFound') |
257 |
|
258 |
def assertMethodNotAllowed(self, response): |
259 |
self.assertFault(response, 400, 'badRequest') |
260 |
try:
|
261 |
error = json.loads(response.content) |
262 |
except ValueError: |
263 |
self.assertTrue(False) |
264 |
self.assertEqual(error['badRequest']['message'], 'Method not allowed') |
265 |
|
266 |
|
267 |
# Imitate unittest assertions new in Python 2.7
|
268 |
|
269 |
class _AssertRaisesContext(object): |
270 |
"""
|
271 |
A context manager used to implement TestCase.assertRaises* methods.
|
272 |
Adapted from unittest2.
|
273 |
"""
|
274 |
|
275 |
def __init__(self, expected): |
276 |
self.expected = expected
|
277 |
|
278 |
def __enter__(self): |
279 |
return self |
280 |
|
281 |
def __exit__(self, exc_type, exc_value, tb): |
282 |
if exc_type is None: |
283 |
try:
|
284 |
exc_name = self.expected.__name__
|
285 |
except AttributeError: |
286 |
exc_name = str(self.expected) |
287 |
raise AssertionError( |
288 |
"%s not raised" % (exc_name,))
|
289 |
if not issubclass(exc_type, self.expected): |
290 |
# let unexpected exceptions pass through
|
291 |
return False |
292 |
self.exception = exc_value # store for later retrieval |
293 |
return True |
294 |
|
295 |
|
296 |
def assertRaises(excClass): |
297 |
return _AssertRaisesContext(excClass)
|
298 |
|
299 |
|
300 |
def assertGreater(x, y): |
301 |
assert x > y
|
302 |
|
303 |
|
304 |
def assertIn(x, y): |
305 |
assert x in y |