update fix

2026-09-22 22:05:38 -05:00
parent 158d3ffe76
commit 0c1c5033e0
5 changed files with 59 additions and 5 deletions
@@ -0,0 +1,42 @@
+#!/usr/bin/env python3
+"""Exercise menu reloads after successful and partially failed updates."""
+from pathlib import Path
+import os
+import subprocess
+import tempfile
+import unittest
+
+ROOT = Path(__file__).resolve().parents[1]
+
+
+class MenuUpdateTests(unittest.TestCase):
+    def test_reload_after_success_or_failure(self):
+        source = (ROOT / 'proxmenu-scripts.sh').read_text()
+        function = source.split('TAPM_INSTALL_MENU_UPDATE() {', 1)[1].split('\n}\n', 1)[0]
+        for status in (0, 1):
+            with self.subTest(update_status=status), tempfile.TemporaryDirectory() as directory:
+                launcher = Path(directory) / 'run.sh'
+                log = Path(directory) / 'calls'
+                launcher.write_text('''#!/bin/bash
+if [[ "${1:-}" == update ]]; then
+    echo update >> "$TEST_LOG"
+    exit "$TEST_STATUS"
+fi
+echo reload >> "$TEST_LOG"
+''')
+                launcher.chmod(0o700)
+                script = 'ENTER2CONTINUE() { echo pause >> "$TEST_LOG"; }\n'
+                script += 'TAPM_INSTALL_MENU_UPDATE() {' + function.replace(
+                    '/opt/idssys/ta-proxmenu/run.sh', '"' + str(launcher) + '"') + '\n}\n'
+                script += 'TAPM_INSTALL_MENU_UPDATE\necho stale-menu >> "$TEST_LOG"\n'
+                result = subprocess.run(['bash', '-c', script], capture_output=True, text=True,
+                                        env=dict(os.environ, TEST_LOG=str(log), TEST_STATUS=str(status)))
+                self.assertEqual(result.returncode, 0, result.stderr)
+                self.assertEqual(log.read_text().splitlines(),
+                                 ['update', 'reload'] if status == 0 else ['update', 'pause', 'reload'])
+                if status:
+                    self.assertIn('reported failures', result.stdout)
+
+
+if __name__ == '__main__':
+    unittest.main()