mirror/pyTelegramBotAPI
1
0
Fork 0
mirror of https://github.com/eternnoir/pyTelegramBotAPI.git synced 2023-08-10 21:12:57 +03:00
Code Releases Activity

Refactor tests

Browse Source
This commit is contained in:
bedilbek 2020-04-13 03:21:10 +05:00
parent 3aec66bc0d
commit 0881e34381
1 changed files with 13 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
tests/test_handler_backends.py
View File

@ -1,3 +1,7 @@
import sys
sys.path.append('../')
import os import os
import time import time
@ -155,12 +159,8 @@ def test_memory_handler_backend_clear_reply_handler(telegram_bot, private_chat,
assert reply_to_message_update_type.message.text == 'entered start' assert reply_to_message_update_type.message.text == 'entered start'
def test_file_handler_backend_register_next_step_handler(private_chat, update_type): def test_file_handler_backend_register_next_step_handler(telegram_bot, private_chat, update_type):
telegram_bot = telebot.TeleBot( telegram_bot.next_step_backend=FileHandlerBackend(filename='./.handler-saves/step1.save', delay=0.1)
token='',
threaded=False,
next_step_backend=FileHandlerBackend(filename='./.handler-saves/step1.save', delay=1)
)
@telegram_bot.message_handler(commands=['start']) @telegram_bot.message_handler(commands=['start'])
def start(message): def start(message):
@ -170,7 +170,7 @@ def test_file_handler_backend_register_next_step_handler(private_chat, update_ty
telegram_bot.process_new_updates([update_type]) telegram_bot.process_new_updates([update_type])
assert update_type.message.text == 'entered start' assert update_type.message.text == 'entered start'
time.sleep(2) time.sleep(0.2)
assert os.path.exists(telegram_bot.next_step_backend.filename) assert os.path.exists(telegram_bot.next_step_backend.filename)
@ -187,17 +187,13 @@ def test_file_handler_backend_register_next_step_handler(private_chat, update_ty
assert private_chat.id not in telegram_bot.next_step_backend.handlers assert private_chat.id not in telegram_bot.next_step_backend.handlers
time.sleep(2) time.sleep(0.2)
if os.path.exists(telegram_bot.next_step_backend.filename): if os.path.exists(telegram_bot.next_step_backend.filename):
os.remove(telegram_bot.next_step_backend.filename) os.remove(telegram_bot.next_step_backend.filename)
def test_file_handler_backend_clear_next_step_handler(private_chat, update_type): def test_file_handler_backend_clear_next_step_handler(telegram_bot, private_chat, update_type):
telegram_bot = telebot.TeleBot( telegram_bot.next_step_backend=FileHandlerBackend(filename='./.handler-saves/step2.save', delay=0.1)
token='',
threaded=False,
next_step_backend=FileHandlerBackend(filename='./.handler-saves/step2.save', delay=1)
)
@telegram_bot.message_handler(commands=['start']) @telegram_bot.message_handler(commands=['start'])
def start(message): def start(message):
@ -209,13 +205,13 @@ def test_file_handler_backend_clear_next_step_handler(private_chat, update_type)
assert len(telegram_bot.next_step_backend.handlers[private_chat.id]) == 1 assert len(telegram_bot.next_step_backend.handlers[private_chat.id]) == 1
time.sleep(2) time.sleep(0.2)
assert os.path.exists(telegram_bot.next_step_backend.filename) assert os.path.exists(telegram_bot.next_step_backend.filename)
telegram_bot.clear_step_handler_by_chat_id(private_chat.id) telegram_bot.clear_step_handler_by_chat_id(private_chat.id)
time.sleep(2) time.sleep(0.2)
telegram_bot.next_step_backend.load_handlers() telegram_bot.next_step_backend.load_handlers()
@ -224,7 +220,7 @@ def test_file_handler_backend_clear_next_step_handler(private_chat, update_type)
telegram_bot.process_new_updates([update_type]) telegram_bot.process_new_updates([update_type])
assert update_type.message.text == 'entered start' assert update_type.message.text == 'entered start'
time.sleep(2) time.sleep(0.2)
if os.path.exists(telegram_bot.next_step_backend.filename): if os.path.exists(telegram_bot.next_step_backend.filename):
os.remove(telegram_bot.next_step_backend.filename) os.remove(telegram_bot.next_step_backend.filename)