mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-16 01:40:15 +08:00
dcaaf6286a
- Add 100+ new test files covering provider sources, platform adapters, agent runners, star/plugin system, knowledge base, core utils, pipeline, computer tools, builtin commands, and dashboard routes - Fix Python type errors in sqlite.py (col() wrappers for SQLModel), astr_agent_tool_exec, core_lifecycle, star context/manager - Fix TypeScript strict mode errors across 30+ dashboard Vue files - Add import smoke tests, auth roundtrip, startup tests - Add compile-all check and CLI entry test for AUR compatibility - Restore BotMessageAccumulator and helpers lost in merge
46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
"""Tests for astrbot.core.conversation_mgr module."""
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from astrbot.core.conversation_mgr import ConversationManager
|
|
from astrbot.core.db import BaseDatabase
|
|
|
|
|
|
class TestConversationManager:
|
|
"""Smoke tests for ConversationManager."""
|
|
|
|
def test_module_import(self):
|
|
"""Verify the conversation_mgr module can be imported."""
|
|
import astrbot.core.conversation_mgr # noqa: F811
|
|
assert hasattr(
|
|
astrbot.core.conversation_mgr,
|
|
"ConversationManager",
|
|
)
|
|
|
|
def test_class_exists(self):
|
|
"""Verify ConversationManager class exists."""
|
|
assert ConversationManager.__name__ == "ConversationManager"
|
|
|
|
def test_init_requires_db_helper(self):
|
|
"""Verify ConversationManager init requires BaseDatabase."""
|
|
db = MagicMock(spec=BaseDatabase)
|
|
mgr = ConversationManager(db)
|
|
assert mgr.db is db
|
|
assert mgr.session_conversations == {}
|
|
assert mgr.save_interval == 60
|
|
|
|
def test_register_on_session_deleted_adds_callback(self):
|
|
"""Verify register_on_session_deleted adds a callback."""
|
|
db = MagicMock(spec=BaseDatabase)
|
|
mgr = ConversationManager(db)
|
|
|
|
async def dummy_callback(unified_msg_origin: str) -> None:
|
|
pass
|
|
|
|
assert len(mgr._on_session_deleted_callbacks) == 0
|
|
mgr.register_on_session_deleted(dummy_callback)
|
|
assert len(mgr._on_session_deleted_callbacks) == 1
|
|
assert mgr._on_session_deleted_callbacks[0] is dummy_callback
|