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
|
from datetime import date
from wtforms import Form
from wtforms.fields import DateField
from wtforms_test import FormTestCase
from tests import MultiDict
from wtforms_components import DateRange
class TestDateRangeValidator(FormTestCase):
def init_form(self, **kwargs):
class ModelTestForm(Form):
date = DateField(validators=[DateRange(**kwargs)])
self.form_class = ModelTestForm
return self.form_class
def test_date_greater_than_validator(self):
form_class = self.init_form(min=date(1990, 1, 1))
form = form_class(MultiDict(date="1980-1-1"))
form.validate()
error_msg = "Date must be equal to or later than 1990-01-01."
assert form.errors["date"] == [error_msg]
def test_date_less_than_validator(self):
form_class = self.init_form(max=date(1990, 1, 1))
form = form_class(MultiDict(date="1991-1-1"))
form.validate()
error_msg = "Date must be equal to or earlier than 1990-01-01."
assert form.errors["date"] == [error_msg]
def test_date_between_validator(self):
form_class = self.init_form(min=date(1990, 1, 1), max=date(1991, 1, 1))
form = form_class(MultiDict(date="1989-1-1"))
form.validate()
error_msg = "Date must be between 1990-01-01 and 1991-01-01."
assert form.errors["date"] == [error_msg]
|
