Lib/distutils/tests/test_dir_util.py

   1 """Tests for distutils.dir_util."""
   2 import unittest
   3 import os
   4 import shutil
   5
   6 from distutils.dir_util import (mkpath, remove_tree, create_tree, copy_tree,
   7                                 ensure_relative)
   8
   9 from distutils import log
  10 from distutils.tests import support
  11
  12 class DirUtilTestCase(support.TempdirManager, unittest.TestCase):
  13
  14     def _log(self, msg, *args):
  15         if len(args) > 0:
  16             self._logs.append(msg % args)
  17         else:
  18             self._logs.append(msg)
  19
  20     def setUp(self):
  21         super(DirUtilTestCase, self).setUp()
  22         self._logs = []
  23         tmp_dir = self.mkdtemp()
  24         self.root_target = os.path.join(tmp_dir, 'deep')
  25         self.target = os.path.join(self.root_target, 'here')
  26         self.target2 = os.path.join(tmp_dir, 'deep2')
  27         self.old_log = log.info
  28         log.info = self._log
  29
  30     def tearDown(self):
  31         log.info = self.old_log
  32         super(DirUtilTestCase, self).tearDown()
  33
  34     def test_mkpath_remove_tree_verbosity(self):
  35
  36         mkpath(self.target, verbose=0)
  37         wanted = []
  38         self.assertEquals(self._logs, wanted)
  39         remove_tree(self.root_target, verbose=0)
  40
  41         mkpath(self.target, verbose=1)
  42         wanted = ['creating %s' % self.root_target,
  43                   'creating %s' % self.target]
  44         self.assertEquals(self._logs, wanted)
  45         self._logs = []
  46
  47         remove_tree(self.root_target, verbose=1)
  48         wanted = ["removing '%s' (and everything under it)" % self.root_target]
  49         self.assertEquals(self._logs, wanted)
  50
  51     def test_create_tree_verbosity(self):
  52
  53         create_tree(self.root_target, ['one', 'two', 'three'], verbose=0)
  54         self.assertEquals(self._logs, [])
  55         remove_tree(self.root_target, verbose=0)
  56
  57         wanted = ['creating %s' % self.root_target]
  58         create_tree(self.root_target, ['one', 'two', 'three'], verbose=1)
  59         self.assertEquals(self._logs, wanted)
  60
  61         remove_tree(self.root_target, verbose=0)
  62
  63
  64     def test_copy_tree_verbosity(self):
  65
  66         mkpath(self.target, verbose=0)
  67
  68         copy_tree(self.target, self.target2, verbose=0)
  69         self.assertEquals(self._logs, [])
  70
  71         remove_tree(self.root_target, verbose=0)
  72
  73         mkpath(self.target, verbose=0)
  74         a_file = os.path.join(self.target, 'ok.txt')
  75         f = open(a_file, 'w')
  76         f.write('some content')
  77         f.close()
  78
  79         wanted = ['copying %s -> %s' % (a_file, self.target2)]
  80         copy_tree(self.target, self.target2, verbose=1)
  81         self.assertEquals(self._logs, wanted)
  82
  83         remove_tree(self.root_target, verbose=0)
  84         remove_tree(self.target2, verbose=0)
  85
  86     def test_ensure_relative(self):
  87         if os.sep == '/':
  88             self.assertEquals(ensure_relative('/home/foo'), 'home/foo')
  89             self.assertEquals(ensure_relative('some/path'), 'some/path')
  90         else:   # \\
  91             self.assertEquals(ensure_relative('c:\\home\\foo'), 'c:home\\foo')
  92             self.assertEquals(ensure_relative('home\\foo'), 'home\\foo')
  93
  94 def test_suite():
  95     return unittest.makeSuite(DirUtilTestCase)
  96
  97 if __name__ == "__main__":
  98     unittest.main(defaultTest="test_suite")