1. import datetime

  2. 

  3. from django.db import models

  4. from django.test import SimpleTestCase, TestCase, override_settings, skipUnlessDBFeature

  5. from django.test.utils import requires_tz_support

  6. from django.utils import timezone

  7. 

  8. from .models import DateTimeModel

  9. 

  10. 

  11. class DateTimeFieldTests(TestCase):

  12.     def test_datetimefield_to_python_microseconds(self):

  13.         """DateTimeField.to_python() supports microseconds."""

  14.         f = models.DateTimeField()

  15.         self.assertEqual(

  16.             f.to_python("2001-01-02 03:04:05.000006"),

  17.             datetime.datetime(2001, 1, 2, 3, 4, 5, 6),

  18.         )

  19.         self.assertEqual(

  20.             f.to_python("2001-01-02 03:04:05.999999"),

  21.             datetime.datetime(2001, 1, 2, 3, 4, 5, 999999),

  22.         )

  23. 

  24.     def test_timefield_to_python_microseconds(self):

  25.         """TimeField.to_python() supports microseconds."""

  26.         f = models.TimeField()

  27.         self.assertEqual(f.to_python("01:02:03.000004"), datetime.time(1, 2, 3, 4))

  28.         self.assertEqual(f.to_python("01:02:03.999999"), datetime.time(1, 2, 3, 999999))

  29. 

  30.     def test_datetimes_save_completely(self):

  31.         dat = datetime.date(2014, 3, 12)

  32.         datetim = datetime.datetime(2014, 3, 12, 21, 22, 23, 240000)

  33.         tim = datetime.time(21, 22, 23, 240000)

  34.         DateTimeModel.objects.create(d=dat, dt=datetim, t=tim)

  35.         obj = DateTimeModel.objects.first()

  36.         self.assertTrue(obj)

  37.         self.assertEqual(obj.d, dat)

  38.         self.assertEqual(obj.dt, datetim)

  39.         self.assertEqual(obj.t, tim)

  40. 

  41.     @override_settings(USE_TZ=False)

  42.     def test_lookup_date_without_use_tz(self):

  43.         d = datetime.date(2014, 3, 12)

  44.         dt1 = datetime.datetime(2014, 3, 12, 21, 22, 23, 240000)

  45.         dt2 = datetime.datetime(2014, 3, 11, 21, 22, 23, 240000)

  46.         t = datetime.time(21, 22, 23, 240000)

  47.         m = DateTimeModel.objects.create(d=d, dt=dt1, t=t)

  48.         # Other model with different datetime.

  49.         DateTimeModel.objects.create(d=d, dt=dt2, t=t)

  50.         self.assertEqual(m, DateTimeModel.objects.get(dt__date=d))

  51. 

  52.     @requires_tz_support

  53.     @skipUnlessDBFeature("has_zoneinfo_database")

  54.     @override_settings(USE_TZ=True, TIME_ZONE="America/Vancouver")

  55.     def test_lookup_date_with_use_tz(self):

  56.         d = datetime.date(2014, 3, 12)

  57.         # The following is equivalent to UTC 2014-03-12 18:34:23.24000.

  58.         dt1 = datetime.datetime(

  59.             2014, 3, 12, 10, 22, 23, 240000, tzinfo=timezone.get_current_timezone()

  60.         )

  61.         # The following is equivalent to UTC 2014-03-13 05:34:23.24000.

  62.         dt2 = datetime.datetime(

  63.             2014, 3, 12, 21, 22, 23, 240000, tzinfo=timezone.get_current_timezone()

  64.         )

  65.         t = datetime.time(21, 22, 23, 240000)

  66.         m1 = DateTimeModel.objects.create(d=d, dt=dt1, t=t)

  67.         m2 = DateTimeModel.objects.create(d=d, dt=dt2, t=t)

  68.         # In Vancouver, we expect both results.

  69.         self.assertCountEqual(

  70.             DateTimeModel.objects.filter(dt__date=d),

  71.             [m1, m2],

  72.         )

  73.         with self.settings(TIME_ZONE="UTC"):

  74.             # But in UTC, the __date only matches one of them.

  75.             self.assertCountEqual(DateTimeModel.objects.filter(dt__date=d), [m1])

  76. 

  77. 

  78. class ValidationTest(SimpleTestCase):

  79.     def test_datefield_cleans_date(self):

  80.         f = models.DateField()

  81.         self.assertEqual(datetime.date(2008, 10, 10), f.clean("2008-10-10", None))
About Sign Up Leave Feedback Contact