Hopefully fix tests

python-telegram-bot · May 9, 2024 · b5d087a · b5d087a
1 parent 2704ee2
commit b5d087a
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 4 deletions.
diff --git a/telegram/_bot.py b/telegram/_bot.py
@@ -6815,7 +6815,7 @@ async def send_poll(
         self,
         chat_id: Union[int, str],
         question: str,
-        options: Sequence[Union[str, InputPollOption]],
+        options: Sequence[Union[str, "InputPollOption"]],
         is_anonymous: Optional[bool] = None,
         type: Optional[str] = None,  # pylint: disable=redefined-builtin
         allows_multiple_answers: Optional[bool] = None,

diff --git a/telegram/_chat.py b/telegram/_chat.py
@@ -57,6 +57,7 @@
         InputMediaDocument,
         InputMediaPhoto,
         InputMediaVideo,
+        InputPollOption,
         LabeledPrice,
         LinkPreviewOptions,
         Location,
@@ -2545,7 +2546,7 @@ async def send_voice(
     async def send_poll(
         self,
         question: str,
-        options: Sequence[str],
+        options: Sequence[Union[str, "InputPollOption"]],
         is_anonymous: Optional[bool] = None,
         type: Optional[str] = None,
         allows_multiple_answers: Optional[bool] = None,

diff --git a/telegram/_user.py b/telegram/_user.py
@@ -40,6 +40,7 @@
         InputMediaDocument,
         InputMediaPhoto,
         InputMediaVideo,
+        InputPollOption,
         LabeledPrice,
         LinkPreviewOptions,
         Location,
@@ -1482,7 +1483,7 @@ async def send_voice(
     async def send_poll(
         self,
         question: str,
-        options: Sequence[str],
+        options: Sequence[Union[str, "InputPollOption"]],
         is_anonymous: Optional[bool] = None,
         type: Optional[str] = None,
         allows_multiple_answers: Optional[bool] = None,

diff --git a/telegram/ext/_extbot.py b/telegram/ext/_extbot.py
@@ -2933,7 +2933,7 @@ async def send_poll(
         self,
         chat_id: Union[int, str],
         question: str,
-        options: Sequence[Union[str, InputPollOption]],
+        options: Sequence[Union[str, "InputPollOption"]],
         is_anonymous: Optional[bool] = None,
         type: Optional[str] = None,  # pylint: disable=redefined-builtin
         allows_multiple_answers: Optional[bool] = None,

diff --git a/tests/auxil/bot_method_checks.py b/tests/auxil/bot_method_checks.py
@@ -314,6 +314,8 @@ def build_kwargs(
             elif name == "ok":
                 kws["ok"] = False
                 kws["error_message"] = "error"
+            elif name == "options":
+                kws[name] = ["option1", "option2"]
             else:
                 kws[name] = True