Тестовый обработчик для бота telegram - PullRequest
Новая
       5

Тестовый обработчик для бота telegram

0 голосов
/ 17 октября 2018

Мне нужно написать unittest для этого кода, это обработчик для бота telegram.Как написать тест с использованием макета?Похоже, я должен сделать поддельное сообщение в функцию, поддельный метод bot.send_message, но я не знаю, как это сделать правильно.Я буду счастлив, если кто-нибудь поможет мне, спасибо. 

@bot.message_handler(commands=['start'])
def cmd_start(message):
    print(message)
    chat_id = message.chat.id
    first_name = message.chat.first_name
    last_name = message.chat.last_name
    username = message.chat.username
    conn = psycopg2.connect(db_conf)
    cursor = conn.cursor()

    cursor.execute('INSERT INTO telegram_users (first_name, last_name,         username, id, days_to_notice)'
                   ' VALUES (%s, %s, %s, %s, %s)'
                   ' ON CONFLICT DO NOTHING',
                   (first_name, last_name, username, chat_id, 7))
    conn.commit()
    conn.close()
    bot.send_message(message.chat.id, start_text)
...