1
# Copyright (C) 2005, 2006, 2008 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.repofmt.knitrepo import RepositoryFormatKnit1
24
from bzrlib.tests.blackbox import ExternalBase
25
from bzrlib.tests import HardlinkFeature
26
from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer
27
from bzrlib.workingtree import WorkingTree
30
class TestBranch(ExternalBase):
32
def example_branch(self, path='.'):
33
tree = self.make_branch_and_tree(path)
34
self.build_tree_contents([(path + '/hello', 'foo')])
36
tree.commit(message='setup')
37
self.build_tree_contents([(path + '/goodbye', 'baz')])
39
tree.commit(message='setup')
41
def test_branch(self):
42
"""Branch from one branch to another."""
43
self.example_branch('a')
44
self.run_bzr('branch a b')
45
b = branch.Branch.open('b')
46
self.run_bzr('branch a c -r 1')
47
# previously was erroneously created by branching
48
self.assertFalse(b._transport.has('branch-name'))
49
b.bzrdir.open_workingtree().commit(message='foo', allow_pointless=True)
51
def test_branch_only_copies_history(self):
52
# Knit branches should only push the history for the current revision.
53
format = bzrdir.BzrDirMetaFormat1()
54
format.repository_format = RepositoryFormatKnit1()
55
shared_repo = self.make_repository('repo', format=format, shared=True)
56
shared_repo.set_make_working_trees(True)
58
def make_shared_tree(path):
59
shared_repo.bzrdir.root_transport.mkdir(path)
60
shared_repo.bzrdir.create_branch_convenience('repo/' + path)
61
return WorkingTree.open('repo/' + path)
62
tree_a = make_shared_tree('a')
63
self.build_tree(['repo/a/file'])
65
tree_a.commit('commit a-1', rev_id='a-1')
66
f = open('repo/a/file', 'ab')
67
f.write('more stuff\n')
69
tree_a.commit('commit a-2', rev_id='a-2')
71
tree_b = make_shared_tree('b')
72
self.build_tree(['repo/b/file'])
74
tree_b.commit('commit b-1', rev_id='b-1')
76
self.assertTrue(shared_repo.has_revision('a-1'))
77
self.assertTrue(shared_repo.has_revision('a-2'))
78
self.assertTrue(shared_repo.has_revision('b-1'))
80
# Now that we have a repository with shared files, make sure
81
# that things aren't copied out by a 'branch'
82
self.run_bzr('branch repo/b branch-b')
83
pushed_tree = WorkingTree.open('branch-b')
84
pushed_repo = pushed_tree.branch.repository
85
self.assertFalse(pushed_repo.has_revision('a-1'))
86
self.assertFalse(pushed_repo.has_revision('a-2'))
87
self.assertTrue(pushed_repo.has_revision('b-1'))
89
def test_branch_hardlink(self):
90
self.requireFeature(HardlinkFeature)
91
source = self.make_branch_and_tree('source')
92
self.build_tree(['source/file1'])
94
source.commit('added file')
95
self.run_bzr(['branch', 'source', 'target', '--hardlink'])
96
source_stat = os.stat('source/file1')
97
target_stat = os.stat('target/file1')
98
self.assertEqual(source_stat, target_stat)
100
def assertShallow(self, branch_revid, stacked_on):
101
"""Assert that the branch 'newbranch' has been published correctly."""
102
new_branch = branch.Branch.open('newbranch')
103
# The branch refers to the mainline
104
self.assertEqual(stacked_on, new_branch.get_stacked_on())
105
# and the branch's work was pushed
106
self.assertTrue(new_branch.repository.has_revision(branch_revid))
107
# but the mainline's was not included
108
repo = new_branch.bzrdir.open_repository()
109
revids = repo.all_revision_ids()
111
self.fail("wrong revisions in shallow repository:\n %r"
114
def test_branch_stacked_branch_also_stacked_same_reference(self):
116
trunk_tree = self.make_branch_and_tree('target',
117
format='development')
118
trunk_tree.commit('mainline')
119
# and a branch from it which is stacked
120
branch_tree = self.make_branch_and_tree('branch',
121
format='development')
122
branch_tree.branch.set_stacked_on(trunk_tree.branch.base)
123
# with some work on it
124
branch_tree.commit('moar work plz')
125
# branching our local branch gives us a new stacked branch pointing at
127
out, err = self.run_bzr(['branch', 'branch', 'newbranch'])
128
self.assertEqual('', out)
129
self.assertEqual('Created new stacked branch referring to %s.\n' %
130
trunk_tree.branch.base, err)
131
self.assertShallow(branch_tree.last_revision(),
132
trunk_tree.branch.base)
134
def test_branch_stacked(self):
136
trunk_tree = self.make_branch_and_tree('mainline',
137
format='development')
138
trunk_tree.commit('mainline')
139
# and a branch from it which is stacked
140
out, err = self.run_bzr(['branch', '--stacked', 'mainline',
142
self.assertEqual('', out)
143
self.assertEqual('Created new stacked branch referring to %s.\n' %
144
trunk_tree.branch.base, err)
145
new_tree = WorkingTree.open('newbranch')
146
new_revid = new_tree.commit('new work')
147
self.assertShallow(new_revid, trunk_tree.branch.base)
149
def test_branch_stacked_from_smart_server(self):
150
# We can branch stacking on a smart server
151
from bzrlib.smart.server import SmartTCPServer_for_testing
152
self.transport_server = SmartTCPServer_for_testing
153
trunk = self.make_branch('mainline', format='development')
154
out, err = self.run_bzr(
155
['branch', '--stacked', self.get_url('mainline'), 'shallow'])
158
class TestRemoteBranch(TestCaseWithSFTPServer):
161
super(TestRemoteBranch, self).setUp()
162
tree = self.make_branch_and_tree('branch')
163
self.build_tree_contents([('branch/file', 'file content\n')])
165
tree.commit('file created')
167
def test_branch_local_remote(self):
168
self.run_bzr(['branch', 'branch', self.get_url('remote')])
169
t = self.get_transport()
170
# Ensure that no working tree what created remotely
171
self.assertFalse(t.has('remote/file'))
173
def test_branch_remote_remote(self):
174
# Light cheat: we access the branch remotely
175
self.run_bzr(['branch', self.get_url('branch'),
176
self.get_url('remote')])
177
t = self.get_transport()
178
# Ensure that no working tree what created remotely
179
self.assertFalse(t.has('remote/file'))
added
removed
bzrlib/tests/blackbox/test_branch.py
