TheLovinator/discord-reminder-bot
1
Fork 0
Code Issues Pull Requests Actions Packages Releases Activity

Reduce the amount of files and functions

Browse Source
This commit is contained in:
Joakim Hellsén
2025-02-24 13:52:00 +01:00
parent 2806e6044d
commit 4aa0bd7837
5 changed files with 59 additions and 293 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
tests/test_misc.py
View File

@ -9,35 +9,12 @@ from apscheduler.triggers.cron import CronTrigger
from apscheduler.triggers.date import DateTrigger
from apscheduler.triggers.interval import IntervalTrigger
from discord_reminder_bot.misc import calc_time, calculate, get_human_time
from discord_reminder_bot.misc import calculate, get_human_time
if TYPE_CHECKING:
from apscheduler.job import Job
def test_calc_time() -> None:
"""Test the calc_time function with various datetime inputs."""
test_datetime: datetime = datetime(2023, 10, 1, 12, 0, 0, tzinfo=timezone.utc)
expected_timestamp: str = f"<t:{int(test_datetime.timestamp())}:R>"
assert_msg = f"Expected {expected_timestamp}, got {calc_time(test_datetime)}"
assert calc_time(test_datetime) == expected_timestamp, assert_msg
now: datetime = datetime.now(tz=timezone.utc)
expected_timestamp_now: str = f"<t:{int(now.timestamp())}:R>"
assert_msg = f"Expected {expected_timestamp_now}, got {calc_time(now)}"
assert calc_time(now) == expected_timestamp_now, assert_msg
past_datetime: datetime = datetime(2000, 1, 1, 0, 0, 0, tzinfo=timezone.utc)
expected_timestamp_past: str = f"<t:{int(past_datetime.timestamp())}:R>"
assert_msg = f"Expected {expected_timestamp_past}, got {calc_time(past_datetime)}"
assert calc_time(past_datetime) == expected_timestamp_past, assert_msg
future_datetime: datetime = datetime(2100, 1, 1, 0, 0, 0, tzinfo=timezone.utc)
expected_timestamp_future: str = f"<t:{int(future_datetime.timestamp())}:R>"
assert_msg: str = f"Expected {expected_timestamp_future}, got {calc_time(future_datetime)}"
assert calc_time(future_datetime) == expected_timestamp_future, assert_msg
def test_get_human_time() -> None:
"""Test the get_human_time function with various timedelta inputs."""
test_timedelta = timedelta(days=1, hours=2, minutes=3, seconds=4)

86
tests/test_settings.py
View File

@ -1,86 +0,0 @@
from __future__ import annotations
import pytest
from apscheduler.jobstores.sqlalchemy import SQLAlchemyJobStore
from apscheduler.schedulers.asyncio import AsyncIOScheduler
from discord_reminder_bot.settings import get_settings
def test_get_settings(monkeypatch: pytest.MonkeyPatch) -> None:
"""Test get_settings function with environment variables."""
monkeypatch.setenv("SQLITE_LOCATION", "/test_jobs.sqlite")
monkeypatch.setenv("TIMEZONE", "UTC")
monkeypatch.setenv("BOT_TOKEN", "test_token")
monkeypatch.setenv("LOG_LEVEL", "DEBUG")
monkeypatch.setenv("WEBHOOK_URL", "http://test_webhook_url")
settings: dict[str, str | dict[str, SQLAlchemyJobStore] | dict[str, bool] | AsyncIOScheduler] = get_settings(use_dotenv=False)
assert_msg = f"Expected /test_jobs.sqlite, got {settings['sqlite_location']}"
assert settings["sqlite_location"] == "/test_jobs.sqlite", assert_msg
assert_msg = f"Expected UTC, got {settings['config_timezone']}"
assert settings["config_timezone"] == "UTC", assert_msg
assert_msg = f"Expected test_token, got {settings['bot_token']}"
assert settings["bot_token"] == "test_token", assert_msg # noqa: S105
assert_msg = f"Expected DEBUG, got {settings['log_level']}"
assert settings["log_level"] == "DEBUG", assert_msg
assert_msg = f"Expected http://test_webhook_url, got {settings['webhook_url']}"
assert settings["webhook_url"] == "http://test_webhook_url", assert_msg
assert_msg = f"Expected dict, got {type(settings['jobstores'])}"
assert isinstance(settings["jobstores"], dict), assert_msg
assert_msg: str = f"Expected SQLAlchemyJobStore, got {type(settings['jobstores']['default'])}"
assert isinstance(settings["jobstores"]["default"], SQLAlchemyJobStore), assert_msg
assert_msg = f"Expected AsyncIOScheduler, got {type(settings['scheduler'])}"
assert isinstance(settings["scheduler"], AsyncIOScheduler), assert_msg
def test_get_settings_missing_bot_token(monkeypatch: pytest.MonkeyPatch) -> None:
"""Test get_settings function with missing bot token."""
monkeypatch.delenv("BOT_TOKEN", raising=False)
with pytest.raises(ValueError, match="Missing bot token"):
get_settings(use_dotenv=False)
def test_get_settings_default_values(monkeypatch: pytest.MonkeyPatch) -> None:
"""Test get_settings function with default values."""
monkeypatch.delenv("SQLITE_LOCATION", raising=False)
monkeypatch.delenv("TIMEZONE", raising=False)
monkeypatch.delenv("BOT_TOKEN", raising=False)
monkeypatch.delenv("LOG_LEVEL", raising=False)
monkeypatch.delenv("WEBHOOK_URL", raising=False)
monkeypatch.setenv("BOT_TOKEN", "default_token")
monkeypatch.setenv("TIMEZONE", "UTC")
settings: dict[str, str | dict[str, SQLAlchemyJobStore] | dict[str, bool] | AsyncIOScheduler] = get_settings(use_dotenv=False)
assert_msg: str = f"Expected /jobs.sqlite, got {settings['sqlite_location']}"
assert settings["sqlite_location"] == "/jobs.sqlite", assert_msg
assert_msg = f"Expected UTC, got {settings['config_timezone']}"
assert settings["config_timezone"] == "UTC", assert_msg
assert_msg = f"Expected default_token, got {settings['bot_token']}"
assert settings["bot_token"] == "default_token", assert_msg # noqa: S105
assert_msg = f"Expected INFO, got {settings['log_level']}"
assert settings["log_level"] == "INFO", assert_msg
assert_msg = f"Expected empty string, got {settings['webhook_url']}"
assert not settings["webhook_url"], assert_msg
assert_msg = f"Expected dict, got {type(settings['jobstores'])}"
assert isinstance(settings["jobstores"], dict), assert_msg
assert_msg = f"Expected SQLAlchemyJobStore, got {type(settings['jobstores']['default'])}"
assert isinstance(settings["jobstores"]["default"], SQLAlchemyJobStore), assert_msg
assert_msg = f"Expected AsyncIOScheduler, got {type(settings['scheduler'])}"
assert isinstance(settings["scheduler"], AsyncIOScheduler), assert_msg