From bb61cced3c99c254f19e6753517b07760235e944 Mon Sep 17 00:00:00 2001 From: lorenzejay Date: Wed, 5 Aug 2026 15:33:49 -0700 Subject: [PATCH] Update LLM event tests to patch the global event bus --- lib/crewai/tests/events/test_llm_finish_reason_response_id.py | 4 ++-- lib/crewai/tests/events/test_llm_usage_event.py | 4 ++-- lib/crewai/tests/llms/google/test_google.py | 4 ++-- lib/crewai/tests/test_llm.py | 4 ++-- lib/crewai/tests/test_llm_streaming_finish_reason.py | 4 ++-- 5 files changed, 10 insertions(+), 10 deletions(-) diff --git a/lib/crewai/tests/events/test_llm_finish_reason_response_id.py b/lib/crewai/tests/events/test_llm_finish_reason_response_id.py index 091875fdf..5a1d4892d 100644 --- a/lib/crewai/tests/events/test_llm_finish_reason_response_id.py +++ b/lib/crewai/tests/events/test_llm_finish_reason_response_id.py @@ -4,7 +4,7 @@ from unittest.mock import MagicMock, patch import pytest -from crewai.events.event_bus import CrewAIEventsBus +from crewai.events.event_bus import crewai_event_bus from crewai.events.types.llm_events import ( LLMCallCompletedEvent, LLMCallStartedEvent, @@ -31,7 +31,7 @@ class _StubLLM(BaseLLM): @pytest.fixture def mock_emit(): - with patch.object(CrewAIEventsBus, "emit") as mock: + with patch.object(crewai_event_bus, "emit") as mock: yield mock diff --git a/lib/crewai/tests/events/test_llm_usage_event.py b/lib/crewai/tests/events/test_llm_usage_event.py index d0b29c863..313ed2ef5 100644 --- a/lib/crewai/tests/events/test_llm_usage_event.py +++ b/lib/crewai/tests/events/test_llm_usage_event.py @@ -4,7 +4,7 @@ from unittest.mock import patch import pytest from pydantic import BaseModel -from crewai.events.event_bus import CrewAIEventsBus +from crewai.events.event_bus import crewai_event_bus from crewai.events.types.llm_events import LLMCallCompletedEvent, LLMCallType from crewai.llm import LLM from crewai.llms.base_llm import BaseLLM @@ -203,7 +203,7 @@ class _StubLLM(BaseLLM): class TestEmitCallCompletedEventPassesUsage: @pytest.fixture def mock_emit(self): - with patch.object(CrewAIEventsBus, "emit") as mock: + with patch.object(crewai_event_bus, "emit") as mock: yield mock @pytest.fixture diff --git a/lib/crewai/tests/llms/google/test_google.py b/lib/crewai/tests/llms/google/test_google.py index 0213eb525..2c03dcd23 100644 --- a/lib/crewai/tests/llms/google/test_google.py +++ b/lib/crewai/tests/llms/google/test_google.py @@ -123,12 +123,12 @@ def test_gemini_completion_initialization_parameters(): def test_gemini_started_event_surfaces_max_output_tokens(): - from crewai.events.event_bus import CrewAIEventsBus + from crewai.events.event_bus import crewai_event_bus from crewai.events.types.llm_events import LLMCallStartedEvent llm = LLM(model="google/gemini-2.0-flash-001", max_output_tokens=2000, api_key="test-key") - with patch.object(CrewAIEventsBus, "emit") as mock_emit: + with patch.object(crewai_event_bus, "emit") as mock_emit: llm._emit_call_started_event(messages="hi") event = mock_emit.call_args[1]["event"] diff --git a/lib/crewai/tests/test_llm.py b/lib/crewai/tests/test_llm.py index 1c98d751e..7e60b18c4 100644 --- a/lib/crewai/tests/test_llm.py +++ b/lib/crewai/tests/test_llm.py @@ -534,9 +534,9 @@ def assert_event_count( @pytest.fixture def mock_emit() -> MagicMock: - from crewai.events.event_bus import CrewAIEventsBus + from crewai.events.event_bus import crewai_event_bus - with patch.object(CrewAIEventsBus, "emit") as mock_emit: + with patch.object(crewai_event_bus, "emit") as mock_emit: yield mock_emit diff --git a/lib/crewai/tests/test_llm_streaming_finish_reason.py b/lib/crewai/tests/test_llm_streaming_finish_reason.py index ff8a94d4e..c90fe77a4 100644 --- a/lib/crewai/tests/test_llm_streaming_finish_reason.py +++ b/lib/crewai/tests/test_llm_streaming_finish_reason.py @@ -11,14 +11,14 @@ from unittest.mock import patch import pytest -from crewai.events.event_bus import CrewAIEventsBus +from crewai.events.event_bus import crewai_event_bus from crewai.events.types.llm_events import LLMCallCompletedEvent from crewai.llm import LLM @pytest.fixture def mock_emit(): - with patch.object(CrewAIEventsBus, "emit") as mock: + with patch.object(crewai_event_bus, "emit") as mock: yield mock