[tda,react,agent] TDA automation for AI Agent (#1164)

* [react] AI Agent Frontend v0 - r2

* [tda,react,agent] TDA automation for AI Agent

* run _tda/test_ai_agent with only 1 browser

* remove _tda/test_about_employees

---------

Co-authored-by: Kosty Maleyev <kosty@mac.lan>

Author: maxkosty

_tda/conftest.py

@@ -512,6 +512,24 @@ def desktop_web_driver(request, se_prefix):
         with _local_browser(request, se) as b:
             yield b
 
+
+@pytest.fixture
+def desktop_web_1browser_driver(request, se_prefix):
+    """Like desktop_web_driver but only uses the first browser from CONFIG.browsers (no parameterization)"""
+    browser_config = CONFIG.browsers[0]
+    
+    import builtins
+    builtins._current_browser_config = browser_config
+    
+    # Create a wrapper that mimics request.param for _sauce_browser
+    request.param = browser_config
+    
+    se = f'{se_prefix}-sauce-{browser_config.param_display}'
+    sentry_sdk.set_tag("se", se)
+    with _sauce_browser(request, se) as b:
+        yield b
+
+
 @pytest.fixture
 def android_react_native_emu_driver(request, se_prefix):
 

_tda/desktop_web/test_about_employees.py

@@ -0,0 +1,83 @@
+import time
+import sentry_sdk
+from selenium.webdriver.common.by import By
+
+
+# Time to wait after conversation completes before closing the chat
+# This allows for proper Sentry trace capture
+WAIT_BEFORE_CLOSE_SECONDS = 80
+
+
+def test_ai_agent(desktop_web_1browser_driver, endpoints):
+    """
+    Test the AI Agent Chat widget on the React homepage.
+    
+    This test:
+    1. Opens the chat widget
+    2. Waits for the welcome messages
+    3. Answers "full sun" to the light question
+    4. Answers "yes" to the maintenance question
+    5. Waits for the API response
+    6. Waits 80 seconds for trace data capture
+    7. Closes the chat by clicking X
+    """
+    
+    endpoint = endpoints.react_endpoint
+    
+    url = endpoint  # use default backend as it doesn't matter for this test
+    
+    try:
+        # Navigate to the homepage
+        desktop_web_1browser_driver.get(url)
+        
+        # Wait for page to load
+        time.sleep(3)
+        
+        # Click the AI Agent button to open chat
+        chat_button = desktop_web_1browser_driver.find_element(By.ID, 'chat-widget-button')
+        chat_button.click()
+        
+        # Wait for the chat to open and show initial messages
+        # The bot will show typing indicator, then welcome message, then typing, then first question
+        time.sleep(5)
+        
+        # Find the input field (indicates first question is ready)
+        chat_input = desktop_web_1browser_driver.find_element(By.ID, 'chat-message-input')
+        
+        # Answer the first question: "How much light does your room get?"
+        chat_input.send_keys("full sun")
+        
+        # Click Send
+        send_button = desktop_web_1browser_driver.find_element(By.ID, 'chat-send-button')
+        send_button.click()
+        
+        # Wait for the second question to appear
+        time.sleep(3)
+        
+        # Find the input field again
+        chat_input = desktop_web_1browser_driver.find_element(By.ID, 'chat-message-input')
+        
+        # Answer the second question: "Are you only looking for low-maintenance plants?"
+        chat_input.send_keys("yes")
+        
+        # Click Send
+        send_button = desktop_web_1browser_driver.find_element(By.ID, 'chat-send-button')
+        send_button.click()
+        
+        # Wait for the API response to appear
+        # The bot will show typing indicator while waiting for the API, then show the response
+        time.sleep(10)
+        
+        # Wait the specified time before closing to allow trace data to be captured
+        print(f"Waiting {WAIT_BEFORE_CLOSE_SECONDS} seconds before closing chat...")
+        time.sleep(WAIT_BEFORE_CLOSE_SECONDS)
+        
+        # Close the chat by clicking the chat button again (chat close button may not be visible)
+        chat_button.click()
+        
+        # Small delay to ensure the close action completes
+        time.sleep(2)
+        
+    except Exception as err:
+        sentry_sdk.capture_exception(err)
+        raise

_tda/desktop_web/test_ai_agent.py

@@ -22,6 +22,7 @@ const ChatWidget = () => {
   const typingSpanRef = useRef(null);
   const typingTimeoutRef = useRef(null);
   const inactivityTimeoutRef = useRef(null);
+  const initTimeoutsRef = useRef([]);
 
   const scrollToBottom = () => {
     messagesEndRef.current?.scrollIntoView({ behavior: 'smooth' });
@@ -40,6 +41,9 @@ const ChatWidget = () => {
       clearTimeout(inactivityTimeoutRef.current);
       inactivityTimeoutRef.current = null;
     }
+    // Clear any pending initialization timeouts
+    initTimeoutsRef.current.forEach(timeoutId => clearTimeout(timeoutId));
+    initTimeoutsRef.current = [];
 
     // Record how the session ended
     if (chatSpanRef.current) {
@@ -116,10 +120,14 @@ const ChatWidget = () => {
 
   useEffect(() => {
     if (isOpen && conversationState === 'initial') {
+      // Clear any existing init timeouts
+      initTimeoutsRef.current.forEach(timeoutId => clearTimeout(timeoutId));
+      initTimeoutsRef.current = [];
+      
       // Show typing indicator
       setMessages([{ type: 'typing', id: generateMessageId() }]);
 
-      setTimeout(() => {
+      const timeout1 = setTimeout(() => {
         // Remove typing indicator and show welcome message
         addBotMessage("Hi, I can help you pick the right plants for your home", () => {
           setMessages([
@@ -132,16 +140,19 @@ const ChatWidget = () => {
         });
 
         // Show second message after a brief pause
-        setTimeout(() => {
+        const timeout2 = setTimeout(() => {
           // Show typing indicator
           setMessages(prev => [...prev, { type: 'typing', id: generateMessageId() }]);
 
-          setTimeout(() => {
+          const timeout3 = setTimeout(() => {
             addBotMessage('How much light does your room get?');
             setConversationState('awaiting_light');
           }, 1000);
+          initTimeoutsRef.current.push(timeout3);
         }, 500);
+        initTimeoutsRef.current.push(timeout2);
       }, 1000);
+      initTimeoutsRef.current.push(timeout1);
     }
   }, [isOpen, conversationState, addBotMessage]);