tests: Add tests for dynamic chat icon

Author: gadenbuie

tests/playwright/shiny/components/chat/icon/app.py

@@ -2,7 +2,7 @@
 
 import faicons
 
-from shiny.express import ui
+from shiny.express import input, ui
 
 ui.page_opts(title="Chat Icons")
 
@@ -27,25 +27,30 @@ async def handle_user_input_default(user_input: str):
         await asyncio.sleep(1)
         await chat_default.append_message(f"You said: {user_input}")
 
-    # Otter Bot -----------------------------------------------------------------------
-    chat_otter = ui.Chat(
-        id="chat_otter",
+    # Animal Bot ----------------------------------------------------------------------
+    chat_animal = ui.Chat(
+        id="chat_animal",
         messages=[
             {
-                "content": "Hello! I'm Otter Bot. How can I help you today?",
+                "content": "Hello! I'm Animal Bot. How can I help you today?",
                 "role": "assistant",
             },
         ],
     )
 
     with ui.div():
-        ui.h2("Otter Bot")
-        chat_otter.ui(icon_assistant=faicons.icon_svg("otter").add_class("icon-otter"))
+        ui.h2("Animal Bot")
+        chat_animal.ui(icon_assistant=faicons.icon_svg("otter").add_class("icon-otter"))
+        ui.input_select("animal", "Animal", choices=["Otter", "Hippo", "Frog", "Dove"])
 
-    @chat_otter.on_user_submit
+    @chat_animal.on_user_submit
     async def handle_user_input_otter(user_input: str):
         await asyncio.sleep(1)
-        await chat_otter.append_message(f"You said: {user_input}")
+        icon_name = input.animal().lower()
+        await chat_animal.append_message(
+            f"{input.animal()} said: {user_input}",
+            icon = faicons.icon_svg(icon_name).add_class(f"icon-{icon_name}")
+        )
 
     # SVG Bot -------------------------------------------------------------------------
     bs_icon_info_circle_fill = """

tests/playwright/shiny/components/chat/icon/test_chat_icon.py

@@ -11,7 +11,7 @@ def __init__(self, page: Page, id: str, classes: str):
         self.chat = controller.Chat(page, f"chat_{id}")
         self.classes = classes
 
-    def expect_last_message_icon_to_have_classes(self):
+    def expect_last_message_icon_to_have_classes(self, classes: str | None = None):
         # TODO: Fix this test to find icon in shadow root
         if self.id == "default":
             last_msg_icon = self.chat.loc_latest_message.locator(
@@ -22,7 +22,7 @@ def expect_last_message_icon_to_have_classes(self):
                 '[slot="icon"]'
             )
 
-        expect(last_msg_icon).to_have_class(self.classes)
+        expect(last_msg_icon).to_have_class(classes or self.classes)
 
 
 @skip_on_webkit
@@ -31,7 +31,7 @@ def test_validate_chat_basic(page: Page, local_app: ShinyAppProc) -> None:
 
     chats: list[ChatModule] = [
         ChatModule(page, "default", "bi bi-robot"),
-        ChatModule(page, "otter", "fa icon-otter"),
+        ChatModule(page, "animal", "fa icon-otter"),
         ChatModule(page, "svg", "bi bi-info-circle-fill icon-svg"),
         ChatModule(page, "shiny", "icon-shiny"),
     ]
@@ -44,3 +44,16 @@ def test_validate_chat_basic(page: Page, local_app: ShinyAppProc) -> None:
         mod.chat.send_user_input()
 
         mod.expect_last_message_icon_to_have_classes()
+
+    animal = controller.InputSelect(page, "animal")
+    chat_animal = chats[1]
+
+    animal.set("Hippo")
+    chat_animal.chat.set_user_input("hello")
+    chat_animal.chat.send_user_input()
+    chat_animal.expect_last_message_icon_to_have_classes("fa icon-hippo")
+
+    animal.set("Frog")
+    chat_animal.chat.set_user_input("hello")
+    chat_animal.chat.send_user_input()
+    chat_animal.expect_last_message_icon_to_have_classes("fa icon-frog")