mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-18 03:34:20 +00:00
huanghuoguoguo
112 lines
3.3 KiB
Python
112 lines
3.3 KiB
Python
"""Unit tests for monitoring trace HTTP routes."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import datetime
|
|
from unittest.mock import AsyncMock, Mock
|
|
|
|
import pytest
|
|
import quart
|
|
|
|
from tests.factories import FakeApp
|
|
from tests.utils.import_isolation import MockLifecycleControlScope, isolated_sys_modules
|
|
|
|
|
|
pytestmark = pytest.mark.asyncio
|
|
|
|
|
|
@pytest.fixture
|
|
async def monitoring_client():
|
|
mock_app = Mock()
|
|
mock_app.Application = type('FakeMinimalApplication', (), {})
|
|
mock_entities = Mock()
|
|
mock_entities.LifecycleControlScope = MockLifecycleControlScope
|
|
|
|
clear = [
|
|
'langbot.pkg.api.http.controller.group',
|
|
'langbot.pkg.api.http.controller.groups',
|
|
'langbot.pkg.api.http.controller.groups.monitoring',
|
|
'langbot.pkg.api.http.controller.main',
|
|
]
|
|
|
|
app = FakeApp()
|
|
app.user_service = Mock()
|
|
app.user_service.verify_jwt_token = AsyncMock(return_value='test@example.com')
|
|
app.user_service.get_user_by_email = AsyncMock(return_value=Mock(email='test@example.com'))
|
|
|
|
app.monitoring_service = Mock()
|
|
app.monitoring_service.get_traces = AsyncMock(return_value=([{'trace_id': 'trace-1'}], 1))
|
|
app.monitoring_service.get_trace_details = AsyncMock(
|
|
side_effect=lambda trace_id: {
|
|
'found': trace_id == 'trace-1',
|
|
'trace_id': trace_id,
|
|
'trace': {'trace_id': trace_id} if trace_id == 'trace-1' else None,
|
|
'spans': [] if trace_id == 'trace-1' else None,
|
|
}
|
|
)
|
|
|
|
with isolated_sys_modules(
|
|
mocks={
|
|
'langbot.pkg.core.app': mock_app,
|
|
'langbot.pkg.core.entities': mock_entities,
|
|
},
|
|
clear=clear,
|
|
):
|
|
from langbot.pkg.api.http.controller.groups.monitoring import MonitoringRouterGroup
|
|
|
|
quart_app = quart.Quart(__name__)
|
|
group = MonitoringRouterGroup(app, quart_app)
|
|
await group.initialize()
|
|
|
|
yield app, quart_app.test_client()
|
|
|
|
|
|
async def test_get_traces_route_forwards_filters(monitoring_client):
|
|
app, client = monitoring_client
|
|
|
|
response = await client.get(
|
|
'/api/v1/monitoring/traces'
|
|
'?botId=bot-1'
|
|
'&pipelineId=pipeline-1'
|
|
'&sessionId=session-1'
|
|
'&status=success'
|
|
'&startTime=2026-01-01T00:00:00Z'
|
|
'&endTime=2026-01-02T00:00:00Z'
|
|
'&limit=25'
|
|
'&offset=5',
|
|
headers={'Authorization': 'Bearer test_token'},
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
data = await response.get_json()
|
|
assert data['data'] == {
|
|
'traces': [{'trace_id': 'trace-1'}],
|
|
'total': 1,
|
|
'limit': 25,
|
|
'offset': 5,
|
|
}
|
|
app.monitoring_service.get_traces.assert_awaited_once_with(
|
|
bot_ids=['bot-1'],
|
|
pipeline_ids=['pipeline-1'],
|
|
session_ids=['session-1'],
|
|
statuses=['success'],
|
|
start_time=datetime.datetime(2026, 1, 1, 0, 0),
|
|
end_time=datetime.datetime(2026, 1, 2, 0, 0),
|
|
limit=25,
|
|
offset=5,
|
|
)
|
|
|
|
|
|
async def test_get_trace_details_route_returns_404_for_missing_trace(monitoring_client):
|
|
_app, client = monitoring_client
|
|
|
|
response = await client.get(
|
|
'/api/v1/monitoring/traces/trace-missing',
|
|
headers={'Authorization': 'Bearer test_token'},
|
|
)
|
|
|
|
assert response.status_code == 404
|
|
data = await response.get_json()
|
|
assert data['code'] == -1
|
|
assert data['msg'] == 'Trace trace-missing not found'
|