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
|
#!/usr/bin/env python
from unittest import TestCase
from wtforms.widgets import html_params, Input
from wtforms.widgets import *
class DummyField(object):
def __init__(self, data, name='f', label='', id='', type='TextField'):
self.data = data
self.name = name
self.label = label
self.id = id
self.type = type
_value = lambda x: x.data
__unicode__ = lambda x: x.data
__str__ = lambda x: x.data
__call__ = lambda x, **k: x.data
__iter__ = lambda x: iter(x.data)
iter_choices = lambda x: iter(x.data)
class HTMLParamsTest(TestCase):
def test(self):
self.assertEqual(html_params(foo=9, k='wuuu'), u'foo="9" k="wuuu"')
self.assertEqual(html_params(class_='foo'), u'class="foo"')
self.assertEqual(html_params(class__='foo'), u'class_="foo"')
self.assertEqual(html_params(for_='foo'), u'for="foo"')
class ListWidgetTest(TestCase):
def test(self):
# ListWidget just expects an iterable of field-like objects as its
# 'field' so that is what we will give it
field = DummyField([DummyField(x, label='l' + x) for x in ['foo', 'bar']], id='hai')
self.assertEqual(ListWidget()(field), u'<ul id="hai"><li>lfoo: foo</li><li>lbar: bar</li></ul>')
w = ListWidget(html_tag='ol', prefix_label=False)
self.assertEqual(w(field), u'<ol id="hai"><li>foo lfoo</li><li>bar lbar</li></ol>')
class TableWidgetTest(TestCase):
def test(self):
inner_fields = [
DummyField('hidden1', type='HiddenField'),
DummyField('foo', label='lfoo'),
DummyField('bar', label='lbar'),
DummyField('hidden2', type='HiddenField'),
]
field = DummyField(inner_fields, id='hai')
self.assertEqual(TableWidget()(field), u'<table id="hai"><tr><th>lfoo</th><td>hidden1foo</td></tr><tr><th>lbar</th><td>bar</td></tr></table>hidden2')
class BasicWidgetsTest(TestCase):
"""Test most of the basic input widget types"""
field = DummyField('foo', name='bar', label='label', id='id')
def test_input_type(self):
a = Input()
self.assertRaises(AttributeError, getattr, a, 'input_type')
b = Input(input_type='test')
self.assertEqual(b.input_type, 'test')
def test_html_marking(self):
html = TextInput()(self.field)
self.assert_(hasattr(html, '__html__'))
self.assert_(html.__html__() is html)
def test_text_input(self):
self.assertEqual(TextInput()(self.field), u'<input id="id" name="bar" type="text" value="foo">')
def test_password_input(self):
self.assert_(u'type="password"' in PasswordInput()(self.field))
self.assert_(u'value=""' in PasswordInput()(self.field))
self.assert_(u'value="foo"' in PasswordInput(hide_value=False)(self.field))
def test_hidden_input(self):
self.assert_(u'type="hidden"' in HiddenInput()(self.field))
def test_checkbox_input(self):
self.assertEqual(CheckboxInput()(self.field, value='v'), '<input checked id="id" name="bar" type="checkbox" value="v">')
field2 = DummyField(False)
self.assert_(u'checked' not in CheckboxInput()(field2))
def test_radio_input(self):
pass # TODO
def test_textarea(self):
# Make sure textareas escape properly and render properly
f = DummyField('hi<>bye')
self.assertEqual(TextArea()(f), '<textarea id="" name="f">hi<>bye</textarea>')
class SelectTest(TestCase):
field = DummyField([('foo', 'lfoo', True), ('bar', 'lbar', False)])
def test(self):
self.assertEqual(Select()(self.field),
u'<select id="" name="f"><option selected value="foo">lfoo</option><option value="bar">lbar</option></select>')
self.assertEqual(Select(multiple=True)(self.field),
'<select id="" multiple name="f"><option selected value="foo">lfoo</option><option value="bar">lbar</option></select>')
if __name__ == '__main__':
from unittest import main
main()
|
