test: fix expected types

Author: zimeg

tests/slack_sdk/models/test_blocks.py

@@ -31,7 +31,8 @@
     StaticSelectElement,
     VideoBlock,
 )
-from slack_sdk.models.blocks.basic_components import SlackFile
+from slack_sdk.models.blocks.basic_components import FeedbackButtonObject, SlackFile
+from slack_sdk.models.blocks.block_elements import FeedbackButtonsElement, IconButtonElement
 
 from . import STRING_3001_CHARS
 
@@ -557,17 +558,17 @@ def test_document(self):
         self.assertDictEqual(input, Block.parse(input).to_dict())
 
     def test_with_feedback_buttons(self):
-        feedback_buttons = FeedbackButtons(
+        feedback_buttons = FeedbackButtonsElement(
             action_id="feedback-action",
-            positive_button={"text": {"type": "plain_text", "text": "Good"}, "value": "positive"},
-            negative_button={"text": {"type": "plain_text", "text": "Bad"}, "value": "negative"},
+            positive_button=FeedbackButtonObject(text="Good", value="positive"),
+            negative_button=FeedbackButtonObject(text="Bad", value="negative"),
         )
         block = ContextActionsBlock(elements=[feedback_buttons])
         self.assertEqual(len(block.elements), 1)
         self.assertEqual(block.elements[0].type, "feedback_buttons")
 
     def test_with_icon_button(self):
-        icon_button = IconButton(
+        icon_button = IconButtonElement(
             action_id="icon-action", icon="star", text=PlainTextObject(text="Favorite"), value="favorite"
         )
         block = ContextActionsBlock(elements=[icon_button])

tests/slack_sdk/models/test_objects.py

@@ -365,13 +365,13 @@ def test_deny_length(self):
 class FeedbackButtonObjectTests(unittest.TestCase):
     def test_basic_json(self):
         feedback_button = FeedbackButtonObject(text="+1", value="positive")
-        expected = {"text": {"type": "plain_text", "text": "+1", "emoji": True}, "value": "positive"}
+        expected = {"text": PlainTextObject(text="+1"), "value": "positive"}
         self.assertDictEqual(expected, feedback_button.to_dict())
 
     def test_with_accessibility_label(self):
         feedback_button = FeedbackButtonObject(text="+1", value="positive", accessibility_label="Positive feedback button")
         expected = {
-            "text": {"type": "plain_text", "text": "+1", "emoji": True},
+            "text": PlainTextObject(text="+1"),
             "value": "positive",
             "accessibility_label": "Positive feedback button",
         }
@@ -380,7 +380,10 @@ def test_with_accessibility_label(self):
     def test_with_plain_text_object(self):
         text_obj = PlainTextObject(text="-1", emoji=False)
         feedback_button = FeedbackButtonObject(text=text_obj, value="negative")
-        expected = {"text": {"type": "plain_text", "text": "-1", "emoji": False}, "value": "negative"}
+        expected = {
+            "text": PlainTextObject(text="-1"),
+            "value": "negative",
+        }
         self.assertDictEqual(expected, feedback_button.to_dict())
 
     def test_text_length_validation(self):
@@ -396,7 +399,7 @@ def test_parse_from_dict(self):
         parsed = FeedbackButtonObject.parse(data)
         self.assertIsInstance(parsed, FeedbackButtonObject)
         expected = {
-            "text": {"type": "plain_text", "text": "+1", "emoji": True},
+            "text": PlainTextObject(text="+1"),
             "value": "positive",
             "accessibility_label": "Positive feedback",
         }