minor: add fake reminder to test

This commit is contained in:
kotolex
2023-12-22 22:46:20 +05:00
parent f874f4754d
commit af606017c4
+7
View File
@@ -8,6 +8,7 @@ from telebot import asyncio_helper, types, logger
os.environ["LUTZPYBOT"] = "00000:AAAAAAAAAAAA"
os.environ["whiteids"] = "100,200,300"
os.environ["rollback"] = "1,2,3"
os.environ["use_reminder"] = "FALSE"
from src import helpers
@@ -45,6 +46,7 @@ def get_update(text, reply_to=None):
chosen_inline_result, callback_query, shipping_query, pre_checkout_query, poll, poll_answer,
my_chat_member, chat_member, chat_join_request)
# сюда будут складываться результаты, перед началом теста очищается
RESULTS = []
@@ -54,15 +56,20 @@ async def custom_sender(token, url, method='get', params=None, files=None, **kwa
result = {"message_id": 1, "date": 1, "chat": {"id": 1, "type": "private"}}
return result
# Чтобы перехватить запросы к телеге
asyncio_helper._process_request = custom_sender
class TestBot(IsolatedAsyncioTestCase):
def setUp(self) -> None:
self.bot = bot
RESULTS.clear()
async def test_start(self):
await self.bot.process_new_updates([get_update('/start')])
self.assertEqual(RESULTS[0]['text'], "Начни с прочтения")