fix: add support init fields from parent object in KeyboardButton (#344)

* fix: add support init fields from parent object in KeyboardButton * fix: add tests
aiogram · May 31, 2020 · 7076711 · 7076711
1 parent cb0a898
commit 7076711
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 2 deletions.
diff --git a/aiogram/types/reply_keyboard.py b/aiogram/types/reply_keyboard.py
@@ -111,11 +111,13 @@ class KeyboardButton(base.TelegramObject):
     def __init__(self, text: base.String,
                  request_contact: base.Boolean = None,
                  request_location: base.Boolean = None,
-                 request_poll: KeyboardButtonPollType = None):
+                 request_poll: KeyboardButtonPollType = None,
+                 **kwargs):
         super(KeyboardButton, self).__init__(text=text,
                                              request_contact=request_contact,
                                              request_location=request_location,
-                                             request_poll=request_poll)
+                                             request_poll=request_poll,
+                                             **kwargs)
 
 
 class ReplyKeyboardRemove(base.TelegramObject):

diff --git a/tests/types/dataset.py b/tests/types/dataset.py
@@ -457,3 +457,8 @@
     "has_custom_certificate": False,
     "pending_update_count": 0,
 }
+
+REPLY_KEYBOARD_MARKUP = {
+    "keyboard": [[{"text": "something here"}]],
+    "resize_keyboard": True,
+}
diff --git a/tests/types/test_reply_keyboard.py b/tests/types/test_reply_keyboard.py
@@ -0,0 +1,12 @@
+from aiogram import types
+from .dataset import REPLY_KEYBOARD_MARKUP
+
+reply_keyboard = types.ReplyKeyboardMarkup(**REPLY_KEYBOARD_MARKUP)
+
+
+def test_serialize():
+    assert reply_keyboard.to_python() == REPLY_KEYBOARD_MARKUP
+
+
+def test_deserialize():
+    assert reply_keyboard.to_object(reply_keyboard.to_python()) == reply_keyboard