-
Notifications
You must be signed in to change notification settings - Fork 16
/
test_formrender.py
164 lines (128 loc) · 5 KB
/
test_formrender.py
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
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
from doajtest.helpers import DoajTestCase
from portality.forms.article_forms import FormContext, Renderer
from wtforms import Form, StringField, SelectMultipleField, widgets
################################################################
# Form context for basic test
################################################################
class TestForm(Form):
__test__ = False # Prevent collection by PyTest
one = StringField("One")
two = StringField("Two")
class TestRenderer(Renderer):
__test__ = False # Prevent collection by PyTest
def __init__(self):
super(TestRenderer, self).__init__()
self.FIELD_GROUPS = {
"test" : [
{"one" : {"class" : "first"}},
{"two" : {"class" : "second"}}
]
}
class TestContext(FormContext):
__test__ = False # Prevent collection by PyTest
def data2form(self):
self.form = TestForm(formdata=self.form_data)
def source2form(self):
self.form = TestForm(data=self.source)
def blank_form(self):
self.form = TestForm()
def make_renderer(self):
self.renderer = TestRenderer()
def set_template(self):
self.template = "test_template.html"
################################################################
# Form context for extra fields test
################################################################
class ExtraFieldsForm(Form):
one = StringField("One")
two = StringField("Two")
extra = SelectMultipleField('Extra', choices = [("alpha", "alpha"), ("beta", "beta")],
option_widget=widgets.CheckboxInput(),
widget=widgets.ListWidget(prefix_label=False)
)
class ExtraFieldsRenderer(Renderer):
def __init__(self):
super(ExtraFieldsRenderer, self).__init__()
self.FIELD_GROUPS = {
"test" : [
{
"extra" : {
"extra_input_fields" : {
"alpha" : "one",
"beta" : "two"
}
}
}
]
}
class ExtraFieldsContext(FormContext):
def data2form(self):
self.form = ExtraFieldsForm(formdata=self.form_data)
def source2form(self):
self.form = ExtraFieldsForm(data=self.source)
def blank_form(self):
self.form = ExtraFieldsForm()
def make_renderer(self):
self.renderer = ExtraFieldsRenderer()
def set_template(self):
self.template = "test_template.html"
class TestRender(DoajTestCase):
def setUp(self): pass
def tearDown(self): pass
def test_01_base_render_init(self):
r = Renderer()
assert len(r.disabled_fields) == 0
assert len(r.error_fields) == 0
r.set_error_fields(["one", "two"])
r.set_disabled_fields(["three", "four"])
assert len(r.error_fields) == 2
assert "one" in r.error_fields
assert "two" in r.error_fields
assert len(r.disabled_fields) == 2
assert "three" in r.disabled_fields
assert "four" in r.disabled_fields
def test_02_subclass_render_field_group(self):
r = TestRenderer()
fc = TestContext()
assert "test" in r.FIELD_GROUPS
html = r.render_field_group(fc, "test")
assert 'name="one"' in html
assert 'name="two"' in html
assert 'class="first"' in html
assert 'class="second"' in html
assert "disabled" not in html
r.set_disabled_fields(["two"])
html = r.render_field_group(fc, "test")
assert "disabled" in html
def test_03_subclass_extra_fields(self):
r = ExtraFieldsRenderer()
fc = ExtraFieldsContext()
# first test the extra fields stuff directly
rewritten = r._rewrite_extra_fields(fc, r.FIELD_GROUPS.get("test")[0].get("extra"))
alpha = rewritten.get("extra_input_fields").get("alpha")
beta = rewritten.get("extra_input_fields").get("beta")
assert isinstance(alpha, StringField)
assert alpha.short_name == "one"
assert isinstance(beta, StringField)
assert beta.short_name == "two"
# now test it indirectly through the render function
html = r.render_field_group(fc, "test")
assert 'name="one"' in html
assert 'name="two"' in html
def test_05_insert_fields(self):
r = TestRenderer()
r.insert_field_after(
field_to_insert={"inserted_in_middle": {}},
after_this_field='one',
field_group='test'
)
r.insert_field_after(
field_to_insert={"inserted_last": {}},
after_this_field='two',
field_group='test'
)
assert len(r.FIELD_GROUPS['test']) == 4
assert list(r.FIELD_GROUPS['test'][0].keys())[0] == 'one'
assert list(r.FIELD_GROUPS['test'][1].keys())[0] == 'inserted_in_middle'
assert list(r.FIELD_GROUPS['test'][2].keys())[0] == 'two'
assert list(r.FIELD_GROUPS['test'][3].keys())[0] == 'inserted_last'