Tests

Author: schloerke

tests/playwright/shiny/bookmark/chat/chatlas/app.py

@@ -0,0 +1,43 @@
+import chatlas
+
+from shiny.express import ui
+
+# Set some Shiny page options
+ui.page_opts(
+    title="Hello Shiny Chat",
+    fillable=True,
+    fillable_mobile=True,
+)
+
+# Create a chat instance
+init_messages = ["""Welcome!"""]
+chat = ui.Chat(
+    id="chat",
+    messages=init_messages,
+)
+
+# Display it
+chat.ui()
+
+# Goal: Test that chatlas is serializied and deserialized correctly.
+#
+# Use ChatOpenAI as it does not need credentials until submission to the server.
+# However, if we use `.set_turns()` and `.get_turns()`, a submission is never made to the server... therefore we don't need credentials.
+chat_client = chatlas.ChatOpenAI(turns=[])  # pyright: ignore[reportUnknownMemberType]
+chat.enable_bookmarking(chat_client, bookmark_store="url")
+
+
+# Define a callback to run when the user submits a message
+@chat.on_user_submit
+async def handle_user_input(user_input: str):
+    mock_msg = f"You said to OpenAI: {user_input}"
+    chat_client.set_turns(  # pyright: ignore[reportUnknownMemberType]
+        [
+            *chat_client.get_turns(),
+            chatlas.Turn(role="user", contents=user_input),
+            chatlas.Turn(role="assistant", contents=mock_msg),
+        ]
+    )
+
+    # Append a response to the chat
+    await chat.append_message(mock_msg)

tests/playwright/shiny/bookmark/chat/chatlas/test_bookmark_chatlas.py

@@ -0,0 +1,26 @@
+from playwright.sync_api import Page
+
+from shiny.playwright.controller import Chat
+from shiny.run import ShinyAppProc
+
+
+def test_bookmark_chatlas(page: Page, local_app: ShinyAppProc):
+
+    page.goto(local_app.url)
+
+    assert "?" not in page.url
+
+    chat_controller = Chat(page, "chat")
+
+    chat_controller.expect_messages("Welcome!")
+
+    chat_controller.set_user_input("Testing")
+    chat_controller.send_user_input()
+
+    chat_controller.expect_messages("Welcome!\nTesting\nYou said to OpenAI: Testing")
+
+    assert "?" in page.url
+
+    page.reload()
+
+    chat_controller.expect_messages("Welcome!\nTesting\nYou said to OpenAI: Testing")

tests/playwright/shiny/bookmark/chat/client_state/app.py

@@ -0,0 +1,65 @@
+from typing import cast
+
+from shiny.express import ui
+from shiny.types import Jsonifiable, JsonifiableDict
+
+# Set some Shiny page options
+ui.page_opts(
+    title="Hello Shiny Chat",
+    fillable=True,
+    fillable_mobile=True,
+)
+
+
+# Create a chat instance
+init_messages = ["""Welcome!"""]
+chat = ui.Chat(
+    id="chat",
+    messages=init_messages,
+)
+
+# Display it
+chat.ui()
+
+
+class RepeaterClient:
+    """
+    A simple chat client repeater that echoes back the user's input.
+    """
+
+    def __init__(self, *, messages: list[str] = []):
+        self.messages = messages
+
+    def append_message(self, message: str) -> str:
+        msg = f"Repeater: {message}"
+        self.messages.append(msg)
+        return msg
+
+    async def get_state(self) -> Jsonifiable:
+        """
+        Get the current state of the chat client.
+        """
+        return cast(JsonifiableDict, {"messages": self.messages})
+
+    async def set_state(self, state: Jsonifiable) -> None:
+        """ "
+        Set the state of the chat client.
+        """
+        assert isinstance(state, dict)
+        assert "messages" in state
+        assert isinstance(state["messages"], list)
+        self.messages = state["messages"]
+
+
+chat_client = RepeaterClient(messages=init_messages)
+
+chat.enable_bookmarking(chat_client, bookmark_store="url")
+
+
+# Define a callback to run when the user submits a message
+@chat.on_user_submit
+async def handle_user_input(user_input: str):
+
+    msg = chat_client.append_message(user_input)
+    # Append a response to the chat
+    await chat.append_message(msg)

tests/playwright/shiny/bookmark/chat/client_state/test_bookmark_chat.py

@@ -0,0 +1,26 @@
+from playwright.sync_api import Page
+
+from shiny.playwright.controller import Chat
+from shiny.run import ShinyAppProc
+
+
+def test_bookmark_chatlas(page: Page, local_app: ShinyAppProc):
+
+    page.goto(local_app.url)
+
+    assert "?" not in page.url
+
+    chat_controller = Chat(page, "chat")
+
+    chat_controller.expect_messages("Welcome!")
+
+    chat_controller.set_user_input("Testing")
+    chat_controller.send_user_input()
+
+    chat_controller.expect_messages("Welcome!\nTesting\nRepeater: Testing")
+
+    assert "?" in page.url
+
+    page.reload()
+
+    chat_controller.expect_messages("Welcome!\nTesting\nRepeater: Testing")