WIP: PAIN

tests/test_live_workflow.py

@@ -5,9 +5,18 @@ import os
 
 # Ensure project root is in path
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
-sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src")))
 
-from api_hook_client import ApiHookClient
+from src.api_hook_client import ApiHookClient
+
+def wait_for_value(client, field, expected, timeout=10):
+ start = time.time()
+ while time.time() - start < timeout:
+  state = client.get_gui_state()
+  val = state.get(field)
+  if val == expected:
+   return True
+  time.sleep(0.5)
+ return False
 
 @pytest.mark.integration
 def test_full_live_workflow(live_gui) -> None:
@@ -17,62 +26,111 @@ def test_full_live_workflow(live_gui) -> None:
  client = ApiHookClient()
  assert client.wait_for_server(timeout=10)
  client.post_session(session_entries=[])
- time.sleep(2)
+ 
  # 1. Reset
+ print("\n[TEST] Clicking Reset...")
  client.click("btn_reset")
  time.sleep(1)
+ 
  # 2. Project Setup
  temp_project_path = os.path.abspath("tests/artifacts/temp_project.toml")
  if os.path.exists(temp_project_path):
-  os.remove(temp_project_path)
+  try: os.remove(temp_project_path)
+  except: pass
+ print(f"[TEST] Creating new project at {temp_project_path}...")
  client.click("btn_project_new_automated", user_data=temp_project_path)
- time.sleep(1) # Wait for project creation and switch
- # Verify metadata update
- proj = client.get_project()
+ 
+ # Wait for project to be active
+ success = False
+ for _ in range(10):
+  proj = client.get_project()
+  # check if name matches 'temp_project'
+  if proj.get('project', {}).get('project', {}).get('name') == 'temp_project':
+   success = True
+   break
+  time.sleep(1)
+ assert success, "Project failed to activate"
+
  test_git = os.path.abspath(".")
+ print(f"[TEST] Setting project_git_dir to {test_git}...")
  client.set_value("project_git_dir", test_git)
+ assert wait_for_value(client, "project_git_dir", test_git)
+ 
  client.click("btn_project_save")
  time.sleep(1)
- proj = client.get_project()
- # flat_config returns {"project": {...}, "output": ...} 
- # so proj is {"project": {"project": {"git_dir": ...}}}
- assert proj['project']['project']['git_dir'] == test_git
+
  # Enable auto-add so the response ends up in history
  client.set_value("auto_add_history", True)
  client.set_value("current_provider", "gemini_cli")
- client.set_value("gcli_path", f'"{sys.executable}" "{os.path.abspath("tests/mock_gemini_cli.py")}"')
+ 
+ mock_path = f'"{sys.executable}" "{os.path.abspath("tests/mock_gemini_cli.py")}"'
+ print(f"[TEST] Setting gcli_path to {mock_path}...")
+ client.set_value("gcli_path", mock_path)
+ assert wait_for_value(client, "gcli_path", mock_path)
+ 
  client.set_value("current_model", "gemini-2.0-flash")
- time.sleep(0.5)
+ time.sleep(1)
+
  # 3. Discussion Turn
+ print("[TEST] Sending AI request...")
  client.set_value("ai_input", "Hello! This is an automated test. Just say 'Acknowledged'.")
  client.click("btn_gen_send")
- time.sleep(2) # Verify thinking indicator appears (might be brief)
- print("\nPolling for thinking indicator...")
- for i in range(40):
-  state = client.get_indicator_state("thinking_indicator")
-  if state.get('shown'):
-   print(f"Thinking indicator seen at poll {i}")
+ 
+ # Verify thinking indicator appears or ai_status changes
+ print("[TEST] Polling for thinking indicator...")
+ success = False
+ for i in range(20):
+  mma = client.get_mma_status()
+  ai_status = mma.get('ai_status')
+  print(f"  Poll {i}: ai_status='{ai_status}'")
+  if ai_status == 'error':
+   state = client.get_gui_state()
+   pytest.fail(f"AI Status went to error during thinking poll. Response: {state.get('ai_response')}")
+  
+  if ai_status == 'sending...' or ai_status == 'streaming...':
+   print(f"  AI is sending/streaming at poll {i}")
+   success = True
+   # Don't break, keep watching for a bit
+  
+  indicator = client.get_indicator_state("thinking_indicator")
+  if indicator.get('shown'):
+   print(f"  Thinking indicator seen at poll {i}")
+   success = True
    break
   time.sleep(0.5)
-  # 4. Wait for response in session
+
+ # 4. Wait for response in session
  success = False
- print("Waiting for AI response in session...")
- for i in range(120):
+ print("[TEST] Waiting for AI response in session history...")
+ for i in range(60):
   session = client.get_session()
   entries = session.get('session', {}).get('entries', [])
   if any(e.get('role') == 'AI' for e in entries):
    success = True
-   print(f"AI response found at second {i}")
+   print(f"  AI response found in history after {i}s")
    break
+  
+  mma = client.get_mma_status()
+  if mma.get('ai_status') == 'error':
+   state = client.get_gui_state()
+   pytest.fail(f"AI Status went to error during response wait. Response: {state.get('ai_response')}")
+   
   time.sleep(1)
- assert success, "AI failed to respond within 120 seconds"
+ assert success, "AI failed to respond or response not added to history"
+
  # 5. Switch Discussion
+ print("[TEST] Creating new discussion 'AutoDisc'...")
  client.set_value("disc_new_name_input", "AutoDisc")
  client.click("btn_disc_create")
- time.sleep(1.0) # Wait for GUI to process creation
+ time.sleep(1.0)
+ 
+ print("[TEST] Switching to 'AutoDisc'...")
  client.select_list_item("disc_listbox", "AutoDisc")
- time.sleep(1.0) # Wait for GUI to switch
+ time.sleep(1.0)
+ 
  # Verify session is empty in new discussion
  session = client.get_session()
- assert len(session.get('session', {}).get('entries', [])) == 0
-
+ entries = session.get('session', {}).get('entries', [])
+ print(f"  New discussion history length: {len(entries)}")
+ assert len(entries) == 0
+ print("[TEST] Workflow completed successfully.")