feat(conductor): Add worker tracking and abort event dictionaries to ConductorEngine

This commit is contained in:
2026-03-07 15:55:39 -05:00
parent beae82860a
commit 5f7909121d
2 changed files with 20 additions and 0 deletions

View File

@@ -89,6 +89,7 @@ class ConductorEngine:
self.pool = WorkerPool(max_workers=max_workers)
self._workers_lock = threading.Lock()
self._active_workers: dict[str, threading.Thread] = {}
self._abort_events: dict[str, threading.Event] = {}
self._tier_usage_lock = threading.Lock()
def update_usage(self, tier: str, input_tokens: int, output_tokens: int) -> None:

View File

@@ -0,0 +1,19 @@
import pytest
from unittest.mock import MagicMock
from src.multi_agent_conductor import ConductorEngine
from src.models import Track
def test_conductor_engine_initializes_empty_worker_and_abort_dicts() -> None:
"""
Test that ConductorEngine correctly initializes _active_workers and _abort_events as empty dictionaries.
"""
# Mock the track object
mock_track = MagicMock(spec=Track)
mock_track.tickets = []
# Initialize ConductorEngine
engine = ConductorEngine(track=mock_track)
# Verify _active_workers and _abort_events are empty dictionaries
assert engine._active_workers == {}
assert engine._abort_events == {}