ComfyUI-Lora-Manager/tests/utils/test_fingerprint_fallback.py

"""Test for modelVersionId fallback in fingerprint calculation."""
import pytest
from py.utils.utils import calculate_recipe_fingerprint, normalize_prompt_for_dedup


def test_calculate_fingerprint_with_model_version_id_fallback():
    """Test that fingerprint uses modelVersionId when hash is empty, even when not deleted."""
    loras = [
        {
            "hash": "",
            "strength": 1.0,
            "modelVersionId": 2639467,
            "isDeleted": False,
            "exclude": False
        }
    ]
    fingerprint = calculate_recipe_fingerprint(loras)
    assert fingerprint == "2639467:1.0"


def test_calculate_fingerprint_with_multiple_model_version_ids():
    """Test fingerprint with multiple loras using modelVersionId fallback."""
    loras = [
        {
            "hash": "",
            "strength": 1.0,
            "modelVersionId": 2639467,
            "isDeleted": False,
            "exclude": False
        },
        {
            "hash": "",
            "strength": 0.8,
            "modelVersionId": 1234567,
            "isDeleted": False,
            "exclude": False
        }
    ]
    fingerprint = calculate_recipe_fingerprint(loras)
    assert fingerprint == "1234567:0.8|2639467:1.0"


def test_calculate_fingerprint_with_deleted_lora():
    """Test that deleted loras with modelVersionId are still included."""
    loras = [
        {
            "hash": "",
            "strength": 1.0,
            "modelVersionId": 2639467,
            "isDeleted": True,
            "exclude": False
        }
    ]
    fingerprint = calculate_recipe_fingerprint(loras)
    assert fingerprint == "2639467:1.0"


def test_calculate_fingerprint_with_excluded_lora():
    """Test that excluded loras are skipped even with modelVersionId."""
    loras = [
        {
            "hash": "",
            "strength": 1.0,
            "modelVersionId": 2639467,
            "isDeleted": False,
            "exclude": True
        }
    ]
    fingerprint = calculate_recipe_fingerprint(loras)
    assert fingerprint == ""


def test_calculate_fingerprint_prefers_hash_over_version_id():
    """Test that hash is used even when modelVersionId is present."""
    loras = [
        {
            "hash": "abc123",
            "strength": 1.0,
            "modelVersionId": 2639467,
            "isDeleted": False,
            "exclude": False
        }
    ]
    fingerprint = calculate_recipe_fingerprint(loras)
    assert fingerprint == "abc123:1.0"


def test_calculate_fingerprint_without_hash_or_version_id():
    """Test that loras without hash or modelVersionId are skipped."""
    loras = [
        {
            "hash": "",
            "strength": 1.0,
            "modelVersionId": 0,
            "isDeleted": False,
            "exclude": False
        }
    ]
    fingerprint = calculate_recipe_fingerprint(loras)
    assert fingerprint == ""


def test_normalize_prompt_casefolds_and_collapses_whitespace():
    assert normalize_prompt_for_dedup("A Girl,   blue hair") == "a girl, blue hair"
    assert normalize_prompt_for_dedup("  landscape \n\t with details  ") == "landscape with details"
    assert normalize_prompt_for_dedup("MASTERPIECE, best quality") == "masterpiece, best quality"


def test_normalize_prompt_handles_missing_or_non_string():
    assert normalize_prompt_for_dedup(None) == ""
    assert normalize_prompt_for_dedup("") == ""
    assert normalize_prompt_for_dedup(12345) == ""