mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-16 13:52:37 +00:00
minimize change
This commit is contained in:
parent
ed78090b8e
commit
06d02bf3de
23 changed files with 161 additions and 197 deletions
|
|
@ -64,9 +64,6 @@ async def test_sqlite_sqlstore():
|
|||
assert result.data == [{"id": 12, "name": "test12"}]
|
||||
assert result.has_more is False
|
||||
|
||||
# cleanup
|
||||
await sqlstore.close()
|
||||
|
||||
|
||||
async def test_sqlstore_pagination_basic():
|
||||
"""Test basic pagination functionality at the SQL store level."""
|
||||
|
|
@ -131,8 +128,6 @@ async def test_sqlstore_pagination_basic():
|
|||
assert result3.data[0]["id"] == "zebra"
|
||||
assert result3.has_more is False
|
||||
|
||||
await store.close()
|
||||
|
||||
|
||||
async def test_sqlstore_pagination_with_filter():
|
||||
"""Test pagination with WHERE conditions."""
|
||||
|
|
@ -185,8 +180,6 @@ async def test_sqlstore_pagination_with_filter():
|
|||
assert result2.data[0]["id"] == "xyz"
|
||||
assert result2.has_more is False
|
||||
|
||||
await store.close()
|
||||
|
||||
|
||||
async def test_sqlstore_pagination_ascending_order():
|
||||
"""Test pagination with ascending order."""
|
||||
|
|
@ -235,8 +228,6 @@ async def test_sqlstore_pagination_ascending_order():
|
|||
assert result2.data[0]["id"] == "alpha"
|
||||
assert result2.has_more is True
|
||||
|
||||
await store.close()
|
||||
|
||||
|
||||
async def test_sqlstore_pagination_multi_column_ordering_error():
|
||||
"""Test that multi-column ordering raises an error when using cursor pagination."""
|
||||
|
|
@ -274,8 +265,6 @@ async def test_sqlstore_pagination_multi_column_ordering_error():
|
|||
assert len(result.data) == 1
|
||||
assert result.data[0]["id"] == "task1"
|
||||
|
||||
await store.close()
|
||||
|
||||
|
||||
async def test_sqlstore_pagination_cursor_requires_order_by():
|
||||
"""Test that cursor pagination requires order_by parameter."""
|
||||
|
|
@ -293,8 +282,6 @@ async def test_sqlstore_pagination_cursor_requires_order_by():
|
|||
cursor=("id", "task1"),
|
||||
)
|
||||
|
||||
await store.close()
|
||||
|
||||
|
||||
async def test_sqlstore_pagination_error_handling():
|
||||
"""Test error handling for invalid columns and cursor IDs."""
|
||||
|
|
@ -427,8 +414,6 @@ async def test_where_operator_edge_cases():
|
|||
with pytest.raises(ValueError, match="Unsupported operator"):
|
||||
await store.fetch_all("events", where={"ts": {"!=": base}})
|
||||
|
||||
await store.close()
|
||||
|
||||
|
||||
async def test_sqlstore_pagination_custom_key_column():
|
||||
"""Test pagination with custom primary key column (not 'id')."""
|
||||
|
|
@ -478,5 +463,3 @@ async def test_sqlstore_pagination_custom_key_column():
|
|||
assert len(result2.data) == 1
|
||||
assert result2.data[0]["uuid"] == "uuid-alpha"
|
||||
assert result2.has_more is False
|
||||
|
||||
await store.close()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue