mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
Paul Hernandez
863e0a4e24
Signed-off-by: phernandez <paul@basicmachines.co> Co-authored-by: Claude Opus 4.5 <noreply@anthropic.com>
53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
"""Test that relation resolution happens in the background."""
|
|
|
|
import pytest
|
|
|
|
from basic_memory.api.routers.knowledge_router import resolve_relations_background
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_resolve_relations_background_success():
|
|
"""Test that background relation resolution calls sync service correctly."""
|
|
|
|
class StubSyncService:
|
|
def __init__(self) -> None:
|
|
self.calls: list[int] = []
|
|
|
|
async def resolve_relations(self, *, entity_id: int) -> None:
|
|
self.calls.append(entity_id)
|
|
|
|
sync_service = StubSyncService()
|
|
|
|
entity_id = 123
|
|
entity_permalink = "test/entity"
|
|
|
|
# Call the background function
|
|
await resolve_relations_background(sync_service, entity_id, entity_permalink)
|
|
|
|
# Verify sync service was called with the entity_id
|
|
assert sync_service.calls == [entity_id]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_resolve_relations_background_handles_errors():
|
|
"""Test that background relation resolution handles errors gracefully."""
|
|
|
|
class StubSyncService:
|
|
def __init__(self) -> None:
|
|
self.calls: list[int] = []
|
|
|
|
async def resolve_relations(self, *, entity_id: int) -> None:
|
|
self.calls.append(entity_id)
|
|
raise Exception("Test error")
|
|
|
|
sync_service = StubSyncService()
|
|
|
|
entity_id = 123
|
|
entity_permalink = "test/entity"
|
|
|
|
# Call should not raise - errors are logged
|
|
await resolve_relations_background(sync_service, entity_id, entity_permalink)
|
|
|
|
# Verify sync service was called
|
|
assert sync_service.calls == [entity_id]
|