mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-15 17:30:13 +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
22 lines
727 B
Python
22 lines
727 B
Python
"""Import smoke tests for the knowledge base helper module."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from astrbot.core.knowledge_base.kb_helper import KBHelper, RateLimiter
|
|
|
|
|
|
class TestKBHelperImports:
|
|
"""Verify that the main classes from kb_helper can be imported."""
|
|
|
|
def test_import_kb_helper(self):
|
|
assert KBHelper is not None
|
|
assert hasattr(KBHelper, "initialize")
|
|
assert hasattr(KBHelper, "upload_document")
|
|
assert hasattr(KBHelper, "list_documents")
|
|
assert hasattr(KBHelper, "delete_document")
|
|
|
|
def test_import_rate_limiter(self):
|
|
assert RateLimiter is not None
|
|
assert hasattr(RateLimiter, "__aenter__")
|
|
assert hasattr(RateLimiter, "__aexit__")
|