Add VedAstro user entrypoint

2026-07-02 23:17:47 +08:00
parent 21212a1b9f
commit ec4598c0d7
3 changed files with 513 additions and 0 deletions
@@ -0,0 +1,178 @@
+from __future__ import annotations
+
+import json
+import os
+import subprocess
+import sys
+from pathlib import Path
+
+
+ROOT = Path(__file__).resolve().parents[1]
+
+
+def test_user_entrypoint_runs_catalog_and_strict_workflow_contract() -> None:
+    catalog_stub = {
+        "available": True,
+        "status": "ok",
+        "capabilities": [
+            {
+                "method": "SearchEvents",
+                "signature": "(birthTime, atTime, eventTagList)",
+                "bucket": "event_time",
+                "parameter_names": ["birthTime", "atTime", "eventTagList"],
+                "callable": True,
+            },
+            {
+                "method": "DasaAtRange",
+                "signature": "(birthTime, startTime, endTime, levels, precisionHours)",
+                "bucket": "dasha_at_range",
+                "parameter_names": ["birthTime", "startTime", "endTime", "levels", "precisionHours"],
+                "callable": True,
+            },
+            {
+                "method": "AllPlanetDashamamshaSign",
+                "signature": "(planetName, time)",
+                "bucket": "planet_time",
+                "parameter_names": ["planetName", "time"],
+                "callable": True,
+            },
+            {
+                "method": "HealthProblemEvent",
+                "signature": "(birthTime, startTime, endTime)",
+                "bucket": "event_range",
+                "parameter_names": ["birthTime", "startTime", "endTime"],
+                "callable": True,
+            },
+        ],
+        "buckets": {
+            "event_time": {"count": 1, "examples": ["SearchEvents"]},
+            "dasha_at_range": {"count": 1, "examples": ["DasaAtRange"]},
+            "planet_time": {"count": 1, "examples": ["AllPlanetDashamamshaSign"]},
+            "event_range": {"count": 1, "examples": ["HealthProblemEvent"]},
+        },
+    }
+    runner_stub = {
+        "SearchEvents": {"available": True, "status": "ok", "result": {"Events": []}},
+        "DasaAtRange": {"available": True, "status": "ok", "result": {"Periods": []}},
+        "AllPlanetDashamamshaSign": {"available": True, "status": "ok", "result": {"Name": "Capricorn"}},
+        "HealthProblemEvent": {"available": True, "status": "ok", "result": {}},
+    }
+
+    completed = subprocess.run(
+        [
+            sys.executable,
+            "scripts/vedastro_user_entrypoint.py",
+            "--year",
+            "REDACTED_YEAR",
+            "--month",
+            "4",
+            "--day",
+            "17",
+            "--hour",
+            "14",
+            "--minute",
+            "49",
+            "--lat",
+            "36.42",
+            "--lon",
+            "114.2",
+            "--tz",
+            "8",
+            "--question",
+            "事业机会什么时候出现",
+            "--themes",
+            "career,health",
+            "--reference-date",
+            "2026-07-02",
+            "--format",
+            "json",
+        ],
+        cwd=ROOT,
+        text=True,
+        capture_output=True,
+        timeout=120,
+        check=False,
+        env={
+            **os.environ,
+            "JYOTISH_SKIP_LOCAL_ENV": "1",
+            "VEDASTRO_API_ENDPOINT": "https://api.vedastro.org/api",
+            "VEDASTRO_ENABLE_NETWORK": "1",
+            "VEDASTRO_TIMEOUT_SECONDS": "20",
+            "VEDASTRO_CACHE_TTL_SECONDS": "600",
+            "VEDASTRO_OFFICIAL_FULL_SNAPSHOT_CACHE_TTL_SECONDS": "600",
+            "VEDASTRO_FREE_TIER_QUEUE": "1",
+            "VEDASTRO_FULL_CATALOG_SAMPLE_LIMIT": "8",
+            "VEDASTRO_OFFICIAL_CAPABILITY_CATALOG_STUB": json.dumps(catalog_stub),
+            "VEDASTRO_OFFICIAL_CAPABILITY_RUNNER_STUB": json.dumps(runner_stub),
+        },
+    )
+
+    assert completed.returncode == 0, completed.stderr or completed.stdout
+    report = json.loads(completed.stdout)
+
+    assert report["scope"] == "vedastro_user_entrypoint"
+    assert report["runtime_mode"]["mode"] == "official_extended"
+    assert report["input"]["themes"] == ["career", "health"]
+    assert report["official_capability_catalog"]["summary"]["catalog_method_count"] == 4
+    assert report["official_capability_catalog"]["summary"]["sample_limit"] == 8
+    assert report["official_capability_catalog"]["dynamic_selection"]["career"]["selected_methods"]
+    assert "health" in report["official_capability_catalog"]["domain_routing"]
+    assert report["cache_and_queue"]["official_full_snapshot_cache_ttl_seconds"] == 600
+    assert report["cache_and_queue"]["range_scan_cache_ttl_seconds"] == 600
+    assert report["cache_and_queue"]["free_tier_queue_enabled"] is True
+    assert report["strict_workflow"]["triggered"] is True
+    assert report["strict_workflow"]["primary_route"] == "career"
+    assert "career" in report["strict_workflow"]["routes_available"]
+    assert report["honesty_boundary"]["all_641_methods_executed"] is False
+    assert "official_full_capability_catalog" in report["user_commands"]["json"]
+
+
+def test_user_entrypoint_markdown_documents_boundaries() -> None:
+    completed = subprocess.run(
+        [
+            sys.executable,
+            "scripts/vedastro_user_entrypoint.py",
+            "--year",
+            "REDACTED_YEAR",
+            "--month",
+            "4",
+            "--day",
+            "17",
+            "--hour",
+            "14",
+            "--minute",
+            "49",
+            "--lat",
+            "36.42",
+            "--lon",
+            "114.2",
+            "--tz",
+            "8",
+            "--question",
+            "婚恋",
+            "--themes",
+            "marriage",
+            "--reference-date",
+            "2026-07-02",
+            "--format",
+            "markdown",
+        ],
+        cwd=ROOT,
+        text=True,
+        capture_output=True,
+        timeout=120,
+        check=False,
+        env={
+            **os.environ,
+            "JYOTISH_SKIP_LOCAL_ENV": "1",
+            "VEDASTRO_API_ENDPOINT": "",
+            "VEDASTRO_ENABLE_NETWORK": "",
+            "VEDASTRO_TIMEOUT_SECONDS": "",
+        },
+    )
+
+    assert completed.returncode == 0, completed.stderr or completed.stdout
+    assert "VedAstro 用户级入口" in completed.stdout
+    assert "fast_local_fallback" in completed.stdout
+    assert "不会把 641 项全部当作已执行" in completed.stdout
+    assert "strict workflow" in completed.stdout