File: test_datetime_safe.py

package info (click to toggle)
python-django 3%3A4.2.27-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 58,916 kB
  • sloc: python: 334,817; javascript: 18,754; xml: 215; makefile: 178; sh: 27
file content (77 lines) | stat: -rw-r--r-- 2,810 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
from datetime import date as original_date
from datetime import datetime as original_datetime

from django.test import SimpleTestCase, ignore_warnings
from django.utils.deprecation import RemovedInDjango50Warning

with ignore_warnings(category=RemovedInDjango50Warning):
    from django.utils.datetime_safe import date, datetime


class DatetimeTests(SimpleTestCase):
    def setUp(self):
        self.percent_y_safe = (1900, 1, 1)  # >= 1900 required on Windows.
        self.just_safe = (1000, 1, 1)
        self.just_unsafe = (999, 12, 31, 23, 59, 59)
        self.really_old = (20, 1, 1)
        self.more_recent = (2006, 1, 1)

    def test_compare_datetimes(self):
        self.assertEqual(
            original_datetime(*self.more_recent), datetime(*self.more_recent)
        )
        self.assertEqual(
            original_datetime(*self.really_old), datetime(*self.really_old)
        )
        self.assertEqual(original_date(*self.more_recent), date(*self.more_recent))
        self.assertEqual(original_date(*self.really_old), date(*self.really_old))

        self.assertEqual(
            original_date(*self.just_safe).strftime("%Y-%m-%d"),
            date(*self.just_safe).strftime("%Y-%m-%d"),
        )
        self.assertEqual(
            original_datetime(*self.just_safe).strftime("%Y-%m-%d"),
            datetime(*self.just_safe).strftime("%Y-%m-%d"),
        )

    def test_safe_strftime(self):
        self.assertEqual(
            date(*self.just_unsafe[:3]).strftime("%Y-%m-%d (weekday %w)"),
            "0999-12-31 (weekday 2)",
        )
        self.assertEqual(
            date(*self.just_safe).strftime("%Y-%m-%d (weekday %w)"),
            "1000-01-01 (weekday 3)",
        )

        self.assertEqual(
            datetime(*self.just_unsafe).strftime("%Y-%m-%d %H:%M:%S (weekday %w)"),
            "0999-12-31 23:59:59 (weekday 2)",
        )
        self.assertEqual(
            datetime(*self.just_safe).strftime("%Y-%m-%d %H:%M:%S (weekday %w)"),
            "1000-01-01 00:00:00 (weekday 3)",
        )

        # %y will error before this date
        self.assertEqual(date(*self.percent_y_safe).strftime("%y"), "00")
        self.assertEqual(datetime(*self.percent_y_safe).strftime("%y"), "00")
        with self.assertRaisesMessage(
            TypeError, "strftime of dates before 1000 does not handle %y"
        ):
            datetime(*self.just_unsafe).strftime("%y")

        self.assertEqual(
            date(1850, 8, 2).strftime("%Y/%m/%d was a %A"), "1850/08/02 was a Friday"
        )

    def test_zero_padding(self):
        """
        Regression for #12524

        Pre-1000AD dates are padded with zeros if necessary
        """
        self.assertEqual(
            date(1, 1, 1).strftime("%Y/%m/%d was a %A"), "0001/01/01 was a Monday"
        )