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
|
# vim:fileencoding=utf8:et:ts=4:sts=4:sw=4:ft=python
from django.conf import settings
from django.contrib.auth.models import AnonymousUser
from django.contrib.messages.middleware import MessageMiddleware
from django.contrib.sessions.backends.cache import SessionStore
from django.contrib.sessions.middleware import SessionMiddleware
from django.db import DatabaseError
from django.test import TestCase, RequestFactory
from django.utils.functional import curry
from okupy.tests import vars
import mock
"""
Decorator that restricts database access
http://codeinthehole.com/writing/disable-database-access-when-writing-unit-
tests-in-django/
"""
no_database = curry(
mock.patch, 'django.db.backends.util.CursorWrapper',
mock.Mock(side_effect=DatabaseError))
def ldap_users(username=None, all=False, clean=False,
directory=vars.DIRECTORY):
"""
Retrieve either a single LDAP user from the fake LDAP directory,
or all the users, or clean up the users from the directory
"""
if (username and all) or (username and clean) or (all and clean):
raise TypeError('Please specify one of username, all or clean')
if username:
dn = settings.AUTH_LDAP_USER_DN_TEMPLATE % {'user': username}
result = (dn, directory[dn])
elif all:
result = []
for dn, attrs in directory.items():
if dn.endswith(settings.AUTH_LDAP_USER_BASE_DN) and \
dn is not settings.AUTH_LDAP_USER_BASE_DN:
result.append((dn, attrs))
elif clean:
result = directory.copy()
for dn in directory.keys():
if dn.endswith(settings.AUTH_LDAP_USER_BASE_DN) and \
dn is not settings.AUTH_LDAP_USER_BASE_DN:
del result[dn]
return result
def set_search_seed(value=None, attr='uid', neg=False):
""" Create the filterstr of the search_s seed part of the mocked
ldap object """
search_item = '(&'
for item in settings.AUTH_LDAP_USER_OBJECTCLASS:
search_item += '(objectClass=%s)' % item
if value:
if neg:
search_item += '(!(%s=%s))' % (attr, value)
else:
search_item += '(%s=%s)' % (attr, value)
return search_item + ')'
def set_request(uri, post=False, user=False, messages=False):
""" Sets a request with RequestFactory """
if post:
if type(post) == bool:
post = {}
request = RequestFactory().post(uri, post)
else:
request = RequestFactory().get(uri)
if user:
request.user = user
request.user.is_verified = lambda: True
else:
request.user = AnonymousUser()
request.session = SessionStore()
if messages:
SessionMiddleware().process_request(request)
MessageMiddleware().process_request(request)
return request
class OkupyTestCase(TestCase):
"""
Custon TestCase class, implements additional assert functions
http://stackoverflow.com/questions/2897609/how-can-i-unit-test-django-
messages
"""
def _get_matches(self, response, text):
""" Get messages that match the given text """
messages = self._get_messages(response)
if messages:
matches = [m for m in messages if text == m.message]
return matches
else:
self.fail('No messages found')
def _get_messages(self, response):
""" Get all messages from the context or the CookieStorage """
try:
messages = response.context['messages']
except (TypeError, KeyError):
return
return messages
def assertMessageCount(self, response, expect_num):
"""
Asserts that exactly the given number of messages have been sent.
"""
messages = self._get_messages(response)
if messages:
actual_num = len(messages)
else:
actual_num = 0
if actual_num != expect_num:
self.fail('Message count was %d, expected %d' %
(actual_num, expect_num))
def assertMessage(self, response, text, level=None):
"""
Asserts that there is exactly one message containing the given text.
"""
matches = self._get_matches(response, text)
if len(matches) == 1:
msg = matches[0]
if level is not None and msg.level != level:
self.fail('There was one matching message but with different '
'level: %s != %s' % (msg.level, level))
elif len(matches) == 0:
messages_str = ", ".join(
'"%s"' % m for m in self._get_messages(response))
self.fail('No message contained text "%s", messages were: %s' %
(text, messages_str))
else:
self.fail('Multiple messages contained text "%s": %s' %
(text, ", ".join(('"%s"' % m) for m in matches)))
def assertNotMessage(self, response, text):
""" Assert that no message contains the given text. """
matches = self._get_matches(response, text)
if len(matches) > 0:
self.fail('Message(s) contained text "%s": %s' %
(text, ", ".join(('"%s"' % m) for m in matches)))
|
GitWeb