mirror of
https://github.com/MacRimi/ProxMenux.git
synced 2026-11-01 18:32:13 +00:00
141 lines
6.5 KiB
Python
141 lines
6.5 KiB
Python
"""Exercise the Cloud-Init menu without network access or guest creation."""
|
|
|
|
import os
|
|
from pathlib import Path
|
|
import re
|
|
import subprocess
|
|
import tempfile
|
|
import unittest
|
|
|
|
ROOT = Path(__file__).resolve().parents[3]
|
|
SELECTOR = ROOT / "scripts/vm/select_linux_iso.sh"
|
|
|
|
|
|
class CloudInitLauncherTests(unittest.TestCase):
|
|
def run_menu(self, selection="2", body=None, curl_exit=0, dialog_exit=0,
|
|
installer_exit=0, stdin="\n"):
|
|
source = SELECTOR.read_text(encoding="utf-8")
|
|
match = re.search(r"(?ms)^function select_linux_cloudinit\(\) \{.*?^\}", source)
|
|
assert match is not None, "Cloud-Init menu function not found"
|
|
with tempfile.TemporaryDirectory(prefix="cloudinit-launcher-") as temporary:
|
|
work = Path(temporary)
|
|
(work / "menus").mkdir()
|
|
(work / "menus/create_vm_menu.sh").write_text(
|
|
'printf "menu-returned\\n" >> "$ACTION_LOG"\n', encoding="utf-8")
|
|
environment = os.environ.copy()
|
|
environment.update({
|
|
"LOCAL_SCRIPTS": str(work), "SELECTION": selection,
|
|
"DIALOG_EXIT": str(dialog_exit), "CURL_EXIT": str(curl_exit),
|
|
"INSTALLER_EXIT": str(installer_exit),
|
|
"CURL_BODY": body if body is not None else
|
|
'printf "helper-executed\\n" >> "$ACTION_LOG"\nexit "$INSTALLER_EXIT"\n',
|
|
"ACTION_LOG": str(work / "actions"),
|
|
"CURL_ARGS": str(work / "curl-args"),
|
|
"DIALOG_LOG": str(work / "dialog-args"),
|
|
})
|
|
prelude = r'''
|
|
translate() { printf '%s' "$1"; }
|
|
msg_success() { printf 'continue-prompt\n' >> "$ACTION_LOG"; }
|
|
whiptail() { printf 'website-prompt\n' >> "$ACTION_LOG"; }
|
|
dialog() {
|
|
printf '%s\n' "$@" >> "$DIALOG_LOG"
|
|
if [[ " $* " == *" --menu "* ]]; then
|
|
printf '%s' "$SELECTION" >&2
|
|
return "$DIALOG_EXIT"
|
|
fi
|
|
}
|
|
curl() {
|
|
printf '%s\n' "$@" > "$CURL_ARGS"
|
|
printf '%s' "$CURL_BODY"
|
|
return "$CURL_EXIT"
|
|
}
|
|
'''
|
|
result = subprocess.run(
|
|
["bash", "--noprofile", "--norc", "-c",
|
|
prelude + match.group() + "\nselect_linux_cloudinit\n"],
|
|
input=stdin, capture_output=True, text=True, env=environment,
|
|
timeout=5)
|
|
def contents(name):
|
|
path = work / name
|
|
return path.read_text(encoding="utf-8") if path.exists() else ""
|
|
return result, contents("curl-args"), contents("dialog-args"), contents("actions")
|
|
|
|
def test_debian_uses_the_maintained_generic_installer(self):
|
|
result, curl_args, _, actions = self.run_menu()
|
|
self.assertEqual(result.returncode, 0, result.stderr)
|
|
self.assertEqual(curl_args.splitlines()[-1],
|
|
"https://raw.githubusercontent.com/community-scripts/ProxmoxVE/main/vm/debian-vm.sh")
|
|
self.assertIn("helper-executed", actions)
|
|
|
|
|
|
def test_ubuntu_uses_one_generic_entry_without_obsolete_release_labels(self):
|
|
result, curl_args, menu, actions = self.run_menu(selection="3")
|
|
self.assertEqual(result.returncode, 0, result.stderr)
|
|
self.assertEqual(curl_args.splitlines()[-1],
|
|
"https://raw.githubusercontent.com/community-scripts/ProxmoxVE/main/vm/ubuntu-vm.sh")
|
|
self.assertIn("helper-executed", actions)
|
|
self.assertEqual(menu.count("(Cloud-Init automated)"), 3)
|
|
for obsolete in ("Debian 13", "Debian 12", "Ubuntu 22.04", "Ubuntu 24.04",
|
|
"Ubuntu 24.10", "Ubuntu 25.04"):
|
|
self.assertNotIn(obsolete, menu)
|
|
|
|
|
|
def test_failed_or_empty_download_never_runs_a_script(self):
|
|
for body, exit_code in (("404: Not Found", 22),
|
|
('printf "helper-executed\\n" >> "$ACTION_LOG"', 28),
|
|
("", 0), (" \n\t", 0)):
|
|
with self.subTest(body=body, exit_code=exit_code):
|
|
result, _, messages, actions = self.run_menu(body=body, curl_exit=exit_code)
|
|
self.assertNotIn("helper-executed", actions)
|
|
self.assertNotEqual(result.returncode, 0)
|
|
self.assertIn("Error: Failed to download the script.", messages)
|
|
self.assertNotIn("menu-returned", actions)
|
|
|
|
|
|
def test_installer_failure_returns_its_status_without_success_navigation(self):
|
|
result, _, _, actions = self.run_menu(installer_exit=17)
|
|
self.assertEqual(result.returncode, 17, result.stderr)
|
|
self.assertIn("helper-executed", actions)
|
|
self.assertNotIn("continue-prompt", actions)
|
|
self.assertNotIn("website-prompt", actions)
|
|
self.assertNotIn("menu-returned", actions)
|
|
|
|
|
|
def test_arch_and_successful_menu_navigation_are_preserved(self):
|
|
result, curl_args, _, actions = self.run_menu(selection="1")
|
|
self.assertEqual(result.returncode, 0, result.stderr)
|
|
self.assertEqual(curl_args.splitlines()[-1],
|
|
"https://raw.githubusercontent.com/community-scripts/ProxmoxVE/main/vm/archlinux-vm.sh")
|
|
self.assertEqual(actions.splitlines(),
|
|
["helper-executed", "continue-prompt", "website-prompt", "menu-returned"])
|
|
arguments = curl_args.splitlines()
|
|
self.assertIn("-fsSL", arguments)
|
|
self.assertEqual(arguments[arguments.index("--connect-timeout") + 1], "5")
|
|
self.assertEqual(arguments[arguments.index("--max-time") + 1], "60")
|
|
|
|
def test_return_and_cancel_do_not_download_or_execute(self):
|
|
for selection, status, expected in (("4", 0, 1), ("2", 1, 0), ("2", 255, 0)):
|
|
with self.subTest(selection=selection, status=status):
|
|
result, curl_args, _, actions = self.run_menu(
|
|
selection=selection, dialog_exit=status)
|
|
self.assertEqual(result.returncode, expected, result.stderr)
|
|
self.assertEqual(curl_args, "")
|
|
self.assertEqual(actions, "")
|
|
|
|
def test_unknown_selection_is_rejected(self):
|
|
result, curl_args, _, actions = self.run_menu(selection="999")
|
|
self.assertNotEqual(result.returncode, 0)
|
|
self.assertEqual(curl_args, "")
|
|
self.assertEqual(actions, "")
|
|
|
|
def test_installer_keeps_interactive_standard_input(self):
|
|
body = 'IFS= read -r answer; printf "input:%s\\n" "$answer" >> "$ACTION_LOG"\n'
|
|
result, _, _, actions = self.run_menu(body=body, stdin="answer with spaces\n\n")
|
|
self.assertEqual(result.returncode, 0, result.stderr)
|
|
self.assertIn("input:answer with spaces\n", actions)
|
|
self.assertIn("menu-returned", actions)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|