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
|
from wtforms import Form
from wtforms.validators import DataRequired
from wtforms_test import FormTestCase
class MultiDict(dict):
def getlist(self, key):
return [self[key]]
class SimpleFieldTestCase(FormTestCase):
field_class = None
def init_form(self, **kwargs):
class TestForm(Form):
test_field = self.field_class(**kwargs)
self.form_class = TestForm
return self.form_class
class FieldTestCase(SimpleFieldTestCase):
def test_assigns_required_from_validator(self):
form_class = self.init_form(validators=[DataRequired()])
form = form_class()
assert ('<input id="test_field"' ' name="test_field" required') in str(
form.test_field
)
def test_renders_autofocus(self):
form_class = self.init_form(
widget=self.field_class.widget.__class__(autofocus=True)
)
form = form_class()
assert "autofocus" in str(form.test_field)
def test_renders_required(self):
form_class = self.init_form(
widget=self.field_class.widget.__class__(required=True)
)
form = form_class()
assert "required" in str(form.test_field)
def test_renders_disabled(self):
form_class = self.init_form(
widget=self.field_class.widget.__class__(disabled=True)
)
form = form_class()
assert "disabled" in str(form.test_field)
def test_renders_readonly(self):
form_class = self.init_form(
widget=self.field_class.widget.__class__(readonly=True)
)
form = form_class()
assert "readonly" in str(form.test_field)
|