aboutsummaryrefslogtreecommitdiffstats
path: root/tests/test_utils.py
blob: 87c8c18697ddf8230dac38be7850363b761452e4 (plain) (blame)
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
from datetime import datetime, timedelta, timezone

from tests import SiteTest
from pysite.utils.time import is_expired, parse_duration


class DurationParsingTests(SiteTest):
    """Tests the `parse_duration` method provided by `pysite.utils.time`."""

    SIMPLE_DURATION_STRINGS = (
        ('42s', timedelta(seconds=42)),
        ('12m', timedelta(minutes=12)),
        ('20h', timedelta(hours=20)),
        ('7d', timedelta(days=7)),
        ('2w', timedelta(weeks=2))
    )
    COMBINED_DURATION_STRINGS = (
        ('12m30s', timedelta(minutes=12, seconds=30)),
        ('20h5m', timedelta(hours=20, minutes=5)),
        ('7d10h12s', timedelta(days=7, hours=10, seconds=12))
    )

    def test_simple_duration_string_parsing(self):
        for duration_string, added_delta in self.SIMPLE_DURATION_STRINGS:
            timezone_aware_now = datetime.now(timezone.utc)
            self.assertAlmostEqual(
                parse_duration(duration_string).timestamp(),
                (timezone_aware_now + added_delta).timestamp(),
                places=-1  # Being off by < 10 seconds is acceptable
            )

    def test_combined_duration_string_parsing(self):
        for duration_string, added_delta in self.COMBINED_DURATION_STRINGS:
            timezone_aware_now = datetime.now(timezone.utc)
            self.assertAlmostEqual(
                parse_duration(duration_string).timestamp(),
                (timezone_aware_now + added_delta).timestamp(),
                places=-1  # Being off by < 10 seconds is acceptable
            )

    def test_empty_duration_raises_valueerror(self):
        with self.assertRaises(ValueError):
            parse_duration('')

    def test_unknown_char_raises_valueerror(self):
        with self.assertRaises(ValueError):
            parse_duration('12l')

    def test_valid_unit_without_digits_raises_valueerror(self):
        with self.assertRaises(ValueError):
            parse_duration('s')


class ExpiryTests(SiteTest):
    """Tests the `is_expired` method provided by `pysite.utils.time`."""

    EXPIRY_DELTAS = (
        timedelta(seconds=30),
        timedelta(minutes=12),
        timedelta(hours=20),
        timedelta(days=5),
        timedelta(weeks=7)
    )

    def test_datetimes_in_the_past_are_expired(self):
        for delta in self.EXPIRY_DELTAS:
            date = datetime.now(timezone.utc) - delta
            self.assertTrue(is_expired(date))

    def test_datetimes_in_the_future_are_not_expired(self):
        for delta in self.EXPIRY_DELTAS:
            date = datetime.now(timezone.utc) + delta
            self.assertFalse(is_expired(date))