mirror of
https://github.com/eternnoir/pyTelegramBotAPI.git
synced 2023-08-10 21:12:57 +03:00
Add some test.
This commit is contained in:
parent
6c8d30b063
commit
56d1dfc045
39
tests/test_telebot.py
Normal file
39
tests/test_telebot.py
Normal file
@ -0,0 +1,39 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
import sys
|
||||||
|
import time
|
||||||
|
|
||||||
|
sys.path.append('../')
|
||||||
|
from telebot import types
|
||||||
|
import telebot
|
||||||
|
|
||||||
|
|
||||||
|
def test_message_listener():
|
||||||
|
msg_list = []
|
||||||
|
for x in range(100):
|
||||||
|
msg_list.append(create_text_message('Message ' + str(x)))
|
||||||
|
|
||||||
|
def listener(messages):
|
||||||
|
assert len(messages) == 100
|
||||||
|
|
||||||
|
tb = telebot.TeleBot('')
|
||||||
|
tb.set_update_listener(listener)
|
||||||
|
|
||||||
|
|
||||||
|
def test_message_handler():
|
||||||
|
tb = telebot.TeleBot('')
|
||||||
|
msg = create_text_message('/help')
|
||||||
|
|
||||||
|
@tb.message_handler(commands=['help', 'start'])
|
||||||
|
def command_handler(message):
|
||||||
|
message.text = 'got'
|
||||||
|
|
||||||
|
tb.process_new_messages([msg])
|
||||||
|
time.sleep(1)
|
||||||
|
assert msg.text == 'got'
|
||||||
|
|
||||||
|
|
||||||
|
def create_text_message(text):
|
||||||
|
params = {}
|
||||||
|
params['text'] = text
|
||||||
|
return types.Message(1, None, None, 1, 'text', params)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user