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.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 = "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()
