Support wildcard handling in emit()

Change `emit()` to call handlers registered for parent classes using
`isinstance()`. Ensures that base event handlers receive derived
events.
This commit is contained in:
Vinicius Brasil
2025-03-20 08:48:21 -03:00
committed by Vini Brasil
parent 90b7937796
commit 032922a4b9
2 changed files with 45 additions and 4 deletions

View File

@@ -0,0 +1,40 @@
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)