mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-06-07 23:10:41 -04:00
39 lines
724 B
Python
39 lines
724 B
Python
import pytest
|
|
from wtforms import Form, StringField
|
|
|
|
from app.main.forms import StripWhitespaceForm, StripWhitespaceStringField
|
|
|
|
|
|
class ExampleForm(StripWhitespaceForm):
|
|
foo = StringField("Foo")
|
|
|
|
|
|
class ExampleFormSpecialField(Form):
|
|
foo = StripWhitespaceStringField("foo")
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"submitted_data",
|
|
[
|
|
"bar",
|
|
" bar ",
|
|
"""
|
|
\t bar
|
|
""",
|
|
" \u180e\u200b \u200c bar \u200d \u2060\ufeff ",
|
|
],
|
|
)
|
|
@pytest.mark.parametrize(
|
|
"form",
|
|
[
|
|
ExampleForm,
|
|
ExampleFormSpecialField,
|
|
],
|
|
)
|
|
def test_form_strips_all_whitespace(
|
|
notify_admin,
|
|
form,
|
|
submitted_data,
|
|
):
|
|
assert form(foo=submitted_data).foo.data == "bar"
|