123 lines
4.8 KiB
Python
123 lines
4.8 KiB
Python
import pytest
|
|
from unittest.mock import MagicMock, patch
|
|
import json
|
|
from typing import Any
|
|
import orchestrator_pm
|
|
import conductor_tech_lead
|
|
import multi_agent_conductor
|
|
from models import Track, Ticket
|
|
|
|
@pytest.fixture
|
|
def mock_ai_client() -> Any:
|
|
with patch("ai_client.send") as mock_send:
|
|
yield mock_send
|
|
|
|
def test_generate_tracks(mock_ai_client: Any) -> None:
|
|
# Tier 1 (PM) response mock
|
|
mock_ai_client.return_value = json.dumps([
|
|
{"id": "track_1", "title": "Infrastructure Setup", "description": "Setup basic project structure"},
|
|
{"id": "track_2", "title": "Feature implementation", "description": "Implement core feature"}
|
|
])
|
|
user_request = "Build a new app"
|
|
project_config = {}
|
|
file_items = []
|
|
tracks = orchestrator_pm.generate_tracks(user_request, project_config, file_items)
|
|
assert len(tracks) == 2
|
|
assert tracks[0]["id"] == "track_1"
|
|
assert tracks[1]["id"] == "track_2"
|
|
mock_ai_client.assert_called_once()
|
|
|
|
def test_generate_tickets(mock_ai_client: Any) -> None:
|
|
mock_ai_client.return_value = json.dumps([
|
|
{"id": "T-001", "description": "Define interfaces", "depends_on": []},
|
|
{"id": "T-002", "description": "Implement interfaces", "depends_on": ["T-001"]}
|
|
])
|
|
track_brief = "Implement a new feature."
|
|
module_skeletons = "class Feature: pass"
|
|
tickets = conductor_tech_lead.generate_tickets(track_brief, module_skeletons)
|
|
assert len(tickets) == 2
|
|
assert tickets[0]["id"] == "T-001"
|
|
assert tickets[1]["id"] == "T-002"
|
|
assert tickets[1]["depends_on"] == ["T-001"]
|
|
|
|
def test_topological_sort() -> None:
|
|
tickets = [
|
|
{"id": "T-002", "description": "Dep on 001", "depends_on": ["T-001"]},
|
|
{"id": "T-001", "description": "Base", "depends_on": []},
|
|
{"id": "T-003", "description": "Dep on 002", "depends_on": ["T-002"]}
|
|
]
|
|
sorted_tickets = conductor_tech_lead.topological_sort(tickets)
|
|
assert sorted_tickets[0]["id"] == "T-001"
|
|
assert sorted_tickets[1]["id"] == "T-002"
|
|
assert sorted_tickets[2]["id"] == "T-003"
|
|
|
|
def test_topological_sort_circular() -> None:
|
|
tickets = [
|
|
{"id": "T-001", "depends_on": ["T-002"]},
|
|
{"id": "T-002", "depends_on": ["T-001"]}
|
|
]
|
|
# Align with conductor_tech_lead.py wrapping of DAG errors
|
|
with pytest.raises(ValueError, match="DAG Validation Error"):
|
|
conductor_tech_lead.topological_sort(tickets)
|
|
|
|
def test_track_executable_tickets() -> None:
|
|
t1 = Ticket(id="T1", description="desc", status="todo", assigned_to="user")
|
|
t2 = Ticket(id="T2", description="desc", status="todo", assigned_to="user", depends_on=["T1"])
|
|
track = Track(id="track_1", description="desc", tickets=[t1, t2])
|
|
executable = track.get_executable_tickets()
|
|
assert len(executable) == 1
|
|
assert executable[0].id == "T1"
|
|
# Complete T1
|
|
t1.status = "completed"
|
|
executable = track.get_executable_tickets()
|
|
assert len(executable) == 1
|
|
assert executable[0].id == "T2"
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_conductor_engine_run(vlogger) -> None:
|
|
t1 = Ticket(id="T1", description="desc", status="todo", assigned_to="user")
|
|
t2 = Ticket(id="T2", description="desc", status="todo", assigned_to="user", depends_on=["T1"])
|
|
track = Track(id="track_1", description="desc", tickets=[t1, t2])
|
|
engine = multi_agent_conductor.ConductorEngine(track, auto_queue=True)
|
|
|
|
vlogger.log_state("T1 Initial Status", "todo", t1.status)
|
|
vlogger.log_state("T2 Initial Status", "todo", t2.status)
|
|
|
|
with patch("multi_agent_conductor.run_worker_lifecycle") as mock_worker:
|
|
# Mock worker to complete tickets
|
|
|
|
def complete_ticket(ticket, context, *args, **kwargs):
|
|
ticket.status = "completed"
|
|
mock_worker.side_effect = complete_ticket
|
|
await engine.run()
|
|
|
|
vlogger.log_state("T1 Final Status", "todo", t1.status)
|
|
vlogger.log_state("T2 Final Status", "todo", t2.status)
|
|
|
|
assert t1.status == "completed"
|
|
assert t2.status == "completed"
|
|
assert mock_worker.call_count == 2
|
|
vlogger.finalize("Orchestration Logic - Conductor Engine", "PASS", "Dependency order honored during run.")
|
|
|
|
def test_conductor_engine_parse_json_tickets() -> None:
|
|
track = Track(id="track_1", description="desc")
|
|
engine = multi_agent_conductor.ConductorEngine(track, auto_queue=True)
|
|
json_data = json.dumps([
|
|
{"id": "T1", "description": "desc 1", "depends_on": []},
|
|
{"id": "T2", "description": "desc 2", "depends_on": ["T1"]}
|
|
])
|
|
engine.parse_json_tickets(json_data)
|
|
assert len(track.tickets) == 2
|
|
assert track.tickets[0].id == "T1"
|
|
assert track.tickets[1].id == "T2"
|
|
assert track.tickets[1].depends_on == ["T1"]
|
|
|
|
def test_run_worker_lifecycle_blocked(mock_ai_client: Any) -> None:
|
|
ticket = Ticket(id="T1", description="desc", status="todo", assigned_to="user")
|
|
context = multi_agent_conductor.WorkerContext(ticket_id="T1", model_name="model", messages=[])
|
|
mock_ai_client.return_value = "BLOCKED because of missing info"
|
|
multi_agent_conductor.run_worker_lifecycle(ticket, context)
|
|
assert ticket.status == "blocked"
|
|
assert ticket.blocked_reason == "BLOCKED because of missing info"
|
|
|