Add choices keyword argument to BaseField.__init__()

This commit is contained in:
Don Spaulding
2010-04-15 17:59:35 -05:00
parent e9c92f30ba
commit ee0c75a26d
2 changed files with 32 additions and 3 deletions

View File

@@ -588,5 +588,26 @@ class FieldTest(unittest.TestCase):
AttachmentRequired.drop_collection()
AttachmentSizeLimit.drop_collection()
def test_choices_validation(self):
"""Ensure that value is in a container of allowed values.
"""
class Shirt(Document):
size = StringField(max_length=3, choices=('S','M','L','XL','XXL'))
Shirt.drop_collection()
shirt = Shirt()
shirt.validate()
shirt.size = "S"
shirt.validate()
shirt.size = "XS"
self.assertRaises(ValidationError, shirt.validate)
Shirt.drop_collection()
if __name__ == '__main__':
unittest.main()