You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
108 lines
3.3 KiB
Python
108 lines
3.3 KiB
Python
|
|
from __future__ import absolute_import
|
|
import os.path
|
|
import subprocess
|
|
import tempfile
|
|
import unittest
|
|
|
|
import mock
|
|
|
|
import ansible_mitogen.target
|
|
import testlib
|
|
|
|
|
|
LOGGER_NAME = ansible_mitogen.target.LOG.name
|
|
|
|
|
|
class NamedTemporaryDirectory(object):
|
|
def __init__(self, **kwargs):
|
|
self.kwargs = kwargs
|
|
|
|
def __enter__(self):
|
|
self.path = tempfile.mkdtemp(**self.kwargs)
|
|
return self.path
|
|
|
|
def __exit__(self, _1, _2, _3):
|
|
subprocess.check_call(['rm', '-rf', self.path])
|
|
|
|
|
|
class FindGoodTempDirTest(testlib.TestCase):
|
|
func = staticmethod(ansible_mitogen.target.find_good_temp_dir)
|
|
|
|
def test_expands_usernames(self):
|
|
with NamedTemporaryDirectory(
|
|
prefix='.ansible_mitogen_test',
|
|
dir=os.environ['HOME']
|
|
) as tmpdir:
|
|
path = self.func(['~'])
|
|
self.assertTrue(path.startswith(os.environ['HOME']))
|
|
|
|
def test_expands_vars(self):
|
|
with NamedTemporaryDirectory(
|
|
prefix='.ansible_mitogen_test',
|
|
dir=os.environ['HOME']
|
|
) as tmpdir:
|
|
os.environ['somevar'] = 'xyz'
|
|
path = self.func([tmpdir + '/$somevar'])
|
|
self.assertTrue(path.startswith('%s/%s' % (tmpdir, 'xyz')))
|
|
|
|
@mock.patch('ansible_mitogen.target.is_good_temp_dir')
|
|
def test_no_good_candidate(self, is_good_temp_dir):
|
|
is_good_temp_dir.return_value = False
|
|
e = self.assertRaises(IOError,
|
|
lambda: self.func([])
|
|
)
|
|
self.assertTrue(str(e).startswith('Unable to find a useable'))
|
|
|
|
|
|
|
|
class ApplyModeSpecTest(unittest.TestCase):
|
|
func = staticmethod(ansible_mitogen.target.apply_mode_spec)
|
|
|
|
def test_simple(self):
|
|
spec = 'u+rwx,go=x'
|
|
self.assertEquals(int('0711', 8), self.func(spec, 0))
|
|
|
|
spec = 'g-rw'
|
|
self.assertEquals(int('0717', 8), self.func(spec, int('0777', 8)))
|
|
|
|
|
|
class IsGoodTempDirTest(unittest.TestCase):
|
|
func = staticmethod(ansible_mitogen.target.is_good_temp_dir)
|
|
|
|
def test_creates(self):
|
|
with NamedTemporaryDirectory() as temp_path:
|
|
bleh = os.path.join(temp_path, 'bleh')
|
|
self.assertFalse(os.path.exists(bleh))
|
|
self.assertTrue(self.func(bleh))
|
|
self.assertTrue(os.path.exists(bleh))
|
|
|
|
def test_file_exists(self):
|
|
with NamedTemporaryDirectory() as temp_path:
|
|
bleh = os.path.join(temp_path, 'bleh')
|
|
with open(bleh, 'w') as fp:
|
|
fp.write('derp')
|
|
self.assertTrue(os.path.isfile(bleh))
|
|
self.assertFalse(self.func(bleh))
|
|
self.assertEquals(open(bleh).read(), 'derp')
|
|
|
|
@unittest.skipIf(
|
|
os.geteuid() == 0, 'writes by root ignore directory permissions')
|
|
def test_unwriteable(self):
|
|
with NamedTemporaryDirectory() as temp_path:
|
|
os.chmod(temp_path, 0)
|
|
self.assertFalse(self.func(temp_path))
|
|
os.chmod(temp_path, int('0700', 8))
|
|
|
|
@mock.patch('os.chmod')
|
|
def test_weird_filesystem(self, os_chmod):
|
|
os_chmod.side_effect = OSError('nope')
|
|
with NamedTemporaryDirectory() as temp_path:
|
|
self.assertFalse(self.func(temp_path))
|
|
|
|
@mock.patch('os.access')
|
|
def test_noexec(self, os_access):
|
|
os_access.return_value = False
|
|
with NamedTemporaryDirectory() as temp_path:
|
|
self.assertFalse(self.func(temp_path))
|