4e495f658d
Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Laure Hugo <201583486+laure0303@users.noreply.github.com> Co-authored-by: Paul VEZIA <166131032+le-codeur-rapide@users.noreply.github.com> Co-authored-by: Peter Evers <peter.evers@mistral.ai> Co-authored-by: Jules YZERD <newtonlormont@gmail.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
123 lines
3.4 KiB
Python
123 lines
3.4 KiB
Python
from __future__ import annotations
|
|
|
|
from types import SimpleNamespace
|
|
from typing import cast
|
|
from unittest.mock import AsyncMock
|
|
|
|
from acp.schema import AgentMessageChunk
|
|
import pytest
|
|
|
|
from tests.stubs.fake_client import FakeClient
|
|
from vibe.acp.acp_agent_loop import VibeAcpAgentLoop
|
|
from vibe.acp.session import AcpSessionLoop
|
|
from vibe.cli.textual_ui.app import VibeApp
|
|
from vibe.cli.textual_ui.widgets.messages import UserCommandMessage
|
|
from vibe.core.config import MCPOAuth, MCPStreamableHttp
|
|
from vibe.core.tools.mcp import MCPRegistry
|
|
|
|
|
|
def _registry_with_uncached_oauth(alias: str, *, disabled: bool = False) -> MCPRegistry:
|
|
registry = MCPRegistry()
|
|
registry.sync_active_servers([
|
|
MCPStreamableHttp(
|
|
name=alias,
|
|
transport="streamable-http",
|
|
url="https://mcp.example.com/mcp",
|
|
auth=MCPOAuth(type="oauth", scopes=["read"]),
|
|
disabled=disabled,
|
|
)
|
|
])
|
|
assert registry.needs_auth == set()
|
|
return registry
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_tui_mcp_auth_notice_uses_status_for_uncached_oauth() -> None:
|
|
mount = AsyncMock()
|
|
app = cast(
|
|
VibeApp,
|
|
SimpleNamespace(
|
|
agent_loop=SimpleNamespace(
|
|
mcp_registry=_registry_with_uncached_oauth("sentry")
|
|
),
|
|
_mount_and_scroll=mount,
|
|
),
|
|
)
|
|
|
|
await VibeApp._show_mcp_auth_required_notice(app)
|
|
|
|
mount.assert_awaited_once()
|
|
args = mount.await_args
|
|
assert args is not None
|
|
message = args.args[0]
|
|
assert isinstance(message, UserCommandMessage)
|
|
assert "sentry" in message._content
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_tui_mcp_auth_notice_skips_disabled_servers() -> None:
|
|
mount = AsyncMock()
|
|
app = cast(
|
|
VibeApp,
|
|
SimpleNamespace(
|
|
agent_loop=SimpleNamespace(
|
|
mcp_registry=_registry_with_uncached_oauth("sentry", disabled=True)
|
|
),
|
|
_mount_and_scroll=mount,
|
|
),
|
|
)
|
|
|
|
await VibeApp._show_mcp_auth_required_notice(app)
|
|
|
|
mount.assert_not_awaited()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_acp_mcp_auth_notice_skips_disabled_servers() -> None:
|
|
agent = VibeAcpAgentLoop()
|
|
client = FakeClient()
|
|
agent.on_connect(client)
|
|
session = cast(
|
|
AcpSessionLoop,
|
|
SimpleNamespace(
|
|
id="session-id",
|
|
agent_loop=SimpleNamespace(
|
|
mcp_registry=_registry_with_uncached_oauth("sentry", disabled=True)
|
|
),
|
|
),
|
|
)
|
|
|
|
await agent._notify_mcp_auth_required(session)
|
|
|
|
messages = [
|
|
notification.update
|
|
for notification in client._session_updates
|
|
if isinstance(notification.update, AgentMessageChunk)
|
|
]
|
|
assert messages == []
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_acp_mcp_auth_notice_uses_status_for_uncached_oauth() -> None:
|
|
agent = VibeAcpAgentLoop()
|
|
client = FakeClient()
|
|
agent.on_connect(client)
|
|
session = cast(
|
|
AcpSessionLoop,
|
|
SimpleNamespace(
|
|
id="session-id",
|
|
agent_loop=SimpleNamespace(
|
|
mcp_registry=_registry_with_uncached_oauth("sentry")
|
|
),
|
|
),
|
|
)
|
|
|
|
await agent._notify_mcp_auth_required(session)
|
|
|
|
messages = [
|
|
notification.update
|
|
for notification in client._session_updates
|
|
if isinstance(notification.update, AgentMessageChunk)
|
|
]
|
|
assert len(messages) == 1
|
|
assert "sentry" in messages[0].content.text
|