732642856
48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
"""Pytest import guardrails for local project modules."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
SCRIPTS = str(ROOT / "scripts")
|
|
WORKBUDDY_SKILL_SCRIPTS = ".workbuddy/skills/jyotish-vedic-astrology/scripts"
|
|
SLOW_API_SECURITY_PREFIXES = (
|
|
"test_vedastro_",
|
|
"test_high_rigor_",
|
|
"test_professional_reading",
|
|
"test_api_prompt_pack",
|
|
"test_consultation_workflow",
|
|
"test_thematic_report",
|
|
"test_capability_audit",
|
|
"test_technique_catalog",
|
|
)
|
|
|
|
|
|
def ensure_project_scripts_first() -> None:
|
|
if SCRIPTS in sys.path:
|
|
sys.path.remove(SCRIPTS)
|
|
sys.path.insert(0, SCRIPTS)
|
|
|
|
|
|
def remove_workbuddy_shadow_modules() -> None:
|
|
for name, module in list(sys.modules.items()):
|
|
module_file = getattr(module, "__file__", "") or ""
|
|
if WORKBUDDY_SKILL_SCRIPTS in module_file:
|
|
del sys.modules[name]
|
|
|
|
|
|
def pytest_runtest_setup() -> None:
|
|
remove_workbuddy_shadow_modules()
|
|
ensure_project_scripts_first()
|
|
|
|
|
|
def pytest_collection_modifyitems(items) -> None:
|
|
for item in items:
|
|
if item.fspath.basename == "test_api_server_security.py" and item.name.startswith(SLOW_API_SECURITY_PREFIXES):
|
|
item.add_marker("slow")
|
|
|
|
|
|
ensure_project_scripts_first()
|