140 lines
4.5 KiB
Python
140 lines
4.5 KiB
Python
|
"""Tests for distutils.dir_util."""
|
||
|
|
import unittest
|
||
|
|
import os
|
||
|
|
import stat
|
||
|
|
import sys
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
from distutils import dir_util, errors
|
||
|
|
from distutils.dir_util import (mkpath, remove_tree, create_tree, copy_tree,
|
||
|
|
ensure_relative)
|
||
|
|
|
||
|
|
from distutils import log
|
||
|
|
from distutils.tests import support
|
||
|
|
from test.support import run_unittest
|
||
|
|
|
||
|
|
|
||
|
|
class DirUtilTestCase(support.TempdirManager, unittest.TestCase):
|
||
|
|
|
||
|
|
def _log(self, msg, *args):
|
||
|
|
if len(args) > 0:
|
||
|
|
self._logs.append(msg % args)
|
||
|
|
else:
|
||
|
|
self._logs.append(msg)
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
super(DirUtilTestCase, self).setUp()
|
||
|
|
self._logs = []
|
||
|
|
tmp_dir = self.mkdtemp()
|
||
|
|
self.root_target = os.path.join(tmp_dir, 'deep')
|
||
|
|
self.target = os.path.join(self.root_target, 'here')
|
||
|
|
self.target2 = os.path.join(tmp_dir, 'deep2')
|
||
|
|
self.old_log = log.info
|
||
|
|
log.info = self._log
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
log.info = self.old_log
|
||
|
|
super(DirUtilTestCase, self).tearDown()
|
||
|
|
|
||
|
|
def test_mkpath_remove_tree_verbosity(self):
|
||
|
|
|
||
|
|
mkpath(self.target, verbose=0)
|
||
|
|
wanted = []
|
||
|
|
self.assertEqual(self._logs, wanted)
|
||
|
|
remove_tree(self.root_target, verbose=0)
|
||
|
|
|
||
|
|
mkpath(self.target, verbose=1)
|
||
|
|
wanted = ['creating %s' % self.root_target,
|
||
|
|
'creating %s' % self.target]
|
||
|
|
self.assertEqual(self._logs, wanted)
|
||
|
|
self._logs = []
|
||
|
|
|
||
|
|
remove_tree(self.root_target, verbose=1)
|
||
|
|
wanted = ["removing '%s' (and everything under it)" % self.root_target]
|
||
|
|
self.assertEqual(self._logs, wanted)
|
||
|
|
|
||
|
|
@unittest.skipIf(sys.platform.startswith('win'),
|
||
|
|
"This test is only appropriate for POSIX-like systems.")
|
||
|
|
def test_mkpath_with_custom_mode(self):
|
||
|
|
# Get and set the current umask value for testing mode bits.
|
||
|
|
umask = os.umask(0o002)
|
||
|
|
os.umask(umask)
|
||
|
|
mkpath(self.target, 0o700)
|
||
|
|
self.assertEqual(
|
||
|
|
stat.S_IMODE(os.stat(self.target).st_mode), 0o700 & ~umask)
|
||
|
|
mkpath(self.target2, 0o555)
|
||
|
|
self.assertEqual(
|
||
|
|
stat.S_IMODE(os.stat(self.target2).st_mode), 0o555 & ~umask)
|
||
|
|
|
||
|
|
def test_create_tree_verbosity(self):
|
||
|
|
|
||
|
|
create_tree(self.root_target, ['one', 'two', 'three'], verbose=0)
|
||
|
|
self.assertEqual(self._logs, [])
|
||
|
|
remove_tree(self.root_target, verbose=0)
|
||
|
|
|
||
|
|
wanted = ['creating %s' % self.root_target]
|
||
|
|
create_tree(self.root_target, ['one', 'two', 'three'], verbose=1)
|
||
|
|
self.assertEqual(self._logs, wanted)
|
||
|
|
|
||
|
|
remove_tree(self.root_target, verbose=0)
|
||
|
|
|
||
|
|
def test_copy_tree_verbosity(self):
|
||
|
|
|
||
|
|
mkpath(self.target, verbose=0)
|
||
|
|
|
||
|
|
copy_tree(self.target, self.target2, verbose=0)
|
||
|
|
self.assertEqual(self._logs, [])
|
||
|
|
|
||
|
|
remove_tree(self.root_target, verbose=0)
|
||
|
|
|
||
|
|
mkpath(self.target, verbose=0)
|
||
|
|
a_file = os.path.join(self.target, 'ok.txt')
|
||
|
|
with open(a_file, 'w') as f:
|
||
|
|
f.write('some content')
|
||
|
|
|
||
|
|
wanted = ['copying %s -> %s' % (a_file, self.target2)]
|
||
|
|
copy_tree(self.target, self.target2, verbose=1)
|
||
|
|
self.assertEqual(self._logs, wanted)
|
||
|
|
|
||
|
|
remove_tree(self.root_target, verbose=0)
|
||
|
|
remove_tree(self.target2, verbose=0)
|
||
|
|
|
||
|
|
def test_copy_tree_skips_nfs_temp_files(self):
|
||
|
|
mkpath(self.target, verbose=0)
|
||
|
|
|
||
|
|
a_file = os.path.join(self.target, 'ok.txt')
|
||
|
|
nfs_file = os.path.join(self.target, '.nfs123abc')
|
||
|
|
for f in a_file, nfs_file:
|
||
|
|
with open(f, 'w') as fh:
|
||
|
|
fh.write('some content')
|
||
|
|
|
||
|
|
copy_tree(self.target, self.target2)
|
||
|
|
self.assertEqual(os.listdir(self.target2), ['ok.txt'])
|
||
|
|
|
||
|
|
remove_tree(self.root_target, verbose=0)
|
||
|
|
remove_tree(self.target2, verbose=0)
|
||
|
|
|
||
|
|
def test_ensure_relative(self):
|
||
|
|
if os.sep == '/':
|
||
|
|
self.assertEqual(ensure_relative('/home/foo'), 'home/foo')
|
||
|
|
self.assertEqual(ensure_relative('some/path'), 'some/path')
|
||
|
|
else: # \\
|
||
|
|
self.assertEqual(ensure_relative('c:\\home\\foo'), 'c:home\\foo')
|
||
|
|
self.assertEqual(ensure_relative('home\\foo'), 'home\\foo')
|
||
|
|
|
||
|
|
def test_copy_tree_exception_in_listdir(self):
|
||
|
|
"""
|
||
|
|
An exception in listdir should raise a DistutilsFileError
|
||
|
|
"""
|
||
|
|
with patch("os.listdir", side_effect=OSError()), \
|
||
|
|
self.assertRaises(errors.DistutilsFileError):
|
||
|
|
src = self.tempdirs[-1]
|
||
|
|
dir_util.copy_tree(src, None)
|
||
|
|
|
||
|
|
|
||
|
|
def test_suite():
|
||
|
|
return unittest.makeSuite(DirUtilTestCase)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
run_unittest(test_suite())
|