Function _make_vm

tests/commands/memory/test_memory_command.py:56–151

Build a mock VM object with all required sub-objects.

()

def _make_vm() -> MagicMock:
    """Build a mock VM object with all required sub-objects."""
    from chuk_ai_session_manager.memory.models import StorageTier
    from chuk_ai_session_manager.memory.models.enums import (
        CompressionLevel,
        Modality,
        PageType,
        VMMode,
    )

    vm = MagicMock()
    vm.mode = VMMode.PASSIVE
    vm.turn = 5

    # metrics
    metrics = MagicMock()
    metrics.tlb_hits = 10
    metrics.tlb_misses = 2
    metrics.tlb_hit_rate = 10 / 12
    metrics.faults_total = 3
    metrics.faults_this_turn = 1
    metrics.evictions_total = 2
    metrics.evictions_this_turn = 0
    vm.metrics = metrics

    # working_set
    ws_stats = MagicMock()
    ws_stats.utilization = 0.5
    ws_stats.l0_pages = 2
    ws_stats.l1_pages = 4
    ws_stats.tokens_used = 8_000
    ws_stats.tokens_available = 8_000
    ws = MagicMock()
    ws.get_stats.return_value = ws_stats
    vm.working_set = ws

    # page_table
    pt_stats = MagicMock()
    # pages_by_tier: a real dict keyed by StorageTier enums
    pt_stats.pages_by_tier = {
        StorageTier.L0: 2,
        StorageTier.L1: 4,
    }
    pt_stats.total_pages = 6
    pt_stats.dirty_pages = 1

    # a couple of page table entries
    def _make_entry(pid: str, tier: StorageTier) -> MagicMock:
        e = MagicMock()
        e.page_id = pid
        e.page_type = PageType.TRANSCRIPT
        e.tier = tier
        e.size_tokens = 100
        e.eviction_priority = 0.5
        e.pinned = False
        e.compression_level = CompressionLevel.FULL
        e.access_count = 3
        e.last_accessed = datetime.now(timezone.utc)