IQ.Lvbs CI [bot]
65 lines
2.9 KiB
Python
65 lines
2.9 KiB
Python
"""
|
|
Copyright © IQ.Lvbs, apart of Project Teal Lvbs, All Rights Reserved, licensed under https://konn3kt.com/tos
|
|
"""
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
from openpilot.iqpilot import _proprietary_loader as loader
|
|
|
|
|
|
def _write(path: Path, content: str = "") -> None:
|
|
path.parent.mkdir(parents=True, exist_ok=True)
|
|
path.write_text(content)
|
|
|
|
|
|
def test_module_root_for_name_prefers_matching_bundle(tmp_path, monkeypatch):
|
|
repo_root = tmp_path / "repo"
|
|
model_root = repo_root / "artifacts" / "iqpilot_model_selector_private" / "python"
|
|
hepha_root = repo_root / "artifacts" / "iqpilot_hephaestusd_private" / "python"
|
|
|
|
_write(model_root / "iqpilot_private" / "__init__.py")
|
|
_write(model_root / "iqpilot_private" / "models" / "__init__.py")
|
|
_write(model_root / "iqpilot_private" / "models" / "manager.py", "VALUE = 'models'\n")
|
|
|
|
_write(hepha_root / "iqpilot_private" / "__init__.py")
|
|
_write(hepha_root / "iqpilot_private" / "konn3kt" / "__init__.py")
|
|
_write(hepha_root / "iqpilot_private" / "konn3kt" / "hephaestus" / "__init__.py")
|
|
_write(hepha_root / "iqpilot_private" / "konn3kt" / "hephaestus" / "hephaestusd.py", "VALUE = 'hepha'\n")
|
|
|
|
monkeypatch.setattr(loader, "__file__", str(repo_root / "iqpilot" / "_proprietary_loader.py"))
|
|
|
|
resolved = loader._module_root_for_name("iqpilot_private.konn3kt.hephaestus.hephaestusd")
|
|
assert resolved == hepha_root
|
|
|
|
|
|
def test_load_private_module_uses_matching_bundle(tmp_path, monkeypatch):
|
|
repo_root = tmp_path / "repo"
|
|
model_root = repo_root / "artifacts" / "iqpilot_model_selector_private" / "python"
|
|
hepha_root = repo_root / "artifacts" / "iqpilot_hephaestusd_private" / "python"
|
|
|
|
_write(model_root / "iqpilot_private" / "__init__.py")
|
|
_write(model_root / "iqpilot_private" / "models" / "__init__.py")
|
|
_write(model_root / "iqpilot_private" / "models" / "manager.py", "VALUE = 'models'\n")
|
|
|
|
_write(hepha_root / "iqpilot_private" / "__init__.py")
|
|
_write(hepha_root / "iqpilot_private" / "konn3kt" / "__init__.py")
|
|
_write(hepha_root / "iqpilot_private" / "konn3kt" / "hephaestus" / "__init__.py")
|
|
_write(hepha_root / "iqpilot_private" / "konn3kt" / "hephaestus" / "hephaestusd.py", "VALUE = 'hepha'\n")
|
|
|
|
monkeypatch.setattr(loader, "__file__", str(repo_root / "iqpilot" / "_proprietary_loader.py"))
|
|
monkeypatch.setenv("IQPILOT_PROPRIETARY_ROOT", "")
|
|
|
|
old_sys_path = list(sys.path)
|
|
for key in [k for k in sys.modules if k.startswith("iqpilot_private") or k == "test_public_module"]:
|
|
sys.modules.pop(key, None)
|
|
|
|
try:
|
|
sys.modules["test_public_module"] = type(sys)("test_public_module")
|
|
private_module = loader.load_private_module("test_public_module", "iqpilot_private.konn3kt.hephaestus.hephaestusd")
|
|
assert private_module.VALUE == "hepha"
|
|
assert sys.modules["test_public_module"].VALUE == "hepha"
|
|
finally:
|
|
sys.path[:] = old_sys_path
|
|
for key in [k for k in sys.modules if k.startswith("iqpilot_private") or k == "test_public_module"]:
|
|
sys.modules.pop(key, None)
|