1
# Copyright (C) 2007 Canonical Ltd
 
 
2
# -*- coding: utf-8 -*-
 
 
4
# This program is free software; you can redistribute it and/or modify
 
 
5
# it under the terms of the GNU General Public License as published by
 
 
6
# the Free Software Foundation; either version 2 of the License, or
 
 
7
# (at your option) any later version.
 
 
9
# This program is distributed in the hope that it will be useful,
 
 
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
 
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
 
12
# GNU General Public License for more details.
 
 
14
# You should have received a copy of the GNU General Public License
 
 
15
# along with this program; if not, write to the Free Software
 
 
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
 
19
"""Tests for the switch command of bzr."""
 
 
23
from bzrlib.workingtree import WorkingTree
 
 
24
from bzrlib.tests.blackbox import ExternalBase
 
 
27
class TestSwitch(ExternalBase):
 
 
29
    def test_switch_up_to_date_light_checkout(self):
 
 
30
        self.make_branch_and_tree('branch')
 
 
31
        self.run_bzr('branch branch branch2')
 
 
32
        self.run_bzr('checkout --lightweight branch checkout')
 
 
34
        out, err = self.run_bzr('switch ../branch2')
 
 
35
        self.assertContainsRe(err, 'Tree is up to date at revision 0.\n')
 
 
36
        self.assertContainsRe(err, 'Switched to branch: .*/branch2.\n')
 
 
37
        self.assertEqual('', out)
 
 
39
    def test_switch_out_of_date_light_checkout(self):
 
 
40
        self.make_branch_and_tree('branch')
 
 
41
        self.run_bzr('branch branch branch2')
 
 
42
        self.build_tree(['branch2/file'])
 
 
43
        self.run_bzr('add branch2/file')
 
 
44
        self.run_bzr('commit -m add-file branch2')
 
 
45
        self.run_bzr('checkout --lightweight branch checkout')
 
 
47
        out, err = self.run_bzr('switch ../branch2')
 
 
48
        #self.assertContainsRe(err, '\+N  file')
 
 
49
        self.assertContainsRe(err, 'Updated to revision 1.\n')
 
 
50
        self.assertContainsRe(err, 'Switched to branch: .*/branch2.\n')
 
 
51
        self.assertEqual('', out)
 
 
53
    def _test_switch_nick(self, lightweight):
 
 
54
        """Check that the nick gets switched too."""
 
 
55
        tree1 = self.make_branch_and_tree('branch1')
 
 
56
        tree2 = self.make_branch_and_tree('branch2')
 
 
57
        tree2.pull(tree1.branch)
 
 
58
        checkout =  tree1.branch.create_checkout('checkout',
 
 
59
            lightweight=lightweight)
 
 
60
        self.assertEqual(checkout.branch.nick, tree1.branch.nick)
 
 
61
        self.assertEqual(checkout.branch.get_config().has_explicit_nickname(),
 
 
63
        self.run_bzr('switch branch2', working_dir='checkout')
 
 
65
        # we need to get the tree again, otherwise we don't get the new branch
 
 
66
        checkout = WorkingTree.open('checkout')
 
 
67
        self.assertEqual(checkout.branch.nick, tree2.branch.nick)
 
 
68
        self.assertEqual(checkout.branch.get_config().has_explicit_nickname(),
 
 
71
    def test_switch_nick(self):
 
 
72
        self._test_switch_nick(lightweight=False)
 
 
74
    def test_switch_nick_lightweight(self):
 
 
75
        self._test_switch_nick(lightweight=True)
 
 
77
    def _test_switch_explicit_nick(self, lightweight):
 
 
78
        """Check that the nick gets switched too."""
 
 
79
        tree1 = self.make_branch_and_tree('branch1')
 
 
80
        tree2 = self.make_branch_and_tree('branch2')
 
 
81
        tree2.pull(tree1.branch)
 
 
82
        checkout =  tree1.branch.create_checkout('checkout',
 
 
83
            lightweight=lightweight)
 
 
84
        self.assertEqual(checkout.branch.nick, tree1.branch.nick)
 
 
85
        checkout.branch.nick = "explicit_nick"
 
 
86
        self.assertEqual(checkout.branch.nick, "explicit_nick")
 
 
87
        self.assertEqual(checkout.branch.get_config()._get_explicit_nickname(),
 
 
89
        self.run_bzr('switch branch2', working_dir='checkout')
 
 
91
        # we need to get the tree again, otherwise we don't get the new branch
 
 
92
        checkout = WorkingTree.open('checkout')
 
 
93
        self.assertEqual(checkout.branch.nick, tree2.branch.nick)
 
 
94
        self.assertEqual(checkout.branch.get_config()._get_explicit_nickname(),
 
 
97
    def test_switch_explicit_nick(self):
 
 
98
        self._test_switch_explicit_nick(lightweight=False)
 
 
100
    def test_switch_explicit_nick_lightweight(self):
 
 
101
        self._test_switch_explicit_nick(lightweight=True)
 
 
103
    def test_switch_finds_relative_branch(self):
 
 
104
        """Switch will find 'foo' relative to the branch the checkout is of."""
 
 
105
        self.build_tree(['repo/'])
 
 
106
        tree1 = self.make_branch_and_tree('repo/brancha')
 
 
108
        tree2 = self.make_branch_and_tree('repo/branchb')
 
 
109
        tree2.pull(tree1.branch)
 
 
110
        branchb_id = tree2.commit('bar')
 
 
111
        checkout =  tree1.branch.create_checkout('checkout', lightweight=True)
 
 
112
        self.run_bzr(['switch', 'branchb'], working_dir='checkout')
 
 
113
        self.assertEqual(branchb_id, checkout.last_revision())
 
 
114
        checkout = checkout.bzrdir.open_workingtree()
 
 
115
        self.assertEqual(tree2.branch.base, checkout.branch.base)
 
 
117
    def test_switch_finds_relative_bound_branch(self):
 
 
118
        """Using switch on a heavy checkout should find master sibling
 
 
120
        The behaviour of lighweight and heavy checkouts should be 
 
 
121
        consistentwhen using the convenient "switch to sibling" feature
 
 
122
        Both should switch to a sibling of the branch
 
 
123
        they are bound to, and not a sibling of themself"""
 
 
125
        self.build_tree(['repo/',
 
 
127
        tree1 = self.make_branch_and_tree('repo/brancha')
 
 
129
        tree2 = self.make_branch_and_tree('repo/branchb')
 
 
130
        tree2.pull(tree1.branch)
 
 
131
        branchb_id = tree2.commit('bar')
 
 
132
        checkout = tree1.branch.create_checkout('heavyco/a', lightweight=False)
 
 
133
        self.run_bzr(['switch', 'branchb'], working_dir='heavyco/a')
 
 
134
        self.assertEqual(branchb_id, checkout.last_revision())
 
 
135
        self.assertEqual(tree2.branch.base, checkout.branch.get_bound_location())