d3302c95b9
* feat: HindsightEmbedded python SDK * feat: HindsightEmbedded python SDK * fixes * improve * ci * improvemnts * fix test * fix test * fix: update tests to use Pydantic model attributes instead of dict access - Fixed test_server_integration.py to access Pydantic model attributes directly - Changed dict-style access (response["field"]) to attribute access (response.field) - Fixed .get() calls on Pydantic models - Updated recall() calls to access .results attribute - Updated reflect() calls to access .text attribute - Fixed test_list_banks to use namespace API instead of deleted default_api - Fixed attribute shadowing in HindsightClient wrapper (renamed _*_api to _*_namespace) * fix: add list() method to BanksAPI namespace * fix: remove leftover async cleanup code from test_list_banks * docs: remove Advanced Configuration section from embed.md
53 lines
2.2 KiB
Python
53 lines
2.2 KiB
Python
"""Tests for EmbedManager interface."""
|
|
|
|
from hindsight_embed import get_embed_manager
|
|
|
|
|
|
def test_sanitize_profile_name_via_db_url():
|
|
"""Test profile name sanitization through database URL generation."""
|
|
manager = get_embed_manager()
|
|
|
|
# Test None defaults to "default"
|
|
assert manager.get_database_url(None) == "pg0://hindsight-embed-default"
|
|
|
|
# Test simple alphanumeric names
|
|
assert manager.get_database_url("myapp") == "pg0://hindsight-embed-myapp"
|
|
assert manager.get_database_url("my-app") == "pg0://hindsight-embed-my-app"
|
|
assert manager.get_database_url("my_app") == "pg0://hindsight-embed-my_app"
|
|
assert manager.get_database_url("app123") == "pg0://hindsight-embed-app123"
|
|
|
|
# Test special characters get replaced with dashes
|
|
assert manager.get_database_url("my app") == "pg0://hindsight-embed-my-app"
|
|
assert manager.get_database_url("my.app") == "pg0://hindsight-embed-my-app"
|
|
assert manager.get_database_url("my@app!") == "pg0://hindsight-embed-my-app-"
|
|
assert manager.get_database_url("My App 2.0!") == "pg0://hindsight-embed-My-App-2-0-"
|
|
|
|
|
|
def test_get_database_url_default():
|
|
"""Test database URL generation with default pg0."""
|
|
manager = get_embed_manager()
|
|
|
|
assert manager.get_database_url("myapp") == "pg0://hindsight-embed-myapp"
|
|
assert manager.get_database_url("myapp", None) == "pg0://hindsight-embed-myapp"
|
|
assert manager.get_database_url("myapp", "pg0") == "pg0://hindsight-embed-myapp"
|
|
|
|
|
|
def test_get_database_url_custom():
|
|
"""Test database URL generation with custom database."""
|
|
manager = get_embed_manager()
|
|
|
|
custom_url = "postgresql://user:pass@localhost/db"
|
|
assert manager.get_database_url("myapp", custom_url) == custom_url
|
|
assert manager.get_database_url("any-profile", custom_url) == custom_url
|
|
|
|
|
|
def test_manager_singleton():
|
|
"""Test that get_embed_manager returns functional instances."""
|
|
manager1 = get_embed_manager()
|
|
manager2 = get_embed_manager()
|
|
|
|
# They should be independent instances but same type
|
|
assert type(manager1) == type(manager2)
|
|
|
|
# They should produce the same results
|
|
assert manager1.get_database_url("test") == manager2.get_database_url("test")
|