Function _make_chat_context

tests/commands/memory/test_memory_command.py:24–53

Build a minimal mock chat_context.

(
    *,
    has_session: bool = False,
    has_vm: bool = False,
    has_store: bool = False,
    vm_budget: int = 128_000,
)

def _make_chat_context(
    *,
    has_session: bool = False,
    has_vm: bool = False,
    has_store: bool = False,
    vm_budget: int = 128_000,
) -> MagicMock:
    """Build a minimal mock chat_context."""
    ctx = MagicMock()
    ctx._vm_budget = vm_budget
    ctx._system_prompt_dirty = False

    if has_vm:
        vm = _make_vm()
        session = MagicMock()
        session.vm = vm
        ctx.session = session
    elif has_session:
        session = MagicMock()
        session.vm = None
        ctx.session = session
    else:
        ctx.session = None

    if has_store:
        ctx.memory_store = _make_store()
    else:
        ctx.memory_store = None

    return ctx


def _make_vm() -> MagicMock: