~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

  • Committer: John Arbash Meinel
  • Date: 2008-09-09 15:09:12 UTC
  • mto: This revision was merged to the branch mainline in revision 3699.
  • Revision ID: john@arbash-meinel.com-20080909150912-wyttm8he1zsls2ck
Use the right timing function on win32

Show diffs side-by-side

added added

removed removed

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