1"""Tests for `chives.dates`."""
2
3import json
4
5import pytest
6
7from chives.dates import date_matches_any_format, find_all_dates, reformat_date
8
9
10def test_find_all_dates() -> None:
11    """find_all_dates finds all the nested dates in a JSON object."""
12    json_value = json.loads("""{
13      "doc1": {"id": "1", "date_created": "2025-10-14T05:34:07+0000"},
14      "shapes": [
15        {"color": "blue", "date_saved": "2015-03-01 23:34:39 +00:00"},
16        {"color": "yellow", "date_saved": "2013-9-21 13:43:00Z", "is_square": true},
17        {"color": "green", "date_saved": null}
18      ],
19      "date_verified": "2024-08-30"
20    }""")
21
22    assert list(find_all_dates(json_value)) == [
23        (
24            {"id": "1", "date_created": "2025-10-14T05:34:07+0000"},
25            "date_created",
26            "2025-10-14T05:34:07+0000",
27        ),
28        (
29            {"color": "blue", "date_saved": "2015-03-01 23:34:39 +00:00"},
30            "date_saved",
31            "2015-03-01 23:34:39 +00:00",
32        ),
33        (
34            {"color": "yellow", "date_saved": "2013-9-21 13:43:00Z", "is_square": True},
35            "date_saved",
36            "2013-9-21 13:43:00Z",
37        ),
38        (json_value, "date_verified", "2024-08-30"),
39    ]
40
41
42def test_date_matches_any_format() -> None:
43    """
44    Tests for `date_matches_any_format`.
45    """
46    assert date_matches_any_format(
47        "2001-01-01", formats=["%Y-%m-%d", "%Y-%m-%dT%H:%M:%S%z"]
48    )
49    assert not date_matches_any_format("2001-01-01", formats=["%Y-%m-%dT%H:%M:%S%z"])
50
51
52@pytest.mark.parametrize(
53    "s, orig_fmt, formatted_date",
54    [
55        ("2025-11-12T15:34:39.570Z", "%Y-%m-%dT%H:%M:%S.%fZ", "2025-11-12T15:34:39Z"),
56        ("2025-03-12 09:57:03", "%Y-%m-%d %H:%M:%S", "2025-03-12T09:57:03Z"),
57        ("2016-02-25 05:28:35 GMT", "%Y-%m-%d %H:%M:%S %Z", "2016-02-25T05:28:35Z"),
58        ("2011-12-06T10:45:15-08:00", "%Y-%m-%dT%H:%M:%S%z", "2011-12-06T18:45:15Z"),
59    ],
60)
61def test_reformat_date(s: str, orig_fmt: str, formatted_date: str) -> None:
62    """Tests for `reformat_date`."""
63    assert reformat_date(s, orig_fmt) == formatted_date