87 lines
3.3 KiB
Python
87 lines
3.3 KiB
Python
from __future__ import annotations
|
|
|
|
import importlib.util
|
|
import json
|
|
from pathlib import Path
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
SCRIPT = ROOT / "tools/checks/runtime-image-smoke.py"
|
|
SPEC = importlib.util.spec_from_file_location("runtime_image_smoke", SCRIPT)
|
|
assert SPEC is not None and SPEC.loader is not None
|
|
MODULE = importlib.util.module_from_spec(SPEC)
|
|
sys.modules[SPEC.name] = MODULE
|
|
SPEC.loader.exec_module(MODULE)
|
|
|
|
|
|
class RuntimeImageSmokeTests(unittest.TestCase):
|
|
def test_selects_the_exact_platform_digest(self) -> None:
|
|
with tempfile.TemporaryDirectory(prefix="govoplan-runtime-smoke-") as value:
|
|
path = Path(value) / "metadata.json"
|
|
path.write_text(
|
|
json.dumps(
|
|
{
|
|
"index": "registry.example/api@sha256:" + "a" * 64,
|
|
"platforms": {
|
|
"linux/amd64": "registry.example/api@sha256:" + "1" * 64,
|
|
"linux/arm64": "registry.example/api@sha256:" + "2" * 64,
|
|
},
|
|
}
|
|
),
|
|
encoding="utf-8",
|
|
)
|
|
|
|
self.assertEqual(
|
|
"registry.example/api@sha256:" + "2" * 64,
|
|
MODULE.platform_image(path, "linux/arm64", "API"),
|
|
)
|
|
|
|
def test_rejects_mutable_or_missing_platform_images(self) -> None:
|
|
with tempfile.TemporaryDirectory(prefix="govoplan-runtime-smoke-") as value:
|
|
path = Path(value) / "metadata.json"
|
|
path.write_text(
|
|
json.dumps({"platforms": {"linux/amd64": "registry.example/api:latest"}}),
|
|
encoding="utf-8",
|
|
)
|
|
|
|
with self.assertRaisesRegex(MODULE.SmokeError, "exact sha256"):
|
|
MODULE.platform_image(path, "linux/amd64", "API")
|
|
with self.assertRaisesRegex(MODULE.SmokeError, "exact sha256"):
|
|
MODULE.platform_image(path, "linux/arm64", "API")
|
|
|
|
def test_readiness_fails_immediately_when_container_exits(self) -> None:
|
|
exited = subprocess.CompletedProcess([], 0, "false\n", "")
|
|
logs = subprocess.CompletedProcess(
|
|
[], 0, "fatal startup error db-secret\n", ""
|
|
)
|
|
with patch.object(MODULE, "_run", side_effect=(exited, logs)):
|
|
with self.assertRaisesRegex(
|
|
MODULE.SmokeError,
|
|
r"container exited before readiness: fatal startup error \[redacted\]",
|
|
):
|
|
MODULE._wait_for(
|
|
"WebUI",
|
|
lambda: self.fail("probe must not run for an exited container"),
|
|
timeout=60,
|
|
container="web",
|
|
redactions=("db-secret",),
|
|
)
|
|
|
|
def test_smoke_supplies_the_packaged_web_upstream_and_schema_contract(self) -> None:
|
|
source = SCRIPT.read_text(encoding="utf-8")
|
|
|
|
self.assertIn('"--network-alias",\n "load-balancer"', source)
|
|
self.assertIn("'core_system_settings'", source)
|
|
self.assertIn("'core_runtime_nodes'", source)
|
|
self.assertIn('if platform == "linux/arm64"', source)
|
|
self.assertIn('"ARM64-COW-BUG"', source)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|