tests: Add tests for auto-submit and focus

Author: gadenbuie

tests/playwright/shiny/components/chat/input-suggestion/app.py

@@ -2,9 +2,12 @@
 
 welcome = ui.HTML(
     """
-Here is the <span class='suggestion'>1st input suggestion</span>.
-And here is a <span data-suggestion='The actual suggestion'>2nd suggestion</span>.
-Finally, a <img data-suggestion='A 3rd, image-based, suggestion' src='shiny-hex.svg' height="50px" alt='Shiny logo'> image suggestion.
+<p>Here is the <span id="first" class='suggestion'>1st input suggestion</span>.
+And here is a <span id="second" data-suggestion='The actual suggestion'>2nd suggestion</span>.
+Finally, a <img id="third" data-suggestion='A 3rd, image-based, suggestion' src='shiny-hex.svg' height="50px" alt='Shiny logo'> image suggestion.</p>
+
+<p>On the other hand, <span id="fourth" class="suggestion submit">this suggestion will auto-submit</span>.
+And <span id="fifth" data-suggestion="another suggestion" data-suggestion-submit="true">this suggestion will also auto-submit</span>.</p>
 """
 )
 

tests/playwright/shiny/components/chat/input-suggestion/test_chat_input_suggestion.py

@@ -17,9 +17,11 @@ def test_validate_chat_input_suggestion(page: Page, local_app: ShinyAppProc) ->
     expect(chat.loc_input_button).to_be_disabled()
 
     # Locate input suggestions
-    first = chat.loc.locator(".suggestion", has_text="1st input suggestion")
-    second = chat.loc.locator("span[data-suggestion]")
-    third = chat.loc.locator("img[data-suggestion]")
+    first = chat.loc.locator("#first", has_text="1st input suggestion")
+    second = chat.loc.locator("#second")
+    third = chat.loc.locator("#third")
+    fourth = chat.loc.locator("#fourth")
+    fifth = chat.loc.locator("#fifth")
 
     # Click on each suggestion and verify the input
     first.click()
@@ -39,3 +41,17 @@ def test_validate_chat_input_suggestion(page: Page, local_app: ShinyAppProc) ->
     chat.expect_latest_message("You said: A 3rd, image-based, suggestion")
     chat.expect_user_input("")
     expect(chat.loc_input_button).to_be_disabled()
+
+    # Test auto-submitting suggestions
+    fourth.click()
+    chat.expect_latest_message("You said: this suggestion will auto-submit")
+    chat.expect_user_input("")
+    expect(chat.loc_input_button).to_be_disabled()
+    expect(chat.loc_input).not_to_be_focused()
+
+    fifth.focus()
+    page.keyboard.press("Enter")
+    chat.expect_latest_message("You said: another suggestion")
+    chat.expect_user_input("")
+    expect(chat.loc_input_button).to_be_disabled()
+    expect(chat.loc_input).not_to_be_focused()