55 lines
2.4 KiB
Python
55 lines
2.4 KiB
Python
|
import importlib.util
|
||
|
|
from pathlib import Path
|
||
|
|
import shlex
|
||
|
|
import unittest
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
ROOT = Path(__file__).resolve().parents[1]
|
||
|
|
spec = importlib.util.spec_from_file_location('kube_route', ROOT / 'scripts/kube_route.py')
|
||
|
|
route = importlib.util.module_from_spec(spec)
|
||
|
|
spec.loader.exec_module(route)
|
||
|
|
|
||
|
|
|
||
|
|
class KubeRouteTest(unittest.TestCase):
|
||
|
|
def test_default_is_agent_without_escalation(self):
|
||
|
|
argv = route.ssh_argv(['kubectl', 'get', 'nodes'], environ={})
|
||
|
|
self.assertEqual(argv[:-1], route.SSH)
|
||
|
|
self.assertEqual(shlex.split(argv[-1]), ['kubectl', 'get', 'nodes'])
|
||
|
|
self.assertNotIn('sudo', argv[-1])
|
||
|
|
|
||
|
|
def test_founder_sudo_never_prompts_and_never_allocates_tty(self):
|
||
|
|
argv = route.ssh_argv(['python3', '-'], environ={'KUBECTL_ROUTE': 'founder-sudo'})
|
||
|
|
self.assertEqual(shlex.split(argv[-1]), ['sudo', '-n', 'python3', '-'])
|
||
|
|
self.assertNotIn('-t', argv)
|
||
|
|
self.assertIn('BatchMode=yes', argv)
|
||
|
|
|
||
|
|
def test_unknown_or_shell_route_refused(self):
|
||
|
|
for value in ('', 'root', 'sudo kubectl', 'founder-sudo; rm -rf /'):
|
||
|
|
with self.assertRaises(ValueError):
|
||
|
|
route.ssh_argv(['kubectl'], environ={'KUBECTL_ROUTE': value})
|
||
|
|
|
||
|
|
def test_remote_arguments_are_quoted(self):
|
||
|
|
argv = route.ssh_argv(['kubectl', 'patch', '-p', '{"a": "b c"}'], environ={})
|
||
|
|
self.assertEqual(shlex.split(argv[-1])[-1], '{"a": "b c"}')
|
||
|
|
|
||
|
|
def test_environment_is_read_when_not_given(self):
|
||
|
|
with patch.dict('os.environ', {'KUBECTL_ROUTE': 'founder-sudo'}):
|
||
|
|
self.assertEqual(route.remote_prefix(), ['sudo', '-n'])
|
||
|
|
|
||
|
|
def test_empty_command_refused(self):
|
||
|
|
with self.assertRaises(ValueError):
|
||
|
|
route.ssh_argv([], environ={})
|
||
|
|
|
||
|
|
def test_bounded_scripts_use_the_route(self):
|
||
|
|
for name in ('argocd_public_transport.py', 'argocd_adoption_proof.py', 'workload_adoption_proof.py',
|
||
|
|
'activity_promotion_readiness.py', 'argocd_child_pin_drift.py', 'capture_service_observation.py'):
|
||
|
|
text = (ROOT / 'scripts' / name).read_text()
|
||
|
|
self.assertIn('ssh_argv(', text, name)
|
||
|
|
self.assertNotIn("'railiance01', shlex.join(['kubectl'", text, name)
|
||
|
|
self.assertNotIn("['ssh', 'railiance01', 'python3'", text, name)
|
||
|
|
self.assertNotIn("['ssh','railiance01','python3'", text, name)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|