phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-15 17:02:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

address comments

Browse source
This commit is contained in:
Raghotham Murthy 2025-10-07 14:19:29 -07:00
parent a424815804
commit ed78090b8e
2 changed files with 14 additions and 97 deletions
Download patch file Download diff file Expand all files Collapse all files

85
tests/unit/utils/kvstore/test_sqlite_memory.py
View file

@ -9,91 +9,6 @@ from llama_stack.providers.utils.kvstore.config import SqliteKVStoreConfig
from llama_stack.providers.utils.kvstore.sqlite.sqlite import SqliteKVStoreImpl
async def test_memory_kvstore_basic_operations():
"""Test basic CRUD operations with :memory: database."""
config = SqliteKVStoreConfig(db_path=":memory:")
store = SqliteKVStoreImpl(config)
await store.initialize()
# Test set and get
await store.set("key1", "value1")
result = await store.get("key1")
assert result == "value1"
# Test get non-existent key
result = await store.get("nonexistent")
assert result is None
# Test update
await store.set("key1", "updated_value")
result = await store.get("key1")
assert result == "updated_value"
# Test delete
await store.delete("key1")
result = await store.get("key1")
assert result is None
await store.close()
async def test_memory_kvstore_range_operations():
"""Test range query operations with :memory: database."""
config = SqliteKVStoreConfig(db_path=":memory:")
store = SqliteKVStoreImpl(config)
await store.initialize()
# Set up test data
await store.set("key_a", "value_a")
await store.set("key_b", "value_b")
await store.set("key_c", "value_c")
await store.set("key_d", "value_d")
# Test values_in_range
values = await store.values_in_range("key_b", "key_c")
assert len(values) == 2
assert "value_b" in values
assert "value_c" in values
# Test keys_in_range
keys = await store.keys_in_range("key_a", "key_c")
assert len(keys) == 3
assert "key_a" in keys
assert "key_b" in keys
assert "key_c" in keys
await store.close()
async def test_memory_kvstore_multiple_instances():
"""Test that multiple :memory: instances are independent."""
config1 = SqliteKVStoreConfig(db_path=":memory:")
config2 = SqliteKVStoreConfig(db_path=":memory:")
store1 = SqliteKVStoreImpl(config1)
store2 = SqliteKVStoreImpl(config2)
await store1.initialize()
await store2.initialize()
# Set data in store1
await store1.set("shared_key", "value_from_store1")
# Verify store2 doesn't see store1's data
result = await store2.get("shared_key")
assert result is None
# Set different value in store2
await store2.set("shared_key", "value_from_store2")
# Verify both stores have independent data
assert await store1.get("shared_key") == "value_from_store1"
assert await store2.get("shared_key") == "value_from_store2"
await store1.close()
await store2.close()
async def test_memory_kvstore_persistence_behavior():
"""Test that :memory: database doesn't persist across instances."""
config = SqliteKVStoreConfig(db_path=":memory:")