# 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA """Tests for the extra cases that WorkingTree.walkdirs can encounter.""" import os from bzrlib import transform from bzrlib.osutils import has_symlinks from bzrlib.tests import TestSkipped from bzrlib.tests.workingtree_implementations import TestCaseWithWorkingTree # tests to write: # type mismatches - file to link, dir, dir to file, link, link to file, dir class TestWalkdirs(TestCaseWithWorkingTree): def get_tree_with_unknowns(self): tree = self.make_branch_and_tree('.') self.build_tree([ 'unknown file', 'unknown dir/', 'unknown dir/a file', ]) u_f_stat = os.lstat('unknown file') u_d_stat = os.lstat('unknown dir') u_d_f_stat = os.lstat('unknown dir/a file') expected_dirblocks = [ (('', tree.path2id('')), [ ('unknown dir', 'unknown dir', 'directory', u_d_stat, None, None), ('unknown file', 'unknown file', 'file', u_f_stat, None, None), ] ), (('unknown dir', None), [('unknown dir/a file', 'a file', 'file', u_d_f_stat, None, None), ] ), ] return tree, expected_dirblocks def test_walkdir_unknowns(self): """unknown files and directories should be reported by walkdirs.""" # test that its iterable by iterating: result = [] tree, expected_dirblocks = self.get_tree_with_unknowns() tree.lock_read() for dirinfo, dirblock in tree.walkdirs(): result.append((dirinfo, list(dirblock))) tree.unlock() # 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_walkdir_from_unknown_dir(self): """Doing a walkdir when the requested prefix is unknown but on disk.""" result = [] tree, expected_dirblocks = self.get_tree_with_unknowns() tree.lock_read() for dirinfo, dirblock in tree.walkdirs('unknown dir'): result.append((dirinfo, list(dirblock))) tree.unlock() # check each return value for debugging ease. for pos, item in enumerate(expected_dirblocks[1:]): self.assertEqual(item, result[pos]) self.assertEqual(len(expected_dirblocks) - 1, len(result)) def get_tree_with_missings(self): tree = self.make_branch_and_tree('.') paths = [ 'missing file', 'missing dir/', 'missing dir/a file', ] ids = [ 'a file', 'a dir', 'a dir-a file', ] self.build_tree(paths) tree.add(paths, ids) # now make the files be missing tree.bzrdir.root_transport.delete_tree('missing dir') tree.bzrdir.root_transport.delete('missing file') expected_dirblocks = [ (('', tree.path2id('')), [ ('missing dir', 'missing dir', 'unknown', None, 'a dir', 'directory'), ('missing file', 'missing file', 'unknown', None, 'a file', 'file'), ] ), (('missing dir', 'a dir'), [('missing dir/a file', 'a file', 'unknown', None, 'a dir-a file', 'file'), ] ), ] return tree, expected_dirblocks def test_walkdir_missings(self): """missing files and directories should be reported by walkdirs.""" # test that its iterable by iterating: result = [] tree, expected_dirblocks = self.get_tree_with_missings() tree.lock_read() for dirinfo, dirblock in tree.walkdirs(): result.append((dirinfo, list(dirblock))) tree.unlock() # 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_walkdir_from_missing_dir(self): """Doing a walkdir when the requested prefix is missing but on disk.""" result = [] tree, expected_dirblocks = self.get_tree_with_missings() tree.lock_read() for dirinfo, dirblock in tree.walkdirs('missing dir'): result.append((dirinfo, list(dirblock))) tree.unlock() # check each return value for debugging ease. for pos, item in enumerate(expected_dirblocks[1:]): self.assertEqual(item, result[pos]) self.assertEqual(len(expected_dirblocks[1:]), len(result)) def test_walkdirs_type_changes(self): """Walkdir shows the actual kinds on disk and the recorded kinds.""" if not has_symlinks(): raise TestSkipped('No symlink support') tree = self.make_branch_and_tree('.') paths = ['file1', 'file2', 'dir1/', 'dir2/'] ids = ['file1', 'file2', 'dir1', 'dir2'] self.build_tree(paths) tree.add(paths, ids) tt = transform.TreeTransform(tree) root_transaction_id = tt.trans_id_tree_path('') tt.new_symlink('link1', root_transaction_id, 'link-target', 'link1') tt.new_symlink('link2', root_transaction_id, 'link-target', 'link2') tt.apply() tree.bzrdir.root_transport.delete_tree('dir1') tree.bzrdir.root_transport.delete_tree('dir2') tree.bzrdir.root_transport.delete('file1') tree.bzrdir.root_transport.delete('file2') tree.bzrdir.root_transport.delete('link1') tree.bzrdir.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 = [ (('', tree.path2id('')), [('dir1', 'dir1', 'file', dir1_stat, 'dir1', 'directory'), ('dir2', 'dir2', 'symlink', dir2_stat, 'dir2', 'directory'), ('file1', 'file1', 'directory', file1_stat, 'file1', 'file'), ('file2', 'file2', 'symlink', file2_stat, 'file2', 'file'), ('link1', 'link1', 'file', link1_stat, 'link1', 'symlink'), ('link2', 'link2', 'directory', link2_stat, 'link2', 'symlink'), ] ), (('dir1', 'dir1'), [ ] ), (('dir2', 'dir2'), [ ] ), (('file1', None), [ ] ), (('link2', None), [ ] ), ] tree.lock_read() result = list(tree.walkdirs()) tree.unlock() # 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/'] ids = ['file1', 'dir1'] self.build_tree(paths) tree.add(paths, ids) tree.bzrdir.root_transport.delete_tree('dir1') tree.bzrdir.root_transport.delete('file1') changed_paths = ['dir1', 'file1/'] self.build_tree(changed_paths) dir1_stat = os.lstat('dir1') file1_stat = os.lstat('file1') expected_dirblocks = [ (('', tree.path2id('')), [('dir1', 'dir1', 'file', dir1_stat, 'dir1', 'directory'), ('file1', 'file1', 'directory', file1_stat, 'file1', 'file'), ] ), (('dir1', 'dir1'), [ ] ), (('file1', None), [ ] ), ] tree.lock_read() result = list(tree.walkdirs()) tree.unlock() # 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))