import zipfile

from check_windows_wheel_install import (
    MAX_PATH,
    WORST_CASE_PREFIX,
    overlong_install_paths,
)


def _wheel(tmp_path, *entry_names):
    path = tmp_path / "pkg.whl"
    with zipfile.ZipFile(path, "w") as zf:
        for name in entry_names:
            zf.writestr(name, "{}")
    return str(path)


def test_flags_entry_one_char_over_budget(tmp_path):
    busts = "a" * (MAX_PATH - WORST_CASE_PREFIX + 1)
    assert overlong_install_paths(_wheel(tmp_path, busts)) == [busts]


def test_allows_entry_exactly_at_budget(tmp_path):
    at_limit = "a" * (MAX_PATH - WORST_CASE_PREFIX)
    assert (
        overlong_install_paths(_wheel(tmp_path, at_limit, "litellm/__init__.py")) == []
    )


def test_orders_offenders_longest_first(tmp_path):
    longer = "a" * (MAX_PATH - WORST_CASE_PREFIX + 5)
    shorter = "b" * (MAX_PATH - WORST_CASE_PREFIX + 1)
    assert overlong_install_paths(_wheel(tmp_path, shorter, longer)) == [
        longer,
        shorter,
    ]