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", "/stop", "/start"], True), ], ) def test_chat_is_active(messages, active): assert chat_is_active(messages) == active @pytest.mark.parametrize( ("messages", "file_path"), [ (["/file /home/user/test"], "/home/user/test"), (["/file /home/user/test", "/stop"], "/home/user/test"), (["I stored your reminders at the file /home/user/test"], "/home/user/test"), (["/start", "/stop", "/start"], None), ], ) def test_get_file_path(messages: [], file_path: str): assert get_file_path(messages) == file_path