Move get_datetime into own file.

Add test cases to cover get_datetime usage and edge cases.
This commit is contained in:
2025-04-24 09:27:05 -04:00
parent af3b8e9e69
commit def9c4aa79
4 changed files with 62 additions and 30 deletions
+31 -3
View File
@@ -4,10 +4,38 @@ import datetime
from freezegun import freeze_time
from dakiny_bot.utils import get_datetime
from dakiny_bot.election_utils import get_datetime
@freeze_time("2025-6-16")
def test_regular_get_datetime() -> None:
date_time = get_datetime("6/16")
assert date_time == datetime.datetime(2025, 6, 16)
"""
Test a regular get_datetime.
Expect the returned datetime to be the frozen date.
Have two different days to ensure that the date is being returned correctly.
"""
date_time = get_datetime("6/15")
assert date_time == datetime.datetime(2025, 6, 15, tzinfo=datetime.UTC)
@freeze_time("2025-1-16")
def test_jan_get_datetime() -> None:
"""
Test previous year get_datetime.
Frozen day is in January 2025, datetime string is in December. Expect datetime to be from 2024.
"""
date_time = get_datetime("12/15")
assert date_time == datetime.datetime(2024, 12, 15, tzinfo=datetime.UTC)
@freeze_time("2025-12-16")
def test_dec_get_datetime() -> None:
"""
Test next year get_datetime.
Frozen day is in December 2025, datetime string is in January. Expect datetime to be from 2026.
"""
date_time = get_datetime("1/15")
assert date_time == datetime.datetime(2026, 1, 15, tzinfo=datetime.UTC)