1. # Unit tests for typecast functions in django.db.backends.util
    
  2. 
    
  3. import datetime
    
  4. import unittest
    
  5. 
    
  6. from django.db.backends import utils as typecasts
    
  7. 
    
  8. TEST_CASES = {
    
  9.     "typecast_date": (
    
  10.         ("", None),
    
  11.         (None, None),
    
  12.         ("2005-08-11", datetime.date(2005, 8, 11)),
    
  13.         ("1990-01-01", datetime.date(1990, 1, 1)),
    
  14.     ),
    
  15.     "typecast_time": (
    
  16.         ("", None),
    
  17.         (None, None),
    
  18.         ("0:00:00", datetime.time(0, 0)),
    
  19.         ("0:30:00", datetime.time(0, 30)),
    
  20.         ("8:50:00", datetime.time(8, 50)),
    
  21.         ("08:50:00", datetime.time(8, 50)),
    
  22.         ("12:00:00", datetime.time(12, 00)),
    
  23.         ("12:30:00", datetime.time(12, 30)),
    
  24.         ("13:00:00", datetime.time(13, 00)),
    
  25.         ("23:59:00", datetime.time(23, 59)),
    
  26.         ("00:00:12", datetime.time(0, 0, 12)),
    
  27.         ("00:00:12.5", datetime.time(0, 0, 12, 500000)),
    
  28.         ("7:22:13.312", datetime.time(7, 22, 13, 312000)),
    
  29.         ("12:45:30.126631", datetime.time(12, 45, 30, 126631)),
    
  30.         ("12:45:30.126630", datetime.time(12, 45, 30, 126630)),
    
  31.         ("12:45:30.123456789", datetime.time(12, 45, 30, 123456)),
    
  32.     ),
    
  33.     "typecast_timestamp": (
    
  34.         ("", None),
    
  35.         (None, None),
    
  36.         ("2005-08-11 0:00:00", datetime.datetime(2005, 8, 11)),
    
  37.         ("2005-08-11 0:30:00", datetime.datetime(2005, 8, 11, 0, 30)),
    
  38.         ("2005-08-11 8:50:30", datetime.datetime(2005, 8, 11, 8, 50, 30)),
    
  39.         ("2005-08-11 8:50:30.123", datetime.datetime(2005, 8, 11, 8, 50, 30, 123000)),
    
  40.         ("2005-08-11 8:50:30.9", datetime.datetime(2005, 8, 11, 8, 50, 30, 900000)),
    
  41.         (
    
  42.             "2005-08-11 8:50:30.312-05",
    
  43.             datetime.datetime(2005, 8, 11, 8, 50, 30, 312000),
    
  44.         ),
    
  45.         (
    
  46.             "2005-08-11 8:50:30.312+02",
    
  47.             datetime.datetime(2005, 8, 11, 8, 50, 30, 312000),
    
  48.         ),
    
  49.         # ticket 14453
    
  50.         (
    
  51.             "2010-10-12 15:29:22.063202",
    
  52.             datetime.datetime(2010, 10, 12, 15, 29, 22, 63202),
    
  53.         ),
    
  54.         (
    
  55.             "2010-10-12 15:29:22.063202-03",
    
  56.             datetime.datetime(2010, 10, 12, 15, 29, 22, 63202),
    
  57.         ),
    
  58.         (
    
  59.             "2010-10-12 15:29:22.063202+04",
    
  60.             datetime.datetime(2010, 10, 12, 15, 29, 22, 63202),
    
  61.         ),
    
  62.         (
    
  63.             "2010-10-12 15:29:22.0632021",
    
  64.             datetime.datetime(2010, 10, 12, 15, 29, 22, 63202),
    
  65.         ),
    
  66.         (
    
  67.             "2010-10-12 15:29:22.0632029",
    
  68.             datetime.datetime(2010, 10, 12, 15, 29, 22, 63202),
    
  69.         ),
    
  70.     ),
    
  71. }
    
  72. 
    
  73. 
    
  74. class DBTypeCasts(unittest.TestCase):
    
  75.     def test_typeCasts(self):
    
  76.         for k, v in TEST_CASES.items():
    
  77.             for inpt, expected in v:
    
  78.                 with self.subTest(k=k, inpt=inpt):
    
  79.                     got = getattr(typecasts, k)(inpt)
    
  80.                     self.assertEqual(
    
  81.                         got,
    
  82.                         expected,
    
  83.                         "In %s: %r doesn't match %r. Got %r instead."
    
  84.                         % (k, inpt, expected, got),
    
  85.                     )