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
|
import pytest
from tests.common import DummyPostData
from wtforms import validators
from wtforms.fields import FormField
from wtforms.fields import StringField
from wtforms.form import Form
class AttrDict:
def __init__(self, *args, **kw):
self.__dict__.update(*args, **kw)
class ClassWithProperty(AttrDict):
@property
def a(self):
return AttrDict(self.a_) if getattr(self, "a_", None) else AttrDict()
@a.setter
def a(self, value):
self.a_ = vars(value)
def make_form(name="F", **fields):
return type(str(name), (Form,), fields)
@pytest.fixture
def F1():
F = make_form(
a=StringField(validators=[validators.DataRequired()]), b=StringField()
)
return make_form("F1", a=FormField(F))
@pytest.fixture
def F2():
F = make_form(
a=StringField(validators=[validators.DataRequired()]), b=StringField()
)
return make_form("F2", a=FormField(F, separator="::"))
def test_formdata(F1):
form = F1(DummyPostData({"a-a": ["moo"]}))
assert form.a.form.a.name == "a-a"
assert form.a["a"].data == "moo"
assert form.a["b"].data is None
assert form.validate()
def test_iteration(F1):
assert [x.name for x in F1().a] == ["a-a", "a-b"]
def test_with_obj(F1):
obj = AttrDict(a=AttrDict(a="mmm"))
form = F1(obj=obj)
assert form.a.form.a.data == "mmm"
assert form.a.form.b.data is None
obj_inner = AttrDict(a=None, b="rawr")
obj2 = AttrDict(a=obj_inner)
form.populate_obj(obj2)
assert obj2.a is obj_inner
assert obj_inner.a == "mmm"
assert obj_inner.b is None
def test_widget(F1):
assert F1().a() == (
'<table id="a">'
'<tr><th><label for="a-a">A</label></th>'
'<td><input id="a-a" name="a-a" required type="text" value=""></td></tr>'
'<tr><th><label for="a-b">B</label></th>'
'<td><input id="a-b" name="a-b" type="text" value=""></td></tr>'
"</table>"
)
def test_separator(F2):
form = F2(DummyPostData({"a-a": "fake", "a::a": "real"}))
assert form.a.a.name == "a::a"
assert form.a.a.data == "real"
assert form.validate()
def test_no_validators_or_filters(F1):
class A(Form):
a = FormField(F1, validators=[validators.DataRequired()])
with pytest.raises(TypeError):
A()
class B(Form):
a = FormField(F1, filters=[str])
with pytest.raises(TypeError):
B()
class C(Form):
a = FormField(F1)
def validate_a(self, field):
pass
form = C()
with pytest.raises(TypeError):
form.validate()
def test_populate_missing_obj(F1):
obj = AttrDict(a=None)
obj2 = AttrDict(a=AttrDict(a="mmm"))
form = F1()
with pytest.raises(TypeError):
form.populate_obj(obj)
form.populate_obj(obj2)
def test_populate_property(F1):
obj1 = ClassWithProperty(a_={"a": "old_a", "b": "old_b"})
form = F1(DummyPostData({"a-a": "new_a", "a-b": "new_b"}))
form.populate_obj(obj1)
assert obj1.a_ == {"a": "new_a", "b": "new_b"}
obj2 = ClassWithProperty()
form.populate_obj(obj2)
assert obj1.a_ == {"a": "new_a", "b": "new_b"}
|
