/tests/regressiontests/datatypes/tests.py

https://code.google.com/p/mango-py/ · Python · 94 lines · 63 code · 19 blank · 12 comment · 1 complexity · bf24db2bf104972cfcd386537d7df28f MD5 · raw file

  1. import datetime
  2. from django.conf import settings
  3. from django.db import DEFAULT_DB_ALIAS
  4. from django.test import TestCase, skipIfDBFeature
  5. from django.utils import tzinfo
  6. from models import Donut, RumBaba
  7. class DataTypesTestCase(TestCase):
  8. def test_boolean_type(self):
  9. d = Donut(name='Apple Fritter')
  10. self.assertFalse(d.is_frosted)
  11. self.assertTrue(d.has_sprinkles is None)
  12. d.has_sprinkles = True
  13. self.assertTrue(d.has_sprinkles)
  14. d.save()
  15. d2 = Donut.objects.get(name='Apple Fritter')
  16. self.assertFalse(d2.is_frosted)
  17. self.assertTrue(d2.has_sprinkles)
  18. def test_date_type(self):
  19. d = Donut(name='Apple Fritter')
  20. d.baked_date = datetime.date(year=1938, month=6, day=4)
  21. d.baked_time = datetime.time(hour=5, minute=30)
  22. d.consumed_at = datetime.datetime(year=2007, month=4, day=20, hour=16, minute=19, second=59)
  23. d.save()
  24. d2 = Donut.objects.get(name='Apple Fritter')
  25. self.assertEqual(d2.baked_date, datetime.date(1938, 6, 4))
  26. self.assertEqual(d2.baked_time, datetime.time(5, 30))
  27. self.assertEqual(d2.consumed_at, datetime.datetime(2007, 4, 20, 16, 19, 59))
  28. def test_time_field(self):
  29. #Test for ticket #12059: TimeField wrongly handling datetime.datetime object.
  30. d = Donut(name='Apple Fritter')
  31. d.baked_time = datetime.datetime(year=2007, month=4, day=20, hour=16, minute=19, second=59)
  32. d.save()
  33. d2 = Donut.objects.get(name='Apple Fritter')
  34. self.assertEqual(d2.baked_time, datetime.time(16, 19, 59))
  35. def test_year_boundaries(self):
  36. """Year boundary tests (ticket #3689)"""
  37. d = Donut.objects.create(name='Date Test 2007',
  38. baked_date=datetime.datetime(year=2007, month=12, day=31),
  39. consumed_at=datetime.datetime(year=2007, month=12, day=31, hour=23, minute=59, second=59))
  40. d1 = Donut.objects.create(name='Date Test 2006',
  41. baked_date=datetime.datetime(year=2006, month=1, day=1),
  42. consumed_at=datetime.datetime(year=2006, month=1, day=1))
  43. self.assertEqual("Date Test 2007",
  44. Donut.objects.filter(baked_date__year=2007)[0].name)
  45. self.assertEqual("Date Test 2006",
  46. Donut.objects.filter(baked_date__year=2006)[0].name)
  47. d2 = Donut.objects.create(name='Apple Fritter',
  48. consumed_at = datetime.datetime(year=2007, month=4, day=20, hour=16, minute=19, second=59))
  49. self.assertEqual([u'Apple Fritter', u'Date Test 2007'],
  50. list(Donut.objects.filter(consumed_at__year=2007).order_by('name').values_list('name', flat=True)))
  51. self.assertEqual(0, Donut.objects.filter(consumed_at__year=2005).count())
  52. self.assertEqual(0, Donut.objects.filter(consumed_at__year=2008).count())
  53. def test_textfields_unicode(self):
  54. """Regression test for #10238: TextField values returned from the
  55. database should be unicode."""
  56. d = Donut.objects.create(name=u'Jelly Donut', review=u'Outstanding')
  57. newd = Donut.objects.get(id=d.id)
  58. self.assertTrue(isinstance(newd.review, unicode))
  59. @skipIfDBFeature('supports_timezones')
  60. def test_error_on_timezone(self):
  61. """Regression test for #8354: the MySQL and Oracle backends should raise
  62. an error if given a timezone-aware datetime object."""
  63. dt = datetime.datetime(2008, 8, 31, 16, 20, tzinfo=tzinfo.FixedOffset(0))
  64. d = Donut(name='Bear claw', consumed_at=dt)
  65. self.assertRaises(ValueError, d.save)
  66. # ValueError: MySQL backend does not support timezone-aware datetimes.
  67. def test_datefield_auto_now_add(self):
  68. """Regression test for #10970, auto_now_add for DateField should store
  69. a Python datetime.date, not a datetime.datetime"""
  70. b = RumBaba.objects.create()
  71. # Verify we didn't break DateTimeField behavior
  72. self.assertTrue(isinstance(b.baked_timestamp, datetime.datetime))
  73. # We need to test this this way because datetime.datetime inherits
  74. # from datetime.date:
  75. self.assertTrue(isinstance(b.baked_date, datetime.date) and not isinstance(b.baked_date, datetime.datetime))