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