90763daf81
Co-authored-by: Bastien <bastien.baret@gmail.com> Co-authored-by: Laure Hugo <201583486+laure0303@users.noreply.github.com> Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com> Co-authored-by: Paul Cacheux <paul.cacheux@mistral.ai> Co-authored-by: Val <102326092+vdeva@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
121 lines
4.2 KiB
Python
121 lines
4.2 KiB
Python
from __future__ import annotations
|
|
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from vibe.core.session.resume_sessions import (
|
|
SHORT_SESSION_ID_LEN,
|
|
list_remote_resume_sessions,
|
|
short_session_id,
|
|
)
|
|
|
|
|
|
class TestShortSessionId:
|
|
def test_local_shortens_to_first_chars(self) -> None:
|
|
sid = "abcdef1234567890"
|
|
result = short_session_id(sid)
|
|
assert result == sid[:SHORT_SESSION_ID_LEN]
|
|
assert len(result) == SHORT_SESSION_ID_LEN
|
|
|
|
def test_local_is_default(self) -> None:
|
|
sid = "abcdef1234567890"
|
|
assert short_session_id(sid) == short_session_id(sid, source="local")
|
|
|
|
def test_remote_shortens_to_last_chars(self) -> None:
|
|
sid = "abcdef1234567890"
|
|
result = short_session_id(sid, source="remote")
|
|
assert result == sid[-SHORT_SESSION_ID_LEN:]
|
|
assert len(result) == SHORT_SESSION_ID_LEN
|
|
|
|
def test_returns_full_id_when_shorter_than_limit(self) -> None:
|
|
sid = "abc"
|
|
assert short_session_id(sid) == "abc"
|
|
assert short_session_id(sid, source="remote") == "abc"
|
|
|
|
def test_empty_string(self) -> None:
|
|
assert short_session_id("") == ""
|
|
|
|
|
|
class TestListRemoteResumeSessions:
|
|
@pytest.mark.asyncio
|
|
async def test_returns_empty_when_nuage_disabled(self) -> None:
|
|
config = MagicMock()
|
|
config.nuage_enabled = False
|
|
config.nuage_api_key = "key"
|
|
result = await list_remote_resume_sessions(config)
|
|
assert result == []
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_returns_empty_when_no_api_key(self) -> None:
|
|
config = MagicMock()
|
|
config.nuage_enabled = True
|
|
config.nuage_api_key = None
|
|
result = await list_remote_resume_sessions(config)
|
|
assert result == []
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_returns_empty_when_both_missing(self) -> None:
|
|
config = MagicMock()
|
|
config.nuage_enabled = False
|
|
config.nuage_api_key = None
|
|
result = await list_remote_resume_sessions(config)
|
|
assert result == []
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_filters_only_active_statuses(self) -> None:
|
|
from datetime import datetime
|
|
|
|
from vibe.core.nuage.workflow import (
|
|
WorkflowExecutionListResponse,
|
|
WorkflowExecutionStatus,
|
|
WorkflowExecutionWithoutResultResponse,
|
|
)
|
|
|
|
running = WorkflowExecutionWithoutResultResponse(
|
|
workflow_name="vibe",
|
|
execution_id="exec-running",
|
|
status=WorkflowExecutionStatus.RUNNING,
|
|
start_time=datetime(2026, 1, 1),
|
|
end_time=None,
|
|
)
|
|
completed = WorkflowExecutionWithoutResultResponse(
|
|
workflow_name="vibe",
|
|
execution_id="exec-completed",
|
|
status=WorkflowExecutionStatus.COMPLETED,
|
|
start_time=datetime(2026, 1, 1),
|
|
end_time=datetime(2026, 1, 2),
|
|
)
|
|
retrying = WorkflowExecutionWithoutResultResponse(
|
|
workflow_name="vibe",
|
|
execution_id="exec-retrying",
|
|
status=WorkflowExecutionStatus.RETRYING_AFTER_ERROR,
|
|
start_time=datetime(2026, 1, 1),
|
|
end_time=None,
|
|
)
|
|
|
|
mock_response = WorkflowExecutionListResponse(
|
|
executions=[running, completed, retrying]
|
|
)
|
|
|
|
config = MagicMock()
|
|
config.nuage_enabled = True
|
|
config.nuage_api_key = "test-key"
|
|
config.nuage_base_url = "https://test.example.com"
|
|
config.api_timeout = 30
|
|
config.nuage_workflow_id = "workflow-1"
|
|
|
|
with patch("vibe.core.session.resume_sessions.WorkflowsClient") as MockClient:
|
|
mock_client = AsyncMock()
|
|
mock_client.get_workflow_runs.return_value = mock_response
|
|
MockClient.return_value.__aenter__ = AsyncMock(return_value=mock_client)
|
|
MockClient.return_value.__aexit__ = AsyncMock(return_value=False)
|
|
|
|
result = await list_remote_resume_sessions(config)
|
|
|
|
assert len(result) == 2
|
|
session_ids = {s.session_id for s in result}
|
|
assert "exec-running" in session_ids
|
|
assert "exec-retrying" in session_ids
|
|
assert "exec-completed" not in session_ids
|
|
assert all(s.source == "remote" for s in result)
|