sh4nks 10 лет назад
Родитель
Сommit
38fb571dd9
1 измененных файлов с 21 добавлено и 0 удалено
  1. 21 0
      tests/unit/utils/test_fields.py

+ 21 - 0
tests/unit/utils/test_fields.py

@@ -0,0 +1,21 @@
+"""Tests for the utils/fields.py file."""
+import pytest
+from wtforms.form import Form
+from flaskbb.utils.fields import BirthdayField
+
+
+def test_birthday_field():
+    class F(Form):
+        birthday = BirthdayField(format='%d %m %Y')
+
+    a = ["04 02 2015"]
+    b = ["None", "None", "2015"]  # this one should fail
+    c = ["None", "None", "None"]
+
+    form = F()
+
+    assert form.birthday.process_formdata(a) is None
+    assert form.birthday.process_formdata(c) is None
+
+    with pytest.raises(ValueError):
+        form.birthday.process_formdata(b)