~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

  • Committer: Martin Pool
  • Date: 2009-08-20 05:47:53 UTC
  • mto: This revision was merged to the branch mainline in revision 4632.
  • Revision ID: mbp@sourcefrog.net-20090820054753-yjio3skxpltbl32z
Remove unnecessary import

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2007 Canonical Ltd
 
1
# Copyright (C) 2005, 2007, 2008 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 push."""
19
19
 
20
 
import os
 
20
import re
21
21
 
22
22
from bzrlib import (
 
23
    branch,
 
24
    bzrdir,
23
25
    errors,
 
26
    osutils,
 
27
    tests,
 
28
    transport,
 
29
    uncommit,
24
30
    urlutils,
 
31
    workingtree
25
32
    )
26
 
from bzrlib.branch import Branch
27
 
from bzrlib.bzrdir import BzrDirMetaFormat1
28
 
from bzrlib.osutils import abspath
29
 
from bzrlib.repofmt.knitrepo import RepositoryFormatKnit1
30
 
from bzrlib.tests.blackbox import ExternalBase
31
 
from bzrlib.uncommit import uncommit
32
 
from bzrlib.urlutils import local_path_from_url
33
 
from bzrlib.workingtree import WorkingTree
34
 
 
35
 
 
36
 
class TestPush(ExternalBase):
 
33
from bzrlib.repofmt import knitrepo
 
34
from bzrlib.tests import http_server
 
35
from bzrlib.transport import memory
 
36
 
 
37
 
 
38
def load_tests(standard_tests, module, loader):
 
39
    """Multiply tests for the push command."""
 
40
    result = loader.suiteClass()
 
41
 
 
42
    # one for each king of change
 
43
    changes_tests, remaining_tests = tests.split_suite_by_condition(
 
44
        standard_tests, tests.condition_isinstance((
 
45
                TestPushStrictWithChanges,
 
46
                )))
 
47
    changes_scenarios = [
 
48
        ('uncommitted',
 
49
         dict(_changes_type= '_uncommitted_changes')),
 
50
        ('pending-merges',
 
51
         dict(_changes_type= '_pending_merges')),
 
52
        ('out-of-sync-trees',
 
53
         dict(_changes_type= '_out_of_sync_trees')),
 
54
        ]
 
55
    tests.multiply_tests(changes_tests, changes_scenarios, result)
 
56
    # No parametrization for the remaining tests
 
57
    result.addTests(remaining_tests)
 
58
 
 
59
    return result
 
60
 
 
61
 
 
62
class TestPush(tests.TestCaseWithTransport):
 
63
 
 
64
    def test_push_error_on_vfs_http(self):
 
65
        """ pushing a branch to a HTTP server fails cleanly. """
 
66
        # the trunk is published on a web server
 
67
        self.transport_readonly_server = http_server.HttpServer
 
68
        self.make_branch('source')
 
69
        public_url = self.get_readonly_url('target')
 
70
        self.run_bzr_error(['http does not support mkdir'],
 
71
                           ['push', public_url],
 
72
                           working_dir='source')
37
73
 
38
74
    def test_push_remember(self):
39
75
        """Push changes from one branch to another and test push location."""
57
93
        self.assertEqual(None, branch_b.get_push_location())
58
94
 
59
95
        # test push for failure without push location set
60
 
        os.chdir('branch_a')
61
 
        out = self.runbzr('push', retcode=3)
 
96
        out = self.run_bzr('push', working_dir='branch_a', retcode=3)
62
97
        self.assertEquals(out,
63
98
                ('','bzr: ERROR: No push location known or specified.\n'))
64
99
 
65
100
        # test not remembered if cannot actually push
66
 
        self.run_bzr('push', '../path/which/doesnt/exist', retcode=3)
67
 
        out = self.run_bzr('push', retcode=3)
 
101
        self.run_bzr('push path/which/doesnt/exist',
 
102
                     working_dir='branch_a', retcode=3)
 
103
        out = self.run_bzr('push', working_dir='branch_a', retcode=3)
68
104
        self.assertEquals(
69
105
                ('', 'bzr: ERROR: No push location known or specified.\n'),
70
106
                out)
71
107
 
72
108
        # test implicit --remember when no push location set, push fails
73
 
        out = self.run_bzr('push', '../branch_b', retcode=3)
 
109
        out = self.run_bzr('push ../branch_b',
 
110
                           working_dir='branch_a', retcode=3)
74
111
        self.assertEquals(out,
75
112
                ('','bzr: ERROR: These branches have diverged.  '
76
 
                    'Try using "merge" and then "push".\n'))
77
 
        self.assertEquals(abspath(branch_a.get_push_location()),
78
 
                          abspath(branch_b.bzrdir.root_transport.base))
 
113
                 'See "bzr help diverged-branches" for more information.\n'))
 
114
        self.assertEquals(osutils.abspath(branch_a.get_push_location()),
 
115
                          osutils.abspath(branch_b.bzrdir.root_transport.base))
79
116
 
80
117
        # test implicit --remember after resolving previous failure
81
 
        uncommit(branch=branch_b, tree=tree_b)
 
118
        uncommit.uncommit(branch=branch_b, tree=tree_b)
82
119
        transport.delete('branch_b/c')
83
 
        out, err = self.run_bzr('push')
 
120
        out, err = self.run_bzr('push', working_dir='branch_a')
84
121
        path = branch_a.get_push_location()
85
122
        self.assertEquals(out,
86
 
                          'Using saved location: %s\n' 
87
 
                          'Pushed up to revision 2.\n'
88
 
                          % local_path_from_url(path))
 
123
                          'Using saved push location: %s\n'
 
124
                          % urlutils.local_path_from_url(path))
89
125
        self.assertEqual(err,
90
 
                         'All changes applied successfully.\n')
 
126
                         'All changes applied successfully.\n'
 
127
                         'Pushed up to revision 2.\n')
91
128
        self.assertEqual(path,
92
129
                         branch_b.bzrdir.root_transport.base)
93
130
        # test explicit --remember
94
 
        self.run_bzr('push', '../branch_c', '--remember')
 
131
        self.run_bzr('push ../branch_c --remember', working_dir='branch_a')
95
132
        self.assertEquals(branch_a.get_push_location(),
96
133
                          branch_c.bzrdir.root_transport.base)
97
 
    
 
134
 
98
135
    def test_push_without_tree(self):
99
136
        # bzr push from a branch that does not have a checkout should work.
 
137
        self.thisFailsStrictLockCheck()
100
138
        b = self.make_branch('.')
101
 
        out, err = self.run_bzr('push', 'pushed-location')
 
139
        out, err = self.run_bzr('push pushed-location')
102
140
        self.assertEqual('', out)
103
141
        self.assertEqual('Created new branch.\n', err)
104
 
        b2 = Branch.open('pushed-location')
 
142
        b2 = branch.Branch.open('pushed-location')
105
143
        self.assertEndsWith(b2.base, 'pushed-location/')
106
144
 
107
145
    def test_push_new_branch_revision_count(self):
108
 
        # bzr push of a branch with revisions to a new location 
109
 
        # should print the number of revisions equal to the length of the 
 
146
        # bzr push of a branch with revisions to a new location
 
147
        # should print the number of revisions equal to the length of the
110
148
        # local branch.
 
149
        self.thisFailsStrictLockCheck()
111
150
        t = self.make_branch_and_tree('tree')
112
151
        self.build_tree(['tree/file'])
113
152
        t.add('file')
114
153
        t.commit('commit 1')
115
 
        os.chdir('tree')
116
 
        out, err = self.run_bzr('push', 'pushed-to')
117
 
        os.chdir('..')
 
154
        out, err = self.run_bzr('push -d tree pushed-to')
118
155
        self.assertEqual('', out)
119
156
        self.assertEqual('Created new branch.\n', err)
120
157
 
121
158
    def test_push_only_pushes_history(self):
122
159
        # Knit branches should only push the history for the current revision.
123
 
        format = BzrDirMetaFormat1()
124
 
        format.repository_format = RepositoryFormatKnit1()
 
160
        self.thisFailsStrictLockCheck()
 
161
        format = bzrdir.BzrDirMetaFormat1()
 
162
        format.repository_format = knitrepo.RepositoryFormatKnit1()
125
163
        shared_repo = self.make_repository('repo', format=format, shared=True)
126
164
        shared_repo.set_make_working_trees(True)
127
165
 
128
166
        def make_shared_tree(path):
129
167
            shared_repo.bzrdir.root_transport.mkdir(path)
130
168
            shared_repo.bzrdir.create_branch_convenience('repo/' + path)
131
 
            return WorkingTree.open('repo/' + path)
 
169
            return workingtree.WorkingTree.open('repo/' + path)
132
170
        tree_a = make_shared_tree('a')
133
171
        self.build_tree(['repo/a/file'])
134
172
        tree_a.add('file')
149
187
 
150
188
        # Now that we have a repository with shared files, make sure
151
189
        # that things aren't copied out by a 'push'
152
 
        os.chdir('repo/b')
153
 
        self.run_bzr('push', '../../push-b')
154
 
        pushed_tree = WorkingTree.open('../../push-b')
 
190
        self.run_bzr('push ../../push-b', working_dir='repo/b')
 
191
        pushed_tree = workingtree.WorkingTree.open('push-b')
155
192
        pushed_repo = pushed_tree.branch.repository
156
193
        self.assertFalse(pushed_repo.has_revision('a-1'))
157
194
        self.assertFalse(pushed_repo.has_revision('a-2'))
158
195
        self.assertTrue(pushed_repo.has_revision('b-1'))
159
196
 
160
197
    def test_push_funky_id(self):
 
198
        self.thisFailsStrictLockCheck()
161
199
        t = self.make_branch_and_tree('tree')
162
 
        os.chdir('tree')
163
 
        self.build_tree(['filename'])
 
200
        self.build_tree(['tree/filename'])
164
201
        t.add('filename', 'funky-chars<>%&;"\'')
165
202
        t.commit('commit filename')
166
 
        self.run_bzr('push', '../new-tree')
 
203
        self.run_bzr('push -d tree new-tree')
167
204
 
168
205
    def test_push_dash_d(self):
 
206
        self.thisFailsStrictLockCheck()
169
207
        t = self.make_branch_and_tree('from')
170
208
        t.commit(allow_pointless=True,
171
209
                message='first commit')
172
 
        self.runbzr('push -d from to-one')
 
210
        self.run_bzr('push -d from to-one')
173
211
        self.failUnlessExists('to-one')
174
 
        self.runbzr('push -d %s %s' 
 
212
        self.run_bzr('push -d %s %s'
175
213
            % tuple(map(urlutils.local_path_to_url, ['from', 'to-two'])))
176
214
        self.failUnlessExists('to-two')
177
215
 
 
216
    def test_push_smart_non_stacked_streaming_acceptance(self):
 
217
        self.setup_smart_server_with_call_log()
 
218
        t = self.make_branch_and_tree('from')
 
219
        t.commit(allow_pointless=True, message='first commit')
 
220
        self.reset_smart_call_log()
 
221
        self.run_bzr(['push', self.get_url('to-one')], working_dir='from')
 
222
        # This figure represent the amount of work to perform this use case. It
 
223
        # is entirely ok to reduce this number if a test fails due to rpc_count
 
224
        # being too low. If rpc_count increases, more network roundtrips have
 
225
        # become necessary for this use case. Please do not adjust this number
 
226
        # upwards without agreement from bzr's network support maintainers.
 
227
        self.assertLength(9, self.hpss_calls)
 
228
 
 
229
    def test_push_smart_stacked_streaming_acceptance(self):
 
230
        self.setup_smart_server_with_call_log()
 
231
        parent = self.make_branch_and_tree('parent', format='1.9')
 
232
        parent.commit(message='first commit')
 
233
        local = parent.bzrdir.sprout('local').open_workingtree()
 
234
        local.commit(message='local commit')
 
235
        self.reset_smart_call_log()
 
236
        self.run_bzr(['push', '--stacked', '--stacked-on', '../parent',
 
237
            self.get_url('public')], working_dir='local')
 
238
        # This figure represent the amount of work to perform this use case. It
 
239
        # is entirely ok to reduce this number if a test fails due to rpc_count
 
240
        # being too low. If rpc_count increases, more network roundtrips have
 
241
        # become necessary for this use case. Please do not adjust this number
 
242
        # upwards without agreement from bzr's network support maintainers.
 
243
        self.assertLength(14, self.hpss_calls)
 
244
        remote = branch.Branch.open('public')
 
245
        self.assertEndsWith(remote.get_stacked_on_url(), '/parent')
 
246
 
 
247
    def test_push_smart_tags_streaming_acceptance(self):
 
248
        self.setup_smart_server_with_call_log()
 
249
        t = self.make_branch_and_tree('from')
 
250
        rev_id = t.commit(allow_pointless=True, message='first commit')
 
251
        t.branch.tags.set_tag('new-tag', rev_id)
 
252
        self.reset_smart_call_log()
 
253
        self.run_bzr(['push', self.get_url('to-one')], working_dir='from')
 
254
        # This figure represent the amount of work to perform this use case. It
 
255
        # is entirely ok to reduce this number if a test fails due to rpc_count
 
256
        # being too low. If rpc_count increases, more network roundtrips have
 
257
        # become necessary for this use case. Please do not adjust this number
 
258
        # upwards without agreement from bzr's network support maintainers.
 
259
        self.assertLength(11, self.hpss_calls)
 
260
 
 
261
    def test_push_smart_with_default_stacking_url_path_segment(self):
 
262
        # If the default stacked-on location is a path element then branches
 
263
        # we push there over the smart server are stacked and their
 
264
        # stacked_on_url is that exact path segment. Added to nail bug 385132.
 
265
        self.setup_smart_server_with_call_log()
 
266
        self.make_branch('stack-on', format='1.9')
 
267
        self.make_bzrdir('.').get_config().set_default_stack_on(
 
268
            '/stack-on')
 
269
        self.make_branch('from', format='1.9')
 
270
        out, err = self.run_bzr(['push', '-d', 'from', self.get_url('to')])
 
271
        b = branch.Branch.open(self.get_url('to'))
 
272
        self.assertEqual('/extra/stack-on', b.get_stacked_on_url())
 
273
 
 
274
    def test_push_smart_with_default_stacking_relative_path(self):
 
275
        # If the default stacked-on location is a relative path then branches
 
276
        # we push there over the smart server are stacked and their
 
277
        # stacked_on_url is a relative path. Added to nail bug 385132.
 
278
        self.setup_smart_server_with_call_log()
 
279
        self.make_branch('stack-on', format='1.9')
 
280
        self.make_bzrdir('.').get_config().set_default_stack_on('stack-on')
 
281
        self.make_branch('from', format='1.9')
 
282
        out, err = self.run_bzr(['push', '-d', 'from', self.get_url('to')])
 
283
        b = branch.Branch.open(self.get_url('to'))
 
284
        self.assertEqual('../stack-on', b.get_stacked_on_url())
 
285
 
178
286
    def create_simple_tree(self):
179
287
        tree = self.make_branch_and_tree('tree')
180
288
        self.build_tree(['tree/a'])
184
292
 
185
293
    def test_push_create_prefix(self):
186
294
        """'bzr push --create-prefix' will create leading directories."""
 
295
        self.thisFailsStrictLockCheck()
187
296
        tree = self.create_simple_tree()
188
297
 
189
298
        self.run_bzr_error(['Parent directory of ../new/tree does not exist'],
190
 
                           'push', '../new/tree',
 
299
                           'push ../new/tree',
191
300
                           working_dir='tree')
192
 
        self.run_bzr('push', '../new/tree', '--create-prefix',
 
301
        self.run_bzr('push ../new/tree --create-prefix',
193
302
                     working_dir='tree')
194
 
        new_tree = WorkingTree.open('new/tree')
 
303
        new_tree = workingtree.WorkingTree.open('new/tree')
195
304
        self.assertEqual(tree.last_revision(), new_tree.last_revision())
196
305
        self.failUnlessExists('new/tree/a')
197
306
 
200
309
 
201
310
        By default, 'bzr push' will not use an existing, non-versioned dir.
202
311
        """
 
312
        self.thisFailsStrictLockCheck()
203
313
        tree = self.create_simple_tree()
204
314
        self.build_tree(['target/'])
205
315
 
206
316
        self.run_bzr_error(['Target directory ../target already exists',
207
317
                            'Supply --use-existing-dir',
208
 
                           ], 'push', '../target',
209
 
                           working_dir='tree')
 
318
                           ],
 
319
                           'push ../target', working_dir='tree')
210
320
 
211
 
        self.run_bzr('push', '--use-existing-dir', '../target',
 
321
        self.run_bzr('push --use-existing-dir ../target',
212
322
                     working_dir='tree')
213
323
 
214
 
        new_tree = WorkingTree.open('target')
 
324
        new_tree = workingtree.WorkingTree.open('target')
215
325
        self.assertEqual(tree.last_revision(), new_tree.last_revision())
216
326
        # The push should have created target/a
217
327
        self.failUnlessExists('target/a')
221
331
        tree = self.create_simple_tree()
222
332
        repo = self.make_repository('repo', shared=True)
223
333
 
224
 
        self.run_bzr('push', '../repo',
 
334
        self.run_bzr('push ../repo',
225
335
                     working_dir='tree')
226
336
 
227
337
        # Pushing onto an existing bzrdir will create a repository and
228
338
        # branch as needed, but will only create a working tree if there was
229
339
        # no BzrDir before.
230
 
        self.assertRaises(errors.NoWorkingTree, WorkingTree.open, 'repo')
231
 
        new_branch = Branch.open('repo')
 
340
        self.assertRaises(errors.NoWorkingTree,
 
341
                          workingtree.WorkingTree.open, 'repo')
 
342
        new_branch = branch.Branch.open('repo')
232
343
        self.assertEqual(tree.last_revision(), new_branch.last_revision())
233
344
 
234
345
    def test_push_onto_just_bzrdir(self):
242
353
        a_bzrdir = self.make_bzrdir('dir')
243
354
 
244
355
        self.run_bzr_error(['At ../dir you have a valid .bzr control'],
245
 
                'push', '../dir',
 
356
                'push ../dir',
246
357
                working_dir='tree')
 
358
 
 
359
    def test_push_with_revisionspec(self):
 
360
        """We should be able to push a revision older than the tip."""
 
361
        self.thisFailsStrictLockCheck()
 
362
        tree_from = self.make_branch_and_tree('from')
 
363
        tree_from.commit("One.", rev_id="from-1")
 
364
        tree_from.commit("Two.", rev_id="from-2")
 
365
 
 
366
        self.run_bzr('push -r1 ../to', working_dir='from')
 
367
 
 
368
        tree_to = workingtree.WorkingTree.open('to')
 
369
        repo_to = tree_to.branch.repository
 
370
        self.assertTrue(repo_to.has_revision('from-1'))
 
371
        self.assertFalse(repo_to.has_revision('from-2'))
 
372
        self.assertEqual(tree_to.branch.last_revision_info()[1], 'from-1')
 
373
 
 
374
        self.run_bzr_error(
 
375
            ['bzr: ERROR: bzr push --revision '
 
376
             'takes exactly one revision identifier\n'],
 
377
            'push -r0..2 ../to', working_dir='from')
 
378
 
 
379
    def create_trunk_and_feature_branch(self):
 
380
        # We have a mainline
 
381
        trunk_tree = self.make_branch_and_tree('target',
 
382
            format='1.9')
 
383
        trunk_tree.commit('mainline')
 
384
        # and a branch from it
 
385
        branch_tree = self.make_branch_and_tree('branch',
 
386
            format='1.9')
 
387
        branch_tree.pull(trunk_tree.branch)
 
388
        branch_tree.branch.set_parent(trunk_tree.branch.base)
 
389
        # with some work on it
 
390
        branch_tree.commit('moar work plz')
 
391
        return trunk_tree, branch_tree
 
392
 
 
393
    def assertPublished(self, branch_revid, stacked_on):
 
394
        """Assert that the branch 'published' has been published correctly."""
 
395
        published_branch = branch.Branch.open('published')
 
396
        # The published branch refers to the mainline
 
397
        self.assertEqual(stacked_on, published_branch.get_stacked_on_url())
 
398
        # and the branch's work was pushed
 
399
        self.assertTrue(published_branch.repository.has_revision(branch_revid))
 
400
 
 
401
    def test_push_new_branch_stacked_on(self):
 
402
        """Pushing a new branch with --stacked-on creates a stacked branch."""
 
403
        self.thisFailsStrictLockCheck()
 
404
        trunk_tree, branch_tree = self.create_trunk_and_feature_branch()
 
405
        # we publish branch_tree with a reference to the mainline.
 
406
        out, err = self.run_bzr(['push', '--stacked-on', trunk_tree.branch.base,
 
407
            self.get_url('published')], working_dir='branch')
 
408
        self.assertEqual('', out)
 
409
        self.assertEqual('Created new stacked branch referring to %s.\n' %
 
410
            trunk_tree.branch.base, err)
 
411
        self.assertPublished(branch_tree.last_revision(),
 
412
            trunk_tree.branch.base)
 
413
 
 
414
    def test_push_new_branch_stacked_uses_parent_when_no_public_url(self):
 
415
        """When the parent has no public url the parent is used as-is."""
 
416
        self.thisFailsStrictLockCheck()
 
417
        trunk_tree, branch_tree = self.create_trunk_and_feature_branch()
 
418
        # now we do a stacked push, which should determine the public location
 
419
        # for us.
 
420
        out, err = self.run_bzr(['push', '--stacked',
 
421
            self.get_url('published')], working_dir='branch')
 
422
        self.assertEqual('', out)
 
423
        self.assertEqual('Created new stacked branch referring to %s.\n' %
 
424
            trunk_tree.branch.base, err)
 
425
        self.assertPublished(branch_tree.last_revision(),
 
426
                             trunk_tree.branch.base)
 
427
 
 
428
    def test_push_new_branch_stacked_uses_parent_public(self):
 
429
        """Pushing a new branch with --stacked creates a stacked branch."""
 
430
        self.thisFailsStrictLockCheck()
 
431
        trunk_tree, branch_tree = self.create_trunk_and_feature_branch()
 
432
        # the trunk is published on a web server
 
433
        self.transport_readonly_server = http_server.HttpServer
 
434
        trunk_public = self.make_branch('public_trunk', format='1.9')
 
435
        trunk_public.pull(trunk_tree.branch)
 
436
        trunk_public_url = self.get_readonly_url('public_trunk')
 
437
        trunk_tree.branch.set_public_branch(trunk_public_url)
 
438
        # now we do a stacked push, which should determine the public location
 
439
        # for us.
 
440
        out, err = self.run_bzr(['push', '--stacked',
 
441
            self.get_url('published')], working_dir='branch')
 
442
        self.assertEqual('', out)
 
443
        self.assertEqual('Created new stacked branch referring to %s.\n' %
 
444
            trunk_public_url, err)
 
445
        self.assertPublished(branch_tree.last_revision(), trunk_public_url)
 
446
 
 
447
    def test_push_new_branch_stacked_no_parent(self):
 
448
        """Pushing with --stacked and no parent branch errors."""
 
449
        branch = self.make_branch_and_tree('branch', format='1.9')
 
450
        # now we do a stacked push, which should fail as the place to refer too
 
451
        # cannot be determined.
 
452
        out, err = self.run_bzr_error(
 
453
            ['Could not determine branch to refer to\\.'], ['push', '--stacked',
 
454
            self.get_url('published')], working_dir='branch')
 
455
        self.assertEqual('', out)
 
456
        self.assertFalse(self.get_transport('published').has('.'))
 
457
 
 
458
    def test_push_notifies_default_stacking(self):
 
459
        self.make_branch('stack_on', format='1.6')
 
460
        self.make_bzrdir('.').get_config().set_default_stack_on('stack_on')
 
461
        self.make_branch('from', format='1.6')
 
462
        out, err = self.run_bzr('push -d from to')
 
463
        self.assertContainsRe(err,
 
464
                              'Using default stacking branch stack_on at .*')
 
465
 
 
466
    def test_push_stacks_with_default_stacking_if_target_is_stackable(self):
 
467
        self.make_branch('stack_on', format='1.6')
 
468
        self.make_bzrdir('.').get_config().set_default_stack_on('stack_on')
 
469
        self.make_branch('from', format='pack-0.92')
 
470
        out, err = self.run_bzr('push -d from to')
 
471
        b = branch.Branch.open('to')
 
472
        self.assertEqual('../stack_on', b.get_stacked_on_url())
 
473
 
 
474
    def test_push_does_not_change_format_with_default_if_target_cannot(self):
 
475
        self.make_branch('stack_on', format='pack-0.92')
 
476
        self.make_bzrdir('.').get_config().set_default_stack_on('stack_on')
 
477
        self.make_branch('from', format='pack-0.92')
 
478
        out, err = self.run_bzr('push -d from to')
 
479
        b = branch.Branch.open('to')
 
480
        self.assertRaises(errors.UnstackableBranchFormat, b.get_stacked_on_url)
 
481
 
 
482
    def test_push_doesnt_create_broken_branch(self):
 
483
        """Pushing a new standalone branch works even when there's a default
 
484
        stacking policy at the destination.
 
485
 
 
486
        The new branch will preserve the repo format (even if it isn't the
 
487
        default for the branch), and will be stacked when the repo format
 
488
        allows (which means that the branch format isn't necessarly preserved).
 
489
        """
 
490
        self.make_repository('repo', shared=True, format='1.6')
 
491
        builder = self.make_branch_builder('repo/local', format='pack-0.92')
 
492
        builder.start_series()
 
493
        builder.build_snapshot('rev-1', None, [
 
494
            ('add', ('', 'root-id', 'directory', '')),
 
495
            ('add', ('filename', 'f-id', 'file', 'content\n'))])
 
496
        builder.build_snapshot('rev-2', ['rev-1'], [])
 
497
        builder.build_snapshot('rev-3', ['rev-2'],
 
498
            [('modify', ('f-id', 'new-content\n'))])
 
499
        builder.finish_series()
 
500
        branch = builder.get_branch()
 
501
        # Push rev-1 to "trunk", so that we can stack on it.
 
502
        self.run_bzr('push -d repo/local trunk -r 1')
 
503
        # Set a default stacking policy so that new branches will automatically
 
504
        # stack on trunk.
 
505
        self.make_bzrdir('.').get_config().set_default_stack_on('trunk')
 
506
        # Push rev-2 to a new branch "remote".  It will be stacked on "trunk".
 
507
        out, err = self.run_bzr('push -d repo/local remote -r 2')
 
508
        self.assertContainsRe(
 
509
            err, 'Using default stacking branch trunk at .*')
 
510
        # Push rev-3 onto "remote".  If "remote" not stacked and is missing the
 
511
        # fulltext record for f-id @ rev-1, then this will fail.
 
512
        out, err = self.run_bzr('push -d repo/local remote -r 3')
 
513
 
 
514
    def test_push_verbose_shows_log(self):
 
515
        self.thisFailsStrictLockCheck()
 
516
        tree = self.make_branch_and_tree('source')
 
517
        tree.commit('rev1')
 
518
        out, err = self.run_bzr('push -v -d source target')
 
519
        # initial push contains log
 
520
        self.assertContainsRe(out, 'rev1')
 
521
        tree.commit('rev2')
 
522
        out, err = self.run_bzr('push -v -d source target')
 
523
        # subsequent push contains log
 
524
        self.assertContainsRe(out, 'rev2')
 
525
        # subsequent log is accurate
 
526
        self.assertNotContainsRe(out, 'rev1')
 
527
 
 
528
    def test_push_from_subdir(self):
 
529
        self.thisFailsStrictLockCheck()
 
530
        t = self.make_branch_and_tree('tree')
 
531
        self.build_tree(['tree/dir/', 'tree/dir/file'])
 
532
        t.add('dir', 'dir/file')
 
533
        t.commit('r1')
 
534
        out, err = self.run_bzr('push ../../pushloc', working_dir='tree/dir')
 
535
        self.assertEqual('', out)
 
536
        self.assertEqual('Created new branch.\n', err)
 
537
 
 
538
 
 
539
class RedirectingMemoryTransport(memory.MemoryTransport):
 
540
 
 
541
    def mkdir(self, relpath, mode=None):
 
542
        if self._cwd == '/source/':
 
543
            raise errors.RedirectRequested(self.abspath(relpath),
 
544
                                           self.abspath('../target'),
 
545
                                           is_permanent=True)
 
546
        elif self._cwd == '/infinite-loop/':
 
547
            raise errors.RedirectRequested(self.abspath(relpath),
 
548
                                           self.abspath('../infinite-loop'),
 
549
                                           is_permanent=True)
 
550
        else:
 
551
            return super(RedirectingMemoryTransport, self).mkdir(
 
552
                relpath, mode)
 
553
 
 
554
    def get(self, relpath):
 
555
        if self.clone(relpath)._cwd == '/infinite-loop/':
 
556
            raise errors.RedirectRequested(self.abspath(relpath),
 
557
                                           self.abspath('../infinite-loop'),
 
558
                                           is_permanent=True)
 
559
        else:
 
560
            return super(RedirectingMemoryTransport, self).get(relpath)
 
561
 
 
562
    def _redirected_to(self, source, target):
 
563
        # We do accept redirections
 
564
        return transport.get_transport(target)
 
565
 
 
566
 
 
567
class RedirectingMemoryServer(memory.MemoryServer):
 
568
 
 
569
    def setUp(self):
 
570
        self._dirs = {'/': None}
 
571
        self._files = {}
 
572
        self._locks = {}
 
573
        self._scheme = 'redirecting-memory+%s:///' % id(self)
 
574
        transport.register_transport(self._scheme, self._memory_factory)
 
575
 
 
576
    def _memory_factory(self, url):
 
577
        result = RedirectingMemoryTransport(url)
 
578
        result._dirs = self._dirs
 
579
        result._files = self._files
 
580
        result._locks = self._locks
 
581
        return result
 
582
 
 
583
    def tearDown(self):
 
584
        transport.unregister_transport(self._scheme, self._memory_factory)
 
585
 
 
586
 
 
587
class TestPushRedirect(tests.TestCaseWithTransport):
 
588
 
 
589
    def setUp(self):
 
590
        tests.TestCaseWithTransport.setUp(self)
 
591
        self.memory_server = RedirectingMemoryServer()
 
592
        self.memory_server.setUp()
 
593
        self.addCleanup(self.memory_server.tearDown)
 
594
 
 
595
        # Make the branch and tree that we'll be pushing.
 
596
        t = self.make_branch_and_tree('tree')
 
597
        self.build_tree(['tree/file'])
 
598
        t.add('file')
 
599
        t.commit('commit 1')
 
600
 
 
601
    def test_push_redirects_on_mkdir(self):
 
602
        """If the push requires a mkdir, push respects redirect requests.
 
603
 
 
604
        This is added primarily to handle lp:/ URI support, so that users can
 
605
        push to new branches by specifying lp:/ URIs.
 
606
        """
 
607
        destination_url = self.memory_server.get_url() + 'source'
 
608
        self.run_bzr(['push', '-d', 'tree', destination_url])
 
609
 
 
610
        local_revision = branch.Branch.open('tree').last_revision()
 
611
        remote_revision = branch.Branch.open(
 
612
            self.memory_server.get_url() + 'target').last_revision()
 
613
        self.assertEqual(remote_revision, local_revision)
 
614
 
 
615
    def test_push_gracefully_handles_too_many_redirects(self):
 
616
        """Push fails gracefully if the mkdir generates a large number of
 
617
        redirects.
 
618
        """
 
619
        destination_url = self.memory_server.get_url() + 'infinite-loop'
 
620
        out, err = self.run_bzr_error(
 
621
            ['Too many redirections trying to make %s\\.\n'
 
622
             % re.escape(destination_url)],
 
623
            ['push', '-d', 'tree', destination_url], retcode=3)
 
624
        self.assertEqual('', out)
 
625
 
 
626
 
 
627
class TestPushStrictMixin(object):
 
628
 
 
629
    def make_local_branch_and_tree(self):
 
630
        self.tree = self.make_branch_and_tree('local')
 
631
        self.build_tree_contents([('local/file', 'initial')])
 
632
        self.tree.add('file')
 
633
        self.tree.commit('adding file', rev_id='added')
 
634
        self.build_tree_contents([('local/file', 'modified')])
 
635
        self.tree.commit('modify file', rev_id='modified')
 
636
 
 
637
    def set_config_push_strict(self, value):
 
638
        # set config var (any of bazaar.conf, locations.conf, branch.conf
 
639
        # should do)
 
640
        conf = self.tree.branch.get_config()
 
641
        conf.set_user_option('push_strict', value)
 
642
 
 
643
    _default_command = ['push', '../to']
 
644
    _default_wd = 'local'
 
645
    _default_errors = ['Working tree ".*/local/" has uncommitted '
 
646
                       'changes \(See bzr status\)\.',]
 
647
    _default_pushed_revid = 'modified'
 
648
 
 
649
    def assertPushFails(self, args):
 
650
        self.run_bzr_error(self._default_errors, self._default_command + args,
 
651
                           working_dir=self._default_wd, retcode=3)
 
652
 
 
653
    def assertPushSucceeds(self, args, pushed_revid=None):
 
654
        self.run_bzr(self._default_command + args,
 
655
                     working_dir=self._default_wd)
 
656
        if pushed_revid is None:
 
657
            pushed_revid = self._default_pushed_revid
 
658
        tree_to = workingtree.WorkingTree.open('to')
 
659
        repo_to = tree_to.branch.repository
 
660
        self.assertTrue(repo_to.has_revision(pushed_revid))
 
661
        self.assertEqual(tree_to.branch.last_revision_info()[1], pushed_revid)
 
662
 
 
663
 
 
664
 
 
665
class TestPushStrictWithoutChanges(tests.TestCaseWithTransport,
 
666
                                   TestPushStrictMixin):
 
667
 
 
668
    def setUp(self):
 
669
        super(TestPushStrictWithoutChanges, self).setUp()
 
670
        self.make_local_branch_and_tree()
 
671
 
 
672
    def test_push_default(self):
 
673
        self.thisFailsStrictLockCheck()
 
674
        self.assertPushSucceeds([])
 
675
 
 
676
    def test_push_strict(self):
 
677
        self.thisFailsStrictLockCheck()
 
678
        self.assertPushSucceeds(['--strict'])
 
679
 
 
680
    def test_push_no_strict(self):
 
681
        self.thisFailsStrictLockCheck()
 
682
        self.assertPushSucceeds(['--no-strict'])
 
683
 
 
684
    def test_push_config_var_strict(self):
 
685
        self.thisFailsStrictLockCheck()
 
686
        self.set_config_push_strict('true')
 
687
        self.assertPushSucceeds([])
 
688
 
 
689
    def test_push_config_var_no_strict(self):
 
690
        self.thisFailsStrictLockCheck()
 
691
        self.set_config_push_strict('false')
 
692
        self.assertPushSucceeds([])
 
693
 
 
694
 
 
695
class TestPushStrictWithChanges(tests.TestCaseWithTransport,
 
696
                                TestPushStrictMixin):
 
697
 
 
698
    _changes_type = None # Set by load_tests
 
699
 
 
700
    def setUp(self):
 
701
        super(TestPushStrictWithChanges, self).setUp()
 
702
        getattr(self, self._changes_type)()
 
703
 
 
704
    def _uncommitted_changes(self):
 
705
        self.make_local_branch_and_tree()
 
706
        # Make a change without committing it
 
707
        self.build_tree_contents([('local/file', 'in progress')])
 
708
 
 
709
    def _pending_merges(self):
 
710
        self.make_local_branch_and_tree()
 
711
        # Create 'other' branch containing a new file
 
712
        other_bzrdir = self.tree.bzrdir.sprout('other')
 
713
        other_tree = other_bzrdir.open_workingtree()
 
714
        self.build_tree_contents([('other/other-file', 'other')])
 
715
        other_tree.add('other-file')
 
716
        other_tree.commit('other commit', rev_id='other')
 
717
        # Merge and revert, leaving a pending merge
 
718
        self.tree.merge_from_branch(other_tree.branch)
 
719
        self.tree.revert(filenames=['other-file'], backups=False)
 
720
 
 
721
    def _out_of_sync_trees(self):
 
722
        self.make_local_branch_and_tree()
 
723
        self.run_bzr(['checkout', '--lightweight', 'local', 'checkout'])
 
724
        # Make a change and commit it
 
725
        self.build_tree_contents([('local/file', 'modified in local')])
 
726
        self.tree.commit('modify file', rev_id='modified-in-local')
 
727
        # Exercise commands from the checkout directory
 
728
        self._default_wd = 'checkout'
 
729
        self._default_errors = ["Working tree is out of date, please run"
 
730
                                " 'bzr update'\.",]
 
731
        self._default_pushed_revid = 'modified-in-local'
 
732
 
 
733
    def test_push_default(self):
 
734
        self.assertPushFails([])
 
735
 
 
736
    def test_push_with_revision(self):
 
737
        self.thisFailsStrictLockCheck()
 
738
        self.assertPushSucceeds(['-r', 'revid:added'], pushed_revid='added')
 
739
 
 
740
    def test_push_no_strict(self):
 
741
        self.thisFailsStrictLockCheck()
 
742
        self.assertPushSucceeds(['--no-strict'])
 
743
 
 
744
    def test_push_strict_with_changes(self):
 
745
        self.assertPushFails(['--strict'])
 
746
 
 
747
    def test_push_respect_config_var_strict(self):
 
748
        self.set_config_push_strict('true')
 
749
        self.assertPushFails([])
 
750
 
 
751
    def test_push_bogus_config_var_ignored(self):
 
752
        self.set_config_push_strict("I don't want you to be strict")
 
753
        self.assertPushFails([])
 
754
 
 
755
    def test_push_no_strict_command_line_override_config(self):
 
756
        self.thisFailsStrictLockCheck()
 
757
        self.set_config_push_strict('yES')
 
758
        self.assertPushFails([])
 
759
        self.assertPushSucceeds(['--no-strict'])
 
760
 
 
761
    def test_push_strict_command_line_override_config(self):
 
762
        self.thisFailsStrictLockCheck()
 
763
        self.set_config_push_strict('oFF')
 
764
        self.assertPushFails(['--strict'])
 
765
        self.assertPushSucceeds([])