from pathlib import Path import pytest from app.agents.graph import AuditGraph from app.config import Settings from app.schemas import AuditReport @pytest.mark.anyio async def test_run_with_progress_yields_real_stages_and_report(tmp_path: Path): source = tmp_path / "app.py" source.write_text("API_KEY = '1234567890abcdef'\nresponse = requests.get(url)\n", encoding="utf-8") (tmp_path / "README.md").write_text( "# Demo\n\n## Quick Start\nInstall and run it.\n## Tests\nRun pytest.\n## Configuration\nCopy .env.example.", encoding="utf-8", ) graph = AuditGraph(Settings(max_files=10, max_file_size_kb=10, max_chars_per_chunk=1000)) graph.crawler.clone_and_scan = lambda repo_url: graph.crawler.scan_local_repo(repo_url, tmp_path) graph.crawler.cleanup = lambda scan_result: None events = [] async for event in graph.run_with_progress("https://github.com/example/project"): events.append(event) assert any("Crawler Agent" in event for event in events if isinstance(event, str)) assert any("Security Agent" in event for event in events if isinstance(event, str)) assert any("Performance Agent" in event for event in events if isinstance(event, str)) assert any("Quality Agent" in event for event in events if isinstance(event, str)) assert any("Docs Agent" in event for event in events if isinstance(event, str)) assert isinstance(events[-1], AuditReport) assert len(events[-1].findings) == 2 assert "Security Agent" in events[-1].agents_run assert "Performance Agent" in events[-1].agents_run assert "Quality Agent" in events[-1].agents_run assert "Docs Agent" in events[-1].agents_run