85 lines
2.5 KiB
Python
85 lines
2.5 KiB
Python
import tempfile
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from haiku.rag.client import HaikuRAG
|
|
from haiku.rag.config import Config
|
|
from haiku.rag.store.engine import Store
|
|
from haiku.rag.store.repositories.settings import (
|
|
ConfigMismatchError,
|
|
SettingsRepository,
|
|
)
|
|
|
|
|
|
def test_settings_table_populated_on_store_init():
|
|
"""Test that settings table is populated with current config when store is initialized."""
|
|
|
|
store = Store(":memory:")
|
|
settings_repo = SettingsRepository(store)
|
|
|
|
db_settings = settings_repo.get()
|
|
config_dict = Config.model_dump(mode="json")
|
|
|
|
assert db_settings == config_dict
|
|
|
|
store.close()
|
|
|
|
|
|
def test_settings_save_and_retrieve():
|
|
"""Test saving and retrieving settings after config change."""
|
|
store = Store(":memory:")
|
|
settings_repo = SettingsRepository(store)
|
|
|
|
original_chunk_size = Config.CHUNK_SIZE
|
|
Config.CHUNK_SIZE = 2 * original_chunk_size
|
|
|
|
settings_repo.save()
|
|
retrieved_settings = settings_repo.get()
|
|
assert retrieved_settings["CHUNK_SIZE"] == 2 * original_chunk_size
|
|
|
|
Config.CHUNK_SIZE = original_chunk_size
|
|
store.close()
|
|
|
|
|
|
async def test_config_validation_on_db_load():
|
|
"""Test that config validation fails when loading db with mismatched settings."""
|
|
# Create a temporary database file
|
|
with tempfile.NamedTemporaryFile(suffix=".sqlite", delete=False) as tmp:
|
|
db_path = Path(tmp.name)
|
|
|
|
try:
|
|
# Create store and save settings
|
|
store1 = Store(db_path)
|
|
SettingsRepository(store1)
|
|
store1.close()
|
|
|
|
# Change config
|
|
original_chunk_size = Config.CHUNK_SIZE
|
|
Config.CHUNK_SIZE = 999
|
|
|
|
# Loading the database should raise ConfigMismatchError
|
|
with pytest.raises(ConfigMismatchError) as exc_info:
|
|
Store(db_path)
|
|
|
|
assert "CHUNK_SIZE" in str(exc_info.value)
|
|
assert "Consider rebuilding" in str(exc_info.value)
|
|
|
|
# Rebuild
|
|
async with HaikuRAG(db_path=db_path, skip_validation=True) as client:
|
|
async for _ in client.rebuild_database():
|
|
pass # Process all documents
|
|
|
|
# Verify we can now load the database without exception (settings were updated)
|
|
store2 = Store(db_path)
|
|
settings_repo2 = SettingsRepository(store2)
|
|
db_settings = settings_repo2.get()
|
|
assert db_settings["CHUNK_SIZE"] == 999
|
|
store2.close()
|
|
|
|
Config.CHUNK_SIZE = original_chunk_size
|
|
|
|
finally:
|
|
# Cleanup
|
|
if db_path.exists():
|
|
db_path.unlink()
|