ed
ada9617308
Batch rename of 22 test files. 62 references renamed total. The full test suite is now GREEN again, matching the pre-rename baseline from Task 1.1. Pure mechanical rename. No behavior change. Files affected: test_ai_cache_tracking, test_ai_client_cli, test_ai_client_result, test_api_events, test_context_pruner, test_deepseek_provider, test_gemini_cli_* (3 files), test_gui2_mcp, test_headless_* (2 files), test_live_gui_integration_v2, test_orchestration_logic, test_phase6_engine, test_rag_integration, test_run_worker_lifecycle_abort, test_spawn_interception_v2, test_symbol_parsing, test_tier4_interceptor, test_tiered_aggregation, test_token_usage. Note: spec estimated 24 files; actual is 22 (test_deprecation_warnings no longer exists, and 1 fewer file than spec's list). Refs: conductor/tracks/send_result_to_send_20260616/
64 lines
2.5 KiB
Python
64 lines
2.5 KiB
Python
import pytest
|
|
from src.models import Persona, Ticket, WorkerContext
|
|
from src import multi_agent_conductor
|
|
from src import app_controller
|
|
from src import aggregate
|
|
from pathlib import Path
|
|
from unittest.mock import MagicMock, patch
|
|
from src.result_types import Result
|
|
|
|
def test_persona_aggregation_strategy():
|
|
p = Persona(name="test_persona", aggregation_strategy="summarize")
|
|
assert p.aggregation_strategy == "summarize"
|
|
|
|
d = p.to_dict()
|
|
assert d.get("aggregation_strategy") == "summarize"
|
|
|
|
p2 = Persona.from_dict("test2", d)
|
|
assert p2.aggregation_strategy == "summarize"
|
|
|
|
@patch("src.aggregate.build_markdown_from_items")
|
|
def test_app_controller_do_generate_uses_persona_strategy(mock_build):
|
|
mock_build.return_value = "fake_md"
|
|
app = app_controller.AppController()
|
|
app.personas = {}
|
|
p = Persona(name="p1", aggregation_strategy="full")
|
|
app.personas["p1"] = p
|
|
app.ui_active_persona = "p1"
|
|
|
|
with patch("src.project_manager.flat_config", return_value={"project": {"name": "test"}, "output": {"output_dir": "."}, "files": {"paths": [], "base_dir": "."}}):
|
|
with patch("src.aggregate.build_file_items", return_value=[]):
|
|
with patch("pathlib.Path.mkdir"):
|
|
with patch("pathlib.Path.write_text"):
|
|
with patch.object(app, "_flush_to_project"):
|
|
with patch.object(app, "_flush_to_config"):
|
|
with patch("src.app_controller.AppController.save_config"):
|
|
full_md, path, file_items, stable_md, disc = app._do_generate()
|
|
|
|
# Verify aggregate.run and build_markdown_no_history received aggregation_strategy="full"
|
|
# Actually mock_build captures the inner calls of build_markdown_no_history
|
|
call_kwargs = mock_build.call_args[1]
|
|
assert call_kwargs.get("aggregation_strategy") == "full"
|
|
|
|
@patch("src.summarize.summarise_file")
|
|
@patch("src.multi_agent_conductor.ai_client.send")
|
|
def test_run_worker_lifecycle_uses_strategy(mock_send, mock_summarise, tmp_path):
|
|
mock_send.return_value = Result(data="fake response")
|
|
mock_summarise.return_value = "fake summary"
|
|
|
|
test_file = tmp_path / "test.py"
|
|
test_file.write_text("def test():\n pass")
|
|
|
|
ticket = Ticket(id="1", description="test")
|
|
context = WorkerContext(ticket_id="1", model_name="test", persona_id="test_persona")
|
|
|
|
p = Persona(name="test_persona", aggregation_strategy="summarize")
|
|
|
|
with patch("src.personas.PersonaManager.load_all", return_value={"test_persona": p}):
|
|
multi_agent_conductor.run_worker_lifecycle(
|
|
ticket, context, context_files=[str(test_file)]
|
|
)
|
|
|
|
# Should have called summarise_file because of the "summarize" strategy
|
|
mock_summarise.assert_called_once()
|