58 lines
2.4 KiB
Python
58 lines
2.4 KiB
Python
from __future__ import annotations
|
|
|
|
import datetime
|
|
from zoneinfo import ZoneInfo
|
|
|
|
from freezegun import freeze_time
|
|
|
|
from discord_reminder_bot import settings
|
|
from discord_reminder_bot.parser import parse_time
|
|
|
|
|
|
def test_parse_time_valid_date() -> None:
|
|
"""Test the `parse_time` function with a valid date string."""
|
|
date_to_parse = "tomorrow at 5pm"
|
|
timezone = "UTC"
|
|
result: datetime.datetime | None = parse_time(date_to_parse, timezone, use_dotenv=False)
|
|
assert result is not None, f"Expected a datetime object, got {result}"
|
|
assert result.tzinfo == ZoneInfo(timezone), f"Expected timezone {timezone}, got {result.tzinfo}"
|
|
|
|
|
|
def test_parse_time_no_date() -> None:
|
|
"""Test the `parse_time` function with no date string."""
|
|
date_to_parse: str = ""
|
|
timezone = "UTC"
|
|
result: datetime.datetime | None = parse_time(date_to_parse, timezone, use_dotenv=False)
|
|
assert result is None, f"Expected None, got {result}"
|
|
|
|
|
|
def test_parse_time_no_timezone() -> None:
|
|
"""Test the `parse_time` function with no timezone."""
|
|
date_to_parse = "tomorrow at 5pm"
|
|
result: datetime.datetime | None = parse_time(date_to_parse, use_dotenv=False)
|
|
assert result is not None, f"Expected a datetime object, got {result}"
|
|
|
|
assert_msg: str = f"Expected timezone {settings.get_timezone(use_dotenv=False)}, got {result.tzinfo}"
|
|
assert result.tzinfo == ZoneInfo(settings.get_timezone(use_dotenv=False)), assert_msg
|
|
|
|
|
|
def test_parse_time_invalid_date() -> None:
|
|
"""Test the `parse_time` function with an invalid date string."""
|
|
date_to_parse = "invalid date"
|
|
timezone = "UTC"
|
|
result: datetime.datetime | None = parse_time(date_to_parse, timezone, use_dotenv=False)
|
|
assert result is None, f"Expected None, got {result}"
|
|
|
|
|
|
@freeze_time("2023-01-01 12:00:00")
|
|
def test_parse_time_invalid_timezone() -> None:
|
|
"""Test the `parse_time` function with an invalid timezone."""
|
|
date_to_parse = "tomorrow at 5pm"
|
|
timezone = "Invalid/Timezone"
|
|
result: datetime.datetime | None = parse_time(date_to_parse, timezone, use_dotenv=False)
|
|
assert result is not None, f"Expected a datetime object, got {result}"
|
|
assert result.tzinfo == ZoneInfo("UTC"), f"Expected timezone UTC, got {result.tzinfo}"
|
|
|
|
assert_msg: str = f"Expected {datetime.datetime(2023, 1, 2, 17, 0, tzinfo=ZoneInfo('UTC'))}, got {result}"
|
|
assert result == datetime.datetime(2023, 1, 2, 17, 0, tzinfo=ZoneInfo("UTC")), assert_msg
|