110 lines
3.1 KiB
Python
110 lines
3.1 KiB
Python
|
|
import os
|
|
|
|
import mitogen
|
|
import mitogen.sudo
|
|
|
|
import unittest2
|
|
|
|
import testlib
|
|
|
|
|
|
class ConstructorTest(testlib.RouterMixin, testlib.TestCase):
|
|
sudo_path = testlib.data_path('stubs/stub-sudo.py')
|
|
|
|
def run_sudo(self, **kwargs):
|
|
context = self.router.sudo(
|
|
sudo_path=self.sudo_path,
|
|
**kwargs
|
|
)
|
|
argv = eval(context.call(os.getenv, 'ORIGINAL_ARGV'))
|
|
return context, argv
|
|
|
|
|
|
def test_basic(self):
|
|
context, argv = self.run_sudo()
|
|
self.assertEquals(argv[:4], [
|
|
self.sudo_path,
|
|
'-u', 'root',
|
|
'--'
|
|
])
|
|
|
|
def test_selinux_type_role(self):
|
|
context, argv = self.run_sudo(
|
|
selinux_type='setype',
|
|
selinux_role='serole',
|
|
)
|
|
self.assertEquals(argv[:8], [
|
|
self.sudo_path,
|
|
'-u', 'root',
|
|
'-r', 'serole',
|
|
'-t', 'setype',
|
|
'--'
|
|
])
|
|
|
|
def test_reparse_args(self):
|
|
context, argv = self.run_sudo(
|
|
sudo_args=['--type', 'setype', '--role', 'serole', '--user', 'user']
|
|
)
|
|
self.assertEquals(argv[:8], [
|
|
self.sudo_path,
|
|
'-u', 'user',
|
|
'-r', 'serole',
|
|
'-t', 'setype',
|
|
'--'
|
|
])
|
|
|
|
def test_tty_preserved(self):
|
|
# issue #481
|
|
os.environ['PREHISTORIC_SUDO'] = '1'
|
|
try:
|
|
context, argv = self.run_sudo()
|
|
self.assertEquals('1', context.call(os.getenv, 'PREHISTORIC_SUDO'))
|
|
finally:
|
|
del os.environ['PREHISTORIC_SUDO']
|
|
|
|
|
|
class NonEnglishPromptTest(testlib.DockerMixin, testlib.TestCase):
|
|
# Only mitogen/debian-test has a properly configured sudo.
|
|
mitogen_test_distro = 'debian'
|
|
|
|
def test_password_required(self):
|
|
ssh = self.docker_ssh(
|
|
username='mitogen__has_sudo',
|
|
password='has_sudo_password',
|
|
)
|
|
ssh.call(os.putenv, 'LANGUAGE', 'fr')
|
|
ssh.call(os.putenv, 'LC_ALL', 'fr_FR.UTF-8')
|
|
e = self.assertRaises(mitogen.core.StreamError,
|
|
lambda: self.router.sudo(via=ssh)
|
|
)
|
|
self.assertTrue(mitogen.sudo.password_required_msg in str(e))
|
|
|
|
def test_password_incorrect(self):
|
|
ssh = self.docker_ssh(
|
|
username='mitogen__has_sudo',
|
|
password='has_sudo_password',
|
|
)
|
|
ssh.call(os.putenv, 'LANGUAGE', 'fr')
|
|
ssh.call(os.putenv, 'LC_ALL', 'fr_FR.UTF-8')
|
|
e = self.assertRaises(mitogen.core.StreamError,
|
|
lambda: self.router.sudo(via=ssh, password='x')
|
|
)
|
|
self.assertTrue(mitogen.sudo.password_incorrect_msg in str(e))
|
|
|
|
def test_password_okay(self):
|
|
ssh = self.docker_ssh(
|
|
username='mitogen__has_sudo',
|
|
password='has_sudo_password',
|
|
)
|
|
ssh.call(os.putenv, 'LANGUAGE', 'fr')
|
|
ssh.call(os.putenv, 'LC_ALL', 'fr_FR.UTF-8')
|
|
e = self.assertRaises(mitogen.core.StreamError,
|
|
lambda: self.router.sudo(via=ssh, password='rootpassword')
|
|
)
|
|
self.assertTrue(mitogen.sudo.password_incorrect_msg in str(e))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest2.main()
|