26
revision as _mod_revision,
22
from bzrlib import (branch, bzrdir, errors, repository)
28
23
from bzrlib.repofmt.knitrepo import RepositoryFormatKnit1
29
from bzrlib.tests import TestCaseWithTransport
30
from bzrlib.tests import (
35
from bzrlib.tests.features import (
38
from bzrlib.tests.blackbox import test_switch
24
from bzrlib.tests.blackbox import ExternalBase
25
from bzrlib.tests import HardlinkFeature
39
26
from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer
40
from bzrlib.tests.script import run_script
41
27
from bzrlib.urlutils import local_path_to_url, strip_trailing_slash
42
28
from bzrlib.workingtree import WorkingTree
45
class TestBranch(TestCaseWithTransport):
31
class TestBranch(ExternalBase):
47
33
def example_branch(self, path='.'):
48
34
tree = self.make_branch_and_tree(path)
63
49
self.assertFalse(b._transport.has('branch-name'))
64
50
b.bzrdir.open_workingtree().commit(message='foo', allow_pointless=True)
66
def test_branch_broken_pack(self):
67
"""branching with a corrupted pack file."""
68
self.example_branch('a')
69
#now add some random corruption
70
fname = 'a/.bzr/repository/packs/' + os.listdir('a/.bzr/repository/packs')[0]
71
with open(fname, 'rb+') as f:
74
self.run_bzr_error(['Corruption while decompressing repository file'],
75
'branch a b', retcode=3)
77
def test_branch_switch_no_branch(self):
78
# No branch in the current directory:
79
# => new branch will be created, but switch fails
80
self.example_branch('a')
81
self.make_repository('current')
82
self.run_bzr_error(['No WorkingTree exists for'],
83
'branch --switch ../a ../b', working_dir='current')
84
a = branch.Branch.open('a')
85
b = branch.Branch.open('b')
86
self.assertEqual(a.last_revision(), b.last_revision())
88
def test_branch_switch_no_wt(self):
89
# No working tree in the current directory:
90
# => new branch will be created, but switch fails and the current
91
# branch is unmodified
92
self.example_branch('a')
93
self.make_branch('current')
94
self.run_bzr_error(['No WorkingTree exists for'],
95
'branch --switch ../a ../b', working_dir='current')
96
a = branch.Branch.open('a')
97
b = branch.Branch.open('b')
98
self.assertEqual(a.last_revision(), b.last_revision())
99
work = branch.Branch.open('current')
100
self.assertEqual(work.last_revision(), _mod_revision.NULL_REVISION)
102
def test_branch_switch_no_checkout(self):
103
# Standalone branch in the current directory:
104
# => new branch will be created, but switch fails and the current
105
# branch is unmodified
106
self.example_branch('a')
107
self.make_branch_and_tree('current')
108
self.run_bzr_error(['Cannot switch a branch, only a checkout'],
109
'branch --switch ../a ../b', working_dir='current')
110
a = branch.Branch.open('a')
111
b = branch.Branch.open('b')
112
self.assertEqual(a.last_revision(), b.last_revision())
113
work = branch.Branch.open('current')
114
self.assertEqual(work.last_revision(), _mod_revision.NULL_REVISION)
116
def test_branch_switch_checkout(self):
117
# Checkout in the current directory:
118
# => new branch will be created and checkout bound to the new branch
119
self.example_branch('a')
120
self.run_bzr('checkout a current')
121
out, err = self.run_bzr('branch --switch ../a ../b', working_dir='current')
122
a = branch.Branch.open('a')
123
b = branch.Branch.open('b')
124
self.assertEqual(a.last_revision(), b.last_revision())
125
work = WorkingTree.open('current')
126
self.assertEndsWith(work.branch.get_bound_location(), '/b/')
127
self.assertContainsRe(err, "Switched to branch: .*/b/")
129
def test_branch_switch_lightweight_checkout(self):
130
# Lightweight checkout in the current directory:
131
# => new branch will be created and lightweight checkout pointed to
133
self.example_branch('a')
134
self.run_bzr('checkout --lightweight a current')
135
out, err = self.run_bzr('branch --switch ../a ../b', working_dir='current')
136
a = branch.Branch.open('a')
137
b = branch.Branch.open('b')
138
self.assertEqual(a.last_revision(), b.last_revision())
139
work = WorkingTree.open('current')
140
self.assertEndsWith(work.branch.base, '/b/')
141
self.assertContainsRe(err, "Switched to branch: .*/b/")
143
52
def test_branch_only_copies_history(self):
144
53
# Knit branches should only push the history for the current revision.
145
54
format = bzrdir.BzrDirMetaFormat1()
184
93
self.build_tree(['source/file1'])
185
94
source.add('file1')
186
95
source.commit('added file')
187
out, err = self.run_bzr(['branch', 'source', 'target', '--hardlink'])
96
self.run_bzr(['branch', 'source', 'target', '--hardlink'])
188
97
source_stat = os.stat('source/file1')
189
98
target_stat = os.stat('target/file1')
190
99
self.assertEqual(source_stat, target_stat)
192
def test_branch_files_from(self):
193
source = self.make_branch_and_tree('source')
194
self.build_tree(['source/file1'])
196
source.commit('added file')
197
out, err = self.run_bzr('branch source target --files-from source')
198
self.assertPathExists('target/file1')
200
def test_branch_files_from_hardlink(self):
201
self.requireFeature(HardlinkFeature)
202
source = self.make_branch_and_tree('source')
203
self.build_tree(['source/file1'])
205
source.commit('added file')
206
source.bzrdir.sprout('second')
207
out, err = self.run_bzr('branch source target --files-from second'
209
source_stat = os.stat('source/file1')
210
second_stat = os.stat('second/file1')
211
target_stat = os.stat('target/file1')
212
self.assertNotEqual(source_stat, target_stat)
213
self.assertEqual(second_stat, target_stat)
215
101
def test_branch_standalone(self):
216
102
shared_repo = self.make_repository('repo', shared=True)
217
103
self.example_branch('source')
241
127
# force operation
242
128
self.run_bzr('branch a b --use-existing-dir')
243
129
# check conflicts
244
self.assertPathExists('b/hello.moved')
245
self.assertPathDoesNotExist('b/godbye.moved')
130
self.failUnlessExists('b/hello.moved')
131
self.failIfExists('b/godbye.moved')
246
132
# we can't branch into branch
247
133
out,err = self.run_bzr('branch a b --use-existing-dir', retcode=3)
248
134
self.assertEqual('', out)
249
135
self.assertEqual('bzr: ERROR: Already a branch: "b".\n', err)
251
def test_branch_bind(self):
252
self.example_branch('a')
253
out, err = self.run_bzr('branch a b --bind')
254
self.assertEndsWith(err, "New branch bound to a\n")
255
b = branch.Branch.open('b')
256
self.assertEndsWith(b.get_bound_location(), '/a/')
258
def test_branch_with_post_branch_init_hook(self):
260
branch.Branch.hooks.install_named_hook('post_branch_init',
262
self.assertLength(0, calls)
263
self.example_branch('a')
264
self.assertLength(1, calls)
265
self.run_bzr('branch a b')
266
self.assertLength(2, calls)
268
def test_checkout_with_post_branch_init_hook(self):
270
branch.Branch.hooks.install_named_hook('post_branch_init',
272
self.assertLength(0, calls)
273
self.example_branch('a')
274
self.assertLength(1, calls)
275
self.run_bzr('checkout a b')
276
self.assertLength(2, calls)
278
def test_lightweight_checkout_with_post_branch_init_hook(self):
280
branch.Branch.hooks.install_named_hook('post_branch_init',
282
self.assertLength(0, calls)
283
self.example_branch('a')
284
self.assertLength(1, calls)
285
self.run_bzr('checkout --lightweight a b')
286
self.assertLength(2, calls)
288
def test_branch_fetches_all_tags(self):
289
builder = self.make_branch_builder('source')
290
source = fixtures.build_branch_with_non_ancestral_rev(builder)
291
source.tags.set_tag('tag-a', 'rev-2')
292
source.get_config().set_user_option('branch.fetch_tags', 'True')
293
# Now source has a tag not in its ancestry. Make a branch from it.
294
self.run_bzr('branch source new-branch')
295
new_branch = branch.Branch.open('new-branch')
296
# The tag is present, and so is its revision.
297
self.assertEqual('rev-2', new_branch.tags.lookup_tag('tag-a'))
298
new_branch.repository.get_revision('rev-2')
301
class TestBranchStacked(TestCaseWithTransport):
138
class TestBranchStacked(ExternalBase):
302
139
"""Tests for branch --stacked"""
141
def check_shallow_branch(self, branch_revid, stacked_on):
142
"""Assert that the branch 'newbranch' has been published correctly.
144
:param stacked_on: url of a branch this one is stacked upon.
145
:param branch_revid: a revision id that should be the only
146
revision present in the stacked branch, and it should not be in
147
the reference branch.
149
new_branch = branch.Branch.open('newbranch')
150
# The branch refers to the mainline
151
self.assertEqual(stacked_on, new_branch.get_stacked_on_url())
152
# and the branch's work was pushed
153
self.assertTrue(new_branch.repository.has_revision(branch_revid))
154
# The newly committed revision shoud be present in the stacked branch,
155
# but not in the stacked-on branch. Because stacking is set up by the
156
# branch object, if we open the stacked branch's repository directly,
157
# bypassing the branch, we see only what's in the stacked repository.
158
stacked_repo = bzrdir.BzrDir.open('newbranch').open_repository()
159
stacked_repo_revisions = set(stacked_repo.all_revision_ids())
160
if len(stacked_repo_revisions) != 1:
161
self.fail("wrong revisions in stacked repository: %r"
162
% (stacked_repo_revisions,))
304
164
def assertRevisionInRepository(self, repo_path, revid):
305
165
"""Check that a revision is in a repository, disregarding stacking."""
306
166
repo = bzrdir.BzrDir.open(repo_path).open_repository()
328
188
branch_tree.branch.set_stacked_on_url(trunk_tree.branch.base)
329
189
# with some work on it
330
work_tree = trunk_tree.branch.bzrdir.sprout('local').open_workingtree()
331
work_tree.commit('moar work plz')
332
work_tree.branch.push(branch_tree.branch)
190
branch_tree.commit('moar work plz')
333
191
# branching our local branch gives us a new stacked branch pointing at
335
193
out, err = self.run_bzr(['branch', 'branch', 'newbranch'])
336
194
self.assertEqual('', out)
337
self.assertEqual('Branched 2 revision(s).\n',
195
self.assertEqual('Branched 1 revision(s).\n',
339
197
# it should have preserved the branch format, and so it should be
340
198
# capable of supporting stacking, but not actually have a stacked_on
384
239
self.assertEqual('Created new stacked branch referring to %s.\n' %
385
240
trunk_tree.branch.base, err)
386
241
self.assertRevisionNotInRepository('newbranch', original_revid)
387
new_branch = branch.Branch.open('newbranch')
388
self.assertEqual(trunk_tree.branch.base, new_branch.get_stacked_on_url())
242
new_tree = WorkingTree.open('newbranch')
243
new_revid = new_tree.commit('new work')
244
self.check_shallow_branch(new_revid, trunk_tree.branch.base)
390
246
def test_branch_stacked_from_smart_server(self):
391
247
# We can branch stacking on a smart server
392
self.transport_server = test_server.SmartTCPServer_for_testing
248
from bzrlib.smart.server import SmartTCPServer_for_testing
249
self.transport_server = SmartTCPServer_for_testing
393
250
trunk = self.make_branch('mainline', format='1.9')
394
251
out, err = self.run_bzr(
395
252
['branch', '--stacked', self.get_url('mainline'), 'shallow'])
478
329
# upwards without agreement from bzr's network support maintainers.
479
330
self.assertLength(15, self.hpss_calls)
481
def test_branch_from_branch_with_tags(self):
482
self.setup_smart_server_with_call_log()
483
builder = self.make_branch_builder('source')
484
source = fixtures.build_branch_with_non_ancestral_rev(builder)
485
source.get_config().set_user_option('branch.fetch_tags', 'True')
486
source.tags.set_tag('tag-a', 'rev-2')
487
source.tags.set_tag('tag-missing', 'missing-rev')
488
# Now source has a tag not in its ancestry. Make a branch from it.
489
self.reset_smart_call_log()
490
out, err = self.run_bzr(['branch', self.get_url('source'), 'target'])
491
# This figure represent the amount of work to perform this use case. It
492
# is entirely ok to reduce this number if a test fails due to rpc_count
493
# being too low. If rpc_count increases, more network roundtrips have
494
# become necessary for this use case. Please do not adjust this number
495
# upwards without agreement from bzr's network support maintainers.
496
self.assertLength(10, self.hpss_calls)
498
def test_branch_to_stacked_from_trivial_branch_streaming_acceptance(self):
499
self.setup_smart_server_with_call_log()
500
t = self.make_branch_and_tree('from')
501
for count in range(9):
502
t.commit(message='commit %d' % count)
503
self.reset_smart_call_log()
504
out, err = self.run_bzr(['branch', '--stacked', self.get_url('from'),
506
# XXX: the number of hpss calls for this case isn't deterministic yet,
507
# so we can't easily assert about the number of calls.
508
#self.assertLength(XXX, self.hpss_calls)
509
# We can assert that none of the calls were readv requests for rix
510
# files, though (demonstrating that at least get_parent_map calls are
511
# not using VFS RPCs).
512
readvs_of_rix_files = [
513
c for c in self.hpss_calls
514
if c.call.method == 'readv' and c.call.args[-1].endswith('.rix')]
515
self.assertLength(0, readvs_of_rix_files)
518
333
class TestRemoteBranch(TestCaseWithSFTPServer):
538
353
# Ensure that no working tree what created remotely
539
354
self.assertFalse(t.has('remote/file'))
542
class TestDeprecatedAliases(TestCaseWithTransport):
544
def test_deprecated_aliases(self):
545
"""bzr branch can be called clone or get, but those names are deprecated.
549
for command in ['clone', 'get']:
551
$ bzr %(command)s A B
552
2>The command 'bzr %(command)s' has been deprecated in bzr 2.4. Please use 'bzr branch' instead.
553
2>bzr: ERROR: Not a branch...
557
class TestBranchParentLocation(test_switch.TestSwitchParentLocationBase):
559
def _checkout_and_branch(self, option=''):
560
self.script_runner.run_script(self, '''
561
$ bzr checkout %(option)s repo/trunk checkout
563
$ bzr branch --switch ../repo/trunk ../repo/branched
564
2>Branched 0 revision(s).
565
2>Tree is up to date at revision 0.
566
2>Switched to branch:...branched...
569
bound_branch = branch.Branch.open_containing('checkout')[0]
570
master_branch = branch.Branch.open_containing('repo/branched')[0]
571
return (bound_branch, master_branch)
573
def test_branch_switch_parent_lightweight(self):
574
"""Lightweight checkout using bzr branch --switch."""
575
bb, mb = self._checkout_and_branch(option='--lightweight')
576
self.assertParent('repo/trunk', bb)
577
self.assertParent('repo/trunk', mb)
579
def test_branch_switch_parent_heavyweight(self):
580
"""Heavyweight checkout using bzr branch --switch."""
581
bb, mb = self._checkout_and_branch()
582
self.assertParent('repo/trunk', bb)
583
self.assertParent('repo/trunk', mb)
added
removed
bzrlib/tests/blackbox/test_branch.py
