remember-remember-bot/tests/test_util.py

from remember_remember_bot.util import (
    check_new_day,
    update_day,
    chat_is_active,
    get_file_path,
)
import pytest


def test_update_day():
    current_day = 0
    assert check_new_day(current_day)
    current_day = update_day()
    assert not check_new_day(current_day)


@pytest.mark.parametrize(
    ("messages", "active"),
    [
        (["/start"], True),
        (["/start", "/stop"], False),
        (["/start", "/stop", "Let's get this party started"], False),
        (
            [
                "/start",
                "Message deletion timer is set to 1 day by",
                "Message deletion timer is disabled by",
                "I will stop sending you messages now.",
            ],
            False,
        ),
        (["/start", "/stop", "/start"], True),
    ],
)
def test_chat_is_active(messages, active):
    assert chat_is_active(messages) == active


@pytest.mark.parametrize(
    "messages, tmp_file_path",
    [
        (["/file %file_path%"], "test1"),
        (["/file %file_path%", "/stop"], "test2"),
        (["I stored your reminders at the file %file_path%"], "test3"),
        (["/start", "/stop", "/start"], None),
    ],
    indirect=["tmp_file_path"],
)
def test_get_file_path(messages: [], tmp_file_path: str):
    if tmp_file_path:
        for listitem in range(len(messages)):
            messages[listitem] = messages[listitem].replace(
                "%file_path%", tmp_file_path
            )
    assert get_file_path(messages) == tmp_file_path