@@ -2459,6 +2459,7 @@ async def test_on_message_send_stream_task_id_provided_but_task_not_found():
24592459def test_init_with_default_queue_manager_issues_deprecation_warning ():
24602460 """Test that initializing with default queue_manager issues deprecation warning."""
24612461 import warnings
2462+
24622463 from unittest .mock import MagicMock
24632464
24642465 with warnings .catch_warnings (record = True ) as w :
@@ -2477,6 +2478,7 @@ def test_init_with_default_queue_manager_issues_deprecation_warning():
24772478def test_init_with_explicit_queue_manager_no_warning ():
24782479 """Test that initializing with explicit queue_manager does not issue warning."""
24792480 import warnings
2481+
24802482 from unittest .mock import MagicMock
24812483
24822484 with warnings .catch_warnings (record = True ) as w :
@@ -2501,6 +2503,7 @@ def test_init_with_explicit_queue_manager_no_warning():
25012503async def test_init_with_disabled_fallback_raises_error ():
25022504 """Test that disabling fallback raises ValueError when queue_manager is None."""
25032505 import os
2506+
25042507 from unittest .mock import MagicMock
25052508
25062509 # Set environment variable to disable fallback
@@ -2525,6 +2528,7 @@ async def test_init_with_disabled_fallback_false_allows_default():
25252528 """Test that setting A2A_DISABLE_QUEUE_MANAGER_FALLBACK=false allows default behavior."""
25262529 import os
25272530 import warnings
2531+
25282532 from unittest .mock import MagicMock
25292533
25302534 # Set environment variable to explicitly allow fallback
@@ -2553,6 +2557,7 @@ async def test_init_with_disabled_fallback_false_allows_default():
25532557def test_environment_variable_parsing ():
25542558 """Test that environment variable accepts various true/false values."""
25552559 import os
2560+
25562561 from unittest .mock import MagicMock
25572562
25582563 test_cases = [
0 commit comments