Refactor calculate function and move it to main.py; remove misc.py and update tests

tests/test_main.py

@@ -1,6 +1,97 @@
 from __future__ import annotations
 
+from datetime import datetime, timezone
+from typing import TYPE_CHECKING
+
+from apscheduler.job import Job
+from apscheduler.schedulers.background import BackgroundScheduler
+from apscheduler.triggers.cron import CronTrigger
+from apscheduler.triggers.date import DateTrigger
+from apscheduler.triggers.interval import IntervalTrigger
+
 from discord_reminder_bot import main
+from discord_reminder_bot.main import calculate
+
+if TYPE_CHECKING:
+    from apscheduler.job import Job
+
+
+def dummy_job() -> None:
+    """Dummy job function for testing."""
+
+
+def test_calculate() -> None:
+    """Test the calculate function with various job inputs."""
+    scheduler = BackgroundScheduler()
+    scheduler.timezone = timezone.utc
+    scheduler.start()
+
+    # Create a job with a DateTrigger
+    run_date = datetime(2270, 10, 1, 12, 0, 0, tzinfo=scheduler.timezone)
+    job: Job = scheduler.add_job(dummy_job, trigger=DateTrigger(run_date=run_date), id="test_job", name="Test Job")
+
+    expected_output = "<t:9490737600:R>"
+    assert_msg: str = f"Expected {expected_output}, got {calculate(job)}\nState:{job.__getstate__()}"
+    assert calculate(job) == expected_output, assert_msg
+
+    # Modify the job to have a next_run_time
+    job.modify(next_run_time=run_date)
+    assert_msg: str = f"Expected {expected_output}, got {calculate(job)}\nState:{job.__getstate__()}"
+    assert calculate(job) == expected_output, assert_msg
+
+    # Paused job should return "Paused"
+    job.pause()
+    assert_msg: str = f"Expected 'Paused', got {calculate(job)}\nState:{job.__getstate__()}"
+    assert calculate(job) == "Paused", assert_msg
+
+    scheduler.shutdown()
+
+
+def test_calculate_cronjob() -> None:
+    """Test the calculate function with a CronTrigger job."""
+    scheduler = BackgroundScheduler()
+    scheduler.start()
+
+    run_date = datetime(2270, 10, 1, 12, 0, 0, tzinfo=scheduler.timezone)
+    job: Job = scheduler.add_job(
+        dummy_job,
+        trigger=CronTrigger(
+            second=run_date.second,
+            minute=run_date.minute,
+            hour=run_date.hour,
+            day=run_date.day,
+            month=run_date.month,
+            year=run_date.year,
+        ),
+    )
+    # Force next_run_time to expected value for testing
+    job.modify(next_run_time=run_date)
+
+    expected_output: str = f"<t:{int(run_date.timestamp())}:R>"
+    assert calculate(job) == expected_output, f"Expected {expected_output}, got {calculate(job)}\nState:{job.__getstate__()}"
+
+    job.pause()
+    assert calculate(job) == "Paused", f"Expected Paused, got {calculate(job)}\nState:{job.__getstate__()}"
+    scheduler.shutdown()
+
+
+def test_calculate_intervaljob() -> None:
+    """Test the calculate function with an IntervalTrigger job."""
+    scheduler = BackgroundScheduler()
+    scheduler.start()
+
+    run_date = datetime(2270, 12, 31, 23, 59, 59, tzinfo=scheduler.timezone)
+    job = scheduler.add_job(dummy_job, trigger=IntervalTrigger(seconds=3600), id="test_interval_job", name="Test Interval Job")
+    # Force next_run_time to expected value for testing
+    job.modify(next_run_time=run_date)
+
+    expected_output = f"<t:{int(run_date.timestamp())}:R>"
+    assert calculate(job) == expected_output, f"Expected {expected_output}, got {calculate(job)}\nState:{job.__getstate__()}"
+
+    # Paused job should return "Paused"
+    job.pause()
+    assert calculate(job) == "Paused", f"Expected Paused, got {calculate(job)}\nState:{job.__getstate__()}"
+    scheduler.shutdown()
 
 
 def test_if_send_to_discord_is_in_main() -> None: