Added some tests for the SelectDateWidget #ep14boat

tests/unit/utils/test_widgets.py

@@ -0,0 +1,35 @@
+"""Tests for the utils/widgets.py file."""
+import datetime
+from flaskbb.utils.widgets import SelectDateWidget
+
+def test_select_date_widget():
+    """Test the SelectDateWidget."""
+    assert SelectDateWidget.FORMAT_CHOICES['%d'] == [(x, str(x)) for x in range(1, 32)]
+    assert SelectDateWidget.FORMAT_CHOICES['%m'] == [(x, str(x)) for x in range(1, 13)]
+
+    assert SelectDateWidget.FORMAT_CLASSES == {
+        '%d': 'select_date_day',
+        '%m': 'select_date_month',
+        '%Y': 'select_date_year'
+    }
+
+    select_date_widget = SelectDateWidget(years=[0, 1])
+
+    assert select_date_widget.FORMAT_CHOICES['%Y'] == [(0, '0'), (1, '1')]
+
+    class Field(object):
+        id = 'world'
+        name = 'helloWorld'
+        format = '%d %m %Y'
+        data = None
+
+    html = select_date_widget(field=Field())
+    assert 'world' in html
+    assert 'helloWorld' in html
+    assert 'class="select_date_day"' in html
+    assert 'class="select_date_month"' in html
+    assert 'class="select_date_year"' in html
+
+
+
+