# Copyright (C) 2007-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 import sys from urllib import quote from bzrlib import ( branch as _mod_branch, bzrdir, info, tests, workingtree, repository as _mod_repository, ) class TestInfo(tests.TestCaseWithTransport): def test_describe_standalone_layout(self): tree = self.make_branch_and_tree('tree') self.assertEqual('Empty control directory', info.describe_layout()) self.assertEqual('Unshared repository with trees', info.describe_layout(tree.branch.repository)) tree.branch.repository.set_make_working_trees(False) self.assertEqual('Unshared repository', info.describe_layout(tree.branch.repository)) self.assertEqual('Standalone branch', info.describe_layout(tree.branch.repository, tree.branch)) self.assertEqual('Standalone branchless tree', info.describe_layout(tree.branch.repository, None, tree)) self.assertEqual('Standalone tree', info.describe_layout(tree.branch.repository, tree.branch, tree)) tree.branch.bind(tree.branch) self.assertEqual('Bound branch', info.describe_layout(tree.branch.repository, tree.branch)) self.assertEqual('Checkout', info.describe_layout(tree.branch.repository, tree.branch, tree)) checkout = tree.branch.create_checkout('checkout', lightweight=True) self.assertEqual('Lightweight checkout', info.describe_layout(checkout.branch.repository, checkout.branch, checkout)) def test_describe_repository_layout(self): repository = self.make_repository('.', shared=True) tree = bzrdir.BzrDir.create_branch_convenience('tree', force_new_tree=True).bzrdir.open_workingtree() self.assertEqual('Shared repository with trees', info.describe_layout(tree.branch.repository)) repository.set_make_working_trees(False) self.assertEqual('Shared repository', info.describe_layout(tree.branch.repository)) self.assertEqual('Repository branch', info.describe_layout(tree.branch.repository, tree.branch)) self.assertEqual('Repository branchless tree', info.describe_layout(tree.branch.repository, None, tree)) self.assertEqual('Repository tree', info.describe_layout(tree.branch.repository, tree.branch, tree)) tree.branch.bind(tree.branch) self.assertEqual('Repository checkout', info.describe_layout(tree.branch.repository, tree.branch, tree)) checkout = tree.branch.create_checkout('checkout', lightweight=True) self.assertEqual('Lightweight checkout', info.describe_layout(checkout.branch.repository, checkout.branch, checkout)) def assertTreeDescription(self, format): """Assert a tree's format description matches expectations""" self.make_branch_and_tree('%s_tree' % format, format=format) tree = workingtree.WorkingTree.open('%s_tree' % format) self.assertEqual(format, info.describe_format(tree.bzrdir, tree.branch.repository, tree.branch, tree)) def assertCheckoutDescription(self, format, expected=None): """Assert a checkout's format description matches expectations""" if expected is None: expected = format branch = self.make_branch('%s_cobranch' % format, format=format) # this ought to be easier... branch.create_checkout('%s_co' % format, lightweight=True).bzrdir.destroy_workingtree() control = bzrdir.BzrDir.open('%s_co' % format) old_format = control._format.workingtree_format try: control._format.workingtree_format = \ bzrdir.format_registry.make_bzrdir(format).workingtree_format control.create_workingtree() tree = workingtree.WorkingTree.open('%s_co' % format) format_description = info.describe_format(tree.bzrdir, tree.branch.repository, tree.branch, tree) self.assertEqual(expected, format_description, "checkout of format called %r was described as %r" % (expected, format_description)) finally: control._format.workingtree_format = old_format def assertBranchDescription(self, format, expected=None): """Assert branch's format description matches expectations""" if expected is None: expected = format self.make_branch('%s_branch' % format, format=format) branch = _mod_branch.Branch.open('%s_branch' % format) self.assertEqual(expected, info.describe_format(branch.bzrdir, branch.repository, branch, None)) def assertRepoDescription(self, format, expected=None): """Assert repository's format description matches expectations""" if expected is None: expected = format self.make_repository('%s_repo' % format, format=format) repo = _mod_repository.Repository.open('%s_repo' % format) self.assertEqual(expected, info.describe_format(repo.bzrdir, repo, None, None)) def test_describe_tree_format(self): for key in bzrdir.format_registry.keys(): if key in bzrdir.format_registry.aliases(): continue self.assertTreeDescription(key) def test_describe_checkout_format(self): for key in bzrdir.format_registry.keys(): if key in bzrdir.format_registry.aliases(): # Aliases will not describe correctly in the UI because the # real format is found. continue # legacy: weave does not support checkouts if key == 'weave': continue if bzrdir.format_registry.get_info(key).experimental: # We don't require that experimental formats support checkouts # or describe correctly in the UI. continue if bzrdir.format_registry.get_info(key).hidden: continue expected = None if key in ('pack-0.92',): expected = 'pack-0.92' elif key in ('knit', 'metaweave'): expected = 'knit or metaweave' elif key in ('1.14', '1.14-rich-root'): expected = '1.14 or 1.14-rich-root' self.assertCheckoutDescription(key, expected) def test_describe_branch_format(self): for key in bzrdir.format_registry.keys(): if key in bzrdir.format_registry.aliases(): continue if bzrdir.format_registry.get_info(key).hidden: continue expected = None if key in ('dirstate', 'knit'): expected = 'dirstate or knit' elif key in ('1.14',): expected = '1.14' elif key in ('1.14-rich-root',): expected = '1.14-rich-root' self.assertBranchDescription(key, expected) def test_describe_repo_format(self): for key in bzrdir.format_registry.keys(): if key in bzrdir.format_registry.aliases(): continue if bzrdir.format_registry.get_info(key).hidden: continue expected = None if key in ('dirstate', 'knit', 'dirstate-tags'): expected = 'dirstate or dirstate-tags or knit' elif key in ('1.14',): expected = '1.14' elif key in ('1.14-rich-root',): expected = '1.14-rich-root' self.assertRepoDescription(key, expected) format = bzrdir.format_registry.make_bzrdir('metaweave') format.set_branch_format(_mod_branch.BzrBranchFormat6()) tree = self.make_branch_and_tree('unknown', format=format) self.assertEqual('unnamed', info.describe_format(tree.bzrdir, tree.branch.repository, tree.branch, tree)) def test_gather_location_standalone(self): tree = self.make_branch_and_tree('tree') self.assertEqual([('branch root', tree.bzrdir.root_transport.base)], info.gather_location_info(tree.branch.repository, tree.branch, tree)) self.assertEqual([('branch root', tree.bzrdir.root_transport.base)], info.gather_location_info(tree.branch.repository, tree.branch)) return tree def test_gather_location_repo(self): srepo = self.make_repository('shared', shared=True) self.assertEqual([('shared repository', srepo.bzrdir.root_transport.base)], info.gather_location_info(srepo)) urepo = self.make_repository('unshared') self.assertEqual([('repository', urepo.bzrdir.root_transport.base)], info.gather_location_info(urepo)) def test_gather_location_repo_branch(self): srepo = self.make_repository('shared', shared=True) self.assertEqual([('shared repository', srepo.bzrdir.root_transport.base)], info.gather_location_info(srepo)) tree = self.make_branch_and_tree('shared/tree') self.assertEqual([('shared repository', srepo.bzrdir.root_transport.base), ('repository branch', tree.branch.base)], info.gather_location_info(srepo, tree.branch, tree)) def test_gather_location_light_checkout(self): tree = self.make_branch_and_tree('tree') lcheckout = tree.branch.create_checkout('lcheckout', lightweight=True) self.assertEqual( [('light checkout root', lcheckout.bzrdir.root_transport.base), ('checkout of branch', tree.bzrdir.root_transport.base)], self.gather_tree_location_info(lcheckout)) def test_gather_location_heavy_checkout(self): tree = self.make_branch_and_tree('tree') checkout = tree.branch.create_checkout('checkout') self.assertEqual( [('checkout root', checkout.bzrdir.root_transport.base), ('checkout of branch', tree.bzrdir.root_transport.base)], self.gather_tree_location_info(checkout)) light_checkout = checkout.branch.create_checkout('light_checkout', lightweight=True) self.assertEqual( [('light checkout root', light_checkout.bzrdir.root_transport.base), ('checkout root', checkout.bzrdir.root_transport.base), ('checkout of branch', tree.bzrdir.root_transport.base)], self.gather_tree_location_info(light_checkout) ) def test_gather_location_shared_repo_checkout(self): tree = self.make_branch_and_tree('tree') srepo = self.make_repository('shared', shared=True) shared_checkout = tree.branch.create_checkout('shared/checkout') self.assertEqual( [('repository checkout root', shared_checkout.bzrdir.root_transport.base), ('checkout of branch', tree.bzrdir.root_transport.base), ('shared repository', srepo.bzrdir.root_transport.base)], self.gather_tree_location_info(shared_checkout)) def gather_tree_location_info(self, tree): return info.gather_location_info(tree.branch.repository, tree.branch, tree) def test_gather_location_bound(self): branch = self.make_branch('branch') bound_branch = self.make_branch('bound_branch') bound_branch.bind(branch) self.assertEqual( [('branch root', bound_branch.bzrdir.root_transport.base), ('bound to branch', branch.bzrdir.root_transport.base)], info.gather_location_info(bound_branch.repository, bound_branch) ) def test_location_list(self): if sys.platform == 'win32': raise tests.TestSkipped('Windows-unfriendly test') locs = info.LocationList('/home/foo') locs.add_url('a', 'file:///home/foo/') locs.add_url('b', 'file:///home/foo/bar/') locs.add_url('c', 'file:///home/bar/bar') locs.add_url('d', 'http://example.com/example/') locs.add_url('e', None) self.assertEqual(locs.locs, [('a', '.'), ('b', 'bar'), ('c', '/home/bar/bar'), ('d', 'http://example.com/example/')]) self.assertEqualDiff(' a: .\n b: bar\n c: /home/bar/bar\n' ' d: http://example.com/example/\n', ''.join(locs.get_lines())) def test_gather_related_braches(self): branch = self.make_branch('.') branch.set_public_branch('baz') branch.set_push_location('bar') branch.set_parent('foo') branch.set_submit_branch('qux') self.assertEqual( [('public branch', 'baz'), ('push branch', 'bar'), ('parent branch', 'foo'), ('submit branch', 'qux')], info._gather_related_branches(branch).locs)