mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 16:18:30 +00:00
032922a4b9
Change `emit()` to call handlers registered for parent classes using `isinstance()`. Ensures that base event handlers receive derived events.
41 lines
955 B
Python
41 lines
955 B
Python
from unittest.mock import Mock
|
|
|
|
import pytest
|
|
|
|
from crewai.utilities.events.base_events import CrewEvent
|
|
from crewai.utilities.events.crewai_event_bus import CrewAIEventsBus
|
|
|
|
|
|
class TestEvent(CrewEvent):
|
|
pass
|
|
|
|
@pytest.fixture
|
|
def event_bus():
|
|
bus = CrewAIEventsBus()
|
|
bus.clear_handlers()
|
|
return bus
|
|
|
|
def test_specific_event_handler(event_bus):
|
|
mock_handler = Mock()
|
|
|
|
@event_bus.on(TestEvent)
|
|
def handler(source, event):
|
|
mock_handler(source, event)
|
|
|
|
event = TestEvent(type="test_event")
|
|
event_bus.emit("source_object", event)
|
|
|
|
mock_handler.assert_called_once_with("source_object", event)
|
|
|
|
def test_wildcard_event_handler(event_bus):
|
|
mock_handler = Mock()
|
|
|
|
@event_bus.on(CrewEvent)
|
|
def handler(source, event):
|
|
mock_handler(source, event)
|
|
|
|
event = TestEvent(type="test_event")
|
|
event_bus.emit("source_object", event)
|
|
|
|
mock_handler.assert_called_once_with("source_object", event)
|