Rename test file
This commit is contained in:
55
tests/test_countdown.py
Normal file
55
tests/test_countdown.py
Normal file
@ -0,0 +1,55 @@
|
||||
"""Test discord-reminder-bot.
|
||||
|
||||
Jobs are stored in memory.
|
||||
"""
|
||||
import re
|
||||
|
||||
import dateparser
|
||||
import pytz
|
||||
from apscheduler.jobstores.sqlalchemy import SQLAlchemyJobStore
|
||||
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
||||
|
||||
from discord_reminder_bot.countdown import calculate
|
||||
from discord_reminder_bot.main import send_to_discord
|
||||
|
||||
|
||||
class TestCountdown:
|
||||
"""This tests everything.
|
||||
|
||||
This sets up sqlite database in memory, changes scheduler timezone
|
||||
to Europe/Stockholm and creates job that runs January 18 2040.
|
||||
"""
|
||||
|
||||
jobstores = {"default": SQLAlchemyJobStore(url="sqlite:///:memory")}
|
||||
job_defaults = {"coalesce": True}
|
||||
scheduler = AsyncIOScheduler(
|
||||
jobstores=jobstores,
|
||||
timezone=pytz.timezone("Europe/Stockholm"),
|
||||
job_defaults=job_defaults,
|
||||
)
|
||||
|
||||
parsed_date = dateparser.parse(
|
||||
"18 January 2040",
|
||||
settings={
|
||||
"PREFER_DATES_FROM": "future",
|
||||
"TO_TIMEZONE": "Europe/Stockholm",
|
||||
},
|
||||
)
|
||||
|
||||
run_date = parsed_date.strftime("%Y-%m-%d %H:%M:%S") # type: ignore
|
||||
job = scheduler.add_job(
|
||||
send_to_discord,
|
||||
run_date=run_date,
|
||||
kwargs={
|
||||
"channel_id": 865712621109772329,
|
||||
"message": "Running PyTest",
|
||||
"author_id": 126462229892694018,
|
||||
},
|
||||
)
|
||||
|
||||
def test_countdown(self):
|
||||
"""Check if calc_countdown returns days, hours and minutes."""
|
||||
# FIXME: This will break when there is 0 seconds/hours/days left
|
||||
pattern = re.compile(r"\d* (day|days), \d* (hour|hours). \d* (minute|minutes)")
|
||||
countdown = calculate(self.job)
|
||||
assert pattern.match(countdown)
|
Reference in New Issue
Block a user