Add some test.

This commit is contained in:
eternnoir 2015-07-15 10:52:32 +08:00
parent 6c8d30b063
commit 56d1dfc045
1 changed files with 39 additions and 0 deletions

39
tests/test_telebot.py Normal file
View 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)