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
|
import collections
from wtforms import Field
from wtforms.validators import (
DataRequired, Length, Email, Optional, NumberRange
)
class FormTestCase(object):
form_class = None
def _make_form(self, *args, **kwargs):
return self.form_class(csrf_enabled=False, *args, **kwargs)
def _get_field(self, field_name):
form = self._make_form()
return getattr(form, field_name)
def _get_validator(self, field, validator_class):
for validator in field.validators:
if isinstance(validator, validator_class):
return validator
def get_validator(self, field_name, validator_class):
return self._get_validator(
self._get_field(field_name),
validator_class
)
def has_field(self, field_name):
form = self._make_form()
return hasattr(form, field_name)
def assert_type(self, field_name, field_type):
self.assert_has(field_name)
assert self._get_field(field_name).__class__ is field_type
def assert_has(self, field_name):
try:
field = self._get_field(field_name)
except AttributeError:
field = None
msg = "Form does not have a field called '%s'." % field_name
assert isinstance(field, Field), msg
def assert_min(self, field_name, min_value):
field = self._get_field(field_name)
found = False
for validator in field.validators:
# we might have multiple NumberRange validators
if isinstance(validator, NumberRange):
if validator.min == min_value:
found = True
assert found, "Field does not have min value of %d" % min_value
def assert_max(self, field_name, max_value):
field = self._get_field(field_name)
found = False
for validator in field.validators:
# we might have multiple NumberRange validators
if isinstance(validator, NumberRange):
if validator.max == max_value:
found = True
assert found, "Field does not have max value of %d" % max_value
def assert_min_length(self, field_name, min_length):
field = self._get_field(field_name)
found = False
for validator in field.validators:
# we might have multiple Length validators
if isinstance(validator, Length):
if validator.min == min_length:
found = True
assert found, "Field does not have min length of %d" % min_length
def assert_max_length(self, field_name, max_length):
field = self._get_field(field_name)
found = False
for validator in field.validators:
# we might have multiple Length validators
if isinstance(validator, Length):
if validator.max == max_length:
found = True
assert found, "Field does not have max length of %d" % max_length
def assert_description(self, field_name, description):
field = self._get_field(field_name)
assert field.description == description
def assert_default(self, field_name, default):
field = self._get_field(field_name)
assert field.default == default
def assert_label(self, field_name, label):
field = self._get_field(field_name)
assert field.label.text == label
def assert_has_validator(self, field_name, validator):
field = self._get_field(field_name)
msg = "Field '%s' does not have validator %r." % (
field_name, validator
)
assert self._get_validator(field, validator), msg
def assert_not_optional(self, field_name):
field = self._get_field(field_name)
msg = "Field '%s' is optional." % field_name
assert not self._get_validator(field, DataRequired), msg
def assert_optional(self, field_name):
field = self._get_field(field_name)
msg = "Field '%s' is not optional." % field_name
assert self._get_validator(field, Optional), msg
def assert_choices(self, field_name, choices):
field = self._get_field(field_name)
assert field.choices == choices
def assert_choice_values(self, field_name, choices):
compare = lambda x, y: collections.Counter(x) == collections.Counter(y)
field = self._get_field(field_name)
assert compare(field.choices, choices)
def assert_not_required(self, field_name):
field = self._get_field(field_name)
msg = "Field '%s' is required." % field_name
assert not self._get_validator(field, DataRequired), msg
def assert_required(self, field_name):
field = self._get_field(field_name)
msg = "Field '%s' is not required." % field_name
assert self._get_validator(field, DataRequired), msg
def assert_email(self, field_name):
field = self._get_field(field_name)
msg = (
"Field '%s' is not required to be a valid email address." %
field_name
)
assert self._get_validator(field, Email), msg
|
