| """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()) |