732642856
77 lines
3.0 KiB
Python
77 lines
3.0 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
def _run_diag(env: dict[str, str]) -> dict:
|
|
completed = subprocess.run(
|
|
[sys.executable, "scripts/diagnose_vedastro_mode.py", "--json"],
|
|
cwd=ROOT,
|
|
text=True,
|
|
capture_output=True,
|
|
timeout=30,
|
|
check=False,
|
|
env={**os.environ, **env, "JYOTISH_SKIP_LOCAL_ENV": "1"},
|
|
)
|
|
assert completed.returncode == 0, completed.stderr or completed.stdout
|
|
return json.loads(completed.stdout)
|
|
|
|
|
|
def test_vedastro_diagnostics_reports_fast_fallback_mode_without_endpoint() -> None:
|
|
report = _run_diag({
|
|
"VEDASTRO_API_ENDPOINT": "",
|
|
"VEDASTRO_ENABLE_NETWORK": "",
|
|
"VEDASTRO_TIMEOUT_SECONDS": "",
|
|
"VEDASTRO_API_KEY": "",
|
|
})
|
|
|
|
assert report["mode"] == "fast_local_fallback"
|
|
assert report["official_ready"] is False
|
|
assert "VEDASTRO_API_ENDPOINT" in report["missing"]
|
|
assert "missing_endpoint" in report["readiness_blockers"]
|
|
assert "network_disabled" in report["readiness_blockers"]
|
|
assert "timeout_too_low" not in report["readiness_blockers"]
|
|
assert report["expected_fallback_status"] == "official_snapshot_budget_exhausted_or_endpoint_blocked"
|
|
assert "VEDASTRO_API_ENDPOINT" in report["official_closure_plan"]["required_env"]
|
|
assert report["official_closure_plan"]["raw_response_acceptance"] == "vedastro_official.raw_response must be present before claiming official cloud closure."
|
|
|
|
|
|
def test_vedastro_diagnostics_reports_official_mode_when_configured() -> None:
|
|
report = _run_diag({
|
|
"VEDASTRO_API_ENDPOINT": "https://api.vedastro.org/api",
|
|
"VEDASTRO_ENABLE_NETWORK": "1",
|
|
"VEDASTRO_TIMEOUT_SECONDS": "20",
|
|
"VEDASTRO_API_KEY": "sample-key",
|
|
})
|
|
|
|
assert report["mode"] == "official_extended"
|
|
assert report["official_ready"] is True
|
|
assert report["timeout_seconds"] == 20.0
|
|
assert report["network_enabled"] is True
|
|
assert report["has_api_key"] is True
|
|
assert report["readiness_blockers"] == []
|
|
assert report["official_closure_plan"]["required_env"]["VEDASTRO_TIMEOUT_SECONDS"] == "20"
|
|
|
|
|
|
def test_vedastro_diagnostics_labels_premium_key_missing_but_free_tier_possible() -> None:
|
|
report = _run_diag({
|
|
"VEDASTRO_API_ENDPOINT": "https://api.vedastro.org/api",
|
|
"VEDASTRO_ENABLE_NETWORK": "1",
|
|
"VEDASTRO_TIMEOUT_SECONDS": "20",
|
|
"VEDASTRO_API_KEY": "",
|
|
"VEDASTRO_FREE_TIER_QUEUE": "1",
|
|
})
|
|
assert report["mode"] == "official_extended"
|
|
assert report["official_ready"] is True
|
|
assert "missing_api_key" not in report["readiness_blockers"]
|
|
assert "premium_key_missing" in report["readiness_blockers"]
|
|
assert report["free_tier_possible_with_cache_queue"] is True
|
|
assert report["official_closure_plan"]["premium_key_policy"] == "API key recommended for stable official full snapshot; free tier may still block or throttle."
|