1. from datetime import date, datetime
    
  2. 
    
  3. from django.core.exceptions import ValidationError
    
  4. from django.forms import DateField, Form, HiddenInput, SelectDateWidget
    
  5. from django.test import SimpleTestCase
    
  6. from django.utils import translation
    
  7. 
    
  8. 
    
  9. class GetDate(Form):
    
  10.     mydate = DateField(widget=SelectDateWidget)
    
  11. 
    
  12. 
    
  13. class DateFieldTest(SimpleTestCase):
    
  14.     def test_form_field(self):
    
  15.         a = GetDate({"mydate_month": "4", "mydate_day": "1", "mydate_year": "2008"})
    
  16.         self.assertTrue(a.is_valid())
    
  17.         self.assertEqual(a.cleaned_data["mydate"], date(2008, 4, 1))
    
  18. 
    
  19.         # As with any widget that implements get_value_from_datadict(), we must
    
  20.         # accept the input from the "as_hidden" rendering as well.
    
  21.         self.assertHTMLEqual(
    
  22.             a["mydate"].as_hidden(),
    
  23.             '<input type="hidden" name="mydate" value="2008-04-01" id="id_mydate">',
    
  24.         )
    
  25. 
    
  26.         b = GetDate({"mydate": "2008-4-1"})
    
  27.         self.assertTrue(b.is_valid())
    
  28.         self.assertEqual(b.cleaned_data["mydate"], date(2008, 4, 1))
    
  29. 
    
  30.         # Invalid dates shouldn't be allowed
    
  31.         c = GetDate({"mydate_month": "2", "mydate_day": "31", "mydate_year": "2010"})
    
  32.         self.assertFalse(c.is_valid())
    
  33.         self.assertEqual(c.errors, {"mydate": ["Enter a valid date."]})
    
  34. 
    
  35.         # label tag is correctly associated with month dropdown
    
  36.         d = GetDate({"mydate_month": "1", "mydate_day": "1", "mydate_year": "2010"})
    
  37.         self.assertIn('<label for="id_mydate_month">', d.as_p())
    
  38. 
    
  39.     @translation.override("nl")
    
  40.     def test_l10n_date_changed(self):
    
  41.         """
    
  42.         DateField.has_changed() with SelectDateWidget works with a localized
    
  43.         date format (#17165).
    
  44.         """
    
  45.         # With Field.show_hidden_initial=False
    
  46.         b = GetDate(
    
  47.             {
    
  48.                 "mydate_year": "2008",
    
  49.                 "mydate_month": "4",
    
  50.                 "mydate_day": "1",
    
  51.             },
    
  52.             initial={"mydate": date(2008, 4, 1)},
    
  53.         )
    
  54.         self.assertFalse(b.has_changed())
    
  55. 
    
  56.         b = GetDate(
    
  57.             {
    
  58.                 "mydate_year": "2008",
    
  59.                 "mydate_month": "4",
    
  60.                 "mydate_day": "2",
    
  61.             },
    
  62.             initial={"mydate": date(2008, 4, 1)},
    
  63.         )
    
  64.         self.assertTrue(b.has_changed())
    
  65. 
    
  66.         # With Field.show_hidden_initial=True
    
  67.         class GetDateShowHiddenInitial(Form):
    
  68.             mydate = DateField(widget=SelectDateWidget, show_hidden_initial=True)
    
  69. 
    
  70.         b = GetDateShowHiddenInitial(
    
  71.             {
    
  72.                 "mydate_year": "2008",
    
  73.                 "mydate_month": "4",
    
  74.                 "mydate_day": "1",
    
  75.                 "initial-mydate": HiddenInput().format_value(date(2008, 4, 1)),
    
  76.             },
    
  77.             initial={"mydate": date(2008, 4, 1)},
    
  78.         )
    
  79.         self.assertFalse(b.has_changed())
    
  80. 
    
  81.         b = GetDateShowHiddenInitial(
    
  82.             {
    
  83.                 "mydate_year": "2008",
    
  84.                 "mydate_month": "4",
    
  85.                 "mydate_day": "22",
    
  86.                 "initial-mydate": HiddenInput().format_value(date(2008, 4, 1)),
    
  87.             },
    
  88.             initial={"mydate": date(2008, 4, 1)},
    
  89.         )
    
  90.         self.assertTrue(b.has_changed())
    
  91. 
    
  92.         b = GetDateShowHiddenInitial(
    
  93.             {
    
  94.                 "mydate_year": "2008",
    
  95.                 "mydate_month": "4",
    
  96.                 "mydate_day": "22",
    
  97.                 "initial-mydate": HiddenInput().format_value(date(2008, 4, 1)),
    
  98.             },
    
  99.             initial={"mydate": date(2008, 4, 22)},
    
  100.         )
    
  101.         self.assertTrue(b.has_changed())
    
  102. 
    
  103.         b = GetDateShowHiddenInitial(
    
  104.             {
    
  105.                 "mydate_year": "2008",
    
  106.                 "mydate_month": "4",
    
  107.                 "mydate_day": "22",
    
  108.                 "initial-mydate": HiddenInput().format_value(date(2008, 4, 22)),
    
  109.             },
    
  110.             initial={"mydate": date(2008, 4, 1)},
    
  111.         )
    
  112.         self.assertFalse(b.has_changed())
    
  113. 
    
  114.     @translation.override("nl")
    
  115.     def test_l10n_invalid_date_in(self):
    
  116.         # Invalid dates shouldn't be allowed
    
  117.         a = GetDate({"mydate_month": "2", "mydate_day": "31", "mydate_year": "2010"})
    
  118.         self.assertFalse(a.is_valid())
    
  119.         # 'Geef een geldige datum op.' = 'Enter a valid date.'
    
  120.         self.assertEqual(a.errors, {"mydate": ["Voer een geldige datum in."]})
    
  121. 
    
  122.     @translation.override("nl")
    
  123.     def test_form_label_association(self):
    
  124.         # label tag is correctly associated with first rendered dropdown
    
  125.         a = GetDate({"mydate_month": "1", "mydate_day": "1", "mydate_year": "2010"})
    
  126.         self.assertIn('<label for="id_mydate_day">', a.as_p())
    
  127. 
    
  128.     def test_datefield_1(self):
    
  129.         f = DateField()
    
  130.         self.assertEqual(date(2006, 10, 25), f.clean(date(2006, 10, 25)))
    
  131.         self.assertEqual(date(2006, 10, 25), f.clean(datetime(2006, 10, 25, 14, 30)))
    
  132.         self.assertEqual(
    
  133.             date(2006, 10, 25), f.clean(datetime(2006, 10, 25, 14, 30, 59))
    
  134.         )
    
  135.         self.assertEqual(
    
  136.             date(2006, 10, 25), f.clean(datetime(2006, 10, 25, 14, 30, 59, 200))
    
  137.         )
    
  138.         self.assertEqual(date(2006, 10, 25), f.clean("2006-10-25"))
    
  139.         self.assertEqual(date(2006, 10, 25), f.clean("10/25/2006"))
    
  140.         self.assertEqual(date(2006, 10, 25), f.clean("10/25/06"))
    
  141.         self.assertEqual(date(2006, 10, 25), f.clean("Oct 25 2006"))
    
  142.         self.assertEqual(date(2006, 10, 25), f.clean("October 25 2006"))
    
  143.         self.assertEqual(date(2006, 10, 25), f.clean("October 25, 2006"))
    
  144.         self.assertEqual(date(2006, 10, 25), f.clean("25 October 2006"))
    
  145.         self.assertEqual(date(2006, 10, 25), f.clean("25 October, 2006"))
    
  146.         with self.assertRaisesMessage(ValidationError, "'Enter a valid date.'"):
    
  147.             f.clean("2006-4-31")
    
  148.         with self.assertRaisesMessage(ValidationError, "'Enter a valid date.'"):
    
  149.             f.clean("200a-10-25")
    
  150.         with self.assertRaisesMessage(ValidationError, "'Enter a valid date.'"):
    
  151.             f.clean("25/10/06")
    
  152.         with self.assertRaisesMessage(ValidationError, "'This field is required.'"):
    
  153.             f.clean(None)
    
  154. 
    
  155.     def test_datefield_2(self):
    
  156.         f = DateField(required=False)
    
  157.         self.assertIsNone(f.clean(None))
    
  158.         self.assertEqual("None", repr(f.clean(None)))
    
  159.         self.assertIsNone(f.clean(""))
    
  160.         self.assertEqual("None", repr(f.clean("")))
    
  161. 
    
  162.     def test_datefield_3(self):
    
  163.         f = DateField(input_formats=["%Y %m %d"])
    
  164.         self.assertEqual(date(2006, 10, 25), f.clean(date(2006, 10, 25)))
    
  165.         self.assertEqual(date(2006, 10, 25), f.clean(datetime(2006, 10, 25, 14, 30)))
    
  166.         self.assertEqual(date(2006, 10, 25), f.clean("2006 10 25"))
    
  167.         with self.assertRaisesMessage(ValidationError, "'Enter a valid date.'"):
    
  168.             f.clean("2006-10-25")
    
  169.         with self.assertRaisesMessage(ValidationError, "'Enter a valid date.'"):
    
  170.             f.clean("10/25/2006")
    
  171.         with self.assertRaisesMessage(ValidationError, "'Enter a valid date.'"):
    
  172.             f.clean("10/25/06")
    
  173. 
    
  174.     def test_datefield_4(self):
    
  175.         # Test whitespace stripping behavior (#5714)
    
  176.         f = DateField()
    
  177.         self.assertEqual(date(2006, 10, 25), f.clean(" 10/25/2006 "))
    
  178.         self.assertEqual(date(2006, 10, 25), f.clean(" 10/25/06 "))
    
  179.         self.assertEqual(date(2006, 10, 25), f.clean(" Oct 25   2006 "))
    
  180.         self.assertEqual(date(2006, 10, 25), f.clean(" October  25 2006 "))
    
  181.         self.assertEqual(date(2006, 10, 25), f.clean(" October 25, 2006 "))
    
  182.         self.assertEqual(date(2006, 10, 25), f.clean(" 25 October 2006 "))
    
  183.         with self.assertRaisesMessage(ValidationError, "'Enter a valid date.'"):
    
  184.             f.clean("   ")
    
  185. 
    
  186.     def test_datefield_5(self):
    
  187.         # Test null bytes (#18982)
    
  188.         f = DateField()
    
  189.         with self.assertRaisesMessage(ValidationError, "'Enter a valid date.'"):
    
  190.             f.clean("a\x00b")
    
  191. 
    
  192.     def test_datefield_changed(self):
    
  193.         format = "%d/%m/%Y"
    
  194.         f = DateField(input_formats=[format])
    
  195.         d = date(2007, 9, 17)
    
  196.         self.assertFalse(f.has_changed(d, "17/09/2007"))
    
  197. 
    
  198.     def test_datefield_strptime(self):
    
  199.         """field.strptime() doesn't raise a UnicodeEncodeError (#16123)"""
    
  200.         f = DateField()
    
  201.         try:
    
  202.             f.strptime("31 мая 2011", "%d-%b-%y")
    
  203.         except Exception as e:
    
  204.             # assertIsInstance or assertRaises cannot be used because UnicodeEncodeError
    
  205.             # is a subclass of ValueError
    
  206.             self.assertEqual(e.__class__, ValueError)