tests: Explicitly test returning to no icon

Author: gadenbuie

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

@@ -47,9 +47,15 @@ async def handle_user_input_default(user_input: str):
     async def handle_user_input_otter(user_input: str):
         await asyncio.sleep(1)
         icon_name = input.animal().lower()
+        if icon_name == "Otter":
+            # Don't include the icon in the custom message, i.e. use default icon
+            icon = None
+        else:
+            icon = faicons.icon_svg(icon_name).add_class(f"icon-{icon_name}")
+
         await chat_animal.append_message(
             f"{input.animal()} said: {user_input}",
-            icon=faicons.icon_svg(icon_name).add_class(f"icon-{icon_name}"),
+            icon=icon,
         )
 
     # SVG Bot -------------------------------------------------------------------------

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

@@ -55,3 +55,10 @@ def test_validate_chat_basic(page: Page, local_app: ShinyAppProc) -> None:
     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")
+
+    # Test that the icon used is the default if no icon is sent
+    # (the message-specific icon is just for that message)
+    animal.set("Otter")
+    chat_animal.chat.set_user_input("hello")
+    chat_animal.chat.send_user_input()
+    chat_animal.expect_last_message_icon_to_have_classes()