feat: list VedAstro official raw archives

This commit is contained in:
732642856
2026-07-08 20:35:42 +08:00
parent 99d00c4124
commit 09dcfb9bcc
5 changed files with 78 additions and 0 deletions
+21
View File
@@ -347,6 +347,27 @@ def test_vedastro_gateway_poll_rejects_invalid_job_id(monkeypatch, tmp_path) ->
assert poller.payload()['error_code'] == 'ERR_NOT_FOUND'
def test_vedastro_gateway_archive_manifest_route(monkeypatch, tmp_path) -> None:
from scripts import vedastro_gateway
monkeypatch.setenv('VEDASTRO_GATEWAY_QUEUE_DIR', str(tmp_path))
job = vedastro_gateway.enqueue_gateway_job({'year': 1955}, question='x')
vedastro_gateway.complete_gateway_job(
job['job_id'],
{'status': 'ok', 'official_raw_response': {'source': 'vedastro_official'}},
)
reader = _ResponseCaptureHandler()
reader.path = '/api/vedastro_gateway/archives'
reader.do_GET()
assert reader.status_code == 200
payload = reader.payload()
assert payload['scope'] == 'vedastro_official_raw_response_archive_manifest'
assert payload['archive_count'] == 1
assert payload['archives'][0]['job_id'] == job['job_id']
def test_vedastro_gateway_job_run_route_executes_worker(monkeypatch, tmp_path) -> None:
from scripts import vedastro_gateway
+20
View File
@@ -138,6 +138,26 @@ def test_gateway_completion_archives_official_raw_response(monkeypatch, tmp_path
assert '"vedastro_official"' in path.read_text(encoding="utf-8")
def test_gateway_lists_official_raw_response_archives(monkeypatch, tmp_path):
from scripts import vedastro_gateway
monkeypatch.setenv("VEDASTRO_GATEWAY_QUEUE_DIR", str(tmp_path))
empty = vedastro_gateway.list_official_raw_response_archives()
assert empty["archive_count"] == 0
job = vedastro_gateway.enqueue_gateway_job({"year": 1955}, question="x")
vedastro_gateway.complete_gateway_job(
job["job_id"],
{"status": "ok", "official_raw_response": {"source": "vedastro_official"}},
)
manifest = vedastro_gateway.list_official_raw_response_archives()
assert manifest["scope"] == "vedastro_official_raw_response_archive_manifest"
assert manifest["archive_count"] == 1
assert manifest["archives"][0]["job_id"] == job["job_id"]
assert manifest["archives"][0]["official_raw_response_available"] is True
def test_gateway_run_job_executes_queued_request(monkeypatch, tmp_path):
from scripts import vedastro_gateway