/brz/remove-bazaar

# Copyright (C) 2008 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 Branch.get_stacked_on and set_stacked_on."""

from bzrlib import errors

from bzrlib.tests import TestNotApplicable

from bzrlib.tests.branch_implementations import TestCaseWithBranch

class TestStacking(TestCaseWithBranch):

    def test_get_set_stacked_on(self):

        # branches must either:

        # raise UnstackableBranchFormat or

        # raise UnstackableRepositoryFormat or

        # permit stacking to be done and then return the stacked location.

        branch = self.make_branch('branch')

        target = self.make_branch('target')

        old_format_errors = (

            errors.UnstackableBranchFormat,

            errors.UnstackableRepositoryFormat,

            )

        try:

            branch.set_stacked_on(target.base)

        except old_format_errors:

            # if the set failed, so must the get

            self.assertRaises(old_format_errors, branch.get_stacked_on)

            return

        # now we have a stacked branch:

        self.assertEqual(target.base, branch.get_stacked_on())

        branch.set_stacked_on(None)

        self.assertRaises(errors.NotStacked, branch.get_stacked_on)

    def test_set_stacked_on_fetches(self):

        # We have a mainline

        trunk_tree = self.make_branch_and_tree('mainline')

        trunk_revid = trunk_tree.commit('mainline')

        # and make branch from it which is shallow

        try:

            new_dir = trunk_tree.bzrdir.sprout('newbranch', shallow=True)

        except (errors.UnstackableBranchFormat, errors.UnstackableRepositoryFormat):

            # not a testable combination.

            return

        new_tree = new_dir.open_workingtree()

        new_tree.commit('something local')