tests: chat.update_user_input() with submit and focus

Author: gadenbuie

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

@@ -0,0 +1,54 @@
+from shiny import reactive
+from shiny.express import input, ui
+
+ui.page_opts(fillable=True)
+
+with ui.layout_columns(fill=False, fillable=True):
+    ui.input_action_button("set_input", "Just set input")
+    ui.input_action_button("set_and_focus", "Set and focus input")
+    ui.input_action_button("submit", "Submit to chat")
+    ui.input_action_button("submit_and_focus", "Submit and focus chat")
+
+chat = ui.Chat("chat")
+
+
+chat.ui()
+
+
+@reactive.effect
+@reactive.event(input.set_input)
+def do_set_input():
+    chat.update_user_input(value="Input was set, but neither focused nor submitted.")
+
+
+@reactive.effect
+@reactive.event(input.set_and_focus)
+def do_set_and_focus():
+    chat.update_user_input(
+        value="Input was set and focused, but not submitted.",
+        focus=True,
+    )
+
+
+@reactive.effect
+@reactive.event(input.submit)
+def do_submit():
+    chat.update_user_input(
+        value="This chat was sent on behalf of the user.",
+        submit=True,
+    )
+
+
+@reactive.effect
+@reactive.event(input.submit_and_focus)
+def do_submit_and_focus():
+    chat.update_user_input(
+        value="This chat was sent on behalf of the user. Input will still be focused.",
+        submit=True,
+        focus=True,
+    )
+
+
+@chat.on_user_submit
+async def on_user_submit(message: str):
+    await chat.append_message(f"You said: {message}")

tests/playwright/shiny/components/chat/update_user_input/test_chat_update_user_input.py

@@ -0,0 +1,54 @@
+from playwright.sync_api import Page, expect
+from utils.deploy_utils import skip_on_webkit
+
+from shiny.playwright import controller
+from shiny.run import ShinyAppProc
+
+
+@skip_on_webkit
+def test_validate_chat_update_user_input(page: Page, local_app: ShinyAppProc) -> None:
+    page.goto(local_app.url)
+
+    chat = controller.Chat(page, "chat")
+
+    # Starting state sanity check
+    expect(chat.loc).to_be_visible(timeout=30 * 1000)
+    chat.expect_user_input("")
+    expect(chat.loc_input_button).to_be_disabled()
+
+    # Locate action buttons
+    set_input = controller.InputActionButton(page, "set_input")
+    set_and_focus = controller.InputActionButton(page, "set_and_focus")
+    submit = controller.InputActionButton(page, "submit")
+    submit_and_focus = controller.InputActionButton(page, "submit_and_focus")
+
+    # Just set the input ----
+    set_input.loc.focus()
+    set_input.click()
+    chat.expect_user_input("Input was set, but neither focused nor submitted.")
+    expect(chat.loc_input_button).to_be_enabled()
+    expect(chat.loc_input).not_to_be_focused()
+
+    # Set the input and move focus ----
+    set_and_focus.click()
+    chat.expect_user_input("Input was set and focused, but not submitted.")
+    expect(chat.loc_input_button).to_be_enabled()
+    expect(chat.loc_input).to_be_focused()
+
+    # Auto submit ----
+    submit.loc.focus()
+    submit.click()
+    chat.expect_user_input("")
+    chat.expect_latest_message("You said: This chat was sent on behalf of the user.")
+    expect(chat.loc_input_button).to_be_disabled()
+    expect(chat.loc_input).not_to_be_focused()
+
+    # Auto submit and move focus ----
+    submit_and_focus.loc.focus()
+    submit_and_focus.click()
+    chat.expect_user_input("")
+    chat.expect_latest_message(
+        "You said: This chat was sent on behalf of the user. Input will still be focused."
+    )
+    expect(chat.loc_input_button).to_be_disabled()
+    expect(chat.loc_input).to_be_focused()