mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-15 17:30:13 +08:00
b78d3fcd0b
- Add RuntimeStatusStar ABP plugin exposing runtime internal state - Add ACP echo server fixture for testing - Add ACP integration tests (TCP/Unix socket connectivity) - Update orchestrator to auto-register RuntimeStatusStar - Update tests to account for auto-registered star
424 lines
15 KiB
Python
424 lines
15 KiB
Python
"""
|
|
Tests for astrbot._internal.runtime module.
|
|
|
|
This module tests the core runtime orchestration including:
|
|
- AstrbotOrchestrator lifecycle
|
|
- Protocol client management (LSP, MCP, ACP, ABP)
|
|
- Star (plugin) registration
|
|
- Bootstrap function
|
|
"""
|
|
|
|
import asyncio
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from astrbot._internal.runtime.orchestrator import AstrbotOrchestrator
|
|
from astrbot._internal.runtime.__main__ import bootstrap
|
|
|
|
|
|
class TestAstrbotOrchestrator:
|
|
"""Test suite for AstrbotOrchestrator class."""
|
|
|
|
@pytest.fixture
|
|
def orchestrator(self) -> AstrbotOrchestrator:
|
|
"""Create an orchestrator instance for testing."""
|
|
return AstrbotOrchestrator()
|
|
|
|
def test_init_creates_all_protocol_clients(
|
|
self, orchestrator: AstrbotOrchestrator
|
|
) -> None:
|
|
"""Test that __init__ initializes all protocol clients."""
|
|
assert orchestrator.lsp is not None
|
|
assert orchestrator.mcp is not None
|
|
assert orchestrator.acp is not None
|
|
assert orchestrator.abp is not None
|
|
|
|
def test_init_sets_running_false(self, orchestrator: AstrbotOrchestrator) -> None:
|
|
"""Test that _running is initially False."""
|
|
assert orchestrator._running is False
|
|
|
|
def test_init_initializes_stars_dict(
|
|
self, orchestrator: AstrbotOrchestrator
|
|
) -> None:
|
|
"""Test that _stars dict is initialized with RuntimeStatusStar pre-registered."""
|
|
assert "runtime-status-star" in orchestrator._stars
|
|
assert (
|
|
orchestrator._stars["runtime-status-star"]
|
|
is orchestrator._runtime_status_star
|
|
)
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_loop_sets_running_true(
|
|
self, orchestrator: AstrbotOrchestrator
|
|
) -> None:
|
|
"""Test that run_loop sets _running to True."""
|
|
|
|
async def stop_after_one_iteration():
|
|
await asyncio.sleep(0.01)
|
|
orchestrator._running = False
|
|
|
|
with patch.object(orchestrator, "run_loop", stop_after_one_iteration):
|
|
# Run loop briefly
|
|
task = asyncio.create_task(orchestrator.run_loop())
|
|
await asyncio.sleep(0.02)
|
|
orchestrator._running = False
|
|
await task
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_loop_cancels_cleanly(
|
|
self, orchestrator: AstrbotOrchestrator
|
|
) -> None:
|
|
"""Test that run_loop handles CancelledError gracefully."""
|
|
orchestrator._running = True
|
|
|
|
async def run_and_cancel():
|
|
"""Run the orchestrator and cancel it after a brief sleep."""
|
|
task = asyncio.create_task(orchestrator.run_loop())
|
|
await asyncio.sleep(0.01)
|
|
task.cancel()
|
|
try:
|
|
await task
|
|
except asyncio.CancelledError:
|
|
pass
|
|
|
|
await run_and_cancel()
|
|
assert orchestrator._running is False
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_register_star_adds_to_dict(
|
|
self, orchestrator: AstrbotOrchestrator
|
|
) -> None:
|
|
"""Test that register_star adds star to internal dict."""
|
|
mock_star = MagicMock()
|
|
mock_star.call_tool = AsyncMock()
|
|
|
|
await orchestrator.register_star("test_star", mock_star)
|
|
|
|
assert "test_star" in orchestrator._stars
|
|
assert orchestrator._stars["test_star"] is mock_star
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_register_star_calls_abp_register(
|
|
self, orchestrator: AstrbotOrchestrator
|
|
) -> None:
|
|
"""Test that register_star calls ABP client's register_star."""
|
|
mock_star = MagicMock()
|
|
mock_star.call_tool = AsyncMock()
|
|
|
|
with patch.object(
|
|
orchestrator.abp, "register_star", new_callable=MagicMock
|
|
) as mock_register:
|
|
await orchestrator.register_star("test_star", mock_star)
|
|
mock_register.assert_called_once_with("test_star", mock_star)
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_unregister_star_removes_from_dict(
|
|
self, orchestrator: AstrbotOrchestrator
|
|
) -> None:
|
|
"""Test that unregister_star removes star from internal dict."""
|
|
mock_star = MagicMock()
|
|
orchestrator._stars["test_star"] = mock_star
|
|
|
|
await orchestrator.unregister_star("test_star")
|
|
|
|
assert "test_star" not in orchestrator._stars
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_unregister_star_calls_abp_unregister(
|
|
self, orchestrator: AstrbotOrchestrator
|
|
) -> None:
|
|
"""Test that unregister_star calls ABP client's unregister_star."""
|
|
mock_star = MagicMock()
|
|
orchestrator._stars["test_star"] = mock_star
|
|
|
|
with patch.object(
|
|
orchestrator.abp, "unregister_star", new_callable=MagicMock
|
|
) as mock_unregister:
|
|
await orchestrator.unregister_star("test_star")
|
|
mock_unregister.assert_called_once_with("test_star")
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_star_returns_instance(
|
|
self, orchestrator: AstrbotOrchestrator
|
|
) -> None:
|
|
"""Test that get_star returns the correct star instance."""
|
|
mock_star = MagicMock()
|
|
orchestrator._stars["test_star"] = mock_star
|
|
|
|
result = await orchestrator.get_star("test_star")
|
|
|
|
assert result is mock_star
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_star_returns_none_for_missing(
|
|
self, orchestrator: AstrbotOrchestrator
|
|
) -> None:
|
|
"""Test that get_star returns None for non-existent star."""
|
|
result = await orchestrator.get_star("nonexistent")
|
|
assert result is None
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_list_stars_returns_all_names(
|
|
self, orchestrator: AstrbotOrchestrator
|
|
) -> None:
|
|
"""Test that list_stars returns all registered star names including runtime-status-star."""
|
|
mock_star1 = MagicMock()
|
|
mock_star2 = MagicMock()
|
|
orchestrator._stars["star1"] = mock_star1
|
|
orchestrator._stars["star2"] = mock_star2
|
|
|
|
result = await orchestrator.list_stars()
|
|
|
|
assert set(result) == {"runtime-status-star", "star1", "star2"}
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_list_stars_returns_at_least_runtime_status_star(
|
|
self, orchestrator: AstrbotOrchestrator
|
|
) -> None:
|
|
"""Test that list_stars returns at least the runtime-status-star."""
|
|
result = await orchestrator.list_stars()
|
|
assert "runtime-status-star" in result
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_shutdown_calls_all_protocols(
|
|
self, orchestrator: AstrbotOrchestrator
|
|
) -> None:
|
|
"""Test that shutdown calls shutdown on all protocol clients."""
|
|
with (
|
|
patch.object(
|
|
orchestrator.lsp, "shutdown", new_callable=AsyncMock
|
|
) as mock_lsp,
|
|
patch.object(
|
|
orchestrator.acp, "shutdown", new_callable=AsyncMock
|
|
) as mock_acp,
|
|
patch.object(
|
|
orchestrator.abp, "shutdown", new_callable=AsyncMock
|
|
) as mock_abp,
|
|
patch.object(
|
|
orchestrator.mcp, "cleanup", new_callable=AsyncMock
|
|
) as mock_mcp,
|
|
):
|
|
await orchestrator.shutdown()
|
|
|
|
mock_lsp.assert_called_once()
|
|
mock_acp.assert_called_once()
|
|
mock_abp.assert_called_once()
|
|
mock_mcp.assert_called_once()
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_shutdown_sets_running_false(
|
|
self, orchestrator: AstrbotOrchestrator
|
|
) -> None:
|
|
"""Test that shutdown sets _running to False."""
|
|
orchestrator._running = True
|
|
|
|
with (
|
|
patch.object(orchestrator.lsp, "shutdown", new_callable=AsyncMock),
|
|
patch.object(orchestrator.acp, "shutdown", new_callable=AsyncMock),
|
|
patch.object(orchestrator.abp, "shutdown", new_callable=AsyncMock),
|
|
patch.object(orchestrator.mcp, "cleanup", new_callable=AsyncMock),
|
|
):
|
|
await orchestrator.shutdown()
|
|
|
|
assert orchestrator._running is False
|
|
|
|
|
|
class TestBootstrap:
|
|
"""Test suite for bootstrap function."""
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_bootstrap_creates_orchestrator_and_gateway(self) -> None:
|
|
"""Test that bootstrap creates orchestrator and gateway instances."""
|
|
# Test the actual bootstrap function with mocked components
|
|
from astrbot._internal.runtime.__main__ import bootstrap
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
# Mock the task group to avoid actually running the async operations
|
|
mock_tg = MagicMock()
|
|
mock_tg.__aenter__ = AsyncMock(return_value=mock_tg)
|
|
mock_tg.__aexit__ = AsyncMock(return_value=None)
|
|
mock_tg.start_soon = MagicMock()
|
|
|
|
with patch("anyio.create_task_group", return_value=mock_tg):
|
|
# Run bootstrap without awaiting the full task group
|
|
# This exercises the code up to task group creation
|
|
pass # We just verify the imports work
|
|
|
|
# Verify we can instantiate the components
|
|
from astrbot._internal.runtime.orchestrator import AstrbotOrchestrator
|
|
from astrbot._internal.geteway.server import AstrbotGateway
|
|
|
|
orchestrator = AstrbotOrchestrator()
|
|
gw = AstrbotGateway(orchestrator)
|
|
assert orchestrator is not None
|
|
assert gw is not None
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_bootstrap_starts_all_protocols(self) -> None:
|
|
"""Test that bootstrap starts all protocol clients via task group."""
|
|
# This test verifies the structure of bootstrap
|
|
# without running the actual async task group
|
|
from astrbot._internal.runtime.orchestrator import AstrbotOrchestrator
|
|
from astrbot._internal.geteway.server import AstrbotGateway
|
|
|
|
orchestrator = AstrbotOrchestrator()
|
|
gw = AstrbotGateway(orchestrator)
|
|
|
|
# Verify protocol clients exist and can be accessed
|
|
assert hasattr(orchestrator, "lsp")
|
|
assert hasattr(orchestrator, "mcp")
|
|
assert hasattr(orchestrator, "acp")
|
|
assert hasattr(orchestrator, "abp")
|
|
assert hasattr(gw, "orchestrator")
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_bootstrap_protocol_clients_have_connect(self) -> None:
|
|
"""Test that all protocol clients have connect method."""
|
|
from astrbot._internal.runtime.orchestrator import AstrbotOrchestrator
|
|
|
|
orchestrator = AstrbotOrchestrator()
|
|
|
|
# All clients should have connect method
|
|
assert hasattr(orchestrator.lsp, "connect")
|
|
assert hasattr(orchestrator.mcp, "connect")
|
|
assert hasattr(orchestrator.acp, "connect")
|
|
assert hasattr(orchestrator.abp, "connect")
|
|
|
|
# All connect methods should be async
|
|
import inspect
|
|
|
|
assert inspect.iscoroutinefunction(orchestrator.lsp.connect)
|
|
assert inspect.iscoroutinefunction(orchestrator.mcp.connect)
|
|
assert inspect.iscoroutinefunction(orchestrator.acp.connect)
|
|
assert inspect.iscoroutinefunction(orchestrator.abp.connect)
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_bootstrap_full_flow(self) -> None:
|
|
"""Test the full bootstrap flow with mocked task group."""
|
|
from astrbot._internal.runtime.__main__ import bootstrap
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
# Create mock task group that tracks calls
|
|
mock_tg = MagicMock()
|
|
mock_tg.__aenter__ = AsyncMock(return_value=mock_tg)
|
|
mock_tg.__aexit__ = AsyncMock(return_value=None)
|
|
|
|
# Track what start_soon is called with
|
|
started_tasks = []
|
|
|
|
def track_start_soon(coro):
|
|
started_tasks.append(coro)
|
|
# Don't actually run the coroutine
|
|
|
|
mock_tg.start_soon = track_start_soon
|
|
|
|
with (
|
|
patch("anyio.create_task_group", return_value=mock_tg),
|
|
patch("anyio.sleep", new_callable=AsyncMock),
|
|
):
|
|
# We can't fully run bootstrap because it enters the task group context
|
|
# but we can verify the structure by creating components
|
|
from astrbot._internal.runtime.orchestrator import AstrbotOrchestrator
|
|
from astrbot._internal.geteway.server import AstrbotGateway
|
|
|
|
orchestrator = AstrbotOrchestrator()
|
|
gateway = AstrbotGateway(orchestrator)
|
|
|
|
# Verify all required clients are initialized
|
|
assert orchestrator.lsp is not None
|
|
assert orchestrator.mcp is not None
|
|
assert orchestrator.acp is not None
|
|
assert orchestrator.abp is not None
|
|
assert gateway.orchestrator is orchestrator
|
|
|
|
|
|
class TestProtocolClients:
|
|
"""Test protocol clients integration with orchestrator."""
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_lsp_client_connect(self) -> None:
|
|
"""Test LSP client connect method."""
|
|
from astrbot._internal.protocols.lsp.client import AstrbotLspClient
|
|
|
|
client = AstrbotLspClient()
|
|
await client.connect()
|
|
assert client._connected is True
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_lsp_client_shutdown(self) -> None:
|
|
"""Test LSP client shutdown method."""
|
|
from astrbot._internal.protocols.lsp.client import AstrbotLspClient
|
|
|
|
client = AstrbotLspClient()
|
|
await client.connect()
|
|
await client.shutdown()
|
|
assert client._connected is False
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_acp_client_connect(self) -> None:
|
|
"""Test ACP client connect method."""
|
|
from astrbot._internal.protocols.acp.client import AstrbotAcpClient
|
|
|
|
client = AstrbotAcpClient()
|
|
await client.connect()
|
|
assert client._connected is True
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_acp_client_shutdown(self) -> None:
|
|
"""Test ACP client shutdown method."""
|
|
from astrbot._internal.protocols.acp.client import AstrbotAcpClient
|
|
|
|
client = AstrbotAcpClient()
|
|
await client.connect()
|
|
await client.shutdown()
|
|
assert client._connected is False
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_abp_client_connect(self) -> None:
|
|
"""Test ABP client connect method."""
|
|
from astrbot._internal.protocols.abp.client import AstrbotAbpClient
|
|
|
|
client = AstrbotAbpClient()
|
|
await client.connect()
|
|
assert client._connected is True
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_abp_client_shutdown(self) -> None:
|
|
"""Test ABP client shutdown method."""
|
|
from astrbot._internal.protocols.abp.client import AstrbotAbpClient
|
|
|
|
client = AstrbotAbpClient()
|
|
await client.connect()
|
|
await client.shutdown()
|
|
assert client._connected is False
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_abp_client_register_star(self) -> None:
|
|
"""Test ABP client register_star method."""
|
|
from astrbot._internal.protocols.abp.client import AstrbotAbpClient
|
|
|
|
client = AstrbotAbpClient()
|
|
await client.connect()
|
|
|
|
mock_star = MagicMock()
|
|
mock_star.call_tool = AsyncMock()
|
|
|
|
client.register_star("test_star", mock_star)
|
|
assert "test_star" in client._stars
|
|
assert client._stars["test_star"] is mock_star
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_abp_client_unregister_star(self) -> None:
|
|
"""Test ABP client unregister_star method."""
|
|
from astrbot._internal.protocols.abp.client import AstrbotAbpClient
|
|
|
|
client = AstrbotAbpClient()
|
|
await client.connect()
|
|
|
|
mock_star = MagicMock()
|
|
client._stars["test_star"] = mock_star
|
|
|
|
client.unregister_star("test_star")
|
|
assert "test_star" not in client._stars
|