~bzr-pqm/bzr/bzr.dev

# 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)