tests(chat-icon): Fix tests

Author: gadenbuie

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

@@ -13,9 +13,15 @@ def __init__(self, page: Page, id: str, classes: str):
 
     def expect_last_message_icon_to_have_classes(self):
         # TODO: Fix this test to find icon in shadow root
-        last_msg_icon = self.chat.loc_latest_message.locator(
-            '.message-icon > slot[name="icon"] > *'
-        )
+        if self.id == "default":
+            last_msg_icon = self.chat.loc_latest_message.locator(
+                'slot[name="icon"] > :first-child'
+            )
+        else:
+            last_msg_icon = self.chat.loc_latest_message.locator(
+                '[slot="icon"]'
+            )
+
         expect(last_msg_icon).to_have_class(self.classes)