电报机器人的测试处理程序

时间:2018-10-17 14:26:53

标签: python-3.x mocking telegram-bot

我需要为此代码编写一个单元测试,这是电报机器人的处理程序。如何使用模拟写测试?看起来我应该将虚假消息转换为功能,即虚假方法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)

0 个答案:

没有答案