1. import datetime

  2. 

  3. from django.test import TestCase, skipIfDBFeature

  4. 

  5. from .models import Donut, RumBaba

  6. 

  7. 

  8. class DataTypesTestCase(TestCase):

  9.     def test_boolean_type(self):

  10.         d = Donut(name="Apple Fritter")

  11.         self.assertFalse(d.is_frosted)

  12.         self.assertIsNone(d.has_sprinkles)

  13.         d.has_sprinkles = True

  14.         self.assertTrue(d.has_sprinkles)

  15. 

  16.         d.save()

  17. 

  18.         d2 = Donut.objects.get(name="Apple Fritter")

  19.         self.assertFalse(d2.is_frosted)

  20.         self.assertTrue(d2.has_sprinkles)

  21. 

  22.     def test_date_type(self):

  23.         d = Donut(name="Apple Fritter")

  24.         d.baked_date = datetime.date(year=1938, month=6, day=4)

  25.         d.baked_time = datetime.time(hour=5, minute=30)

  26.         d.consumed_at = datetime.datetime(

  27.             year=2007, month=4, day=20, hour=16, minute=19, second=59

  28.         )

  29.         d.save()

  30. 

  31.         d2 = Donut.objects.get(name="Apple Fritter")

  32.         self.assertEqual(d2.baked_date, datetime.date(1938, 6, 4))

  33.         self.assertEqual(d2.baked_time, datetime.time(5, 30))

  34.         self.assertEqual(d2.consumed_at, datetime.datetime(2007, 4, 20, 16, 19, 59))

  35. 

  36.     def test_time_field(self):

  37.         # Test for ticket #12059: TimeField wrongly handling datetime.datetime object.

  38.         d = Donut(name="Apple Fritter")

  39.         d.baked_time = datetime.datetime(

  40.             year=2007, month=4, day=20, hour=16, minute=19, second=59

  41.         )

  42.         d.save()

  43. 

  44.         d2 = Donut.objects.get(name="Apple Fritter")

  45.         self.assertEqual(d2.baked_time, datetime.time(16, 19, 59))

  46. 

  47.     def test_year_boundaries(self):

  48.         """Year boundary tests (ticket #3689)"""

  49.         Donut.objects.create(

  50.             name="Date Test 2007",

  51.             baked_date=datetime.datetime(year=2007, month=12, day=31),

  52.             consumed_at=datetime.datetime(

  53.                 year=2007, month=12, day=31, hour=23, minute=59, second=59

  54.             ),

  55.         )

  56.         Donut.objects.create(

  57.             name="Date Test 2006",

  58.             baked_date=datetime.datetime(year=2006, month=1, day=1),

  59.             consumed_at=datetime.datetime(year=2006, month=1, day=1),

  60.         )

  61.         self.assertEqual(

  62.             "Date Test 2007", Donut.objects.filter(baked_date__year=2007)[0].name

  63.         )

  64.         self.assertEqual(

  65.             "Date Test 2006", Donut.objects.filter(baked_date__year=2006)[0].name

  66.         )

  67. 

  68.         Donut.objects.create(

  69.             name="Apple Fritter",

  70.             consumed_at=datetime.datetime(

  71.                 year=2007, month=4, day=20, hour=16, minute=19, second=59

  72.             ),

  73.         )

  74. 

  75.         self.assertEqual(

  76.             ["Apple Fritter", "Date Test 2007"],

  77.             list(

  78.                 Donut.objects.filter(consumed_at__year=2007)

  79.                 .order_by("name")

  80.                 .values_list("name", flat=True)

  81.             ),

  82.         )

  83.         self.assertEqual(0, Donut.objects.filter(consumed_at__year=2005).count())

  84.         self.assertEqual(0, Donut.objects.filter(consumed_at__year=2008).count())

  85. 

  86.     def test_textfields_str(self):

  87.         """TextField values returned from the database should be str."""

  88.         d = Donut.objects.create(name="Jelly Donut", review="Outstanding")

  89.         newd = Donut.objects.get(id=d.id)

  90.         self.assertIsInstance(newd.review, str)

  91. 

  92.     @skipIfDBFeature("supports_timezones")

  93.     def test_error_on_timezone(self):

  94.         """Regression test for #8354: the MySQL and Oracle backends should raise

  95.         an error if given a timezone-aware datetime object."""

  96.         dt = datetime.datetime(2008, 8, 31, 16, 20, tzinfo=datetime.timezone.utc)

  97.         d = Donut(name="Bear claw", consumed_at=dt)

  98.         # MySQL backend does not support timezone-aware datetimes.

  99.         with self.assertRaises(ValueError):

  100.             d.save()

  101. 

  102.     def test_datefield_auto_now_add(self):

  103.         """Regression test for #10970, auto_now_add for DateField should store

  104.         a Python datetime.date, not a datetime.datetime"""

  105.         b = RumBaba.objects.create()

  106.         # Verify we didn't break DateTimeField behavior

  107.         self.assertIsInstance(b.baked_timestamp, datetime.datetime)

  108.         # We need to test this way because datetime.datetime inherits

  109.         # from datetime.date:

  110.         self.assertIsInstance(b.baked_date, datetime.date)

  111.         self.assertNotIsInstance(b.baked_date, datetime.datetime)
About Sign Up Leave Feedback Contact