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
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
import pytest
|
|
|
|
from astrbot.core.agent.runners.deerflow.deerflow_api_client import (
|
|
DeerFlowAPIClient,
|
|
DeerFlowAPIError,
|
|
)
|
|
|
|
|
|
class _FakeDeleteResponse:
|
|
def __init__(self, status: int, body: str):
|
|
self.status = status
|
|
self._body = body
|
|
|
|
async def __aenter__(self):
|
|
return self
|
|
|
|
async def __aexit__(self, exc_type, exc, tb):
|
|
_ = exc_type, exc, tb
|
|
|
|
async def text(self) -> str:
|
|
return self._body
|
|
|
|
|
|
class _FakeSession:
|
|
def __init__(self, response: _FakeDeleteResponse):
|
|
self.closed = False
|
|
self._response = response
|
|
|
|
def delete(self, *args, **kwargs):
|
|
_ = args, kwargs
|
|
return self._response
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_delete_thread_raises_api_error_with_thread_context():
|
|
client = DeerFlowAPIClient(api_base="http://127.0.0.1:2026")
|
|
client._session = _FakeSession(
|
|
_FakeDeleteResponse(status=500, body="thread cleanup failed"),
|
|
)
|
|
|
|
with pytest.raises(DeerFlowAPIError) as exc_info:
|
|
await client.delete_thread("thread-123")
|
|
client._closed = True
|
|
|
|
assert exc_info.value.status == 500
|
|
assert exc_info.value.thread_id == "thread-123"
|
|
assert "/api/threads/thread-123" in str(exc_info.value)
|
|
assert "thread cleanup failed" in str(exc_info.value)
|