Files
awoooi/scripts/reboot-recovery/tests/test_momo_source_arrival_gate.py
Your Name 5a1c507671
Some checks failed
CD Pipeline / workflow-shape (push) Successful in 0s
CD Pipeline / cancel-stale-cd (push) Has been skipped
CD Pipeline / tests (push) Failing after 1m44s
CD Pipeline / build-and-deploy (push) Has been skipped
CD Pipeline / post-deploy-checks (push) Has been skipped
fix(recovery): clear cold-start service blockers
2026-07-01 19:16:58 +08:00

122 lines
3.8 KiB
Python

from __future__ import annotations
import importlib.util
from pathlib import Path
ROOT = Path(__file__).resolve().parents[3]
SCRIPT = ROOT / "scripts" / "reboot-recovery" / "momo-source-arrival-gate.py"
def load_module():
spec = importlib.util.spec_from_file_location("momo_source_arrival_gate", SCRIPT)
assert spec and spec.loader
module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(module)
return module
BASE_PREFLIGHT = """
HOST ollama
MOMO_HEALTH_CODE 200
MOMO_PUBLIC_HEALTH_CODE 200
MOMO_APP_HEALTH healthy
SCHEDULER_RUNNING true
SCHEDULER_HEALTH healthy
DRIVE_ARCHIVE_LATEST_MODIFIED 2026-06-25T04:21:47.000Z
DRIVE_GLOBAL_LATEST_MODIFIED 2026-06-25T04:21:47.000Z
DRIVE_FAILED_COUNT 0
DB_MONTHLY_SYNC 15383|15383|2026-06-01|2026-06-24|2026-06-01|2026-06-24
DB_LATEST_DAILY_IMPORT_JOB 57|completed|即時業績_當日.xlsx|2026-06-25T13:16:47.359958|2026-06-25T13:18:02.964985|15383|15383|0
IMPORT_CONFIG 當日業績匯入|即時業績_當日
MOMO_DRIVE_TOKEN_SOURCE_PREFLIGHT PASS=20 WARN=3 BLOCKED=2 HOST=ollama@192.168.0.188 FRESHNESS_MAX_DAYS=2
"""
def classify(extra_lines: str):
module = load_module()
return module.classify(module.parse_preflight(BASE_PREFLIGHT + extra_lines))
def assert_no_write_authorization(result):
assert result["runtime_write_authorized"] is False
assert result["db_write_authorized"] is False
assert result["drive_move_authorized"] is False
assert result["manual_import_authorized"] is False
assert result["secret_value_collection_allowed"] is False
def test_source_absent_without_newer_drive_waits_without_runtime_write():
result = classify(
"""
DRIVE_INTAKE_COUNT 0
DB_DAILY_FRESHNESS 3|2026-06-24
"""
)
assert result["status"] == "source_absent_no_newer_drive_waiting"
assert result["exit_code"] == 1
assert result["safe_import_preflight_allowed"] is False
assert result["source_absent_without_newer_drive"] is True
assert_no_write_authorization(result)
def test_zero_current_month_rows_are_not_blocking_when_no_newer_source():
result = classify(
"""
DRIVE_INTAKE_COUNT 0
DB_MONTHLY_SYNC 0|0|-|-|-|-
DB_DAILY_FRESHNESS 3|2026-06-24
"""
)
assert result["status"] == "source_absent_no_newer_drive_waiting"
assert result["current_month_sync_status"] == "not_applicable_no_current_month_source"
assert "current_month_snapshot_realtime_sync_not_proven" not in result["blockers"]
assert result["exit_code"] == 1
assert_no_write_authorization(result)
def test_source_absent_fail_closed_when_latest_import_not_clean():
result = classify(
"""
DRIVE_INTAKE_COUNT 0
DB_DAILY_FRESHNESS 3|2026-06-24
DB_LATEST_DAILY_IMPORT_JOB 58|failed|即時業績_當日.xlsx|2026-06-25T13:16:47.359958|2026-06-25T13:18:02.964985|15383|100|15283
"""
)
assert result["status"] == "blocked_source_absent_fail_closed"
assert result["exit_code"] == 2
assert result["source_absent_without_newer_drive"] is False
assert_no_write_authorization(result)
def test_source_arrived_allows_only_safe_import_preflight():
result = classify(
"""
DRIVE_INTAKE_COUNT 1
DB_DAILY_FRESHNESS 3|2026-06-24
"""
)
assert result["status"] == "source_arrived_ready_for_safe_import_preflight"
assert result["exit_code"] == 0
assert result["safe_import_preflight_allowed"] is True
assert_no_write_authorization(result)
def test_freshness_green_requires_cold_start_recheck():
result = classify(
"""
DRIVE_INTAKE_COUNT 0
DB_DAILY_FRESHNESS 1|2026-06-26
"""
)
assert result["status"] == "freshness_already_green_recheck_cold_start"
assert result["next_step"] == "rerun_post_reboot_readiness_summary_with_same_evidence_chain"
assert result["exit_code"] == 0
assert result["safe_import_preflight_allowed"] is False
assert_no_write_authorization(result)