Merge pull request #109 from jimmyruann/master

Fix RichText type issue with views
imryche · Feb 12, 2024 · 54fbefa · 54fbefa
2 parents 1371b6a + 72ab51a
commit 54fbefa
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 5 deletions.
diff --git a/blockkit/surfaces.py b/blockkit/surfaces.py
@@ -9,6 +9,7 @@
     Header,
     ImageBlock,
     Input,
+    RichText,
     Section,
 )
 from blockkit.components import Component
@@ -17,7 +18,7 @@
 
 __all__ = ["Home", "Message", "Modal", "WorkflowStep"]
 
-Block = Union[Actions, Context, Divider, Header, ImageBlock, Input, Section]
+Block = Union[Actions, Context, Divider, Header, ImageBlock, Input, RichText, Section]
 
 
 class View(Component):

diff --git a/tests/test_surfaces.py b/tests/test_surfaces.py
@@ -1,7 +1,7 @@
 import pytest
-from blockkit.blocks import Input, Section
+from blockkit.blocks import Input, RichText, RichTextSection, Section
 from blockkit.elements import Button, PlainTextInput
-from blockkit.objects import MarkdownText, PlainText
+from blockkit.objects import MarkdownText, PlainText, Text
 from blockkit.surfaces import Home, Message, Modal, WorkflowStep
 from pydantic import ValidationError
 
@@ -199,7 +199,10 @@ def test_builds_message():
             Section(
                 text=MarkdownText(text="*markdown* text"),
                 accessory=Button(text=PlainText(text="button"), action_id="action_id"),
-            )
+            ),
+            RichText(
+                elements=[RichTextSection(elements=[Text(text="test_rich_text")])]
+            ),
         ],
     ).build() == {
         "text": "message text",
@@ -212,7 +215,16 @@ def test_builds_message():
                     "text": {"type": "plain_text", "text": "button"},
                     "action_id": "action_id",
                 },
-            }
+            },
+            {
+                "type": "rich_text",
+                "elements": [
+                    {
+                        "type": "rich_text_section",
+                        "elements": [{"type": "text", "text": "test_rich_text"}],
+                    }
+                ],
+            },
         ],
     }