File length validation for task order upload

This commit is contained in:
dandds
2019-06-06 13:04:39 -04:00
parent f5e208ccc5
commit 318257e32c
4 changed files with 28 additions and 9 deletions

View File

@@ -1,13 +1,7 @@
from wtforms.validators import ValidationError, StopValidation
import pytest
from atst.forms.validators import (
Name,
IsNumber,
PhoneNumber,
ListItemsUnique,
RequiredIf,
)
from atst.forms.validators import *
class TestIsNumber:
@@ -97,3 +91,12 @@ class TestRequiredIf:
with pytest.raises(StopValidation):
validator(dummy_form, dummy_field)
class TestFileLength:
def test_FileLength(self, dummy_form, dummy_field, pdf_upload):
validator = FileLength(max_length=1)
dummy_field.data = pdf_upload
with pytest.raises(ValidationError):
validator(dummy_form, dummy_field)