Fix tests for the new handler

eternnoir · Dec 29, 2023 · 8fee136 · 8fee136
1 parent 974395e
commit 8fee136
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/tests/test_handler_backends.py b/tests/test_handler_backends.py
@@ -65,9 +65,10 @@ def update_type(message):
     my_chat_member = None
     chat_member = None
     chat_join_request = None
+    message_reaction = None
     return types.Update(1001234038283, message, edited_message, channel_post, edited_channel_post, inline_query,
                         chosen_inline_result, callback_query, shipping_query, pre_checkout_query, poll, poll_answer,
-                        my_chat_member, chat_member, chat_join_request)
+                        my_chat_member, chat_member, chat_join_request, message_reaction)
 
 
 @pytest.fixture()
@@ -85,9 +86,10 @@ def reply_to_message_update_type(reply_to_message):
     my_chat_member = None
     chat_member = None
     chat_join_request = None
+    message_reaction = None
     return types.Update(1001234038284, reply_to_message, edited_message, channel_post, edited_channel_post,
                         inline_query, chosen_inline_result, callback_query, shipping_query, pre_checkout_query, 
-                        poll, poll_answer, my_chat_member, chat_member, chat_join_request)
+                        poll, poll_answer, my_chat_member, chat_member, chat_join_request, message_reaction)
 
 
 def next_handler(message):

diff --git a/tests/test_telebot.py b/tests/test_telebot.py
@@ -542,9 +542,10 @@ def create_message_update(text):
         my_chat_member = None
         chat_member = None
         chat_join_request = None
+        message_reaction = None
         return types.Update(-1001234038283, message, edited_message, channel_post, edited_channel_post, inline_query,
                             chosen_inline_result, callback_query, shipping_query, pre_checkout_query, poll, poll_answer,
-                            my_chat_member, chat_member, chat_join_request)
+                            my_chat_member, chat_member, chat_join_request, message_reaction)
 
     def test_is_string_unicode(self):
         s1 = u'string'