Function _make_store

tests/commands/memory/test_memory_command.py:154–183

Build a mock MemoryScopeStore.

()

def _make_store() -> MagicMock:
    """Build a mock MemoryScopeStore."""
    from mcp_cli.memory.models import MemoryEntry, MemoryScope

    store = MagicMock()

    ws_entry = MemoryEntry(
        key="db_type",
        content="postgresql",
    )
    gl_entry = MemoryEntry(
        key="test_framework",
        content="always use pytest for testing Python projects",
    )

    def _list(scope):
        if scope == MemoryScope.WORKSPACE:
            return [ws_entry]
        elif scope == MemoryScope.GLOBAL:
            return [gl_entry]
        return []

    store.list_entries.side_effect = _list

    remembered = MemoryEntry(key="new_key", content="new content")
    store.remember.return_value = remembered
    store.forget.return_value = True
    store.clear.return_value = 3

    return store


# ---------------------------------------------------------------------------