mirror of
https://github.com/python/cpython.git
synced 2025-07-27 21:24:32 +00:00
3e4a3dcb23
These files are created by some NFS clients a file is edited and removed concurrently (see added link in doc for more info). If such a file is removed between distutils calls listdir and copy, it will get confused. Other special files are ignored in sdist (namely VCS directories), but this has to be filtered out earlier.
127 lines
4.2 KiB
Python
127 lines
4.2 KiB
Python
"""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.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_suite():
|
|
return unittest.makeSuite(DirUtilTestCase)
|
|
|
|
if __name__ == "__main__":
|
|
run_unittest(test_suite())
|