Refactor Tests for TelegramObject Classes with Subclasses (python-t…

…elegram-bot#4654)
octoleo · Feb 6, 2025 · f9f1533 · f9f1533
1 parent dfb0ae3
commit f9f1533
Show file tree

Hide file tree

Showing 8 changed files with 2,203 additions and 1,287 deletions.
diff --git a/telegram/_chatmember.py b/telegram/_chatmember.py
@@ -24,6 +24,7 @@
 from telegram import constants
 from telegram._telegramobject import TelegramObject
 from telegram._user import User
+from telegram._utils import enum
 from telegram._utils.argumentparsing import de_json_optional
 from telegram._utils.datetime import extract_tzinfo_from_defaults, from_timestamp
 from telegram._utils.types import JSONDict
@@ -99,7 +100,7 @@ def __init__(
         super().__init__(api_kwargs=api_kwargs)
         # Required by all subclasses
         self.user: User = user
-        self.status: str = status
+        self.status: str = enum.get_member(constants.ChatMemberStatus, status, status)
 
         self._id_attrs = (self.user, self.status)