Files
Claude-Code-Workflow/codex-lens/examples/test_wait_for_analysis.py

100 lines
3.3 KiB
Python

"""Test with longer wait time for Pyright analysis."""
import asyncio
import json
from pathlib import Path
from codexlens.lsp.standalone_manager import StandaloneLspManager
async def test_with_wait():
"""Test prepareCallHierarchy with longer wait for analysis."""
workspace_root = Path("D:/Claude_dms3/codex-lens")
test_file = workspace_root / "test_simple_function.py"
print("Testing with Wait for Analysis")
print("="*80)
manager = StandaloneLspManager(
workspace_root=str(workspace_root),
timeout=30.0,
)
try:
print("\n1. Starting LSP manager...")
await manager.start()
state = await manager._get_server(str(test_file))
if not state:
print(" [ERROR] No server state")
return
print(" [OK] Server ready")
print(f" Workspace: {manager.workspace_root}")
# Open document
print("\n2. Opening document...")
await manager._open_document(state, str(test_file))
print(" [OK] Document opened")
# Wait longer for analysis
print("\n3. Waiting for Pyright to analyze (5 seconds)...")
await asyncio.sleep(5)
print(" [OK] Wait complete")
# Check diagnostics first to verify file is analyzed
print("\n4. Checking if document symbols work (to verify analysis)...")
symbols = await manager._send_request(
state,
"textDocument/documentSymbol",
{"textDocument": {"uri": test_file.resolve().as_uri()}}
)
if symbols:
print(f" [OK] Found {len(symbols)} symbols:")
for s in symbols:
name = s.get('name', 'unknown')
kind = s.get('kind', 0)
range_info = s.get('range', {}).get('start', {})
line = range_info.get('line', 0) + 1
print(f" - {name} (kind={kind}) at line {line}")
else:
print(" [WARN] No symbols found!")
# Now try call hierarchy on different lines
print("\n5. Testing prepareCallHierarchy on each symbol...")
if symbols:
for s in symbols:
name = s.get('name', 'unknown')
range_info = s.get('range', {}).get('start', {})
line = range_info.get('line', 0)
char = range_info.get('character', 0)
params = {
"textDocument": {"uri": test_file.resolve().as_uri()},
"position": {"line": line, "character": char + 4} # offset into name
}
result = await manager._send_request(
state,
"textDocument/prepareCallHierarchy",
params
)
status = f"[OK] {len(result)} items" if result else "[NONE]"
print(f" {name} (line {line+1}, char {char+4}): {status}")
if result:
for item in result:
print(f" - {item.get('name')}")
except Exception as e:
print(f"\n[ERROR] {e}")
import traceback
traceback.print_exc()
finally:
print("\n6. Cleanup...")
await manager.stop()
print(" [OK]")
if __name__ == "__main__":
asyncio.run(test_with_wait())