MajoorWaldi-ComfyUI-Majoor-AssetsManager/tests/features/test_index_runtime_helpers.py

from pathlib import Path
from types import SimpleNamespace

from mjr_am_backend.features.index import index_runtime_helpers as irh
from mjr_am_backend.shared import Result


def test_record_refresh_outcome_updates_stats_and_enrich_list() -> None:
    called = {"n": 0}

    def _append_to_enrich(**kwargs):
        called["n"] += 1

    scanner = SimpleNamespace(_append_to_enrich=_append_to_enrich)
    stats = {"skipped": 0, "errors": 1}
    irh.record_refresh_outcome(
        scanner,
        stats=stats,
        fallback_mode=True,
        refreshed=True,
        entry={"filepath": "x"},
        to_enrich=[],
        respect_enrich_limit=True,
    )
    assert stats["skipped"] == 1
    assert stats["errors"] == 0
    assert called["n"] == 1


def test_record_index_entry_success_tracks_added_ids_when_not_fallback() -> None:
    scanner = SimpleNamespace(_append_to_enrich=lambda **kwargs: None)
    stats = {"added": 0}
    added_ids: list[int] = []
    irh.record_index_entry_success(
        scanner,
        stats=stats,
        fallback_mode=False,
        added_ids=added_ids,
        added_asset_id=12,
        entry={"filepath": "x"},
        action="added",
        to_enrich=[],
        respect_enrich_limit=True,
    )
    assert stats["added"] == 1
    assert added_ids == [12]


def test_entry_state_drifted_returns_false_when_no_file_path() -> None:
    assert not irh.entry_state_drifted(SimpleNamespace(), entry={}, stats={"skipped": 0})


def test_append_to_enrich_honors_flags_and_limits() -> None:
    scanner = SimpleNamespace(_max_to_enrich_items=1)
    target: list[str] = []
    irh.append_to_enrich(scanner, entry={"fast": True, "filepath": "a"}, to_enrich=target, respect_limit=True)
    irh.append_to_enrich(scanner, entry={"fast": True, "filepath": "b"}, to_enrich=target, respect_limit=True)
    assert target == ["a"]
    irh.append_to_enrich(scanner, entry={"fast": False, "filepath": "c"}, to_enrich=target, respect_limit=False)
    assert target == ["a"]


def test_maybe_store_entry_cache_noop_when_cache_disabled() -> None:
    # smoke branch: should return without touching MetadataHelpers
    scanner = SimpleNamespace(db=object())
    import asyncio

    asyncio.run(irh.maybe_store_entry_cache(scanner, entry={"cache_store": False}, metadata_result=Result.Ok({})))


def test_maybe_store_entry_cache_calls_store_when_enabled(monkeypatch) -> None:
    """Lines 18-24 — cache_store=True triggers MetadataHelpers.store_metadata_cache."""
    import asyncio

    from mjr_am_backend.features.index import metadata_helpers as mh

    calls = []

    async def _fake_store(db, filepath, state_hash, metadata_result):
        calls.append((filepath, state_hash))

    monkeypatch.setattr(mh.MetadataHelpers, "store_metadata_cache", staticmethod(_fake_store))
    scanner = SimpleNamespace(db=object())
    asyncio.run(
        irh.maybe_store_entry_cache(
            scanner,
            entry={"cache_store": True, "filepath": "f.png", "state_hash": "h1"},
            metadata_result=Result.Ok({}),
        )
    )
    assert calls == [("f.png", "h1")]


def test_maybe_store_entry_cache_exception_suppressed(monkeypatch) -> None:
    """Lines 25-26 — exception in store_metadata_cache is caught silently."""
    import asyncio

    from mjr_am_backend.features.index import metadata_helpers as mh

    async def _bad_store(db, filepath, state_hash, metadata_result):
        raise RuntimeError("boom")

    monkeypatch.setattr(mh.MetadataHelpers, "store_metadata_cache", staticmethod(_bad_store))
    scanner = SimpleNamespace(db=object())
    # Should not raise
    asyncio.run(
        irh.maybe_store_entry_cache(
            scanner,
            entry={"cache_store": True, "filepath": "f.png", "state_hash": "h1"},
            metadata_result=Result.Ok({}),
        )
    )


def test_record_index_entry_success_with_fallback_mode() -> None:
    """Line 64 — fallback_mode=True calls fallback_correct_error."""
    scanner = SimpleNamespace(_append_to_enrich=lambda **kwargs: None)
    stats = {"added": 0, "errors": 2}
    irh.record_index_entry_success(
        scanner,
        stats=stats,
        fallback_mode=True,
        added_ids=None,
        added_asset_id=None,
        entry={"filepath": "y"},
        action="added",
        to_enrich=[],
        respect_enrich_limit=False,
    )
    assert stats["added"] == 1
    # fallback_correct_error decrements errors by 1
    assert stats["errors"] == 1


def test_entry_state_drifted_no_drift(tmp_path: Path) -> None:
    """Lines 87-90 — file exists with matching mtime_ns/size → returns False."""
    fp = tmp_path / "a.png"
    fp.write_bytes(b"hello")
    st = fp.stat()
    stats = {"skipped": 0}
    result = irh.entry_state_drifted(
        SimpleNamespace(),
        entry={"file_path": fp, "mtime_ns": st.st_mtime_ns, "size": st.st_size},
        stats=stats,
    )
    assert result is False
    assert stats["skipped"] == 0


def test_entry_state_drifted_with_drift(tmp_path: Path) -> None:
    """Lines 91-93 — file mtime/size changed → returns True and updates stats."""
    fp = tmp_path / "b.png"
    fp.write_bytes(b"hello")
    stats = {"skipped": 0, "skipped_state_changed": 0}
    result = irh.entry_state_drifted(
        SimpleNamespace(),
        entry={"file_path": fp, "mtime_ns": 0, "size": 99999},
        stats=stats,
    )
    assert result is True
    assert stats["skipped"] == 1
    assert stats["skipped_state_changed"] == 1