~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/blackbox/test_branch.py

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006 by Canonical Ltd
 
1
# Copyright (C) 2005, 2006, 2008, 2009 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
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
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
 
18
18
"""Black-box tests for bzr branch."""
19
19
 
20
20
import os
21
21
 
22
 
from bzrlib import branch, bzrdir
23
 
from bzrlib.repository import RepositoryFormatKnit1
 
22
from bzrlib import (
 
23
    branch,
 
24
    bzrdir,
 
25
    errors,
 
26
    repository,
 
27
    revision as _mod_revision,
 
28
    )
 
29
from bzrlib.repofmt.knitrepo import RepositoryFormatKnit1
24
30
from bzrlib.tests.blackbox import ExternalBase
 
31
from bzrlib.tests import (
 
32
    KnownFailure,
 
33
    HardlinkFeature,
 
34
    )
 
35
from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer
 
36
from bzrlib.urlutils import local_path_to_url, strip_trailing_slash
25
37
from bzrlib.workingtree import WorkingTree
26
38
 
27
39
 
28
40
class TestBranch(ExternalBase):
29
41
 
30
 
    def example_branch(test):
31
 
        test.runbzr('init')
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')
 
42
    def example_branch(self, path='.'):
 
43
        tree = self.make_branch_and_tree(path)
 
44
        self.build_tree_contents([(path + '/hello', 'foo')])
 
45
        tree.add('hello')
 
46
        tree.commit(message='setup')
 
47
        self.build_tree_contents([(path + '/goodbye', 'baz')])
 
48
        tree.add('goodbye')
 
49
        tree.commit(message='setup')
38
50
 
39
51
    def test_branch(self):
40
52
        """Branch from one branch to another."""
41
 
        os.mkdir('a')
42
 
        os.chdir('a')
43
 
        self.example_branch()
44
 
        os.chdir('..')
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')
49
 
        os.chdir('b')
50
 
        self.runbzr('commit -m foo --unchanged')
51
 
        os.chdir('..')
52
 
 
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('..'))
57
 
        tree.add('foo')
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'))
66
 
        dir = source.bzrdir
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().last_revision())
71
 
        self.assertTrue(target.open_branch().repository.has_revision('2'))
 
53
        self.example_branch('a')
 
54
        self.run_bzr('branch a b')
 
55
        b = branch.Branch.open('b')
 
56
        self.run_bzr('branch a c -r 1')
 
57
        # previously was erroneously created by branching
 
58
        self.assertFalse(b._transport.has('branch-name'))
 
59
        b.bzrdir.open_workingtree().commit(message='foo', allow_pointless=True)
 
60
 
 
61
    def test_branch_switch_no_branch(self):
 
62
        # No branch in the current directory:
 
63
        #  => new branch will be created, but switch fails
 
64
        self.example_branch('a')
 
65
        self.make_repository('current')
 
66
        self.run_bzr_error(['No WorkingTree exists for'],
 
67
            'branch --switch ../a ../b', working_dir='current')
 
68
        a = branch.Branch.open('a')
 
69
        b = branch.Branch.open('b')
 
70
        self.assertEqual(a.last_revision(), b.last_revision())
 
71
 
 
72
    def test_branch_switch_no_wt(self):
 
73
        # No working tree in the current directory:
 
74
        #  => new branch will be created, but switch fails and the current
 
75
        #     branch is unmodified
 
76
        self.example_branch('a')
 
77
        self.make_branch('current')
 
78
        self.run_bzr_error(['No WorkingTree exists for'],
 
79
            'branch --switch ../a ../b', working_dir='current')
 
80
        a = branch.Branch.open('a')
 
81
        b = branch.Branch.open('b')
 
82
        self.assertEqual(a.last_revision(), b.last_revision())
 
83
        work = branch.Branch.open('current')
 
84
        self.assertEqual(work.last_revision(), _mod_revision.NULL_REVISION)
 
85
 
 
86
    def test_branch_switch_no_checkout(self):
 
87
        # Standalone branch in the current directory:
 
88
        #  => new branch will be created, but switch fails and the current
 
89
        #     branch is unmodified
 
90
        self.example_branch('a')
 
91
        self.make_branch_and_tree('current')
 
92
        self.run_bzr_error(['Cannot switch a branch, only a checkout'],
 
93
            'branch --switch ../a ../b', working_dir='current')
 
94
        a = branch.Branch.open('a')
 
95
        b = branch.Branch.open('b')
 
96
        self.assertEqual(a.last_revision(), b.last_revision())
 
97
        work = branch.Branch.open('current')
 
98
        self.assertEqual(work.last_revision(), _mod_revision.NULL_REVISION)
 
99
 
 
100
    def test_branch_switch_checkout(self):
 
101
        # Checkout in the current directory:
 
102
        #  => new branch will be created and checkout bound to the new branch
 
103
        self.example_branch('a')
 
104
        self.run_bzr('checkout a current')
 
105
        out, err = self.run_bzr('branch --switch ../a ../b', working_dir='current')
 
106
        a = branch.Branch.open('a')
 
107
        b = branch.Branch.open('b')
 
108
        self.assertEqual(a.last_revision(), b.last_revision())
 
109
        work = WorkingTree.open('current')
 
110
        self.assertEndsWith(work.branch.get_bound_location(), '/b/')
 
111
        self.assertContainsRe(err, "Switched to branch: .*/b/")
 
112
 
 
113
    def test_branch_switch_lightweight_checkout(self):
 
114
        # Lightweight checkout in the current directory:
 
115
        #  => new branch will be created and lightweight checkout pointed to
 
116
        #     the new branch
 
117
        self.example_branch('a')
 
118
        self.run_bzr('checkout --lightweight a current')
 
119
        out, err = self.run_bzr('branch --switch ../a ../b', working_dir='current')
 
120
        a = branch.Branch.open('a')
 
121
        b = branch.Branch.open('b')
 
122
        self.assertEqual(a.last_revision(), b.last_revision())
 
123
        work = WorkingTree.open('current')
 
124
        self.assertEndsWith(work.branch.base, '/b/')
 
125
        self.assertContainsRe(err, "Switched to branch: .*/b/")
72
126
 
73
127
    def test_branch_only_copies_history(self):
74
128
        # Knit branches should only push the history for the current revision.
101
155
 
102
156
        # Now that we have a repository with shared files, make sure
103
157
        # that things aren't copied out by a 'branch'
104
 
        self.run_bzr('branch', 'repo/b', 'branch-b')
 
158
        self.run_bzr('branch repo/b branch-b')
105
159
        pushed_tree = WorkingTree.open('branch-b')
106
160
        pushed_repo = pushed_tree.branch.repository
107
161
        self.assertFalse(pushed_repo.has_revision('a-1'))
108
162
        self.assertFalse(pushed_repo.has_revision('a-2'))
109
163
        self.assertTrue(pushed_repo.has_revision('b-1'))
110
164
 
 
165
    def test_branch_hardlink(self):
 
166
        self.requireFeature(HardlinkFeature)
 
167
        source = self.make_branch_and_tree('source')
 
168
        self.build_tree(['source/file1'])
 
169
        source.add('file1')
 
170
        source.commit('added file')
 
171
        out, err = self.run_bzr(['branch', 'source', 'target', '--hardlink'])
 
172
        source_stat = os.stat('source/file1')
 
173
        target_stat = os.stat('target/file1')
 
174
        same_file = (source_stat == target_stat)
 
175
        if same_file:
 
176
            pass
 
177
        else:
 
178
            # https://bugs.edge.launchpad.net/bzr/+bug/408193
 
179
            self.assertContainsRe(err, "hardlinking working copy files is "
 
180
                "not currently supported")
 
181
            raise KnownFailure("--hardlink doesn't work in formats "
 
182
                "that support content filtering (#408193)")
 
183
 
 
184
    def test_branch_standalone(self):
 
185
        shared_repo = self.make_repository('repo', shared=True)
 
186
        self.example_branch('source')
 
187
        self.run_bzr('branch --standalone source repo/target')
 
188
        b = branch.Branch.open('repo/target')
 
189
        expected_repo_path = os.path.abspath('repo/target/.bzr/repository')
 
190
        self.assertEqual(strip_trailing_slash(b.repository.base),
 
191
            strip_trailing_slash(local_path_to_url(expected_repo_path)))
 
192
 
 
193
    def test_branch_no_tree(self):
 
194
        self.example_branch('source')
 
195
        self.run_bzr('branch --no-tree source target')
 
196
        self.failIfExists('target/hello')
 
197
        self.failIfExists('target/goodbye')
 
198
 
 
199
    def test_branch_into_existing_dir(self):
 
200
        self.example_branch('a')
 
201
        # existing dir with similar files but no .bzr dir
 
202
        self.build_tree_contents([('b/',)])
 
203
        self.build_tree_contents([('b/hello', 'bar')])  # different content
 
204
        self.build_tree_contents([('b/goodbye', 'baz')])# same content
 
205
        # fails without --use-existing-dir
 
206
        out,err = self.run_bzr('branch a b', retcode=3)
 
207
        self.assertEqual('', out)
 
208
        self.assertEqual('bzr: ERROR: Target directory "b" already exists.\n',
 
209
            err)
 
210
        # force operation
 
211
        self.run_bzr('branch a b --use-existing-dir')
 
212
        # check conflicts
 
213
        self.failUnlessExists('b/hello.moved')
 
214
        self.failIfExists('b/godbye.moved')
 
215
        # we can't branch into branch
 
216
        out,err = self.run_bzr('branch a b --use-existing-dir', retcode=3)
 
217
        self.assertEqual('', out)
 
218
        self.assertEqual('bzr: ERROR: Already a branch: "b".\n', err)
 
219
 
 
220
 
 
221
class TestBranchStacked(ExternalBase):
 
222
    """Tests for branch --stacked"""
 
223
 
 
224
    def assertRevisionInRepository(self, repo_path, revid):
 
225
        """Check that a revision is in a repository, disregarding stacking."""
 
226
        repo = bzrdir.BzrDir.open(repo_path).open_repository()
 
227
        self.assertTrue(repo.has_revision(revid))
 
228
 
 
229
    def assertRevisionNotInRepository(self, repo_path, revid):
 
230
        """Check that a revision is not in a repository, disregarding stacking."""
 
231
        repo = bzrdir.BzrDir.open(repo_path).open_repository()
 
232
        self.assertFalse(repo.has_revision(revid))
 
233
 
 
234
    def assertRevisionsInBranchRepository(self, revid_list, branch_path):
 
235
        repo = branch.Branch.open(branch_path).repository
 
236
        self.assertEqual(set(revid_list),
 
237
            repo.has_revisions(revid_list))
 
238
 
 
239
    def test_branch_stacked_branch_not_stacked(self):
 
240
        """Branching a stacked branch is not stacked by default"""
 
241
        # We have a mainline
 
242
        trunk_tree = self.make_branch_and_tree('target',
 
243
            format='1.9')
 
244
        trunk_tree.commit('mainline')
 
245
        # and a branch from it which is stacked
 
246
        branch_tree = self.make_branch_and_tree('branch',
 
247
            format='1.9')
 
248
        branch_tree.branch.set_stacked_on_url(trunk_tree.branch.base)
 
249
        # with some work on it
 
250
        work_tree = trunk_tree.branch.bzrdir.sprout('local').open_workingtree()
 
251
        work_tree.commit('moar work plz')
 
252
        work_tree.branch.push(branch_tree.branch)
 
253
        # branching our local branch gives us a new stacked branch pointing at
 
254
        # mainline.
 
255
        out, err = self.run_bzr(['branch', 'branch', 'newbranch'])
 
256
        self.assertEqual('', out)
 
257
        self.assertEqual('Branched 2 revision(s).\n',
 
258
            err)
 
259
        # it should have preserved the branch format, and so it should be
 
260
        # capable of supporting stacking, but not actually have a stacked_on
 
261
        # branch configured
 
262
        self.assertRaises(errors.NotStacked,
 
263
            bzrdir.BzrDir.open('newbranch').open_branch().get_stacked_on_url)
 
264
 
 
265
    def test_branch_stacked_branch_stacked(self):
 
266
        """Asking to stack on a stacked branch does work"""
 
267
        # We have a mainline
 
268
        trunk_tree = self.make_branch_and_tree('target',
 
269
            format='1.9')
 
270
        trunk_revid = trunk_tree.commit('mainline')
 
271
        # and a branch from it which is stacked
 
272
        branch_tree = self.make_branch_and_tree('branch',
 
273
            format='1.9')
 
274
        branch_tree.branch.set_stacked_on_url(trunk_tree.branch.base)
 
275
        # with some work on it
 
276
        work_tree = trunk_tree.branch.bzrdir.sprout('local').open_workingtree()
 
277
        branch_revid = work_tree.commit('moar work plz')
 
278
        work_tree.branch.push(branch_tree.branch)
 
279
        # you can chain branches on from there
 
280
        out, err = self.run_bzr(['branch', 'branch', '--stacked', 'branch2'])
 
281
        self.assertEqual('', out)
 
282
        self.assertEqual('Created new stacked branch referring to %s.\n' %
 
283
            branch_tree.branch.base, err)
 
284
        self.assertEqual(branch_tree.branch.base,
 
285
            branch.Branch.open('branch2').get_stacked_on_url())
 
286
        branch2_tree = WorkingTree.open('branch2')
 
287
        branch2_revid = work_tree.commit('work on second stacked branch')
 
288
        work_tree.branch.push(branch2_tree.branch)
 
289
        self.assertRevisionInRepository('branch2', branch2_revid)
 
290
        self.assertRevisionsInBranchRepository(
 
291
            [trunk_revid, branch_revid, branch2_revid],
 
292
            'branch2')
 
293
 
 
294
    def test_branch_stacked(self):
 
295
        # We have a mainline
 
296
        trunk_tree = self.make_branch_and_tree('mainline',
 
297
            format='1.9')
 
298
        original_revid = trunk_tree.commit('mainline')
 
299
        self.assertRevisionInRepository('mainline', original_revid)
 
300
        # and a branch from it which is stacked
 
301
        out, err = self.run_bzr(['branch', '--stacked', 'mainline',
 
302
            'newbranch'])
 
303
        self.assertEqual('', out)
 
304
        self.assertEqual('Created new stacked branch referring to %s.\n' %
 
305
            trunk_tree.branch.base, err)
 
306
        self.assertRevisionNotInRepository('newbranch', original_revid)
 
307
        new_branch = branch.Branch.open('newbranch')
 
308
        self.assertEqual(trunk_tree.branch.base, new_branch.get_stacked_on_url())
 
309
 
 
310
    def test_branch_stacked_from_smart_server(self):
 
311
        # We can branch stacking on a smart server
 
312
        from bzrlib.smart.server import SmartTCPServer_for_testing
 
313
        self.transport_server = SmartTCPServer_for_testing
 
314
        trunk = self.make_branch('mainline', format='1.9')
 
315
        out, err = self.run_bzr(
 
316
            ['branch', '--stacked', self.get_url('mainline'), 'shallow'])
 
317
 
 
318
    def test_branch_stacked_from_non_stacked_format(self):
 
319
        """The origin format doesn't support stacking"""
 
320
        trunk = self.make_branch('trunk', format='pack-0.92')
 
321
        out, err = self.run_bzr(
 
322
            ['branch', '--stacked', 'trunk', 'shallow'])
 
323
        # We should notify the user that we upgraded their format
 
324
        self.assertEqualDiff(
 
325
            'Source repository format does not support stacking, using format:\n'
 
326
            '  Packs 5 (adds stacking support, requires bzr 1.6)\n'
 
327
            'Source branch format does not support stacking, using format:\n'
 
328
            '  Branch format 7\n'
 
329
            'Created new stacked branch referring to %s.\n' % (trunk.base,),
 
330
            err)
 
331
 
 
332
    def test_branch_stacked_from_rich_root_non_stackable(self):
 
333
        trunk = self.make_branch('trunk', format='rich-root-pack')
 
334
        out, err = self.run_bzr(
 
335
            ['branch', '--stacked', 'trunk', 'shallow'])
 
336
        # We should notify the user that we upgraded their format
 
337
        self.assertEqualDiff(
 
338
            'Source repository format does not support stacking, using format:\n'
 
339
            '  Packs 5 rich-root (adds stacking support, requires bzr 1.6.1)\n'
 
340
            'Source branch format does not support stacking, using format:\n'
 
341
            '  Branch format 7\n'
 
342
            'Created new stacked branch referring to %s.\n' % (trunk.base,),
 
343
            err)
 
344
 
 
345
 
 
346
class TestSmartServerBranching(ExternalBase):
 
347
 
 
348
    def test_branch_from_trivial_branch_to_same_server_branch_acceptance(self):
 
349
        self.setup_smart_server_with_call_log()
 
350
        t = self.make_branch_and_tree('from')
 
351
        for count in range(9):
 
352
            t.commit(message='commit %d' % count)
 
353
        self.reset_smart_call_log()
 
354
        out, err = self.run_bzr(['branch', self.get_url('from'),
 
355
            self.get_url('target')])
 
356
        # This figure represent the amount of work to perform this use case. It
 
357
        # is entirely ok to reduce this number if a test fails due to rpc_count
 
358
        # being too low. If rpc_count increases, more network roundtrips have
 
359
        # become necessary for this use case. Please do not adjust this number
 
360
        # upwards without agreement from bzr's network support maintainers.
 
361
        self.assertLength(38, self.hpss_calls)
 
362
 
 
363
    def test_branch_from_trivial_branch_streaming_acceptance(self):
 
364
        self.setup_smart_server_with_call_log()
 
365
        t = self.make_branch_and_tree('from')
 
366
        for count in range(9):
 
367
            t.commit(message='commit %d' % count)
 
368
        self.reset_smart_call_log()
 
369
        out, err = self.run_bzr(['branch', self.get_url('from'),
 
370
            'local-target'])
 
371
        # This figure represent the amount of work to perform this use case. It
 
372
        # is entirely ok to reduce this number if a test fails due to rpc_count
 
373
        # being too low. If rpc_count increases, more network roundtrips have
 
374
        # become necessary for this use case. Please do not adjust this number
 
375
        # upwards without agreement from bzr's network support maintainers.
 
376
        self.assertLength(10, self.hpss_calls)
 
377
 
 
378
    def test_branch_from_trivial_stacked_branch_streaming_acceptance(self):
 
379
        self.setup_smart_server_with_call_log()
 
380
        t = self.make_branch_and_tree('trunk')
 
381
        for count in range(8):
 
382
            t.commit(message='commit %d' % count)
 
383
        tree2 = t.branch.bzrdir.sprout('feature', stacked=True
 
384
            ).open_workingtree()
 
385
        local_tree = t.branch.bzrdir.sprout('local-working').open_workingtree()
 
386
        local_tree.commit('feature change')
 
387
        local_tree.branch.push(tree2.branch)
 
388
        self.reset_smart_call_log()
 
389
        out, err = self.run_bzr(['branch', self.get_url('feature'),
 
390
            'local-target'])
 
391
        # This figure represent the amount of work to perform this use case. It
 
392
        # is entirely ok to reduce this number if a test fails due to rpc_count
 
393
        # being too low. If rpc_count increases, more network roundtrips have
 
394
        # become necessary for this use case. Please do not adjust this number
 
395
        # upwards without agreement from bzr's network support maintainers.
 
396
        self.assertLength(15, self.hpss_calls)
 
397
 
 
398
 
 
399
class TestRemoteBranch(TestCaseWithSFTPServer):
 
400
 
 
401
    def setUp(self):
 
402
        super(TestRemoteBranch, self).setUp()
 
403
        tree = self.make_branch_and_tree('branch')
 
404
        self.build_tree_contents([('branch/file', 'file content\n')])
 
405
        tree.add('file')
 
406
        tree.commit('file created')
 
407
 
 
408
    def test_branch_local_remote(self):
 
409
        self.run_bzr(['branch', 'branch', self.get_url('remote')])
 
410
        t = self.get_transport()
 
411
        # Ensure that no working tree what created remotely
 
412
        self.assertFalse(t.has('remote/file'))
 
413
 
 
414
    def test_branch_remote_remote(self):
 
415
        # Light cheat: we access the branch remotely
 
416
        self.run_bzr(['branch', self.get_url('branch'),
 
417
                      self.get_url('remote')])
 
418
        t = self.get_transport()
 
419
        # Ensure that no working tree what created remotely
 
420
        self.assertFalse(t.has('remote/file'))
111
421