def test_process_with_one_format(self):
"""One date format is specified."""
fmt = "%Y-%m-%d %H:%M:%S"
field = MultiFormatDateField(formats=[fmt],
_form=mock.MagicMock(),
_name="test")
field.process_formdata(["2012-01-02 05:55:02"])
self.assertIsInstance(field.data, date, "Should parse successfully")
def test_value_with_one_format(self):
"""One date format is specified."""
fmt = '%Y-%m-%d %H:%M:%S'
value = datetime.now()
field = MultiFormatDateField(formats=[fmt],
_form=mock.MagicMock(),
_name='test')
field.data = value
self.assertEqual(field._value(), value.strftime(fmt),
"Should use the first (only) format to render value")
def test_process_with_several_formats(self):
"""Several date formats are specified."""
field = MultiFormatDateField(formats=['%Y-%m-%d', '%Y-%m', '%Y'],
_form=mock.MagicMock(),
_name='test')
field.process_formdata(['2012-03-02'])
self.assertIsInstance(field.data, date, "Should parse successfully")
self.assertEqual(field.data.day, 2)
self.assertEqual(field.data.month, 3)
self.assertEqual(field.data.year, 2012)
field.process_formdata(['2014-05'])
self.assertIsInstance(field.data, date, "Should parse successfully")
self.assertEqual(field.data.day, 1)
self.assertEqual(field.data.month, 5)
self.assertEqual(field.data.year, 2014)
field.process_formdata(['2011'])
self.assertIsInstance(field.data, date, "Should parse successfully")
self.assertEqual(field.data.day, 1)
self.assertEqual(field.data.month, 1)
self.assertEqual(field.data.year, 2011)