import pytest
from app.memory.store import MemoryManager


@pytest.mark.asyncio
async def test_memory_manager_init():
    mgr = MemoryManager()
    assert mgr is not None
    assert mgr.memory_vectorstore is not None


@pytest.mark.asyncio
async def test_working_memory():
    mgr = MemoryManager()
    mem = mgr.get_working_memory()
    assert mem is not None
    assert mem.memory_key == "history"


@pytest.mark.asyncio
async def test_longterm_memory():
    mgr = MemoryManager()
    mem = mgr.get_longterm_memory(user_id=1)
    assert mem is not None
    assert mem.memory_key == "long_term_memory"