mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-07-16 09:58:10 +00:00
test: add test for PaginatedResponse url field population in dynamic routes
This commit is contained in:
parent
b5047db685
commit
200f10b093
1 changed files with 24 additions and 1 deletions
|
@ -5,10 +5,12 @@
|
||||||
# the root directory of this source tree.
|
# the root directory of this source tree.
|
||||||
|
|
||||||
import asyncio
|
import asyncio
|
||||||
|
from unittest.mock import AsyncMock, MagicMock
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
from llama_stack.distribution.server.server import create_sse_event, sse_generator
|
from llama_stack.apis.common.responses import PaginatedResponse
|
||||||
|
from llama_stack.distribution.server.server import create_dynamic_typed_route, create_sse_event, sse_generator
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.asyncio
|
@pytest.mark.asyncio
|
||||||
|
@ -89,3 +91,24 @@ async def test_sse_generator_error_before_response_starts():
|
||||||
# We should have 1 error event
|
# We should have 1 error event
|
||||||
assert len(seen_events) == 1
|
assert len(seen_events) == 1
|
||||||
assert 'data: {"error":' in seen_events[0]
|
assert 'data: {"error":' in seen_events[0]
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_paginated_response_url_setting():
|
||||||
|
"""Test that PaginatedResponse gets url set to route path."""
|
||||||
|
|
||||||
|
async def mock_api_method():
|
||||||
|
return PaginatedResponse(data=[], has_more=False, url=None)
|
||||||
|
|
||||||
|
route_handler = create_dynamic_typed_route(mock_api_method, "get", "/test/route")
|
||||||
|
|
||||||
|
# Mock minimal request
|
||||||
|
request = MagicMock()
|
||||||
|
request.scope = {"user_attributes": {}, "principal": ""}
|
||||||
|
request.headers = {}
|
||||||
|
request.body = AsyncMock(return_value=b"")
|
||||||
|
|
||||||
|
result = await route_handler(request)
|
||||||
|
|
||||||
|
assert isinstance(result, PaginatedResponse)
|
||||||
|
assert result.url == "/test/route"
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue