""" test_yuanbao_integration.py - Yuanbao 模块集成测试 验证各模块能正确组装和交互： - YuanbaoAdapter 初始化 - Config / Platform 枚举 - get_connected_platforms 逻辑 - Proto 编解码 round-trip - Markdown 分块 - API / Media 模块 import - Toolset 注册 """ import sys import os # 确保 hermes-agent 根目录在 sys.path 中 _REPO_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) if _REPO_ROOT not in sys.path: sys.path.insert(0, _REPO_ROOT) import pytest from unittest.mock import AsyncMock, MagicMock, patch from gateway.config import Platform, PlatformConfig, GatewayConfig from gateway.platforms.yuanbao import YuanbaoAdapter def make_config(**kwargs): extra = kwargs.pop("extra", {}) extra.setdefault("app_id", "test_key") extra.setdefault("app_secret", "test_secret") extra.setdefault("ws_url", "wss://test.example.com/ws") extra.setdefault("api_domain", "https://test.example.com") return PlatformConfig( extra=extra, **kwargs, ) # =========================================================== # 1. Adapter 初始化 # =========================================================== class TestYuanbaoAdapterInit: def test_create_adapter(self): config = make_config() adapter = YuanbaoAdapter(config) assert adapter is not None assert adapter.PLATFORM == Platform.YUANBAO def test_initial_state(self): config = make_config() adapter = YuanbaoAdapter(config) status = adapter.get_status() assert status["connected"] == False assert status["bot_id"] is None # =========================================================== # 2. Config / Platform 枚举 # =========================================================== class TestYuanbaoConfig: def test_platform_enum(self): assert Platform.YUANBAO.value == "yuanbao" def test_config_fields(self): config = make_config() assert config.extra["app_id"] == "test_key" assert config.extra["app_secret"] == "test_secret" def test_get_connected_platforms_requires_key_and_secret(self): # Only key, no secret → not in connected list gw_only_key = GatewayConfig( platforms={ Platform.YUANBAO: PlatformConfig( enabled=True, extra={"app_id": "key"}, ) } ) platforms = gw_only_key.get_connected_platforms() assert Platform.YUANBAO not in platforms # key + secret both present → in connected list gw_full = GatewayConfig( platforms={ Platform.YUANBAO: PlatformConfig( enabled=True, extra={"app_id": "key", "app_secret": "secret"}, ) } ) platforms2 = gw_full.get_connected_platforms() assert Platform.YUANBAO in platforms2 # =========================================================== # 3. GatewayRunner 注册 # =========================================================== class TestGatewayRunnerRegistration: def test_yuanbao_in_platform_enum(self): """Platform 枚举包含 YUANBAO""" assert hasattr(Platform, "YUANBAO") assert Platform.YUANBAO.value == "yuanbao" def _make_minimal_runner(self, config): """通过 __new__ + 最小初始化绕过 run.py 的模块级 dotenv/ssl 副作用""" import sys from unittest.mock import MagicMock # Stub out heavy dependencies if not already present stubs = [ "dotenv", "hermes_cli.env_loader", "hermes_cli.config", "hermes_constants", ] _orig = {} for mod in stubs: if mod not in sys.modules: _orig[mod] = None sys.modules[mod] = MagicMock() try: from gateway.run import GatewayRunner finally: # Restore only the ones we injected for mod, orig in _orig.items(): if orig is None: sys.modules.pop(mod, None) runner = GatewayRunner.__new__(GatewayRunner) runner.config = config runner.adapters = {} runner._failed_platforms = {} runner._session_model_overrides = {} return runner, GatewayRunner def test_runner_creates_yuanbao_adapter(self): """GatewayRunner._create_adapter 能为 YUANBAO 返回 YuanbaoAdapter 实例""" from gateway.config import GatewayConfig from unittest.mock import patch config = make_config(enabled=True) gw_config = GatewayConfig(platforms={Platform.YUANBAO: config}) try: runner, _ = self._make_minimal_runner(gw_config) # websockets 在测试环境可能未安装，mock 掉 WEBSOCKETS_AVAILABLE with patch("gateway.platforms.yuanbao.WEBSOCKETS_AVAILABLE", True): adapter = runner._create_adapter(Platform.YUANBAO, config) except ImportError as e: pytest.skip(f"run.py import unavailable in test env: {e}") assert adapter is not None assert isinstance(adapter, YuanbaoAdapter) def test_runner_adapter_platform_attr(self): """创建的 adapter.PLATFORM 为 Platform.YUANBAO""" from gateway.config import GatewayConfig from unittest.mock import patch config = make_config(enabled=True) gw_config = GatewayConfig(platforms={Platform.YUANBAO: config}) try: runner, _ = self._make_minimal_runner(gw_config) with patch("gateway.platforms.yuanbao.WEBSOCKETS_AVAILABLE", True): adapter = runner._create_adapter(Platform.YUANBAO, config) except ImportError as e: pytest.skip(f"run.py import unavailable in test env: {e}") assert adapter is not None assert adapter.PLATFORM == Platform.YUANBAO # =========================================================== # 4. Proto round-trip # =========================================================== class TestProtoRoundTrip: """验证 proto 编解码基本功能""" def test_conn_msg_roundtrip(self): from gateway.platforms.yuanbao_proto import encode_conn_msg, decode_conn_msg encoded = encode_conn_msg(msg_type=1, seq_no=42, data=b"hello") decoded = decode_conn_msg(encoded) assert decoded["seq_no"] == 42 assert decoded["data"] == b"hello" def test_text_elem_encoding(self): from gateway.platforms.yuanbao_proto import encode_send_c2c_message msg = encode_send_c2c_message( to_account="user123", msg_body=[{"msg_type": "TIMTextElem", "msg_content": {"text": "hello"}}], from_account="bot456", ) assert isinstance(msg, bytes) assert len(msg) > 0 # =========================================================== # 5. Markdown 分块 # =========================================================== class TestMarkdownChunking: def test_chunks_are_sent_separately(self): from gateway.platforms.yuanbao import MarkdownProcessor long_text = "paragraph\n\n" * 100 chunks = MarkdownProcessor.chunk_markdown_text(long_text, 200) assert len(chunks) > 1 for c in chunks: # 段落原子块允许轻微超限，仅验证不崩溃 assert isinstance(c, str) assert len(c) > 0 def test_chunk_short_text_no_split(self): from gateway.platforms.yuanbao import MarkdownProcessor text = "hello world" chunks = MarkdownProcessor.chunk_markdown_text(text, 3000) assert chunks == [text] # =========================================================== # 6. Sign Token 模块 # =========================================================== class TestSignToken: def test_import_ok(self): from gateway.platforms.yuanbao import SignManager assert callable(SignManager.get_token) assert callable(SignManager.force_refresh) # =========================================================== # 6b. ConnectionManager / OutboundManager # =========================================================== class TestManagerImports: def test_connection_manager_import(self): from gateway.platforms.yuanbao import ConnectionManager assert ConnectionManager is not None def test_outbound_manager_import(self): from gateway.platforms.yuanbao import OutboundManager assert OutboundManager is not None def test_message_sender_import(self): from gateway.platforms.yuanbao import MessageSender assert MessageSender is not None def test_heartbeat_manager_import(self): from gateway.platforms.yuanbao import HeartbeatManager assert HeartbeatManager is not None def test_slow_response_notifier_import(self): from gateway.platforms.yuanbao import SlowResponseNotifier assert SlowResponseNotifier is not None def test_adapter_has_outbound_manager(self): adapter = YuanbaoAdapter(make_config()) from gateway.platforms.yuanbao import ConnectionManager, OutboundManager assert isinstance(adapter._connection, ConnectionManager) assert isinstance(adapter._outbound, OutboundManager) def test_outbound_composes_sub_managers(self): adapter = YuanbaoAdapter(make_config()) from gateway.platforms.yuanbao import MessageSender, HeartbeatManager, SlowResponseNotifier assert isinstance(adapter._outbound.sender, MessageSender) assert isinstance(adapter._outbound.heartbeat, HeartbeatManager) assert isinstance(adapter._outbound.slow_notifier, SlowResponseNotifier) # =========================================================== # 7. Media 模块 # =========================================================== class TestMediaModule: def test_import_ok(self): from gateway.platforms.yuanbao_media import upload_to_cos, download_url assert callable(upload_to_cos) assert callable(download_url) # =========================================================== # 8. Toolset 注册 # =========================================================== class TestToolset: def test_yuanbao_toolset_registered(self): """toolsets.py 中存在 hermes-yuanbao 键""" import importlib ts = importlib.import_module("toolsets") assert hasattr(ts, "TOOLSETS") or hasattr(ts, "toolsets") toolsets_dict = getattr(ts, "TOOLSETS", getattr(ts, "toolsets", {})) assert "hermes-yuanbao" in toolsets_dict def test_tools_import(self): from tools.yuanbao_tools import ( get_group_info, query_group_members, send_dm, ) assert all(callable(f) for f in [ get_group_info, query_group_members, send_dm, ]) # =========================================================== # 9. platforms/__init__.py 导出 # =========================================================== class TestPlatformInit: def test_yuanbao_adapter_exported(self): """gateway.platforms.__init__.py 应导出 YuanbaoAdapter""" from gateway.platforms import YuanbaoAdapter as _YuanbaoAdapter assert _YuanbaoAdapter is YuanbaoAdapter # =========================================================== # 10. P0 fixes verification # =========================================================== import asyncio import collections class TestP0ReconnectGuard: """P0-1: _reconnecting flag prevents concurrent reconnect attempts.""" def test_reconnecting_flag_initialized(self): adapter = YuanbaoAdapter(make_config()) assert hasattr(adapter._connection, '_reconnecting') assert adapter._connection._reconnecting is False def test_schedule_reconnect_skips_when_not_running(self): adapter = YuanbaoAdapter(make_config()) adapter._running = False adapter._connection._reconnecting = False adapter._connection.schedule_reconnect() # No task should be created because _running is False def test_schedule_reconnect_skips_when_already_reconnecting(self): adapter = YuanbaoAdapter(make_config()) adapter._running = True adapter._connection._reconnecting = True adapter._connection.schedule_reconnect() # No new task should be created because already reconnecting class TestP0InboundTaskTracking: """P0-2: _inbound_tasks set is initialized and usable.""" def test_inbound_tasks_initialized(self): adapter = YuanbaoAdapter(make_config()) assert hasattr(adapter, '_inbound_tasks') assert isinstance(adapter._inbound_tasks, set) assert len(adapter._inbound_tasks) == 0 class TestP0ChatLockEviction: """P0-3: get_chat_lock uses OrderedDict and safe eviction.""" def test_chat_locks_is_ordered_dict(self): adapter = YuanbaoAdapter(make_config()) assert isinstance(adapter._outbound._chat_locks, collections.OrderedDict) def test_eviction_skips_locked(self): """When eviction is needed, locked entries are skipped.""" adapter = YuanbaoAdapter(make_config()) from gateway.platforms.yuanbao import OutboundManager # Fill to capacity with unlocked locks for i in range(OutboundManager.CHAT_DICT_MAX_SIZE): adapter._outbound._chat_locks[f"chat_{i}"] = asyncio.Lock() # Lock the oldest entry oldest_key = next(iter(adapter._outbound._chat_locks)) oldest_lock = adapter._outbound._chat_locks[oldest_key] # Simulate a held lock by acquiring it in a non-async way (set _locked) # asyncio.Lock is not held until actually acquired; so we test the # method logic by acquiring the first lock manually. # For a sync test, we check that get_chat_lock doesn't crash. new_lock = adapter._outbound.get_chat_lock("new_chat") assert "new_chat" in adapter._outbound._chat_locks assert isinstance(new_lock, asyncio.Lock) # The oldest unlocked entry should have been evicted assert len(adapter._outbound._chat_locks) == OutboundManager.CHAT_DICT_MAX_SIZE def test_move_to_end_on_access(self): """Accessing an existing key moves it to the end (MRU).""" adapter = YuanbaoAdapter(make_config()) adapter._outbound._chat_locks["a"] = asyncio.Lock() adapter._outbound._chat_locks["b"] = asyncio.Lock() adapter._outbound._chat_locks["c"] = asyncio.Lock() # Access "a" — should move to end adapter._outbound.get_chat_lock("a") keys = list(adapter._outbound._chat_locks.keys()) assert keys[-1] == "a" assert keys[0] == "b" class TestP0PlatformScopedLock: """P0-4: connect() calls _acquire_platform_lock.""" def test_adapter_has_platform_lock_methods(self): adapter = YuanbaoAdapter(make_config()) assert hasattr(adapter, '_acquire_platform_lock') assert hasattr(adapter, '_release_platform_lock') if __name__ == "__main__": pytest.main([__file__, "-v"])