mirror of
https://github.com/python/cpython.git
synced 2024-11-24 10:24:35 +08:00
40348acc18
Instead of explicitly enumerate test classes for run_unittest() use the unittest ability to discover tests. This also makes these tests discoverable and runnable with unittest. load_tests() can be used for dynamic generating tests and adding doctests. setUpModule(), tearDownModule() and addModuleCleanup() can be used for running code before and after all module tests.
251 lines
10 KiB
Python
251 lines
10 KiB
Python
import filecmp
|
|
import os
|
|
import shutil
|
|
import tempfile
|
|
import unittest
|
|
|
|
from test import support
|
|
from test.support import os_helper
|
|
|
|
|
|
class FileCompareTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
self.name = os_helper.TESTFN
|
|
self.name_same = os_helper.TESTFN + '-same'
|
|
self.name_diff = os_helper.TESTFN + '-diff'
|
|
data = 'Contents of file go here.\n'
|
|
for name in [self.name, self.name_same, self.name_diff]:
|
|
with open(name, 'w', encoding="utf-8") as output:
|
|
output.write(data)
|
|
|
|
with open(self.name_diff, 'a+', encoding="utf-8") as output:
|
|
output.write('An extra line.\n')
|
|
self.dir = tempfile.gettempdir()
|
|
|
|
def tearDown(self):
|
|
os.unlink(self.name)
|
|
os.unlink(self.name_same)
|
|
os.unlink(self.name_diff)
|
|
|
|
def test_matching(self):
|
|
self.assertTrue(filecmp.cmp(self.name, self.name),
|
|
"Comparing file to itself fails")
|
|
self.assertTrue(filecmp.cmp(self.name, self.name, shallow=False),
|
|
"Comparing file to itself fails")
|
|
self.assertTrue(filecmp.cmp(self.name, self.name_same),
|
|
"Comparing file to identical file fails")
|
|
self.assertTrue(filecmp.cmp(self.name, self.name_same, shallow=False),
|
|
"Comparing file to identical file fails")
|
|
|
|
def test_different(self):
|
|
self.assertFalse(filecmp.cmp(self.name, self.name_diff),
|
|
"Mismatched files compare as equal")
|
|
self.assertFalse(filecmp.cmp(self.name, self.dir),
|
|
"File and directory compare as equal")
|
|
|
|
def test_cache_clear(self):
|
|
first_compare = filecmp.cmp(self.name, self.name_same, shallow=False)
|
|
second_compare = filecmp.cmp(self.name, self.name_diff, shallow=False)
|
|
filecmp.clear_cache()
|
|
self.assertTrue(len(filecmp._cache) == 0,
|
|
"Cache not cleared after calling clear_cache")
|
|
|
|
class DirCompareTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
tmpdir = tempfile.gettempdir()
|
|
self.dir = os.path.join(tmpdir, 'dir')
|
|
self.dir_same = os.path.join(tmpdir, 'dir-same')
|
|
self.dir_diff = os.path.join(tmpdir, 'dir-diff')
|
|
|
|
# Another dir is created under dir_same, but it has a name from the
|
|
# ignored list so it should not affect testing results.
|
|
self.dir_ignored = os.path.join(self.dir_same, '.hg')
|
|
|
|
self.caseinsensitive = os.path.normcase('A') == os.path.normcase('a')
|
|
data = 'Contents of file go here.\n'
|
|
for dir in (self.dir, self.dir_same, self.dir_diff, self.dir_ignored):
|
|
shutil.rmtree(dir, True)
|
|
os.mkdir(dir)
|
|
subdir_path = os.path.join(dir, 'subdir')
|
|
os.mkdir(subdir_path)
|
|
if self.caseinsensitive and dir is self.dir_same:
|
|
fn = 'FiLe' # Verify case-insensitive comparison
|
|
else:
|
|
fn = 'file'
|
|
with open(os.path.join(dir, fn), 'w', encoding="utf-8") as output:
|
|
output.write(data)
|
|
|
|
with open(os.path.join(self.dir_diff, 'file2'), 'w', encoding="utf-8") as output:
|
|
output.write('An extra file.\n')
|
|
|
|
def tearDown(self):
|
|
for dir in (self.dir, self.dir_same, self.dir_diff):
|
|
shutil.rmtree(dir)
|
|
|
|
def test_default_ignores(self):
|
|
self.assertIn('.hg', filecmp.DEFAULT_IGNORES)
|
|
|
|
def test_cmpfiles(self):
|
|
self.assertTrue(filecmp.cmpfiles(self.dir, self.dir, ['file']) ==
|
|
(['file'], [], []),
|
|
"Comparing directory to itself fails")
|
|
self.assertTrue(filecmp.cmpfiles(self.dir, self.dir_same, ['file']) ==
|
|
(['file'], [], []),
|
|
"Comparing directory to same fails")
|
|
|
|
# Try it with shallow=False
|
|
self.assertTrue(filecmp.cmpfiles(self.dir, self.dir, ['file'],
|
|
shallow=False) ==
|
|
(['file'], [], []),
|
|
"Comparing directory to itself fails")
|
|
self.assertTrue(filecmp.cmpfiles(self.dir, self.dir_same, ['file'],
|
|
shallow=False),
|
|
"Comparing directory to same fails")
|
|
|
|
# Add different file2
|
|
with open(os.path.join(self.dir, 'file2'), 'w', encoding="utf-8") as output:
|
|
output.write('Different contents.\n')
|
|
|
|
self.assertFalse(filecmp.cmpfiles(self.dir, self.dir_same,
|
|
['file', 'file2']) ==
|
|
(['file'], ['file2'], []),
|
|
"Comparing mismatched directories fails")
|
|
|
|
|
|
def _assert_lists(self, actual, expected):
|
|
"""Assert that two lists are equal, up to ordering."""
|
|
self.assertEqual(sorted(actual), sorted(expected))
|
|
|
|
|
|
def test_dircmp(self):
|
|
# Check attributes for comparison of two identical directories
|
|
left_dir, right_dir = self.dir, self.dir_same
|
|
d = filecmp.dircmp(left_dir, right_dir)
|
|
self.assertEqual(d.left, left_dir)
|
|
self.assertEqual(d.right, right_dir)
|
|
if self.caseinsensitive:
|
|
self._assert_lists(d.left_list, ['file', 'subdir'])
|
|
self._assert_lists(d.right_list, ['FiLe', 'subdir'])
|
|
else:
|
|
self._assert_lists(d.left_list, ['file', 'subdir'])
|
|
self._assert_lists(d.right_list, ['file', 'subdir'])
|
|
self._assert_lists(d.common, ['file', 'subdir'])
|
|
self._assert_lists(d.common_dirs, ['subdir'])
|
|
self.assertEqual(d.left_only, [])
|
|
self.assertEqual(d.right_only, [])
|
|
self.assertEqual(d.same_files, ['file'])
|
|
self.assertEqual(d.diff_files, [])
|
|
expected_report = [
|
|
"diff {} {}".format(self.dir, self.dir_same),
|
|
"Identical files : ['file']",
|
|
"Common subdirectories : ['subdir']",
|
|
]
|
|
self._assert_report(d.report, expected_report)
|
|
|
|
# Check attributes for comparison of two different directories (right)
|
|
left_dir, right_dir = self.dir, self.dir_diff
|
|
d = filecmp.dircmp(left_dir, right_dir)
|
|
self.assertEqual(d.left, left_dir)
|
|
self.assertEqual(d.right, right_dir)
|
|
self._assert_lists(d.left_list, ['file', 'subdir'])
|
|
self._assert_lists(d.right_list, ['file', 'file2', 'subdir'])
|
|
self._assert_lists(d.common, ['file', 'subdir'])
|
|
self._assert_lists(d.common_dirs, ['subdir'])
|
|
self.assertEqual(d.left_only, [])
|
|
self.assertEqual(d.right_only, ['file2'])
|
|
self.assertEqual(d.same_files, ['file'])
|
|
self.assertEqual(d.diff_files, [])
|
|
expected_report = [
|
|
"diff {} {}".format(self.dir, self.dir_diff),
|
|
"Only in {} : ['file2']".format(self.dir_diff),
|
|
"Identical files : ['file']",
|
|
"Common subdirectories : ['subdir']",
|
|
]
|
|
self._assert_report(d.report, expected_report)
|
|
|
|
# Check attributes for comparison of two different directories (left)
|
|
left_dir, right_dir = self.dir, self.dir_diff
|
|
shutil.move(
|
|
os.path.join(self.dir_diff, 'file2'),
|
|
os.path.join(self.dir, 'file2')
|
|
)
|
|
d = filecmp.dircmp(left_dir, right_dir)
|
|
self.assertEqual(d.left, left_dir)
|
|
self.assertEqual(d.right, right_dir)
|
|
self._assert_lists(d.left_list, ['file', 'file2', 'subdir'])
|
|
self._assert_lists(d.right_list, ['file', 'subdir'])
|
|
self._assert_lists(d.common, ['file', 'subdir'])
|
|
self.assertEqual(d.left_only, ['file2'])
|
|
self.assertEqual(d.right_only, [])
|
|
self.assertEqual(d.same_files, ['file'])
|
|
self.assertEqual(d.diff_files, [])
|
|
expected_report = [
|
|
"diff {} {}".format(self.dir, self.dir_diff),
|
|
"Only in {} : ['file2']".format(self.dir),
|
|
"Identical files : ['file']",
|
|
"Common subdirectories : ['subdir']",
|
|
]
|
|
self._assert_report(d.report, expected_report)
|
|
|
|
# Add different file2
|
|
with open(os.path.join(self.dir_diff, 'file2'), 'w', encoding="utf-8") as output:
|
|
output.write('Different contents.\n')
|
|
d = filecmp.dircmp(self.dir, self.dir_diff)
|
|
self.assertEqual(d.same_files, ['file'])
|
|
self.assertEqual(d.diff_files, ['file2'])
|
|
expected_report = [
|
|
"diff {} {}".format(self.dir, self.dir_diff),
|
|
"Identical files : ['file']",
|
|
"Differing files : ['file2']",
|
|
"Common subdirectories : ['subdir']",
|
|
]
|
|
self._assert_report(d.report, expected_report)
|
|
|
|
def test_dircmp_subdirs_type(self):
|
|
"""Check that dircmp.subdirs respects subclassing."""
|
|
class MyDirCmp(filecmp.dircmp):
|
|
pass
|
|
d = MyDirCmp(self.dir, self.dir_diff)
|
|
sub_dirs = d.subdirs
|
|
self.assertEqual(list(sub_dirs.keys()), ['subdir'])
|
|
sub_dcmp = sub_dirs['subdir']
|
|
self.assertEqual(type(sub_dcmp), MyDirCmp)
|
|
|
|
def test_report_partial_closure(self):
|
|
left_dir, right_dir = self.dir, self.dir_same
|
|
d = filecmp.dircmp(left_dir, right_dir)
|
|
left_subdir = os.path.join(left_dir, 'subdir')
|
|
right_subdir = os.path.join(right_dir, 'subdir')
|
|
expected_report = [
|
|
"diff {} {}".format(self.dir, self.dir_same),
|
|
"Identical files : ['file']",
|
|
"Common subdirectories : ['subdir']",
|
|
'',
|
|
"diff {} {}".format(left_subdir, right_subdir),
|
|
]
|
|
self._assert_report(d.report_partial_closure, expected_report)
|
|
|
|
def test_report_full_closure(self):
|
|
left_dir, right_dir = self.dir, self.dir_same
|
|
d = filecmp.dircmp(left_dir, right_dir)
|
|
left_subdir = os.path.join(left_dir, 'subdir')
|
|
right_subdir = os.path.join(right_dir, 'subdir')
|
|
expected_report = [
|
|
"diff {} {}".format(self.dir, self.dir_same),
|
|
"Identical files : ['file']",
|
|
"Common subdirectories : ['subdir']",
|
|
'',
|
|
"diff {} {}".format(left_subdir, right_subdir),
|
|
]
|
|
self._assert_report(d.report_full_closure, expected_report)
|
|
|
|
def _assert_report(self, dircmp_report, expected_report_lines):
|
|
with support.captured_stdout() as stdout:
|
|
dircmp_report()
|
|
report_lines = stdout.getvalue().strip().split('\n')
|
|
self.assertEqual(report_lines, expected_report_lines)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|