v2.17.0 (#822)
Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai> Co-authored-by: Guillaume LE GOFF <guillaume.lgf@gmail.com> Co-authored-by: Hdandria <henri.dandria@mistral.ai> Co-authored-by: Ivana Dunisijevic <ivana.dunisijevic@mistral.ai> Co-authored-by: Jean Burellier <sheplu@users.noreply.github.com> Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai> Co-authored-by: Mert Unsal <mert.unsal@mistral.ai> Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com> Co-authored-by: Paul VEZIA <166131032+le-codeur-rapide@users.noreply.github.com> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Val <102326092+vdeva@users.noreply.github.com> Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com> Co-authored-by: renovate-mistral[bot] <253709520+renovate-mistral[bot]@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
This commit is contained in:
parent
564a14365e
commit
6bedf271ce
223 changed files with 10533 additions and 6947 deletions
79
tests/test_mcp_auth_notices.py
Normal file
79
tests/test_mcp_auth_notices.py
Normal file
|
|
@ -0,0 +1,79 @@
|
|||
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) -> 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"]),
|
||||
)
|
||||
])
|
||||
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_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
|
||||
Loading…
Add table
Add a link
Reference in a new issue