1. import unittest
    
  2. from datetime import date, datetime, time, timedelta
    
  3. 
    
  4. from django.utils.dateparse import (
    
  5.     parse_date,
    
  6.     parse_datetime,
    
  7.     parse_duration,
    
  8.     parse_time,
    
  9. )
    
  10. from django.utils.timezone import get_fixed_timezone
    
  11. from django.utils.version import PY311
    
  12. 
    
  13. 
    
  14. class DateParseTests(unittest.TestCase):
    
  15.     def test_parse_date(self):
    
  16.         # Valid inputs
    
  17.         self.assertEqual(parse_date("2012-04-23"), date(2012, 4, 23))
    
  18.         self.assertEqual(parse_date("2012-4-9"), date(2012, 4, 9))
    
  19.         if PY311:
    
  20.             self.assertEqual(parse_date("20120423"), date(2012, 4, 23))
    
  21.         # Invalid inputs
    
  22.         self.assertIsNone(parse_date("2012423"))
    
  23.         with self.assertRaises(ValueError):
    
  24.             parse_date("2012-04-56")
    
  25. 
    
  26.     def test_parse_time(self):
    
  27.         # Valid inputs
    
  28.         self.assertEqual(parse_time("09:15:00"), time(9, 15))
    
  29.         if PY311:
    
  30.             self.assertEqual(parse_time("091500"), time(9, 15))
    
  31.         self.assertEqual(parse_time("10:10"), time(10, 10))
    
  32.         self.assertEqual(parse_time("10:20:30.400"), time(10, 20, 30, 400000))
    
  33.         self.assertEqual(parse_time("10:20:30,400"), time(10, 20, 30, 400000))
    
  34.         self.assertEqual(parse_time("4:8:16"), time(4, 8, 16))
    
  35.         # Time zone offset is ignored.
    
  36.         self.assertEqual(parse_time("00:05:23+04:00"), time(0, 5, 23))
    
  37.         # Invalid inputs
    
  38.         self.assertIsNone(parse_time("00:05:"))
    
  39.         self.assertIsNone(parse_time("00:05:23,"))
    
  40.         self.assertIsNone(parse_time("00:05:23+"))
    
  41.         self.assertIsNone(parse_time("00:05:23+25:00"))
    
  42.         self.assertIsNone(parse_time("4:18:101"))
    
  43.         self.assertIsNone(parse_time("91500"))
    
  44.         with self.assertRaises(ValueError):
    
  45.             parse_time("09:15:90")
    
  46. 
    
  47.     def test_parse_datetime(self):
    
  48.         valid_inputs = (
    
  49.             ("2012-04-23T09:15:00", datetime(2012, 4, 23, 9, 15)),
    
  50.             ("2012-4-9 4:8:16", datetime(2012, 4, 9, 4, 8, 16)),
    
  51.             (
    
  52.                 "2012-04-23T09:15:00Z",
    
  53.                 datetime(2012, 4, 23, 9, 15, 0, 0, get_fixed_timezone(0)),
    
  54.             ),
    
  55.             (
    
  56.                 "2012-4-9 4:8:16-0320",
    
  57.                 datetime(2012, 4, 9, 4, 8, 16, 0, get_fixed_timezone(-200)),
    
  58.             ),
    
  59.             (
    
  60.                 "2012-04-23T10:20:30.400+02:30",
    
  61.                 datetime(2012, 4, 23, 10, 20, 30, 400000, get_fixed_timezone(150)),
    
  62.             ),
    
  63.             (
    
  64.                 "2012-04-23T10:20:30.400+02",
    
  65.                 datetime(2012, 4, 23, 10, 20, 30, 400000, get_fixed_timezone(120)),
    
  66.             ),
    
  67.             (
    
  68.                 "2012-04-23T10:20:30.400-02",
    
  69.                 datetime(2012, 4, 23, 10, 20, 30, 400000, get_fixed_timezone(-120)),
    
  70.             ),
    
  71.             (
    
  72.                 "2012-04-23T10:20:30,400-02",
    
  73.                 datetime(2012, 4, 23, 10, 20, 30, 400000, get_fixed_timezone(-120)),
    
  74.             ),
    
  75.             (
    
  76.                 "2012-04-23T10:20:30.400 +0230",
    
  77.                 datetime(2012, 4, 23, 10, 20, 30, 400000, get_fixed_timezone(150)),
    
  78.             ),
    
  79.             (
    
  80.                 "2012-04-23T10:20:30,400 +00",
    
  81.                 datetime(2012, 4, 23, 10, 20, 30, 400000, get_fixed_timezone(0)),
    
  82.             ),
    
  83.             (
    
  84.                 "2012-04-23T10:20:30   -02",
    
  85.                 datetime(2012, 4, 23, 10, 20, 30, 0, get_fixed_timezone(-120)),
    
  86.             ),
    
  87.         )
    
  88.         for source, expected in valid_inputs:
    
  89.             with self.subTest(source=source):
    
  90.                 self.assertEqual(parse_datetime(source), expected)
    
  91. 
    
  92.         # Invalid inputs
    
  93.         self.assertIsNone(parse_datetime("20120423091500"))
    
  94.         with self.assertRaises(ValueError):
    
  95.             parse_datetime("2012-04-56T09:15:90")
    
  96. 
    
  97. 
    
  98. class DurationParseTests(unittest.TestCase):
    
  99.     def test_parse_python_format(self):
    
  100.         timedeltas = [
    
  101.             timedelta(
    
  102.                 days=4, minutes=15, seconds=30, milliseconds=100
    
  103.             ),  # fractions of seconds
    
  104.             timedelta(hours=10, minutes=15, seconds=30),  # hours, minutes, seconds
    
  105.             timedelta(days=4, minutes=15, seconds=30),  # multiple days
    
  106.             timedelta(days=1, minutes=00, seconds=00),  # single day
    
  107.             timedelta(days=-4, minutes=15, seconds=30),  # negative durations
    
  108.             timedelta(minutes=15, seconds=30),  # minute & seconds
    
  109.             timedelta(seconds=30),  # seconds
    
  110.         ]
    
  111.         for delta in timedeltas:
    
  112.             with self.subTest(delta=delta):
    
  113.                 self.assertEqual(parse_duration(format(delta)), delta)
    
  114. 
    
  115.     def test_parse_postgresql_format(self):
    
  116.         test_values = (
    
  117.             ("1 day", timedelta(1)),
    
  118.             ("-1 day", timedelta(-1)),
    
  119.             ("1 day 0:00:01", timedelta(days=1, seconds=1)),
    
  120.             ("1 day -0:00:01", timedelta(days=1, seconds=-1)),
    
  121.             ("-1 day -0:00:01", timedelta(days=-1, seconds=-1)),
    
  122.             ("-1 day +0:00:01", timedelta(days=-1, seconds=1)),
    
  123.             (
    
  124.                 "4 days 0:15:30.1",
    
  125.                 timedelta(days=4, minutes=15, seconds=30, milliseconds=100),
    
  126.             ),
    
  127.             (
    
  128.                 "4 days 0:15:30.0001",
    
  129.                 timedelta(days=4, minutes=15, seconds=30, microseconds=100),
    
  130.             ),
    
  131.             ("-4 days -15:00:30", timedelta(days=-4, hours=-15, seconds=-30)),
    
  132.         )
    
  133.         for source, expected in test_values:
    
  134.             with self.subTest(source=source):
    
  135.                 self.assertEqual(parse_duration(source), expected)
    
  136. 
    
  137.     def test_seconds(self):
    
  138.         self.assertEqual(parse_duration("30"), timedelta(seconds=30))
    
  139. 
    
  140.     def test_minutes_seconds(self):
    
  141.         self.assertEqual(parse_duration("15:30"), timedelta(minutes=15, seconds=30))
    
  142.         self.assertEqual(parse_duration("5:30"), timedelta(minutes=5, seconds=30))
    
  143. 
    
  144.     def test_hours_minutes_seconds(self):
    
  145.         self.assertEqual(
    
  146.             parse_duration("10:15:30"), timedelta(hours=10, minutes=15, seconds=30)
    
  147.         )
    
  148.         self.assertEqual(
    
  149.             parse_duration("1:15:30"), timedelta(hours=1, minutes=15, seconds=30)
    
  150.         )
    
  151.         self.assertEqual(
    
  152.             parse_duration("100:200:300"),
    
  153.             timedelta(hours=100, minutes=200, seconds=300),
    
  154.         )
    
  155. 
    
  156.     def test_days(self):
    
  157.         self.assertEqual(
    
  158.             parse_duration("4 15:30"), timedelta(days=4, minutes=15, seconds=30)
    
  159.         )
    
  160.         self.assertEqual(
    
  161.             parse_duration("4 10:15:30"),
    
  162.             timedelta(days=4, hours=10, minutes=15, seconds=30),
    
  163.         )
    
  164. 
    
  165.     def test_fractions_of_seconds(self):
    
  166.         test_values = (
    
  167.             ("15:30.1", timedelta(minutes=15, seconds=30, milliseconds=100)),
    
  168.             ("15:30.01", timedelta(minutes=15, seconds=30, milliseconds=10)),
    
  169.             ("15:30.001", timedelta(minutes=15, seconds=30, milliseconds=1)),
    
  170.             ("15:30.0001", timedelta(minutes=15, seconds=30, microseconds=100)),
    
  171.             ("15:30.00001", timedelta(minutes=15, seconds=30, microseconds=10)),
    
  172.             ("15:30.000001", timedelta(minutes=15, seconds=30, microseconds=1)),
    
  173.             ("15:30,000001", timedelta(minutes=15, seconds=30, microseconds=1)),
    
  174.         )
    
  175.         for source, expected in test_values:
    
  176.             with self.subTest(source=source):
    
  177.                 self.assertEqual(parse_duration(source), expected)
    
  178. 
    
  179.     def test_negative(self):
    
  180.         test_values = (
    
  181.             ("-4 15:30", timedelta(days=-4, minutes=15, seconds=30)),
    
  182.             ("-172800", timedelta(days=-2)),
    
  183.             ("-15:30", timedelta(minutes=-15, seconds=-30)),
    
  184.             ("-1:15:30", timedelta(hours=-1, minutes=-15, seconds=-30)),
    
  185.             ("-30.1", timedelta(seconds=-30, milliseconds=-100)),
    
  186.             ("-30,1", timedelta(seconds=-30, milliseconds=-100)),
    
  187.             ("-00:01:01", timedelta(minutes=-1, seconds=-1)),
    
  188.             ("-01:01", timedelta(seconds=-61)),
    
  189.             ("-01:-01", None),
    
  190.         )
    
  191.         for source, expected in test_values:
    
  192.             with self.subTest(source=source):
    
  193.                 self.assertEqual(parse_duration(source), expected)
    
  194. 
    
  195.     def test_iso_8601(self):
    
  196.         test_values = (
    
  197.             ("P4Y", None),
    
  198.             ("P4M", None),
    
  199.             ("P4W", None),
    
  200.             ("P4D", timedelta(days=4)),
    
  201.             ("-P1D", timedelta(days=-1)),
    
  202.             ("P0.5D", timedelta(hours=12)),
    
  203.             ("P0,5D", timedelta(hours=12)),
    
  204.             ("-P0.5D", timedelta(hours=-12)),
    
  205.             ("-P0,5D", timedelta(hours=-12)),
    
  206.             ("PT5H", timedelta(hours=5)),
    
  207.             ("-PT5H", timedelta(hours=-5)),
    
  208.             ("PT5M", timedelta(minutes=5)),
    
  209.             ("-PT5M", timedelta(minutes=-5)),
    
  210.             ("PT5S", timedelta(seconds=5)),
    
  211.             ("-PT5S", timedelta(seconds=-5)),
    
  212.             ("PT0.000005S", timedelta(microseconds=5)),
    
  213.             ("PT0,000005S", timedelta(microseconds=5)),
    
  214.             ("-PT0.000005S", timedelta(microseconds=-5)),
    
  215.             ("-PT0,000005S", timedelta(microseconds=-5)),
    
  216.             ("-P4DT1H", timedelta(days=-4, hours=-1)),
    
  217.             # Invalid separators for decimal fractions.
    
  218.             ("P3(3D", None),
    
  219.             ("PT3)3H", None),
    
  220.             ("PT3|3M", None),
    
  221.             ("PT3/3S", None),
    
  222.         )
    
  223.         for source, expected in test_values:
    
  224.             with self.subTest(source=source):
    
  225.                 self.assertEqual(parse_duration(source), expected)