# Copyright (C) 2006, 2007 Canonical Ltd # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 2 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA """Tests for the extra cases that WorkingTree.walkdirs can encounter.""" import os from breezy.tests.features import SymlinkFeature from breezy.tests.per_workingtree import TestCaseWithWorkingTree # tests to write: # type mismatches - file to link, dir, dir to file, link, link to file, dir class DirBlock: """Object representation of the tuples returned by dirstate.""" def __init__(self, tree, file_path, file_name=None, inventory_kind=None, stat=None, disk_kind='unknown'): self.file_path = file_path self.abspath = tree.abspath(file_path) self.relpath = tree.relpath(file_path) if file_name is None: file_name = os.path.split(file_path)[-1] if len(file_name) == 0: file_name = os.path.split(file_path)[-2] self.file_name = file_name self.inventory_kind = inventory_kind self.stat = stat self.disk_kind = disk_kind def as_tuple(self): return (self.relpath, self.file_name, self.disk_kind, self.stat, self.inventory_kind) def as_dir_tuple(self): return self.relpath def __str__(self): return """ file_path = %r abspath = %r relpath = %r file_name = %r inventory_kind = %r stat = %r disk_kind = %r""" % (self.file_path, self.abspath, self.relpath, self.file_name, self.inventory_kind, self.stat, self.disk_kind) class TestWalkdirs(TestCaseWithWorkingTree): added = 'added' missing = 'missing' unknown = 'unknown' def get_tree(self, file_status, prefix=None): tree = self.make_branch_and_tree('.') dirblocks = [] paths = [ file_status + ' file', file_status + ' dir/', file_status + ' dir/a file', file_status + ' empty dir/', ] self.build_tree(paths) def add_dirblock(path, kind): dirblock = DirBlock(tree, path) if file_status != self.unknown: dirblock.inventory_kind = kind if file_status != self.missing: dirblock.disk_kind = kind dirblock.stat = os.lstat(dirblock.relpath) dirblocks.append(dirblock) add_dirblock(paths[0], 'file') add_dirblock(paths[1], 'directory') add_dirblock(paths[2], 'file') add_dirblock(paths[3], 'directory') if file_status != self.unknown: tree.add(paths) if file_status == self.missing: # now make the files be missing tree.controldir.root_transport.delete(dirblocks[0].relpath) tree.controldir.root_transport.delete_tree(dirblocks[1].relpath) tree.controldir.root_transport.delete_tree(dirblocks[3].relpath) expected_dirblocks = [ ('', [dirblocks[1].as_tuple()] + ([dirblocks[3].as_tuple()] if (tree.has_versioned_directories() or file_status == self.unknown) else []) + [dirblocks[0].as_tuple()] ), (dirblocks[1].as_dir_tuple(), [dirblocks[2].as_tuple()] ), ] if (tree.has_versioned_directories() or file_status != self.missing): expected_dirblocks.append((dirblocks[3].as_dir_tuple(), [])) if prefix: expected_dirblocks = [e for e in expected_dirblocks if len(e) > 0 and len(e[0]) > 0 and e[0] == prefix] return tree, expected_dirblocks def _test_walkdir(self, file_status, prefix=""): result = [] tree, expected_dirblocks = self.get_tree(file_status, prefix) with tree.lock_read(): for dirpath, dirblock in tree.walkdirs(prefix): result.append((dirpath, list(dirblock))) # check each return value for debugging ease. for pos, item in enumerate(expected_dirblocks): result_pos = [] if len(result) > pos: result_pos = result[pos] self.assertEqual(item, result_pos) self.assertEqual(expected_dirblocks, result) def test_walkdir_unknowns(self): """unknown files and directories should be reported by walkdirs.""" self._test_walkdir(self.unknown) def test_walkdir_from_unknown_dir(self): """Doing a walkdir when the requested prefix is unknown but on disk.""" self._test_walkdir(self.unknown, 'unknown dir') def test_walkdir_missings(self): """missing files and directories should be reported by walkdirs.""" self._test_walkdir(self.missing) def test_walkdir_from_dir(self): """Doing a walkdir when the requested prefix is known and on disk.""" self._test_walkdir(self.added, 'added dir') def test_walkdir_from_empty_dir(self): """Doing a walkdir when the requested prefix is empty dir.""" self._test_walkdir(self.added, 'added empty dir') def test_walkdir_from_missing_dir(self): """Doing a walkdir when the requested prefix is missing but on disk.""" self._test_walkdir(self.missing, 'missing dir') def test_walkdirs_type_changes(self): """Walkdir shows the actual kinds on disk and the recorded kinds.""" self.requireFeature(SymlinkFeature) tree = self.make_branch_and_tree('.') paths = ['file1', 'file2', 'dir1/', 'dir2/'] self.build_tree(paths) tree.add(paths) tt = tree.transform() root_transaction_id = tt.trans_id_tree_path('') tt.new_symlink('link1', root_transaction_id, 'link-target', b'link1') tt.new_symlink('link2', root_transaction_id, 'link-target', b'link2') tt.apply() tree.controldir.root_transport.delete_tree('dir1') tree.controldir.root_transport.delete_tree('dir2') tree.controldir.root_transport.delete('file1') tree.controldir.root_transport.delete('file2') tree.controldir.root_transport.delete('link1') tree.controldir.root_transport.delete('link2') changed_paths = ['dir1', 'file1/', 'link1', 'link2/'] self.build_tree(changed_paths) os.symlink('target', 'dir2') os.symlink('target', 'file2') dir1_stat = os.lstat('dir1') dir2_stat = os.lstat('dir2') file1_stat = os.lstat('file1') file2_stat = os.lstat('file2') link1_stat = os.lstat('link1') link2_stat = os.lstat('link2') expected_dirblocks = [ ('', [('dir1', 'dir1', 'file', dir1_stat, 'directory' if tree.has_versioned_directories() else None), ('dir2', 'dir2', 'symlink', dir2_stat, 'directory' if tree.has_versioned_directories() else None), ('file1', 'file1', 'directory', file1_stat, 'file'), ('file2', 'file2', 'symlink', file2_stat, 'file'), ('link1', 'link1', 'file', link1_stat, 'symlink'), ('link2', 'link2', 'directory', link2_stat, 'symlink'), ] )] if tree.has_versioned_directories(): expected_dirblocks.extend([ ('dir1', []), ('dir2', [])]) expected_dirblocks.extend([ ('file1', []), ('link2', []), ]) with tree.lock_read(): result = list(tree.walkdirs()) # check each return value for debugging ease. for pos, item in enumerate(expected_dirblocks): self.assertEqual(item, result[pos]) self.assertEqual(len(expected_dirblocks), len(result)) def test_walkdirs_type_changes_wo_symlinks(self): # similar to test_walkdirs_type_changes # but don't use symlinks for safe testing on win32 tree = self.make_branch_and_tree('.') paths = ['file1', 'dir1/'] self.build_tree(paths) tree.add(paths) tree.controldir.root_transport.delete_tree('dir1') tree.controldir.root_transport.delete('file1') changed_paths = ['dir1', 'file1/'] self.build_tree(changed_paths) dir1_stat = os.lstat('dir1') file1_stat = os.lstat('file1') if tree.has_versioned_directories(): expected_dirblocks = [ ('', [('dir1', 'dir1', 'file', dir1_stat, 'directory'), ('file1', 'file1', 'directory', file1_stat, 'file'), ]), ('dir1', []), ('file1', []), ] else: expected_dirblocks = [ ('', [('dir1', 'dir1', 'file', dir1_stat, None), ('file1', 'file1', 'directory', file1_stat, 'file'), ] ), ('file1', []), ] with tree.lock_read(): result = list(tree.walkdirs()) # check each return value for debugging ease. for pos, item in enumerate(expected_dirblocks): self.assertEqual(item, result[pos]) self.assertEqual(len(expected_dirblocks), len(result))