mirror of
https://github.com/MacRimi/ProxMenux.git
synced 2026-11-01 18:32:13 +00:00
42 lines
1.9 KiB
Python
42 lines
1.9 KiB
Python
|
"""Compose aliases remain single, safe /etc/hosts tokens, including underscores."""
|
||
|
|
from pathlib import Path
|
||
|
|
import re
|
||
|
|
import subprocess
|
||
|
|
import sys
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
ROOT = Path(__file__).resolve().parents[1]
|
||
|
|
sys.path.insert(0, str(ROOT / 'src'))
|
||
|
|
from proxmenux_oci.converter import _extra_hosts
|
||
|
|
|
||
|
|
|
||
|
|
class ComposeHostAliasTests(unittest.TestCase):
|
||
|
|
@classmethod
|
||
|
|
def setUpClass(cls):
|
||
|
|
source = (ROOT / 'remote/install_oci.sh').read_text()
|
||
|
|
cls.shell = re.search(r'^oci_valid_extra_hostname\(\) \{\n.*?^\}', source, re.M | re.S)[0]
|
||
|
|
|
||
|
|
def shell_accepts(self, value):
|
||
|
|
return subprocess.run(['bash', '-c', self.shell + '\noci_valid_extra_hostname "$1"', 'alias-test', value],
|
||
|
|
capture_output=True).returncode == 0
|
||
|
|
|
||
|
|
def test_compose_container_aliases_are_accepted_by_import_and_installer(self):
|
||
|
|
for value in ('affine_postgres', 'affine_migration_job', 'db-1', 'db.internal', 'a', 'x' * 253):
|
||
|
|
with self.subTest(value=value):
|
||
|
|
self.assertTrue(self.shell_accepts(value))
|
||
|
|
self.assertEqual(_extra_hosts({value: '10.77.8.2'}),
|
||
|
|
[{'hostname': value, 'address': '10.77.8.2'}])
|
||
|
|
|
||
|
|
def test_multiline_whitespace_and_shell_syntax_are_refused(self):
|
||
|
|
for value in ('', 'db name', 'db\tname', 'db\n127.0.0.1 injected', '#comment',
|
||
|
|
'-db', '_db', '$(id)', 'db;id', 'db/../host', 'x' * 254):
|
||
|
|
with self.subTest(value=value):
|
||
|
|
self.assertFalse(self.shell_accepts(value))
|
||
|
|
self.assertIsNone(_extra_hosts({value: '10.77.8.2'}))
|
||
|
|
|
||
|
|
def test_apply_extra_hosts_uses_validated_token_before_writing(self):
|
||
|
|
source = (ROOT / 'remote/install_oci.sh').read_text()
|
||
|
|
apply = source.split('apply_extra_hosts() {', 1)[1].split('\n}', 1)[0]
|
||
|
|
self.assertLess(apply.index('oci_valid_extra_hostname "$hostname"'),
|
||
|
|
apply.index("printf '%s %s\\n'"))
|