~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

  • Committer: Tarmac
  • Author(s): Vincent Ladeuil, Patch Queue Manager, Jelmer Vernooij
  • Date: 2017-01-17 16:20:41 UTC
  • mfrom: (6619.1.2 trunk)
  • Revision ID: tarmac-20170117162041-oo62uk1qsmgc9j31
Merge 2.7 into trunk including fixes for bugs #1622039, #1644003, #1579093 and #1645017. [r=vila]

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2006-2012, 2016 Canonical Ltd
 
2
#
 
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.
 
7
#
 
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.
 
12
#
 
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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
 
 
17
 
 
18
"""Black-box tests for bzr push."""
 
19
 
 
20
import re
 
21
 
 
22
from bzrlib import (
 
23
    branch,
 
24
    bzrdir,
 
25
    controldir,
 
26
    errors,
 
27
    osutils,
 
28
    tests,
 
29
    transport,
 
30
    uncommit,
 
31
    urlutils,
 
32
    workingtree
 
33
    )
 
34
from bzrlib.repofmt import knitrepo
 
35
from bzrlib.tests import (
 
36
    http_server,
 
37
    scenarios,
 
38
    script,
 
39
    test_foreign,
 
40
    )
 
41
from bzrlib.tests.matchers import ContainsNoVfsCalls
 
42
from bzrlib.transport import memory
 
43
 
 
44
 
 
45
load_tests = scenarios.load_tests_apply_scenarios
 
46
 
 
47
 
 
48
class TestPush(tests.TestCaseWithTransport):
 
49
 
 
50
    def test_push_error_on_vfs_http(self):
 
51
        """ pushing a branch to a HTTP server fails cleanly. """
 
52
        # the trunk is published on a web server
 
53
        self.transport_readonly_server = http_server.HttpServer
 
54
        self.make_branch('source')
 
55
        public_url = self.get_readonly_url('target')
 
56
        self.run_bzr_error(['http does not support mkdir'],
 
57
                           ['push', public_url],
 
58
                           working_dir='source')
 
59
 
 
60
    def test_push_suggests_parent_alias(self):
 
61
        """Push suggests using :parent if there is a known parent branch."""
 
62
        tree_a = self.make_branch_and_tree('a')
 
63
        tree_a.commit('this is a commit')
 
64
        tree_b = self.make_branch_and_tree('b')
 
65
 
 
66
        # If there is no parent location set, :parent isn't mentioned.
 
67
        out = self.run_bzr('push', working_dir='a', retcode=3)
 
68
        self.assertEqual(out,
 
69
                ('','bzr: ERROR: No push location known or specified.\n'))
 
70
 
 
71
        # If there is a parent location set, the error suggests :parent.
 
72
        tree_a.branch.set_parent(tree_b.branch.base)
 
73
        out = self.run_bzr('push', working_dir='a', retcode=3)
 
74
        self.assertEqual(out,
 
75
            ('','bzr: ERROR: No push location known or specified. '
 
76
                'To push to the parent branch '
 
77
                '(at %s), use \'bzr push :parent\'.\n' %
 
78
                urlutils.unescape_for_display(tree_b.branch.base, 'utf-8')))
 
79
 
 
80
    def test_push_remember(self):
 
81
        """Push changes from one branch to another and test push location."""
 
82
        transport = self.get_transport()
 
83
        tree_a = self.make_branch_and_tree('branch_a')
 
84
        branch_a = tree_a.branch
 
85
        self.build_tree(['branch_a/a'])
 
86
        tree_a.add('a')
 
87
        tree_a.commit('commit a')
 
88
        tree_b = branch_a.bzrdir.sprout('branch_b').open_workingtree()
 
89
        branch_b = tree_b.branch
 
90
        tree_c = branch_a.bzrdir.sprout('branch_c').open_workingtree()
 
91
        branch_c = tree_c.branch
 
92
        self.build_tree(['branch_a/b'])
 
93
        tree_a.add('b')
 
94
        tree_a.commit('commit b')
 
95
        self.build_tree(['branch_b/c'])
 
96
        tree_b.add('c')
 
97
        tree_b.commit('commit c')
 
98
        # initial push location must be empty
 
99
        self.assertEqual(None, branch_b.get_push_location())
 
100
 
 
101
        # test push for failure without push location set
 
102
        out = self.run_bzr('push', working_dir='branch_a', retcode=3)
 
103
        self.assertEqual(out,
 
104
                ('','bzr: ERROR: No push location known or specified.\n'))
 
105
 
 
106
        # test not remembered if cannot actually push
 
107
        self.run_bzr('push path/which/doesnt/exist',
 
108
                     working_dir='branch_a', retcode=3)
 
109
        out = self.run_bzr('push', working_dir='branch_a', retcode=3)
 
110
        self.assertEqual(
 
111
                ('', 'bzr: ERROR: No push location known or specified.\n'),
 
112
                out)
 
113
 
 
114
        # test implicit --remember when no push location set, push fails
 
115
        out = self.run_bzr('push ../branch_b',
 
116
                           working_dir='branch_a', retcode=3)
 
117
        self.assertEqual(out,
 
118
                ('','bzr: ERROR: These branches have diverged.  '
 
119
                 'See "bzr help diverged-branches" for more information.\n'))
 
120
        # Refresh the branch as 'push' modified it
 
121
        branch_a = branch_a.bzrdir.open_branch()
 
122
        self.assertEqual(osutils.abspath(branch_a.get_push_location()),
 
123
                          osutils.abspath(branch_b.bzrdir.root_transport.base))
 
124
 
 
125
        # test implicit --remember after resolving previous failure
 
126
        uncommit.uncommit(branch=branch_b, tree=tree_b)
 
127
        transport.delete('branch_b/c')
 
128
        out, err = self.run_bzr('push', working_dir='branch_a')
 
129
        # Refresh the branch as 'push' modified it
 
130
        branch_a = branch_a.bzrdir.open_branch()
 
131
        path = branch_a.get_push_location()
 
132
        self.assertEqual(err,
 
133
                         'Using saved push location: %s\n'
 
134
                         'All changes applied successfully.\n'
 
135
                         'Pushed up to revision 2.\n'
 
136
                         % urlutils.local_path_from_url(path))
 
137
        self.assertEqual(path,
 
138
                         branch_b.bzrdir.root_transport.base)
 
139
        # test explicit --remember
 
140
        self.run_bzr('push ../branch_c --remember', working_dir='branch_a')
 
141
        # Refresh the branch as 'push' modified it
 
142
        branch_a = branch_a.bzrdir.open_branch()
 
143
        self.assertEqual(branch_a.get_push_location(),
 
144
                          branch_c.bzrdir.root_transport.base)
 
145
 
 
146
    def test_push_without_tree(self):
 
147
        # bzr push from a branch that does not have a checkout should work.
 
148
        b = self.make_branch('.')
 
149
        out, err = self.run_bzr('push pushed-location')
 
150
        self.assertEqual('', out)
 
151
        self.assertEqual('Created new branch.\n', err)
 
152
        b2 = branch.Branch.open('pushed-location')
 
153
        self.assertEndsWith(b2.base, 'pushed-location/')
 
154
 
 
155
    def test_push_no_tree(self):
 
156
        # bzr push --no-tree of a branch with working trees
 
157
        b = self.make_branch_and_tree('push-from')
 
158
        self.build_tree(['push-from/file'])
 
159
        b.add('file')
 
160
        b.commit('commit 1')
 
161
        out, err = self.run_bzr('push --no-tree -d push-from push-to')
 
162
        self.assertEqual('', out)
 
163
        self.assertEqual('Created new branch.\n', err)
 
164
        self.assertPathDoesNotExist('push-to/file')
 
165
 
 
166
    def test_push_new_branch_revision_count(self):
 
167
        # bzr push of a branch with revisions to a new location
 
168
        # should print the number of revisions equal to the length of the
 
169
        # local branch.
 
170
        t = self.make_branch_and_tree('tree')
 
171
        self.build_tree(['tree/file'])
 
172
        t.add('file')
 
173
        t.commit('commit 1')
 
174
        out, err = self.run_bzr('push -d tree pushed-to')
 
175
        self.assertEqual('', out)
 
176
        self.assertEqual('Created new branch.\n', err)
 
177
 
 
178
    def test_push_quiet(self):
 
179
        # test that using -q makes output quiet
 
180
        t = self.make_branch_and_tree('tree')
 
181
        self.build_tree(['tree/file'])
 
182
        t.add('file')
 
183
        t.commit('commit 1')
 
184
        self.run_bzr('push -d tree pushed-to')
 
185
        # Refresh the branch as 'push' modified it and get the push location
 
186
        push_loc = t.branch.bzrdir.open_branch().get_push_location()
 
187
        out, err = self.run_bzr('push', working_dir="tree")
 
188
        self.assertEqual('Using saved push location: %s\n'
 
189
                         'No new revisions or tags to push.\n' %
 
190
                         urlutils.local_path_from_url(push_loc), err)
 
191
        out, err = self.run_bzr('push -q', working_dir="tree")
 
192
        self.assertEqual('', out)
 
193
        self.assertEqual('', err)
 
194
 
 
195
    def test_push_only_pushes_history(self):
 
196
        # Knit branches should only push the history for the current revision.
 
197
        format = bzrdir.BzrDirMetaFormat1()
 
198
        format.repository_format = knitrepo.RepositoryFormatKnit1()
 
199
        shared_repo = self.make_repository('repo', format=format, shared=True)
 
200
        shared_repo.set_make_working_trees(True)
 
201
 
 
202
        def make_shared_tree(path):
 
203
            shared_repo.bzrdir.root_transport.mkdir(path)
 
204
            controldir.ControlDir.create_branch_convenience('repo/' + path)
 
205
            return workingtree.WorkingTree.open('repo/' + path)
 
206
        tree_a = make_shared_tree('a')
 
207
        self.build_tree(['repo/a/file'])
 
208
        tree_a.add('file')
 
209
        tree_a.commit('commit a-1', rev_id='a-1')
 
210
        f = open('repo/a/file', 'ab')
 
211
        f.write('more stuff\n')
 
212
        f.close()
 
213
        tree_a.commit('commit a-2', rev_id='a-2')
 
214
 
 
215
        tree_b = make_shared_tree('b')
 
216
        self.build_tree(['repo/b/file'])
 
217
        tree_b.add('file')
 
218
        tree_b.commit('commit b-1', rev_id='b-1')
 
219
 
 
220
        self.assertTrue(shared_repo.has_revision('a-1'))
 
221
        self.assertTrue(shared_repo.has_revision('a-2'))
 
222
        self.assertTrue(shared_repo.has_revision('b-1'))
 
223
 
 
224
        # Now that we have a repository with shared files, make sure
 
225
        # that things aren't copied out by a 'push'
 
226
        self.run_bzr('push ../../push-b', working_dir='repo/b')
 
227
        pushed_tree = workingtree.WorkingTree.open('push-b')
 
228
        pushed_repo = pushed_tree.branch.repository
 
229
        self.assertFalse(pushed_repo.has_revision('a-1'))
 
230
        self.assertFalse(pushed_repo.has_revision('a-2'))
 
231
        self.assertTrue(pushed_repo.has_revision('b-1'))
 
232
 
 
233
    def test_push_funky_id(self):
 
234
        t = self.make_branch_and_tree('tree')
 
235
        self.build_tree(['tree/filename'])
 
236
        t.add('filename', 'funky-chars<>%&;"\'')
 
237
        t.commit('commit filename')
 
238
        self.run_bzr('push -d tree new-tree')
 
239
 
 
240
    def test_push_dash_d(self):
 
241
        t = self.make_branch_and_tree('from')
 
242
        t.commit(allow_pointless=True,
 
243
                message='first commit')
 
244
        self.run_bzr('push -d from to-one')
 
245
        self.assertPathExists('to-one')
 
246
        self.run_bzr('push -d %s %s'
 
247
            % tuple(map(urlutils.local_path_to_url, ['from', 'to-two'])))
 
248
        self.assertPathExists('to-two')
 
249
 
 
250
    def test_push_repository_no_branch_doesnt_fetch_all_revs(self):
 
251
        # See https://bugs.launchpad.net/bzr/+bug/465517
 
252
        target_repo = self.make_repository('target')
 
253
        source = self.make_branch_builder('source')
 
254
        source.start_series()
 
255
        source.build_snapshot('A', None, [
 
256
            ('add', ('', 'root-id', 'directory', None))])
 
257
        source.build_snapshot('B', ['A'], [])
 
258
        source.build_snapshot('C', ['A'], [])
 
259
        source.finish_series()
 
260
        self.run_bzr('push target -d source')
 
261
        self.addCleanup(target_repo.lock_read().unlock)
 
262
        # We should have pushed 'C', but not 'B', since it isn't in the
 
263
        # ancestry
 
264
        self.assertEqual([('A',), ('C',)], sorted(target_repo.revisions.keys()))
 
265
 
 
266
    def test_push_smart_non_stacked_streaming_acceptance(self):
 
267
        self.setup_smart_server_with_call_log()
 
268
        t = self.make_branch_and_tree('from')
 
269
        t.commit(allow_pointless=True, message='first commit')
 
270
        self.reset_smart_call_log()
 
271
        self.run_bzr(['push', self.get_url('to-one')], working_dir='from')
 
272
        # This figure represent the amount of work to perform this use case. It
 
273
        # is entirely ok to reduce this number if a test fails due to rpc_count
 
274
        # being too low. If rpc_count increases, more network roundtrips have
 
275
        # become necessary for this use case. Please do not adjust this number
 
276
        # upwards without agreement from bzr's network support maintainers.
 
277
        self.assertLength(9, self.hpss_calls)
 
278
        self.assertLength(1, self.hpss_connections)
 
279
        self.assertThat(self.hpss_calls, ContainsNoVfsCalls)
 
280
 
 
281
    def test_push_smart_stacked_streaming_acceptance(self):
 
282
        self.setup_smart_server_with_call_log()
 
283
        parent = self.make_branch_and_tree('parent', format='1.9')
 
284
        parent.commit(message='first commit')
 
285
        local = parent.bzrdir.sprout('local').open_workingtree()
 
286
        local.commit(message='local commit')
 
287
        self.reset_smart_call_log()
 
288
        self.run_bzr(['push', '--stacked', '--stacked-on', '../parent',
 
289
            self.get_url('public')], working_dir='local')
 
290
        # This figure represent the amount of work to perform this use case. It
 
291
        # is entirely ok to reduce this number if a test fails due to rpc_count
 
292
        # being too low. If rpc_count increases, more network roundtrips have
 
293
        # become necessary for this use case. Please do not adjust this number
 
294
        # upwards without agreement from bzr's network support maintainers.
 
295
        self.assertLength(15, self.hpss_calls)
 
296
        self.assertLength(1, self.hpss_connections)
 
297
        self.assertThat(self.hpss_calls, ContainsNoVfsCalls)
 
298
        remote = branch.Branch.open('public')
 
299
        self.assertEndsWith(remote.get_stacked_on_url(), '/parent')
 
300
 
 
301
    def test_push_smart_tags_streaming_acceptance(self):
 
302
        self.setup_smart_server_with_call_log()
 
303
        t = self.make_branch_and_tree('from')
 
304
        rev_id = t.commit(allow_pointless=True, message='first commit')
 
305
        t.branch.tags.set_tag('new-tag', rev_id)
 
306
        self.reset_smart_call_log()
 
307
        self.run_bzr(['push', self.get_url('to-one')], working_dir='from')
 
308
        # This figure represent the amount of work to perform this use case. It
 
309
        # is entirely ok to reduce this number if a test fails due to rpc_count
 
310
        # being too low. If rpc_count increases, more network roundtrips have
 
311
        # become necessary for this use case. Please do not adjust this number
 
312
        # upwards without agreement from bzr's network support maintainers.
 
313
        self.assertLength(11, self.hpss_calls)
 
314
        self.assertLength(1, self.hpss_connections)
 
315
        self.assertThat(self.hpss_calls, ContainsNoVfsCalls)
 
316
 
 
317
    def test_push_smart_incremental_acceptance(self):
 
318
        self.setup_smart_server_with_call_log()
 
319
        t = self.make_branch_and_tree('from')
 
320
        rev_id1 = t.commit(allow_pointless=True, message='first commit')
 
321
        rev_id2 = t.commit(allow_pointless=True, message='second commit')
 
322
        self.run_bzr(
 
323
            ['push', self.get_url('to-one'), '-r1'], working_dir='from')
 
324
        self.reset_smart_call_log()
 
325
        self.run_bzr(['push', self.get_url('to-one')], working_dir='from')
 
326
        # This figure represent the amount of work to perform this use case. It
 
327
        # is entirely ok to reduce this number if a test fails due to rpc_count
 
328
        # being too low. If rpc_count increases, more network roundtrips have
 
329
        # become necessary for this use case. Please do not adjust this number
 
330
        # upwards without agreement from bzr's network support maintainers.
 
331
        self.assertLength(11, self.hpss_calls)
 
332
        self.assertLength(1, self.hpss_connections)
 
333
        self.assertThat(self.hpss_calls, ContainsNoVfsCalls)
 
334
 
 
335
    def test_push_smart_with_default_stacking_url_path_segment(self):
 
336
        # If the default stacked-on location is a path element then branches
 
337
        # we push there over the smart server are stacked and their
 
338
        # stacked_on_url is that exact path segment. Added to nail bug 385132.
 
339
        self.setup_smart_server_with_call_log()
 
340
        self.make_branch('stack-on', format='1.9')
 
341
        self.make_bzrdir('.').get_config().set_default_stack_on(
 
342
            '/stack-on')
 
343
        self.make_branch('from', format='1.9')
 
344
        out, err = self.run_bzr(['push', '-d', 'from', self.get_url('to')])
 
345
        b = branch.Branch.open(self.get_url('to'))
 
346
        self.assertEqual('/extra/stack-on', b.get_stacked_on_url())
 
347
 
 
348
    def test_push_smart_with_default_stacking_relative_path(self):
 
349
        # If the default stacked-on location is a relative path then branches
 
350
        # we push there over the smart server are stacked and their
 
351
        # stacked_on_url is a relative path. Added to nail bug 385132.
 
352
        self.setup_smart_server_with_call_log()
 
353
        self.make_branch('stack-on', format='1.9')
 
354
        self.make_bzrdir('.').get_config().set_default_stack_on('stack-on')
 
355
        self.make_branch('from', format='1.9')
 
356
        out, err = self.run_bzr(['push', '-d', 'from', self.get_url('to')])
 
357
        b = branch.Branch.open(self.get_url('to'))
 
358
        self.assertEqual('../stack-on', b.get_stacked_on_url())
 
359
 
 
360
    def create_simple_tree(self):
 
361
        tree = self.make_branch_and_tree('tree')
 
362
        self.build_tree(['tree/a'])
 
363
        tree.add(['a'], ['a-id'])
 
364
        tree.commit('one', rev_id='r1')
 
365
        return tree
 
366
 
 
367
    def test_push_create_prefix(self):
 
368
        """'bzr push --create-prefix' will create leading directories."""
 
369
        tree = self.create_simple_tree()
 
370
 
 
371
        self.run_bzr_error(['Parent directory of ../new/tree does not exist'],
 
372
                           'push ../new/tree',
 
373
                           working_dir='tree')
 
374
        self.run_bzr('push ../new/tree --create-prefix',
 
375
                     working_dir='tree')
 
376
        new_tree = workingtree.WorkingTree.open('new/tree')
 
377
        self.assertEqual(tree.last_revision(), new_tree.last_revision())
 
378
        self.assertPathExists('new/tree/a')
 
379
 
 
380
    def test_push_use_existing(self):
 
381
        """'bzr push --use-existing-dir' can push into an existing dir.
 
382
 
 
383
        By default, 'bzr push' will not use an existing, non-versioned dir.
 
384
        """
 
385
        tree = self.create_simple_tree()
 
386
        self.build_tree(['target/'])
 
387
 
 
388
        self.run_bzr_error(['Target directory ../target already exists',
 
389
                            'Supply --use-existing-dir',
 
390
                           ],
 
391
                           'push ../target', working_dir='tree')
 
392
 
 
393
        self.run_bzr('push --use-existing-dir ../target',
 
394
                     working_dir='tree')
 
395
 
 
396
        new_tree = workingtree.WorkingTree.open('target')
 
397
        self.assertEqual(tree.last_revision(), new_tree.last_revision())
 
398
        # The push should have created target/a
 
399
        self.assertPathExists('target/a')
 
400
 
 
401
    def test_push_use_existing_into_empty_bzrdir(self):
 
402
        """'bzr push --use-existing-dir' into a dir with an empty .bzr dir
 
403
        fails.
 
404
        """
 
405
        tree = self.create_simple_tree()
 
406
        self.build_tree(['target/', 'target/.bzr/'])
 
407
        self.run_bzr_error(
 
408
            ['Target directory ../target already contains a .bzr directory, '
 
409
             'but it is not valid.'],
 
410
            'push ../target --use-existing-dir', working_dir='tree')
 
411
 
 
412
    def test_push_onto_repo(self):
 
413
        """We should be able to 'bzr push' into an existing bzrdir."""
 
414
        tree = self.create_simple_tree()
 
415
        repo = self.make_repository('repo', shared=True)
 
416
 
 
417
        self.run_bzr('push ../repo',
 
418
                     working_dir='tree')
 
419
 
 
420
        # Pushing onto an existing bzrdir will create a repository and
 
421
        # branch as needed, but will only create a working tree if there was
 
422
        # no BzrDir before.
 
423
        self.assertRaises(errors.NoWorkingTree,
 
424
                          workingtree.WorkingTree.open, 'repo')
 
425
        new_branch = branch.Branch.open('repo')
 
426
        self.assertEqual(tree.last_revision(), new_branch.last_revision())
 
427
 
 
428
    def test_push_onto_just_bzrdir(self):
 
429
        """We don't handle when the target is just a bzrdir.
 
430
 
 
431
        Because you shouldn't be able to create *just* a bzrdir in the wild.
 
432
        """
 
433
        # TODO: jam 20070109 Maybe it would be better to create the repository
 
434
        #       if at this point
 
435
        tree = self.create_simple_tree()
 
436
        a_bzrdir = self.make_bzrdir('dir')
 
437
 
 
438
        self.run_bzr_error(['At ../dir you have a valid .bzr control'],
 
439
                'push ../dir',
 
440
                working_dir='tree')
 
441
 
 
442
    def test_push_with_revisionspec(self):
 
443
        """We should be able to push a revision older than the tip."""
 
444
        tree_from = self.make_branch_and_tree('from')
 
445
        tree_from.commit("One.", rev_id="from-1")
 
446
        tree_from.commit("Two.", rev_id="from-2")
 
447
 
 
448
        self.run_bzr('push -r1 ../to', working_dir='from')
 
449
 
 
450
        tree_to = workingtree.WorkingTree.open('to')
 
451
        repo_to = tree_to.branch.repository
 
452
        self.assertTrue(repo_to.has_revision('from-1'))
 
453
        self.assertFalse(repo_to.has_revision('from-2'))
 
454
        self.assertEqual(tree_to.branch.last_revision_info()[1], 'from-1')
 
455
        self.assertFalse(
 
456
            tree_to.changes_from(tree_to.basis_tree()).has_changed())
 
457
 
 
458
        self.run_bzr_error(
 
459
            ['bzr: ERROR: bzr push --revision '
 
460
             'takes exactly one revision identifier\n'],
 
461
            'push -r0..2 ../to', working_dir='from')
 
462
 
 
463
    def create_trunk_and_feature_branch(self):
 
464
        # We have a mainline
 
465
        trunk_tree = self.make_branch_and_tree('target',
 
466
            format='1.9')
 
467
        trunk_tree.commit('mainline')
 
468
        # and a branch from it
 
469
        branch_tree = self.make_branch_and_tree('branch',
 
470
            format='1.9')
 
471
        branch_tree.pull(trunk_tree.branch)
 
472
        branch_tree.branch.set_parent(trunk_tree.branch.base)
 
473
        # with some work on it
 
474
        branch_tree.commit('moar work plz')
 
475
        return trunk_tree, branch_tree
 
476
 
 
477
    def assertPublished(self, branch_revid, stacked_on):
 
478
        """Assert that the branch 'published' has been published correctly."""
 
479
        published_branch = branch.Branch.open('published')
 
480
        # The published branch refers to the mainline
 
481
        self.assertEqual(stacked_on, published_branch.get_stacked_on_url())
 
482
        # and the branch's work was pushed
 
483
        self.assertTrue(published_branch.repository.has_revision(branch_revid))
 
484
 
 
485
    def test_push_new_branch_stacked_on(self):
 
486
        """Pushing a new branch with --stacked-on creates a stacked branch."""
 
487
        trunk_tree, branch_tree = self.create_trunk_and_feature_branch()
 
488
        # we publish branch_tree with a reference to the mainline.
 
489
        out, err = self.run_bzr(['push', '--stacked-on', trunk_tree.branch.base,
 
490
            self.get_url('published')], working_dir='branch')
 
491
        self.assertEqual('', out)
 
492
        self.assertEqual('Created new stacked branch referring to %s.\n' %
 
493
            trunk_tree.branch.base, err)
 
494
        self.assertPublished(branch_tree.last_revision(),
 
495
            trunk_tree.branch.base)
 
496
 
 
497
    def test_push_new_branch_stacked_uses_parent_when_no_public_url(self):
 
498
        """When the parent has no public url the parent is used as-is."""
 
499
        trunk_tree, branch_tree = self.create_trunk_and_feature_branch()
 
500
        # now we do a stacked push, which should determine the public location
 
501
        # for us.
 
502
        out, err = self.run_bzr(['push', '--stacked',
 
503
            self.get_url('published')], working_dir='branch')
 
504
        self.assertEqual('', out)
 
505
        self.assertEqual('Created new stacked branch referring to %s.\n' %
 
506
            trunk_tree.branch.base, err)
 
507
        self.assertPublished(branch_tree.last_revision(),
 
508
                             trunk_tree.branch.base)
 
509
 
 
510
    def test_push_new_branch_stacked_uses_parent_public(self):
 
511
        """Pushing a new branch with --stacked creates a stacked branch."""
 
512
        trunk_tree, branch_tree = self.create_trunk_and_feature_branch()
 
513
        # the trunk is published on a web server
 
514
        self.transport_readonly_server = http_server.HttpServer
 
515
        trunk_public = self.make_branch('public_trunk', format='1.9')
 
516
        trunk_public.pull(trunk_tree.branch)
 
517
        trunk_public_url = self.get_readonly_url('public_trunk')
 
518
        br = trunk_tree.branch
 
519
        br.set_public_branch(trunk_public_url)
 
520
        # now we do a stacked push, which should determine the public location
 
521
        # for us.
 
522
        out, err = self.run_bzr(['push', '--stacked',
 
523
            self.get_url('published')], working_dir='branch')
 
524
        self.assertEqual('', out)
 
525
        self.assertEqual('Created new stacked branch referring to %s.\n' %
 
526
            trunk_public_url, err)
 
527
        self.assertPublished(branch_tree.last_revision(), trunk_public_url)
 
528
 
 
529
    def test_push_new_branch_stacked_no_parent(self):
 
530
        """Pushing with --stacked and no parent branch errors."""
 
531
        branch = self.make_branch_and_tree('branch', format='1.9')
 
532
        # now we do a stacked push, which should fail as the place to refer too
 
533
        # cannot be determined.
 
534
        out, err = self.run_bzr_error(
 
535
            ['Could not determine branch to refer to\\.'], ['push', '--stacked',
 
536
            self.get_url('published')], working_dir='branch')
 
537
        self.assertEqual('', out)
 
538
        self.assertFalse(self.get_transport('published').has('.'))
 
539
 
 
540
    def test_push_notifies_default_stacking(self):
 
541
        self.make_branch('stack_on', format='1.6')
 
542
        self.make_bzrdir('.').get_config().set_default_stack_on('stack_on')
 
543
        self.make_branch('from', format='1.6')
 
544
        out, err = self.run_bzr('push -d from to')
 
545
        self.assertContainsRe(err,
 
546
                              'Using default stacking branch stack_on at .*')
 
547
 
 
548
    def test_push_stacks_with_default_stacking_if_target_is_stackable(self):
 
549
        self.make_branch('stack_on', format='1.6')
 
550
        self.make_bzrdir('.').get_config().set_default_stack_on('stack_on')
 
551
        self.make_branch('from', format='pack-0.92')
 
552
        out, err = self.run_bzr('push -d from to')
 
553
        b = branch.Branch.open('to')
 
554
        self.assertEqual('../stack_on', b.get_stacked_on_url())
 
555
 
 
556
    def test_push_does_not_change_format_with_default_if_target_cannot(self):
 
557
        self.make_branch('stack_on', format='pack-0.92')
 
558
        self.make_bzrdir('.').get_config().set_default_stack_on('stack_on')
 
559
        self.make_branch('from', format='pack-0.92')
 
560
        out, err = self.run_bzr('push -d from to')
 
561
        b = branch.Branch.open('to')
 
562
        self.assertRaises(errors.UnstackableBranchFormat, b.get_stacked_on_url)
 
563
 
 
564
    def test_push_doesnt_create_broken_branch(self):
 
565
        """Pushing a new standalone branch works even when there's a default
 
566
        stacking policy at the destination.
 
567
 
 
568
        The new branch will preserve the repo format (even if it isn't the
 
569
        default for the branch), and will be stacked when the repo format
 
570
        allows (which means that the branch format isn't necessarly preserved).
 
571
        """
 
572
        self.make_repository('repo', shared=True, format='1.6')
 
573
        builder = self.make_branch_builder('repo/local', format='pack-0.92')
 
574
        builder.start_series()
 
575
        builder.build_snapshot('rev-1', None, [
 
576
            ('add', ('', 'root-id', 'directory', '')),
 
577
            ('add', ('filename', 'f-id', 'file', 'content\n'))])
 
578
        builder.build_snapshot('rev-2', ['rev-1'], [])
 
579
        builder.build_snapshot('rev-3', ['rev-2'],
 
580
            [('modify', ('f-id', 'new-content\n'))])
 
581
        builder.finish_series()
 
582
        branch = builder.get_branch()
 
583
        # Push rev-1 to "trunk", so that we can stack on it.
 
584
        self.run_bzr('push -d repo/local trunk -r 1')
 
585
        # Set a default stacking policy so that new branches will automatically
 
586
        # stack on trunk.
 
587
        self.make_bzrdir('.').get_config().set_default_stack_on('trunk')
 
588
        # Push rev-2 to a new branch "remote".  It will be stacked on "trunk".
 
589
        out, err = self.run_bzr('push -d repo/local remote -r 2')
 
590
        self.assertContainsRe(
 
591
            err, 'Using default stacking branch trunk at .*')
 
592
        # Push rev-3 onto "remote".  If "remote" not stacked and is missing the
 
593
        # fulltext record for f-id @ rev-1, then this will fail.
 
594
        out, err = self.run_bzr('push -d repo/local remote -r 3')
 
595
 
 
596
    def test_push_verbose_shows_log(self):
 
597
        tree = self.make_branch_and_tree('source')
 
598
        tree.commit('rev1')
 
599
        out, err = self.run_bzr('push -v -d source target')
 
600
        # initial push contains log
 
601
        self.assertContainsRe(out, 'rev1')
 
602
        tree.commit('rev2')
 
603
        out, err = self.run_bzr('push -v -d source target')
 
604
        # subsequent push contains log
 
605
        self.assertContainsRe(out, 'rev2')
 
606
        # subsequent log is accurate
 
607
        self.assertNotContainsRe(out, 'rev1')
 
608
 
 
609
    def test_push_from_subdir(self):
 
610
        t = self.make_branch_and_tree('tree')
 
611
        self.build_tree(['tree/dir/', 'tree/dir/file'])
 
612
        t.add('dir', 'dir/file')
 
613
        t.commit('r1')
 
614
        out, err = self.run_bzr('push ../../pushloc', working_dir='tree/dir')
 
615
        self.assertEqual('', out)
 
616
        self.assertEqual('Created new branch.\n', err)
 
617
 
 
618
    def test_overwrite_tags(self):
 
619
        """--overwrite-tags only overwrites tags, not revisions."""
 
620
        from_tree = self.make_branch_and_tree('from')
 
621
        from_tree.branch.tags.set_tag("mytag", "somerevid")
 
622
        to_tree = self.make_branch_and_tree('to')
 
623
        to_tree.branch.tags.set_tag("mytag", "anotherrevid")
 
624
        revid1 = to_tree.commit('my commit')
 
625
        out = self.run_bzr(['push', '-d', 'from', 'to'])
 
626
        self.assertEqual(out,
 
627
            ('Conflicting tags:\n    mytag\n', 'No new revisions to push.\n'))
 
628
        out = self.run_bzr(['push', '-d', 'from', '--overwrite-tags', 'to'])
 
629
        self.assertEqual(out, ('', '1 tag updated.\n'))
 
630
        self.assertEqual(to_tree.branch.tags.lookup_tag('mytag'),
 
631
                          'somerevid')
 
632
        self.assertEqual(to_tree.branch.last_revision(), revid1)
 
633
 
 
634
 
 
635
class RedirectingMemoryTransport(memory.MemoryTransport):
 
636
 
 
637
    def mkdir(self, relpath, mode=None):
 
638
        if self._cwd == '/source/':
 
639
            raise errors.RedirectRequested(self.abspath(relpath),
 
640
                                           self.abspath('../target'),
 
641
                                           is_permanent=True)
 
642
        elif self._cwd == '/infinite-loop/':
 
643
            raise errors.RedirectRequested(self.abspath(relpath),
 
644
                                           self.abspath('../infinite-loop'),
 
645
                                           is_permanent=True)
 
646
        else:
 
647
            return super(RedirectingMemoryTransport, self).mkdir(
 
648
                relpath, mode)
 
649
 
 
650
    def get(self, relpath):
 
651
        if self.clone(relpath)._cwd == '/infinite-loop/':
 
652
            raise errors.RedirectRequested(self.abspath(relpath),
 
653
                                           self.abspath('../infinite-loop'),
 
654
                                           is_permanent=True)
 
655
        else:
 
656
            return super(RedirectingMemoryTransport, self).get(relpath)
 
657
 
 
658
    def _redirected_to(self, source, target):
 
659
        # We do accept redirections
 
660
        return transport.get_transport(target)
 
661
 
 
662
 
 
663
class RedirectingMemoryServer(memory.MemoryServer):
 
664
 
 
665
    def start_server(self):
 
666
        self._dirs = {'/': None}
 
667
        self._files = {}
 
668
        self._locks = {}
 
669
        self._scheme = 'redirecting-memory+%s:///' % id(self)
 
670
        transport.register_transport(self._scheme, self._memory_factory)
 
671
 
 
672
    def _memory_factory(self, url):
 
673
        result = RedirectingMemoryTransport(url)
 
674
        result._dirs = self._dirs
 
675
        result._files = self._files
 
676
        result._locks = self._locks
 
677
        return result
 
678
 
 
679
    def stop_server(self):
 
680
        transport.unregister_transport(self._scheme, self._memory_factory)
 
681
 
 
682
 
 
683
class TestPushRedirect(tests.TestCaseWithTransport):
 
684
 
 
685
    def setUp(self):
 
686
        super(TestPushRedirect, self).setUp()
 
687
        self.memory_server = RedirectingMemoryServer()
 
688
        self.start_server(self.memory_server)
 
689
        # Make the branch and tree that we'll be pushing.
 
690
        t = self.make_branch_and_tree('tree')
 
691
        self.build_tree(['tree/file'])
 
692
        t.add('file')
 
693
        t.commit('commit 1')
 
694
 
 
695
    def test_push_redirects_on_mkdir(self):
 
696
        """If the push requires a mkdir, push respects redirect requests.
 
697
 
 
698
        This is added primarily to handle lp:/ URI support, so that users can
 
699
        push to new branches by specifying lp:/ URIs.
 
700
        """
 
701
        destination_url = self.memory_server.get_url() + 'source'
 
702
        self.run_bzr(['push', '-d', 'tree', destination_url])
 
703
 
 
704
        local_revision = branch.Branch.open('tree').last_revision()
 
705
        remote_revision = branch.Branch.open(
 
706
            self.memory_server.get_url() + 'target').last_revision()
 
707
        self.assertEqual(remote_revision, local_revision)
 
708
 
 
709
    def test_push_gracefully_handles_too_many_redirects(self):
 
710
        """Push fails gracefully if the mkdir generates a large number of
 
711
        redirects.
 
712
        """
 
713
        destination_url = self.memory_server.get_url() + 'infinite-loop'
 
714
        out, err = self.run_bzr_error(
 
715
            ['Too many redirections trying to make %s\\.\n'
 
716
             % re.escape(destination_url)],
 
717
            ['push', '-d', 'tree', destination_url], retcode=3)
 
718
        self.assertEqual('', out)
 
719
 
 
720
 
 
721
class TestPushStrictMixin(object):
 
722
 
 
723
    def make_local_branch_and_tree(self):
 
724
        self.tree = self.make_branch_and_tree('local')
 
725
        self.build_tree_contents([('local/file', 'initial')])
 
726
        self.tree.add('file')
 
727
        self.tree.commit('adding file', rev_id='added')
 
728
        self.build_tree_contents([('local/file', 'modified')])
 
729
        self.tree.commit('modify file', rev_id='modified')
 
730
 
 
731
    def set_config_push_strict(self, value):
 
732
        br = branch.Branch.open('local')
 
733
        br.get_config_stack().set('push_strict', value)
 
734
 
 
735
    _default_command = ['push', '../to']
 
736
    _default_wd = 'local'
 
737
    _default_errors = ['Working tree ".*/local/" has uncommitted '
 
738
                       'changes \(See bzr status\)\.',]
 
739
    _default_additional_error = 'Use --no-strict to force the push.\n'
 
740
    _default_additional_warning = 'Uncommitted changes will not be pushed.'
 
741
 
 
742
 
 
743
    def assertPushFails(self, args):
 
744
        out, err = self.run_bzr_error(self._default_errors,
 
745
                                      self._default_command + args,
 
746
                                      working_dir=self._default_wd, retcode=3)
 
747
        self.assertContainsRe(err, self._default_additional_error)
 
748
 
 
749
    def assertPushSucceeds(self, args, with_warning=False, revid_to_push=None):
 
750
        if with_warning:
 
751
            error_regexes = self._default_errors
 
752
        else:
 
753
            error_regexes = []
 
754
        out, err = self.run_bzr(self._default_command + args,
 
755
                                working_dir=self._default_wd,
 
756
                                error_regexes=error_regexes)
 
757
        if with_warning:
 
758
            self.assertContainsRe(err, self._default_additional_warning)
 
759
        else:
 
760
            self.assertNotContainsRe(err, self._default_additional_warning)
 
761
        branch_from = branch.Branch.open(self._default_wd)
 
762
        if revid_to_push is None:
 
763
            revid_to_push = branch_from.last_revision()
 
764
        branch_to = branch.Branch.open('to')
 
765
        repo_to = branch_to.repository
 
766
        self.assertTrue(repo_to.has_revision(revid_to_push))
 
767
        self.assertEqual(revid_to_push, branch_to.last_revision())
 
768
 
 
769
 
 
770
 
 
771
class TestPushStrictWithoutChanges(tests.TestCaseWithTransport,
 
772
                                   TestPushStrictMixin):
 
773
 
 
774
    def setUp(self):
 
775
        super(TestPushStrictWithoutChanges, self).setUp()
 
776
        self.make_local_branch_and_tree()
 
777
 
 
778
    def test_push_default(self):
 
779
        self.assertPushSucceeds([])
 
780
 
 
781
    def test_push_strict(self):
 
782
        self.assertPushSucceeds(['--strict'])
 
783
 
 
784
    def test_push_no_strict(self):
 
785
        self.assertPushSucceeds(['--no-strict'])
 
786
 
 
787
    def test_push_config_var_strict(self):
 
788
        self.set_config_push_strict('true')
 
789
        self.assertPushSucceeds([])
 
790
 
 
791
    def test_push_config_var_no_strict(self):
 
792
        self.set_config_push_strict('false')
 
793
        self.assertPushSucceeds([])
 
794
 
 
795
 
 
796
strict_push_change_scenarios = [
 
797
    ('uncommitted',
 
798
        dict(_changes_type= '_uncommitted_changes')),
 
799
    ('pending-merges',
 
800
        dict(_changes_type= '_pending_merges')),
 
801
    ('out-of-sync-trees',
 
802
        dict(_changes_type= '_out_of_sync_trees')),
 
803
    ]
 
804
 
 
805
 
 
806
class TestPushStrictWithChanges(tests.TestCaseWithTransport,
 
807
                                TestPushStrictMixin):
 
808
 
 
809
    scenarios = strict_push_change_scenarios 
 
810
    _changes_type = None # Set by load_tests
 
811
 
 
812
    def setUp(self):
 
813
        super(TestPushStrictWithChanges, self).setUp()
 
814
        # Apply the changes defined in load_tests: one of _uncommitted_changes,
 
815
        # _pending_merges or _out_of_sync_trees
 
816
        getattr(self, self._changes_type)()
 
817
 
 
818
    def _uncommitted_changes(self):
 
819
        self.make_local_branch_and_tree()
 
820
        # Make a change without committing it
 
821
        self.build_tree_contents([('local/file', 'in progress')])
 
822
 
 
823
    def _pending_merges(self):
 
824
        self.make_local_branch_and_tree()
 
825
        # Create 'other' branch containing a new file
 
826
        other_bzrdir = self.tree.bzrdir.sprout('other')
 
827
        other_tree = other_bzrdir.open_workingtree()
 
828
        self.build_tree_contents([('other/other-file', 'other')])
 
829
        other_tree.add('other-file')
 
830
        other_tree.commit('other commit', rev_id='other')
 
831
        # Merge and revert, leaving a pending merge
 
832
        self.tree.merge_from_branch(other_tree.branch)
 
833
        self.tree.revert(filenames=['other-file'], backups=False)
 
834
 
 
835
    def _out_of_sync_trees(self):
 
836
        self.make_local_branch_and_tree()
 
837
        self.run_bzr(['checkout', '--lightweight', 'local', 'checkout'])
 
838
        # Make a change and commit it
 
839
        self.build_tree_contents([('local/file', 'modified in local')])
 
840
        self.tree.commit('modify file', rev_id='modified-in-local')
 
841
        # Exercise commands from the checkout directory
 
842
        self._default_wd = 'checkout'
 
843
        self._default_errors = ["Working tree is out of date, please run"
 
844
                                " 'bzr update'\.",]
 
845
 
 
846
    def test_push_default(self):
 
847
        self.assertPushSucceeds([], with_warning=True)
 
848
 
 
849
    def test_push_with_revision(self):
 
850
        self.assertPushSucceeds(['-r', 'revid:added'], revid_to_push='added')
 
851
 
 
852
    def test_push_no_strict(self):
 
853
        self.assertPushSucceeds(['--no-strict'])
 
854
 
 
855
    def test_push_strict_with_changes(self):
 
856
        self.assertPushFails(['--strict'])
 
857
 
 
858
    def test_push_respect_config_var_strict(self):
 
859
        self.set_config_push_strict('true')
 
860
        self.assertPushFails([])
 
861
 
 
862
    def test_push_bogus_config_var_ignored(self):
 
863
        self.set_config_push_strict("I don't want you to be strict")
 
864
        self.assertPushSucceeds([], with_warning=True)
 
865
 
 
866
    def test_push_no_strict_command_line_override_config(self):
 
867
        self.set_config_push_strict('yES')
 
868
        self.assertPushFails([])
 
869
        self.assertPushSucceeds(['--no-strict'])
 
870
 
 
871
    def test_push_strict_command_line_override_config(self):
 
872
        self.set_config_push_strict('oFF')
 
873
        self.assertPushFails(['--strict'])
 
874
        self.assertPushSucceeds([])
 
875
 
 
876
 
 
877
class TestPushForeign(tests.TestCaseWithTransport):
 
878
 
 
879
    def setUp(self):
 
880
        super(TestPushForeign, self).setUp()
 
881
        test_foreign.register_dummy_foreign_for_test(self)
 
882
 
 
883
    def make_dummy_builder(self, relpath):
 
884
        builder = self.make_branch_builder(
 
885
            relpath, format=test_foreign.DummyForeignVcsDirFormat())
 
886
        builder.build_snapshot('revid', None,
 
887
            [('add', ('', 'TREE_ROOT', 'directory', None)),
 
888
             ('add', ('foo', 'fooid', 'file', 'bar'))])
 
889
        return builder
 
890
 
 
891
    def test_no_roundtripping(self):
 
892
        target_branch = self.make_dummy_builder('dp').get_branch()
 
893
        source_tree = self.make_branch_and_tree("dc")
 
894
        output, error = self.run_bzr("push -d dc dp", retcode=3)
 
895
        self.assertEqual("", output)
 
896
        self.assertEqual(error, "bzr: ERROR: It is not possible to losslessly"
 
897
            " push to dummy. You may want to use dpush instead.\n")
 
898
 
 
899
 
 
900
class TestPushOutput(script.TestCaseWithTransportAndScript):
 
901
 
 
902
    def test_push_log_format(self):
 
903
        self.run_script("""
 
904
            $ bzr init trunk
 
905
            Created a standalone tree (format: 2a)
 
906
            $ cd trunk
 
907
            $ echo foo > file
 
908
            $ bzr add
 
909
            adding file
 
910
            $ bzr commit -m 'we need some foo'
 
911
            2>Committing to:...trunk/
 
912
            2>added file
 
913
            2>Committed revision 1.
 
914
            $ bzr init ../feature
 
915
            Created a standalone tree (format: 2a)
 
916
            $ bzr push -v ../feature -Olog_format=line
 
917
            Added Revisions:
 
918
            1: jrandom@example.com ...we need some foo
 
919
            2>All changes applied successfully.
 
920
            2>Pushed up to revision 1.
 
921
            """)
 
922
 
 
923
    def test_push_with_revspec(self):
 
924
        self.run_script("""
 
925
            $ bzr init-repo .
 
926
            Shared repository with trees (format: 2a)
 
927
            Location:
 
928
              shared repository: .
 
929
            $ bzr init trunk
 
930
            Created a repository tree (format: 2a)
 
931
            Using shared repository...
 
932
            $ cd trunk
 
933
            $ bzr commit -m 'first rev' --unchanged
 
934
            2>Committing to:...trunk/
 
935
            2>Committed revision 1.
 
936
            $ echo foo > file
 
937
            $ bzr add
 
938
            adding file
 
939
            $ bzr commit -m 'we need some foo'
 
940
            2>Committing to:...trunk/
 
941
            2>added file
 
942
            2>Committed revision 2.
 
943
            $ bzr push -r 1 ../other
 
944
            2>Created new branch.
 
945
            $ bzr st ../other # checking that file is not created (#484516)
 
946
            """)