Files
stack/packages/mosaic/framework/tools/quality/scripts/test-framework-drift-check.py
T
2026-08-13 01:42:32 -05:00

77 lines
3.2 KiB
Python

#!/usr/bin/env python3
from __future__ import annotations
import os
from pathlib import Path
import subprocess
import sys
import tempfile
import unittest
CHECKER = Path(__file__).with_name("framework-drift-check.py")
class FrameworkDriftCheckTests(unittest.TestCase):
def setUp(self) -> None:
self.temp = tempfile.TemporaryDirectory()
root = Path(self.temp.name)
self.source = root / "framework" / "tools"
self.installed = root / "home" / "tools"
for directory in (self.source / "git", self.source / "_lib", self.installed / "git", self.installed / "_lib"):
directory.mkdir(parents=True, exist_ok=True)
(self.source / "git" / "guard.sh").write_text("fixed\n")
(self.source / "git" / "new-wrapper.sh").write_text("new\n")
(self.source / "_lib" / "credentials.json").write_text("source-placeholder\n")
def tearDown(self) -> None:
self.temp.cleanup()
def run_check(self, *extra: str) -> subprocess.CompletedProcess[str]:
return subprocess.run(
[sys.executable, str(CHECKER), "--source-root", str(self.source.parent), "--installed-root", str(self.installed), *extra],
text=True,
capture_output=True,
check=False,
env={**os.environ, "PYTHONDONTWRITEBYTECODE": "1"},
)
def test_fails_loudly_and_classifies_stale_missing_and_installed_only(self) -> None:
(self.installed / "git" / "guard.sh").write_text("broken\n")
(self.installed / "local-helper.sh").write_text("operator\n")
(self.installed / "_lib" / "credentials.json").write_text("secret\n")
result = self.run_check("--verbose")
self.assertEqual(result.returncode, 1)
self.assertIn("STALE git/guard.sh", result.stdout)
self.assertIn("NOT_INSTALLED git/new-wrapper.sh", result.stdout)
self.assertIn("INSTALLED_ONLY operator-or-unknown local-helper.sh", result.stdout)
self.assertNotIn("STALE _lib/credentials.json", result.stdout)
self.assertNotIn("NOT_INSTALLED _lib/credentials.json", result.stdout)
self.assertIn("in-sync=0 stale=1 not-installed=1 installed-only=1", result.stdout)
self.assertIn("FAIL deployed framework tools", result.stderr)
def test_passes_only_when_every_framework_owned_source_file_matches(self) -> None:
(self.installed / "git" / "guard.sh").write_text("fixed\n")
(self.installed / "git" / "new-wrapper.sh").write_text("new\n")
(self.installed / "_lib" / "credentials.json").write_text("different-operator-secret\n")
result = self.run_check()
self.assertEqual(result.returncode, 0, result.stderr)
self.assertIn("in-sync=2 stale=0 not-installed=0 installed-only=0", result.stdout)
def test_refuses_self_comparison_that_would_make_drift_unobservable(self) -> None:
result = subprocess.run(
[sys.executable, str(CHECKER), "--source-root", str(self.source), "--installed-root", str(self.source)],
text=True,
capture_output=True,
check=False,
)
self.assertEqual(result.returncode, 2)
self.assertIn("source and installed roots are identical", result.stderr)
if __name__ == "__main__":
unittest.main()