# Copyright (C) 2007-2010 Canonical Ltd
# -*- coding: utf-8 -*-
#
# 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
"""Tests for the switch command of bzr."""
import os
from bzrlib import osutils
from bzrlib.workingtree import WorkingTree
from bzrlib.tests.blackbox import ExternalBase
from bzrlib.directory_service import directories
class TestSwitch(ExternalBase):
def _create_sample_tree(self):
tree = self.make_branch_and_tree('branch-1')
self.build_tree(['branch-1/file-1', 'branch-1/file-2'])
tree.add('file-1')
tree.commit('rev1')
tree.add('file-2')
tree.commit('rev2')
return tree
def test_switch_up_to_date_light_checkout(self):
self.make_branch_and_tree('branch')
self.run_bzr('branch branch branch2')
self.run_bzr('checkout --lightweight branch checkout')
os.chdir('checkout')
out, err = self.run_bzr('switch ../branch2')
self.assertContainsRe(err, 'Tree is up to date at revision 0.\n')
self.assertContainsRe(err, 'Switched to branch: .*/branch2.\n')
self.assertEqual('', out)
def test_switch_out_of_date_light_checkout(self):
self.make_branch_and_tree('branch')
self.run_bzr('branch branch branch2')
self.build_tree(['branch2/file'])
self.run_bzr('add branch2/file')
self.run_bzr('commit -m add-file branch2')
self.run_bzr('checkout --lightweight branch checkout')
os.chdir('checkout')
out, err = self.run_bzr('switch ../branch2')
#self.assertContainsRe(err, '\+N file')
self.assertContainsRe(err, 'Updated to revision 1.\n')
self.assertContainsRe(err, 'Switched to branch: .*/branch2.\n')
self.assertEqual('', out)
def _test_switch_nick(self, lightweight):
"""Check that the nick gets switched too."""
tree1 = self.make_branch_and_tree('branch1')
tree2 = self.make_branch_and_tree('branch2')
tree2.pull(tree1.branch)
checkout = tree1.branch.create_checkout('checkout',
lightweight=lightweight)
self.assertEqual(checkout.branch.nick, tree1.branch.nick)
self.assertEqual(checkout.branch.get_config().has_explicit_nickname(),
False)
self.run_bzr('switch branch2', working_dir='checkout')
# we need to get the tree again, otherwise we don't get the new branch
checkout = WorkingTree.open('checkout')
self.assertEqual(checkout.branch.nick, tree2.branch.nick)
self.assertEqual(checkout.branch.get_config().has_explicit_nickname(),
False)
def test_switch_nick(self):
self._test_switch_nick(lightweight=False)
def test_switch_nick_lightweight(self):
self._test_switch_nick(lightweight=True)
def _test_switch_explicit_nick(self, lightweight):
"""Check that the nick gets switched too."""
tree1 = self.make_branch_and_tree('branch1')
tree2 = self.make_branch_and_tree('branch2')
tree2.pull(tree1.branch)
checkout = tree1.branch.create_checkout('checkout',
lightweight=lightweight)
self.assertEqual(checkout.branch.nick, tree1.branch.nick)
checkout.branch.nick = "explicit_nick"
self.assertEqual(checkout.branch.nick, "explicit_nick")
self.assertEqual(checkout.branch.get_config()._get_explicit_nickname(),
"explicit_nick")
self.run_bzr('switch branch2', working_dir='checkout')
# we need to get the tree again, otherwise we don't get the new branch
checkout = WorkingTree.open('checkout')
self.assertEqual(checkout.branch.nick, tree2.branch.nick)
self.assertEqual(checkout.branch.get_config()._get_explicit_nickname(),
tree2.branch.nick)
def test_switch_explicit_nick(self):
self._test_switch_explicit_nick(lightweight=False)
def test_switch_explicit_nick_lightweight(self):
self._test_switch_explicit_nick(lightweight=True)
def test_switch_finds_relative_branch(self):
"""Switch will find 'foo' relative to the branch the checkout is of."""
self.build_tree(['repo/'])
tree1 = self.make_branch_and_tree('repo/brancha')
tree1.commit('foo')
tree2 = self.make_branch_and_tree('repo/branchb')
tree2.pull(tree1.branch)
branchb_id = tree2.commit('bar')
checkout = tree1.branch.create_checkout('checkout', lightweight=True)
self.run_bzr(['switch', 'branchb'], working_dir='checkout')
self.assertEqual(branchb_id, checkout.last_revision())
checkout = checkout.bzrdir.open_workingtree()
self.assertEqual(tree2.branch.base, checkout.branch.base)
def test_switch_finds_relative_bound_branch(self):
"""Using switch on a heavy checkout should find master sibling
The behaviour of lighweight and heavy checkouts should be
consistentwhen using the convenient "switch to sibling" feature
Both should switch to a sibling of the branch
they are bound to, and not a sibling of themself"""
self.build_tree(['repo/',
'heavyco/'])
tree1 = self.make_branch_and_tree('repo/brancha')
tree1.commit('foo')
tree2 = self.make_branch_and_tree('repo/branchb')
tree2.pull(tree1.branch)
branchb_id = tree2.commit('bar')
checkout = tree1.branch.create_checkout('heavyco/a', lightweight=False)
self.run_bzr(['switch', 'branchb'], working_dir='heavyco/a')
self.assertEqual(branchb_id, checkout.last_revision())
self.assertEqual(tree2.branch.base, checkout.branch.get_bound_location())
def test_switch_revision(self):
tree = self._create_sample_tree()
checkout = tree.branch.create_checkout('checkout', lightweight=True)
self.run_bzr(['switch', 'branch-1', '-r1'], working_dir='checkout')
self.failUnlessExists('checkout/file-1')
self.failIfExists('checkout/file-2')
def test_switch_only_revision(self):
tree = self._create_sample_tree()
checkout = tree.branch.create_checkout('checkout', lightweight=True)
self.failUnlessExists('checkout/file-1')
self.failUnlessExists('checkout/file-2')
self.run_bzr(['switch', '-r1'], working_dir='checkout')
self.failUnlessExists('checkout/file-1')
self.failIfExists('checkout/file-2')
# Check that we don't accept a range
self.run_bzr_error(
['bzr switch --revision takes exactly one revision identifier'],
['switch', '-r0..2'], working_dir='checkout')
def prepare_lightweight_switch(self):
branch = self.make_branch('branch')
branch.create_checkout('tree', lightweight=True)
osutils.rename('branch', 'branch1')
def test_switch_lightweight_after_branch_moved(self):
self.prepare_lightweight_switch()
self.run_bzr('switch --force ../branch1', working_dir='tree')
branch_location = WorkingTree.open('tree').branch.base
self.assertEndsWith(branch_location, 'branch1/')
def test_switch_lightweight_after_branch_moved_relative(self):
self.prepare_lightweight_switch()
self.run_bzr('switch --force branch1', working_dir='tree')
branch_location = WorkingTree.open('tree').branch.base
self.assertEndsWith(branch_location, 'branch1/')
def test_create_branch_no_branch(self):
self.prepare_lightweight_switch()
self.run_bzr_error(['cannot create branch without source branch'],
'switch --create-branch ../branch2', working_dir='tree')
def test_create_branch(self):
branch = self.make_branch('branch')
tree = branch.create_checkout('tree', lightweight=True)
tree.commit('one', rev_id='rev-1')
self.run_bzr('switch --create-branch ../branch2', working_dir='tree')
tree = WorkingTree.open('tree')
self.assertEndsWith(tree.branch.base, '/branch2/')
def test_create_branch_local(self):
branch = self.make_branch('branch')
tree = branch.create_checkout('tree', lightweight=True)
tree.commit('one', rev_id='rev-1')
self.run_bzr('switch --create-branch branch2', working_dir='tree')
tree = WorkingTree.open('tree')
# The new branch should have been created at the same level as
# 'branch', because we did not have a '/' segment
self.assertEqual(branch.base[:-1] + '2/', tree.branch.base)
def test_create_branch_short_name(self):
branch = self.make_branch('branch')
tree = branch.create_checkout('tree', lightweight=True)
tree.commit('one', rev_id='rev-1')
self.run_bzr('switch -b branch2', working_dir='tree')
tree = WorkingTree.open('tree')
# The new branch should have been created at the same level as
# 'branch', because we did not have a '/' segment
self.assertEqual(branch.base[:-1] + '2/', tree.branch.base)
def test_create_branch_directory_services(self):
branch = self.make_branch('branch')
tree = branch.create_checkout('tree', lightweight=True)
class FooLookup(object):
def look_up(self, name, url):
return 'foo-'+name
directories.register('foo:', FooLookup, 'Create branches named foo-')
self.addCleanup(directories.remove, 'foo:')
self.run_bzr('switch -b foo:branch2', working_dir='tree')
tree = WorkingTree.open('tree')
self.assertEndsWith(tree.branch.base, 'foo-branch2/')
def test_switch_with_post_switch_hook(self):
from bzrlib import branch
calls = []
_mod_branch.Branch.hooks.install_named_hook('post_switch',
calls.append, None)
self.make_branch_and_tree('branch')
self.run_bzr('branch branch branch2')
self.run_bzr('checkout branch checkout')
os.chdir('checkout')
self.assertLength(0, calls)
out, err = self.run_bzr('switch ../branch2')
self.assertLength(1, calls)
def test_switch_lightweight_co_with_post_switch_hook(self):
from bzrlib import branch
calls = []
_mod_branch.Branch.hooks.install_named_hook('post_switch',
calls.append, None)
self.make_branch_and_tree('branch')
self.run_bzr('branch branch branch2')
self.run_bzr('checkout --lightweight branch checkout')
os.chdir('checkout')
self.assertLength(0, calls)
out, err = self.run_bzr('switch ../branch2')
self.assertLength(1, calls)
|