1
# Copyright (C) 2005, 2006 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18
"""Black-box tests for bzr branch."""
22
from bzrlib import branch, bzrdir
23
from bzrlib.repository import RepositoryFormatKnit1
24
from bzrlib.tests.blackbox import ExternalBase
25
from bzrlib.workingtree import WorkingTree
28
class TestBranch(ExternalBase):
30
def example_branch(test):
32
file('hello', 'wt').write('foo')
33
test.runbzr('add hello')
34
test.runbzr('commit -m setup hello')
35
file('goodbye', 'wt').write('baz')
36
test.runbzr('add goodbye')
37
test.runbzr('commit -m setup goodbye')
39
def test_branch(self):
40
"""Branch from one branch to another."""
45
self.runbzr('branch a b')
46
b = branch.Branch.open('b')
47
self.assertEqual('b\n', b.control_files.get_utf8('branch-name').read())
48
self.runbzr('branch a c -r 1')
50
self.runbzr('commit -m foo --unchanged')
53
def test_branch_basis(self):
54
# ensure that basis really does grab from the basis by having incomplete source
55
tree = self.make_branch_and_tree('commit_tree')
56
self.build_tree(['foo'], transport=tree.bzrdir.transport.clone('..'))
58
tree.commit('revision 1', rev_id='1')
59
source = self.make_branch_and_tree('source')
60
# this gives us an incomplete repository
61
tree.bzrdir.open_repository().copy_content_into(source.branch.repository)
62
tree.commit('revision 2', rev_id='2', allow_pointless=True)
63
tree.bzrdir.open_branch().copy_content_into(source.branch)
64
tree.copy_content_into(source)
65
self.assertFalse(source.branch.repository.has_revision('2'))
67
self.runbzr('branch source target --basis commit_tree')
68
target = bzrdir.BzrDir.open('target')
69
self.assertEqual('2', target.open_branch().last_revision())
70
self.assertEqual(['2'], target.open_workingtree().get_parent_ids())
71
self.assertTrue(target.open_branch().repository.has_revision('2'))
73
def test_branch_only_copies_history(self):
74
# Knit branches should only push the history for the current revision.
75
format = bzrdir.BzrDirMetaFormat1()
76
format.repository_format = RepositoryFormatKnit1()
77
shared_repo = self.make_repository('repo', format=format, shared=True)
78
shared_repo.set_make_working_trees(True)
80
def make_shared_tree(path):
81
shared_repo.bzrdir.root_transport.mkdir(path)
82
shared_repo.bzrdir.create_branch_convenience('repo/' + path)
83
return WorkingTree.open('repo/' + path)
84
tree_a = make_shared_tree('a')
85
self.build_tree(['repo/a/file'])
87
tree_a.commit('commit a-1', rev_id='a-1')
88
f = open('repo/a/file', 'ab')
89
f.write('more stuff\n')
91
tree_a.commit('commit a-2', rev_id='a-2')
93
tree_b = make_shared_tree('b')
94
self.build_tree(['repo/b/file'])
96
tree_b.commit('commit b-1', rev_id='b-1')
98
self.assertTrue(shared_repo.has_revision('a-1'))
99
self.assertTrue(shared_repo.has_revision('a-2'))
100
self.assertTrue(shared_repo.has_revision('b-1'))
102
# Now that we have a repository with shared files, make sure
103
# that things aren't copied out by a 'branch'
104
self.run_bzr('branch', 'repo/b', 'branch-b')
105
pushed_tree = WorkingTree.open('branch-b')
106
pushed_repo = pushed_tree.branch.repository
107
self.assertFalse(pushed_repo.has_revision('a-1'))
108
self.assertFalse(pushed_repo.has_revision('a-2'))
109
self.assertTrue(pushed_repo.has_revision('b-1'))