59 lines
1.4 KiB
Python
59 lines
1.4 KiB
Python
import pytest
|
|
from wtforms import Form
|
|
import pendulum
|
|
from werkzeug.datastructures import ImmutableMultiDict
|
|
|
|
from atst.forms.fields import DateField, NewlineListField
|
|
|
|
|
|
class DateForm(Form):
|
|
date = DateField()
|
|
|
|
|
|
class NewlineListForm(Form):
|
|
newline_list = NewlineListField()
|
|
|
|
|
|
def test_date_ie_format():
|
|
form = DateForm(data={"date": "12/24/2018"})
|
|
assert form.date._value() == pendulum.date(2018, 12, 24)
|
|
|
|
|
|
def test_date_sane_format():
|
|
form = DateForm(data={"date": "2018-12-24"})
|
|
assert form.date._value() == pendulum.date(2018, 12, 24)
|
|
|
|
|
|
def test_date_insane_format():
|
|
form = DateForm(data={"date": "hello"})
|
|
with pytest.raises(ValueError):
|
|
form.date._value()
|
|
|
|
|
|
@pytest.mark.parametrize("input_,expected", [
|
|
("", []),
|
|
("hello", ["hello"]),
|
|
("hello\n", ["hello"]),
|
|
("hello\nworld", ["hello", "world"]),
|
|
("hello\nworld\n", ["hello", "world"])
|
|
])
|
|
def test_newline_list_process(input_, expected):
|
|
form_data = ImmutableMultiDict({"newline_list": input_})
|
|
form = NewlineListForm(form_data)
|
|
|
|
assert form.validate()
|
|
assert form.data == {"newline_list": expected}
|
|
|
|
|
|
@pytest.mark.parametrize("input_,expected", [
|
|
([], ""),
|
|
(["hello"], "hello"),
|
|
(["hello", "world"], "hello\nworld")
|
|
])
|
|
def test_newline_list_value(input_, expected):
|
|
form_data = {"newline_list": input_}
|
|
form = NewlineListForm(data=form_data)
|
|
|
|
assert form.validate()
|
|
assert form.newline_list._value() == expected
|