notifications-admin/tests/app/main/forms/test_strip_whitespace.py

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'