feat(conductor): Add pause/resume mechanism to ConductorEngine
This commit is contained in:
24
tests/test_pipeline_pause.py
Normal file
24
tests/test_pipeline_pause.py
Normal file
@@ -0,0 +1,24 @@
|
||||
import pytest
|
||||
from unittest.mock import MagicMock, patch
|
||||
from src.models import Ticket, Track
|
||||
from src.multi_agent_conductor import ConductorEngine
|
||||
|
||||
def test_conductor_engine_has_pause_event():
|
||||
track = Track(id="test", description="Test", tickets=[])
|
||||
engine = ConductorEngine(track)
|
||||
assert hasattr(engine, '_pause_event'), "ConductorEngine must have _pause_event"
|
||||
assert engine._pause_event.is_set() == False, "Pause event should start unset (not paused)"
|
||||
|
||||
def test_pause_method():
|
||||
track = Track(id="test", description="Test", tickets=[])
|
||||
engine = ConductorEngine(track)
|
||||
engine.pause()
|
||||
assert engine._pause_event.is_set() == True, "Pause should set the event"
|
||||
|
||||
def test_resume_method():
|
||||
track = Track(id="test", description="Test", tickets=[])
|
||||
engine = ConductorEngine(track)
|
||||
engine.pause()
|
||||
assert engine._pause_event.is_set() == True
|
||||
engine.resume()
|
||||
assert engine._pause_event.is_set() == False, "Resume should clear the event"
|
||||
Reference in New Issue
Block a user