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
|
import re
import pytest
from wtforms.validators import regexp
from wtforms.validators import ValidationError
def grab_error_message(callable, form, field):
try:
callable(form, field)
except ValidationError as e:
return e.args[0]
@pytest.mark.parametrize(
"re_pattern, re_flags, test_v, expected_v",
[
("^a", None, "abcd", "a"),
("^a", re.I, "ABcd", "A"),
(re.compile("^a"), None, "abcd", "a"),
(re.compile("^a", re.I), None, "ABcd", "A"),
],
)
def test_regex_passes(
re_pattern, re_flags, test_v, expected_v, dummy_form, dummy_field
):
"""
Regex should pass if there is a match.
Should work for complie regex too
"""
validator = regexp(re_pattern, re_flags) if re_flags else regexp(re_pattern)
dummy_field.data = test_v
assert validator(dummy_form, dummy_field).group(0) == expected_v
@pytest.mark.parametrize(
"re_pattern, re_flags, test_v",
[
("^a", None, "ABC"),
("^a", re.I, "foo"),
("^a", None, None),
(re.compile("^a"), None, "foo"),
(re.compile("^a", re.I), None, None),
],
)
def test_regex_raises(re_pattern, re_flags, test_v, dummy_form, dummy_field):
"""
Regex should raise ValidationError if there is no match
Should work for complie regex too
"""
validator = regexp(re_pattern, re_flags) if re_flags else regexp(re_pattern)
dummy_field.data = test_v
with pytest.raises(ValidationError):
validator(dummy_form, dummy_field)
def test_regexp_message(dummy_form, dummy_field):
"""
Regexp validator should return given message
"""
validator = regexp("^a", message="foo")
dummy_field.data = "f"
assert grab_error_message(validator, dummy_form, dummy_field) == "foo"
|
