732642856
121 lines
4.1 KiB
Python
121 lines
4.1 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 test_vedastro_method_catalog_sync_schema_is_declared() -> None:
|
|
completed = subprocess.run(
|
|
[sys.executable, "scripts/vedastro_method_catalog_sync.py", "--print-schema"],
|
|
cwd=ROOT,
|
|
text=True,
|
|
capture_output=True,
|
|
timeout=120,
|
|
check=False,
|
|
)
|
|
|
|
assert completed.returncode == 0, completed.stderr or completed.stdout
|
|
report = json.loads(completed.stdout)
|
|
assert report["sync"] == "vedastro_method_catalog_sync"
|
|
assert report["scope"] == "official_vedastro_method_catalog"
|
|
assert "sync_tags" in report["operations"]
|
|
assert "write_snapshot" in report["operations"]
|
|
assert "sync_python_capabilities" in report["operations"]
|
|
|
|
|
|
def test_vedastro_method_catalog_sync_can_write_stubbed_snapshot() -> None:
|
|
env = os.environ.copy()
|
|
env["VEDASTRO_METHOD_CATALOG_STUB"] = json.dumps(
|
|
{
|
|
"source": "stubbed_catalog",
|
|
"tag_groups": {
|
|
"Marriage": [{"Name": "GoodForMarriage", "CalculatorMethod": "GoodForMarriage"}],
|
|
"LendingMoney": [{"Name": "GoodForLendingMoney", "CalculatorMethod": "GoodForLendingMoney"}],
|
|
},
|
|
}
|
|
)
|
|
output_path = ROOT / "scratch" / "local" / "vedastro_adapter" / "test_method_catalog_snapshot.json"
|
|
completed = subprocess.run(
|
|
[
|
|
sys.executable,
|
|
"scripts/vedastro_method_catalog_sync.py",
|
|
"--write",
|
|
"--output",
|
|
str(output_path),
|
|
],
|
|
cwd=ROOT,
|
|
text=True,
|
|
capture_output=True,
|
|
timeout=120,
|
|
check=False,
|
|
env=env,
|
|
)
|
|
|
|
assert completed.returncode == 0, completed.stderr or completed.stdout
|
|
report = json.loads(completed.stdout)
|
|
assert report["status"] == "ok"
|
|
assert report["source"] == "stubbed_catalog"
|
|
assert report["summary"]["tag_count"] == 2
|
|
assert report["summary"]["method_count"] == 2
|
|
snapshot = json.loads(output_path.read_text(encoding="utf-8"))
|
|
assert "Marriage" in snapshot["tag_groups"]
|
|
assert snapshot["summary"]["method_count"] == 2
|
|
|
|
|
|
def test_vedastro_method_catalog_sync_can_build_stubbed_python_capability_registry() -> None:
|
|
env = os.environ.copy()
|
|
env["VEDASTRO_METHOD_CATALOG_STUB"] = json.dumps(
|
|
{
|
|
"source": "stubbed_catalog",
|
|
"tag_groups": {},
|
|
"python_capabilities": [
|
|
{
|
|
"method": "AllPlanetData",
|
|
"signature": "(planetName, time)",
|
|
"bucket": "planet_time",
|
|
"parameter_names": ["planetName", "time"],
|
|
"callable": True,
|
|
},
|
|
{
|
|
"method": "AllHouseData",
|
|
"signature": "(houseName, time)",
|
|
"bucket": "house_name_time",
|
|
"parameter_names": ["houseName", "time"],
|
|
"callable": True,
|
|
},
|
|
{
|
|
"method": "GetAllEventDataGroupedByTag",
|
|
"signature": "()",
|
|
"bucket": "zero_arg",
|
|
"parameter_names": [],
|
|
"callable": True,
|
|
},
|
|
],
|
|
}
|
|
)
|
|
completed = subprocess.run(
|
|
[sys.executable, "scripts/vedastro_method_catalog_sync.py"],
|
|
cwd=ROOT,
|
|
text=True,
|
|
capture_output=True,
|
|
timeout=120,
|
|
check=False,
|
|
env=env,
|
|
)
|
|
|
|
assert completed.returncode == 0, completed.stderr or completed.stdout
|
|
report = json.loads(completed.stdout)
|
|
assert report["summary"]["python_capability_count"] == 3
|
|
assert report["summary"]["python_callable_count"] == 3
|
|
assert report["summary"]["python_signature_bucket_count"] >= 2
|
|
assert any(item["method"] == "AllPlanetData" for item in report["python_capabilities"])
|
|
buckets = report["python_signature_buckets"]
|
|
assert buckets["planet_time"]["count"] == 1
|
|
assert "AllPlanetData" in buckets["planet_time"]["examples"]
|