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
|
from contextlib import contextmanager
from wtforms.validators import ValidationError, StopValidation
class DummyTranslations(object):
def gettext(self, string):
return string
def ngettext(self, singular, plural, n):
if n == 1:
return singular
return plural
class DummyField(object):
_translations = DummyTranslations()
def __init__(self, data, errors=(), raw_data=None):
self.data = data
self.errors = list(errors)
self.raw_data = raw_data
def gettext(self, string):
return self._translations.gettext(string)
def ngettext(self, singular, plural, n):
return self._translations.ngettext(singular, plural, n)
def grab_error_message(callable, form, field):
try:
callable(form, field)
except ValidationError as e:
return e.args[0]
def grab_stop_message(callable, form, field):
try:
callable(form, field)
except StopValidation as e:
return e.args[0]
def contains_validator(field, v_type):
for v in field.validators:
if isinstance(v, v_type):
return True
return False
class DummyPostData(dict):
def getlist(self, key):
v = self[key]
if not isinstance(v, (list, tuple)):
v = [v]
return v
@contextmanager
def assert_raises_text(e_type, text):
import re
try:
yield
except e_type as e:
if not re.match(text, e.args[0]):
raise AssertionError('Exception raised: %r but text %r did not match pattern %r' % (e, e.args[0], text))
else:
raise AssertionError('Expected Exception %r, did not get it' % (e_type, ))
|
