icloud_photos_downloader/tests/test_string_helpers.py at 565b74621b507382d01ed1a41294a5f8b6407cea · AndreyNikiforov/icloud_photos_downloader · GitHub

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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
import datetime
import random
import string
import sys
from unittest import TestCase

import pytest

from icloudpd.string_helpers import parse_timestamp_or_timedelta, truncate_middle


class TruncateMiddleTestCase(TestCase):
    def test_truncate_middle(self) -> None:
        assert truncate_middle("test_filename.jpg", 50) == "test_filename.jpg"
        assert truncate_middle("test_filename.jpg", 17) == "test_filename.jpg"
        assert truncate_middle("test_filename.jpg", 16) == "test_f...me.jpg"
        assert truncate_middle("test_filename.jpg", 10) == "tes...jpg"
        assert truncate_middle("test_filename.jpg", 5) == "t...g"
        assert truncate_middle("test_filename.jpg", 4) == "...g"
        assert truncate_middle("test_filename.jpg", 3) == "..."
        assert truncate_middle("test_filename.jpg", 2) == ".."
        assert truncate_middle("test_filename.jpg", 1) == "."
        assert truncate_middle("test_filename.jpg", 0) == ""
        with self.assertRaises(ValueError):
            truncate_middle("test_filename.jpg", -1)


class ParseTimestampeOrTimeDeltaTestCase(TestCase):
    def test_totality(self) -> None:
        characters = string.ascii_letters + string.digits
        for _case in range(500):
            target_length = random.randint(0, 100)
            target_string = "".join(random.choice(characters) for i in range(target_length))
            _result = parse_timestamp_or_timedelta(target_string)
            #  not throwing is okay

    def test_naive(self) -> None:
        assert parse_timestamp_or_timedelta("2025-01-02T03:04:05.000600") == datetime.datetime(
            2025, 1, 2, 3, 4, 5, 600
        )
        assert parse_timestamp_or_timedelta("2025-01-02T03:04:05.006") == datetime.datetime(
            2025, 1, 2, 3, 4, 5, 6000
        )
        assert parse_timestamp_or_timedelta("2025-01-02") == datetime.datetime(
            2025, 1, 2, 0, 0, 0, 0
        )

    @pytest.mark.skipif(
        sys.version_info < (3, 11), reason="Requires Python 3.11 or higher for full parsing support"
    )
    def test_naive_311plus(self) -> None:
        #  short
        assert parse_timestamp_or_timedelta("2025-01-02T03:04:05.0006") == datetime.datetime(
            2025, 1, 2, 3, 4, 5, 600
        )

    def test_aware(self) -> None:
        assert parse_timestamp_or_timedelta(
            "2025-01-02T03:04:05.000600+08:00"
        ) == datetime.datetime(
            2025, 1, 2, 3, 4, 5, 600, datetime.timezone(datetime.timedelta(hours=8))
        )
        assert parse_timestamp_or_timedelta("2025-01-02T03:04:05.006+08:00") == datetime.datetime(
            2025, 1, 2, 3, 4, 5, 6000, datetime.timezone(datetime.timedelta(hours=8))
        )

    @pytest.mark.skipif(
        sys.version_info < (3, 11), reason="Requires Python 3.11 or higher for full parsing support"
    )
    def test_aware_311plus(self) -> None:
        #  short
        assert parse_timestamp_or_timedelta("2025-01-02T03:04:05.0006Z") == datetime.datetime(
            2025, 1, 2, 3, 4, 5, 600, datetime.timezone.utc
        )
        assert parse_timestamp_or_timedelta("2025-01-02T03:04:05.0006+08:00") == datetime.datetime(
            2025, 1, 2, 3, 4, 5, 600, datetime.timezone(datetime.timedelta(hours=8))
        )

    def test_timestamp_invalid(self) -> None:
        assert parse_timestamp_or_timedelta("2025-01") is None
        assert parse_timestamp_or_timedelta("") is None

    def test_delta_totality(self) -> None:
        digits = string.digits
        characters = ["d", "D"]
        for _case in range(500):
            target_length = random.randint(1, 5)
            target_string = "".join(
                random.choice(digits) for i in range(target_length)
            ) + random.choice(characters)
            result = parse_timestamp_or_timedelta(target_string)
            assert isinstance(result, datetime.timedelta)

    def test_delta(self) -> None:
        assert parse_timestamp_or_timedelta("1d") == datetime.timedelta(days=1)
        assert parse_timestamp_or_timedelta("0d") == datetime.timedelta(days=0)

    def test_delta_invalid(self) -> None:
        assert parse_timestamp_or_timedelta("-1d") is None
        assert parse_timestamp_or_timedelta("") is None