Activity Reminder Bot

from datetime import datetime

from activity_reminder_bot.models import ReminderRepeat
from activity_reminder_bot.services.repeat import next_repeat_due_at


def test_daily_repeat_next_due_at() -> None:
    assert next_repeat_due_at(datetime(2026, 6, 10, 8, 0), ReminderRepeat.daily) == datetime(2026, 6, 11, 8, 0)


def test_weekly_repeat_next_due_at() -> None:
    assert next_repeat_due_at(datetime(2026, 6, 10, 8, 0), ReminderRepeat.weekly) == datetime(2026, 6, 17, 8, 0)


def test_no_repeat_has_no_next_due_at() -> None:
    assert next_repeat_due_at(datetime(2026, 6, 10, 8, 0), ReminderRepeat.none) is None