mirror of
https://gitee.com/wanwujie/deer-flow
synced 2026-04-05 15:10:20 +08:00
34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
|
"""Test configuration for the backend test suite.
|
||
|
|
|
||
|
|
Sets up sys.path and pre-mocks modules that would cause circular import
|
||
|
|
issues when unit-testing lightweight config/registry code in isolation.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import sys
|
||
|
|
from pathlib import Path
|
||
|
|
from unittest.mock import MagicMock
|
||
|
|
|
||
|
|
# Make 'src' importable from any working directory
|
||
|
|
sys.path.insert(0, str(Path(__file__).parent.parent))
|
||
|
|
|
||
|
|
# Break the circular import chain that exists in production code:
|
||
|
|
# src.subagents.__init__
|
||
|
|
# -> .executor (SubagentExecutor, SubagentResult)
|
||
|
|
# -> src.agents.thread_state
|
||
|
|
# -> src.agents.__init__
|
||
|
|
# -> lead_agent.agent
|
||
|
|
# -> subagent_limit_middleware
|
||
|
|
# -> src.subagents.executor <-- circular!
|
||
|
|
#
|
||
|
|
# By injecting a mock for src.subagents.executor *before* any test module
|
||
|
|
# triggers the import, __init__.py's "from .executor import ..." succeeds
|
||
|
|
# immediately without running the real executor module.
|
||
|
|
_executor_mock = MagicMock()
|
||
|
|
_executor_mock.SubagentExecutor = MagicMock
|
||
|
|
_executor_mock.SubagentResult = MagicMock
|
||
|
|
_executor_mock.SubagentStatus = MagicMock
|
||
|
|
_executor_mock.MAX_CONCURRENT_SUBAGENTS = 3
|
||
|
|
_executor_mock.get_background_task_result = MagicMock()
|
||
|
|
|
||
|
|
sys.modules["src.subagents.executor"] = _executor_mock
|