63 lines
2.7 KiB
Python
63 lines
2.7 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
|
|
|
|
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.models.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 = "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()
|