# Copyright (C) 2007, 2009, 2010 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 """Test that all Trees implement path_content_summary.""" import os from breezy import ( osutils, tests, ) from breezy.transform import ( PreviewTree, ) from breezy.tests import ( features, per_tree, ) from breezy.tests.features import ( SymlinkFeature, ) class TestPathContentSummary(per_tree.TestCaseWithTree): def _convert_tree(self, tree): result = per_tree.TestCaseWithTree._convert_tree(self, tree) result.lock_read() self.addCleanup(result.unlock) return result def check_content_summary_size(self, tree, summary, expected_size): # if the tree supports content filters, then it's allowed to leave out # the size because it might be difficult to compute. otherwise, it # must be present and correct returned_size = summary[1] if returned_size == expected_size or ( tree.supports_content_filtering() and returned_size is None): pass else: self.fail("invalid size in summary: %r" % (returned_size,)) def test_symlink_content_summary(self): self.requireFeature(SymlinkFeature) tree = self.make_branch_and_tree('tree') os.symlink('target', 'tree/path') tree.add(['path']) summary = self._convert_tree(tree).path_content_summary('path') self.assertEqual(('symlink', None, None, 'target'), summary) def test_unicode_symlink_content_summary(self): self.requireFeature(features.SymlinkFeature) self.requireFeature(features.UnicodeFilenameFeature) tree = self.make_branch_and_tree('tree') os.symlink('target', u'tree/\u03b2-path'.encode(osutils._fs_enc)) tree.add([u'\u03b2-path']) summary = self._convert_tree(tree).path_content_summary(u'\u03b2-path') self.assertEqual(('symlink', None, None, 'target'), summary) def test_unicode_symlink_target_summary(self): self.requireFeature(features.SymlinkFeature) self.requireFeature(features.UnicodeFilenameFeature) tree = self.make_branch_and_tree('tree') os.symlink(u'tree/\u03b2-path'.encode(osutils._fs_enc), 'tree/link') tree.add(['link']) summary = self._convert_tree(tree).path_content_summary('link') self.assertEqual(('symlink', None, None, u'tree/\u03b2-path'), summary) def test_missing_content_summary(self): tree = self.make_branch_and_tree('tree') summary = self._convert_tree(tree).path_content_summary('path') self.assertEqual(('missing', None, None, None), summary) def test_file_content_summary_executable(self): tree = self.make_branch_and_tree('tree') self.build_tree(['tree/path']) tree.add(['path']) with tree.transform() as tt: tt.set_executability(True, tt.trans_id_tree_path('path')) tt.apply() summary = self._convert_tree(tree).path_content_summary('path') self.assertEqual(4, len(summary)) self.assertEqual('file', summary[0]) self.check_content_summary_size(tree, summary, 22) # executable self.assertEqual(True, summary[2]) # may have hash, self.assertSubset((summary[3],), (None, b'0c352290ae1c26ca7f97d5b2906c4624784abd60')) def test_file_content_summary_not_versioned(self): tree = self.make_branch_and_tree('tree') self.build_tree(['tree/path']) tree = self._convert_tree(tree) summary = tree.path_content_summary('path') self.assertEqual(4, len(summary)) if isinstance(tree, (per_tree.DirStateRevisionTree, per_tree.RevisionTree)): self.assertEqual('missing', summary[0]) self.assertIs(None, summary[2]) self.assertIs(None, summary[3]) elif isinstance(tree, PreviewTree): self.expectFailure('PreviewTree returns "missing" for unversioned' 'files', self.assertEqual, 'file', summary[0]) self.assertEqual('file', summary[0]) else: self.assertEqual('file', summary[0]) self.check_content_summary_size(tree, summary, 22) self.assertEqual(False, summary[2]) self.assertSubset((summary[3],), (None, b'0c352290ae1c26ca7f97d5b2906c4624784abd60')) def test_file_content_summary_non_exec(self): tree = self.make_branch_and_tree('tree') self.build_tree(['tree/path']) tree.add(['path']) summary = self._convert_tree(tree).path_content_summary('path') self.assertEqual(4, len(summary)) self.assertEqual('file', summary[0]) self.check_content_summary_size(tree, summary, 22) # not executable self.assertEqual(False, summary[2]) # may have hash, self.assertSubset((summary[3],), (None, b'0c352290ae1c26ca7f97d5b2906c4624784abd60')) def test_dir_content_summary(self): tree = self.make_branch_and_tree('tree') self.build_tree(['tree/path/']) tree.add(['path']) converted_tree = self._convert_tree(tree) summary = converted_tree.path_content_summary('path') if converted_tree.has_versioned_directories() or converted_tree.has_filename('path'): self.assertEqual(('directory', None, None, None), summary) else: self.assertEqual(('missing', None, None, None), summary) def test_tree_content_summary(self): tree = self.make_branch_and_tree('tree') if not tree.branch.repository._format.supports_tree_reference: raise tests.TestNotApplicable("Tree references not supported.") subtree = self.make_branch_and_tree('tree/path') subtree.commit('') tree.add(['path']) summary = self._convert_tree(tree).path_content_summary('path') self.assertEqual(4, len(summary)) self.assertEqual('tree-reference', summary[0])