| """Tests for distutils.dir_util.""" | |
| import unittest | |
| import os | |
| import stat | |
| import shutil | |
| import sys | |
| 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.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') | |
| f = open(a_file, 'w') | |
| try: | |
| f.write('some content') | |
| finally: | |
| f.close() | |
| 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_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_suite(): | |
| return unittest.makeSuite(DirUtilTestCase) | |
| if __name__ == "__main__": | |
| run_unittest(test_suite()) |