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