Prepare branch for remote review; preserve documented validation gaps and protected historical packages. Co-Authored-By: Claude Code <noreply@anthropic.com>
62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
"""Pytest import guardrails for local project modules."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
# Deliberately fictional birth data; not a real person or an oracle golden.
|
|
FIXTURE_BIRTH_DATE = "1990-06-15"
|
|
FIXTURE_BIRTH_TIME = "10:20"
|
|
FIXTURE_BIRTH_LAT = 31.2
|
|
FIXTURE_BIRTH_LON = 121.5
|
|
|
|
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()
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _isolate_vedastro_snapshot_cache(tmp_path_factory, monkeypatch) -> None:
|
|
cache_dir = tmp_path_factory.mktemp("vedastro_snapshot_cache")
|
|
monkeypatch.setenv("JYOTISH_VEDASTRO_SNAPSHOT_CACHE_DIR", str(cache_dir))
|
|
|
|
|
|
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()
|