Files
NODEDC_PLATFORM/infra/deploy-runner/test_bim_tasker_cad_ops_artifacts.py
T

106 lines
3.9 KiB
Python

#!/usr/bin/env python3
import hashlib
import json
import os
import subprocess
import tarfile
import tempfile
import unittest
from pathlib import Path
SCRIPT_DIR = Path(__file__).resolve().parent
BUILDER = SCRIPT_DIR / "build-bim-tasker-cad-ops-artifacts.mjs"
EXPECTED = {
"bim-viewer": {
"files": [
"converter/worker.py",
"server/cad-formats.js",
"server/index.js",
],
"manifest": (
"id=bim-viewer-cad-ops-20990101-001\n"
"component=bim-viewer\n"
"type=app-overlay\n"
),
},
"tasker": {
"files": [
"plane-src/apps/web/helpers/beam-viewer-config.ts",
"plane-src/apps/web/helpers/beam-viewer.ts",
],
"manifest": (
"id=tasker-cad-ops-20990101-001\n"
"component=tasker\n"
"type=app-overlay\n"
),
},
}
class BimTaskerCadOpsArtifactsTest(unittest.TestCase):
def build(self, artifact_dir: Path) -> dict:
environment = os.environ.copy()
environment["NODEDC_DEPLOY_ARTIFACT_DIR"] = str(artifact_dir)
result = subprocess.run(
["node", str(BUILDER), "20990101-001"],
check=True,
capture_output=True,
text=True,
env=environment,
)
return json.loads(result.stdout)
def test_artifacts_are_exact_deterministic_and_runtime_only(self) -> None:
with (
tempfile.TemporaryDirectory(prefix="nodedc-cad-ops-first-") as first_dir,
tempfile.TemporaryDirectory(prefix="nodedc-cad-ops-second-") as second_dir,
):
first = self.build(Path(first_dir))
second = self.build(Path(second_dir))
first_by_component = {item["component"]: item for item in first["artifacts"]}
second_by_component = {item["component"]: item for item in second["artifacts"]}
self.assertEqual(set(first_by_component), set(EXPECTED))
self.assertEqual(set(second_by_component), set(EXPECTED))
for component, contract in EXPECTED.items():
first_item = first_by_component[component]
second_item = second_by_component[component]
first_bytes = Path(first_item["artifact"]).read_bytes()
second_bytes = Path(second_item["artifact"]).read_bytes()
self.assertEqual(first_bytes, second_bytes)
self.assertEqual(first_item["sha256"], hashlib.sha256(first_bytes).hexdigest())
self.assertEqual(first_item["files"], contract["files"])
self.assertRegex(first_item["sourceCommit"], r"^[0-9a-f]{40}$")
with tarfile.open(first_item["artifact"], "r:gz") as archive:
members = archive.getmembers()
names = [member.name for member in members]
self.assertEqual(
archive.extractfile("manifest.env").read().decode("utf-8"),
contract["manifest"],
)
self.assertEqual(
archive.extractfile("files.txt").read().decode("utf-8").splitlines(),
contract["files"],
)
self.assertEqual(names[:3], ["manifest.env", "files.txt", "payload"])
self.assertTrue(all(member.isfile() or member.isdir() for member in members))
self.assertFalse(any(member.issym() or member.islnk() for member in members))
self.assertFalse(any("test" in name.lower() for name in names))
self.assertFalse(
any(
name.startswith("payload/") and ("/.env" in name or name.endswith(".env"))
for name in names
)
)
self.assertFalse(any(name.startswith("._") or "/._" in name for name in names))
if __name__ == "__main__":
unittest.main()