~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

Cleaned up some long lines

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
2
 
#
 
1
# Copyright (C) 2005 by Canonical Ltd
 
2
# -*- coding: utf-8 -*-
 
3
 
3
4
# This program is free software; you can redistribute it and/or modify
4
5
# it under the terms of the GNU General Public License as published by
5
6
# the Free Software Foundation; either version 2 of the License, or
6
7
# (at your option) any later version.
7
 
#
 
8
 
8
9
# This program is distributed in the hope that it will be useful,
9
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
12
# GNU General Public License for more details.
12
 
#
 
13
 
13
14
# You should have received a copy of the GNU General Public License
14
15
# 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 pull."""
 
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
17
 
 
18
 
 
19
"""Black-box tests for bzr pull.
 
20
"""
19
21
 
20
22
import os
21
23
import sys
22
24
 
23
 
from bzrlib import (
24
 
    debug,
25
 
    remote,
26
 
    tests,
27
 
    urlutils,
28
 
    )
29
 
 
30
25
from bzrlib.branch import Branch
31
 
from bzrlib.directory_service import directories
32
 
from bzrlib.osutils import pathjoin
33
 
from bzrlib.tests import (
34
 
    fixtures,
35
 
    script,
36
 
    )
37
 
from bzrlib.uncommit import uncommit
38
 
from bzrlib.workingtree import WorkingTree
39
 
 
40
 
 
41
 
class TestPull(tests.TestCaseWithTransport):
42
 
 
43
 
    def example_branch(self, path='.'):
44
 
        tree = self.make_branch_and_tree(path)
45
 
        self.build_tree_contents([
46
 
            (pathjoin(path, 'hello'),   'foo'),
47
 
            (pathjoin(path, 'goodbye'), 'baz')])
48
 
        tree.add('hello')
49
 
        tree.commit(message='setup')
50
 
        tree.add('goodbye')
51
 
        tree.commit(message='setup')
52
 
        return tree
 
26
from bzrlib.tests.blackbox import ExternalBase
 
27
 
 
28
class TestPull(ExternalBase):
 
29
 
 
30
    def example_branch(test):
 
31
        test.runbzr('init')
 
32
        file('hello', 'wt').write('foo')
 
33
        test.runbzr('add hello')
 
34
        test.runbzr('commit -m setup hello')
 
35
        file('goodbye', 'wt').write('baz')
 
36
        test.runbzr('add goodbye')
 
37
        test.runbzr('commit -m setup goodbye')
53
38
 
54
39
    def test_pull(self):
55
40
        """Pull changes from one branch to another."""
56
 
        a_tree = self.example_branch('a')
57
 
        base_rev = a_tree.branch.last_revision()
 
41
        os.mkdir('a')
58
42
        os.chdir('a')
59
 
        self.run_bzr('pull', retcode=3)
60
 
        self.run_bzr('missing', retcode=3)
61
 
        self.run_bzr('missing .')
62
 
        self.run_bzr('missing')
63
 
        # this will work on windows because we check for the same branch
64
 
        # in pull - if it fails, it is a regression
65
 
        self.run_bzr('pull')
66
 
        self.run_bzr('pull /', retcode=3)
67
 
        if sys.platform not in ('win32', 'cygwin'):
68
 
            self.run_bzr('pull')
 
43
 
 
44
        self.example_branch()
 
45
        self.runbzr('pull', retcode=3)
 
46
        self.runbzr('missing', retcode=3)
 
47
        self.runbzr('missing .')
 
48
        self.runbzr('missing')
 
49
        if sys.platform not in ('win32', 'cygwin'):
 
50
            # This is equivalent to doing "bzr pull ."
 
51
            # Which means that bzr creates 2 branches grabbing
 
52
            # the same location, and tries to pull.
 
53
            # However, 2 branches mean 2 locks on the same file
 
54
            # which ultimately implies a deadlock.
 
55
            # (non windows platforms allow multiple locks on the
 
56
            # same file by the same calling process)
 
57
            self.runbzr('pull')
 
58
        self.runbzr('pull /', retcode=3)
 
59
        if sys.platform not in ('win32', 'cygwin'):
 
60
            self.runbzr('pull')
69
61
 
70
62
        os.chdir('..')
71
 
        b_tree = a_tree.bzrdir.sprout('b').open_workingtree()
 
63
        self.runbzr('branch a b')
72
64
        os.chdir('b')
73
 
        self.run_bzr('pull')
 
65
        self.runbzr('pull')
74
66
        os.mkdir('subdir')
75
 
        b_tree.add('subdir')
76
 
        new_rev = b_tree.commit(message='blah', allow_pointless=True)
77
 
 
78
 
        os.chdir('..')
79
 
        a = Branch.open('a')
80
 
        b = Branch.open('b')
81
 
        self.assertEqual(a.last_revision(), base_rev)
82
 
        self.assertEqual(b.last_revision(), new_rev)
83
 
 
84
 
        os.chdir('a')
85
 
        self.run_bzr('pull ../b')
86
 
        self.assertEqual(a.last_revision(), b.last_revision())
87
 
        a_tree.commit(message='blah2', allow_pointless=True)
88
 
        b_tree.commit(message='blah3', allow_pointless=True)
 
67
        self.runbzr('add subdir')
 
68
        self.runbzr('commit -m blah --unchanged')
 
69
        os.chdir('../a')
 
70
        a = Branch.open('.')
 
71
        b = Branch.open('../b')
 
72
        self.assertEquals(a.revision_history(), b.revision_history()[:-1])
 
73
        self.runbzr('pull ../b')
 
74
        self.assertEquals(a.revision_history(), b.revision_history())
 
75
        self.runbzr('commit -m blah2 --unchanged')
 
76
        os.chdir('../b')
 
77
        self.runbzr('commit -m blah3 --unchanged')
89
78
        # no overwrite
90
 
        os.chdir('../b')
91
 
        self.run_bzr('pull ../a', retcode=3)
 
79
        self.runbzr('pull ../a', retcode=3)
92
80
        os.chdir('..')
93
 
        b_tree.bzrdir.sprout('overwriteme')
 
81
        self.runbzr('branch b overwriteme')
94
82
        os.chdir('overwriteme')
95
 
        self.run_bzr('pull --overwrite ../a')
 
83
        self.runbzr('pull --overwrite ../a')
96
84
        overwritten = Branch.open('.')
97
 
        self.assertEqual(overwritten.last_revision(),
98
 
                         a.last_revision())
99
 
        a_tree.merge_from_branch(b_tree.branch)
100
 
        a_tree.commit(message="blah4", allow_pointless=True)
 
85
        self.assertEqual(overwritten.revision_history(),
 
86
                         a.revision_history())
 
87
        os.chdir('../a')
 
88
        self.runbzr('merge ../b')
 
89
        self.runbzr('commit -m blah4 --unchanged')
101
90
        os.chdir('../b/subdir')
102
 
        self.run_bzr('pull ../../a')
103
 
        self.assertEqual(a.last_revision(), b.last_revision())
104
 
        sub_tree = WorkingTree.open_containing('.')[0]
105
 
        sub_tree.commit(message="blah5", allow_pointless=True)
106
 
        sub_tree.commit(message="blah6", allow_pointless=True)
 
91
        self.runbzr('pull ../../a')
 
92
        self.assertEquals(a.revision_history()[-1], b.revision_history()[-1])
 
93
        self.runbzr('commit -m blah5 --unchanged')
 
94
        self.runbzr('commit -m blah6 --unchanged')
107
95
        os.chdir('..')
108
 
        self.run_bzr('pull ../a')
 
96
        self.runbzr('pull ../a')
109
97
        os.chdir('../a')
110
 
        a_tree.commit(message="blah7", allow_pointless=True)
111
 
        a_tree.merge_from_branch(b_tree.branch)
112
 
        a_tree.commit(message="blah8", allow_pointless=True)
113
 
        self.run_bzr('pull ../b')
114
 
        self.run_bzr('pull ../b')
115
 
 
116
 
    def test_pull_dash_d(self):
117
 
        self.example_branch('a')
118
 
        self.make_branch_and_tree('b')
119
 
        self.make_branch_and_tree('c')
120
 
        # pull into that branch
121
 
        self.run_bzr('pull -d b a')
122
 
        # pull into a branch specified by a url
123
 
        c_url = urlutils.local_path_to_url('c')
124
 
        self.assertStartsWith(c_url, 'file://')
125
 
        self.run_bzr(['pull', '-d', c_url, 'a'])
 
98
        self.runbzr('commit -m blah7 --unchanged')
 
99
        self.runbzr('merge ../b')
 
100
        self.runbzr('commit -m blah8 --unchanged')
 
101
        self.runbzr('pull ../b')
 
102
        self.runbzr('pull ../b')
126
103
 
127
104
    def test_pull_revision(self):
128
105
        """Pull some changes from one branch to another."""
129
 
        a_tree = self.example_branch('a')
130
 
        self.build_tree_contents([
131
 
            ('a/hello2',   'foo'),
132
 
            ('a/goodbye2', 'baz')])
133
 
        a_tree.add('hello2')
134
 
        a_tree.commit(message="setup")
135
 
        a_tree.add('goodbye2')
136
 
        a_tree.commit(message="setup")
137
 
 
138
 
        b_tree = a_tree.bzrdir.sprout('b',
139
 
                   revision_id=a_tree.branch.get_rev_id(1)).open_workingtree()
 
106
        os.mkdir('a')
 
107
        os.chdir('a')
 
108
 
 
109
        self.example_branch()
 
110
        file('hello2', 'wt').write('foo')
 
111
        self.runbzr('add hello2')
 
112
        self.runbzr('commit -m setup hello2')
 
113
        file('goodbye2', 'wt').write('baz')
 
114
        self.runbzr('add goodbye2')
 
115
        self.runbzr('commit -m setup goodbye2')
 
116
 
 
117
        os.chdir('..')
 
118
        self.runbzr('branch -r 1 a b')
140
119
        os.chdir('b')
141
 
        self.run_bzr('pull -r 2')
 
120
        self.runbzr('pull -r 2')
142
121
        a = Branch.open('../a')
143
122
        b = Branch.open('.')
144
 
        self.assertEqual(a.revno(),4)
145
 
        self.assertEqual(b.revno(),2)
146
 
        self.run_bzr('pull -r 3')
147
 
        self.assertEqual(b.revno(),3)
148
 
        self.run_bzr('pull -r 4')
149
 
        self.assertEqual(a.last_revision(), b.last_revision())
 
123
        self.assertEquals(a.revno(),4)
 
124
        self.assertEquals(b.revno(),2)
 
125
        self.runbzr('pull -r 3')
 
126
        self.assertEquals(b.revno(),3)
 
127
        self.runbzr('pull -r 4')
 
128
        self.assertEquals(a.revision_history(), b.revision_history())
150
129
 
151
 
    def test_pull_tags(self):
152
 
        """Tags are updated by pull, and revisions named in those tags are
153
 
        fetched.
154
 
        """
155
 
        # Make a source, sprout a target off it
156
 
        builder = self.make_branch_builder('source')
157
 
        source = fixtures.build_branch_with_non_ancestral_rev(builder)
158
 
        source.get_config().set_user_option('branch.fetch_tags', 'True')
159
 
        target_bzrdir = source.bzrdir.sprout('target')
160
 
        source.tags.set_tag('tag-a', 'rev-2')
161
 
        # Pull from source
162
 
        self.run_bzr('pull -d target source')
163
 
        target = target_bzrdir.open_branch()
164
 
        # The tag is present, and so is its revision.
165
 
        self.assertEqual('rev-2', target.tags.lookup_tag('tag-a'))
166
 
        target.repository.get_revision('rev-2')
167
130
 
168
131
    def test_overwrite_uptodate(self):
169
132
        # Make sure pull --overwrite overwrites
170
133
        # even if the target branch has merged
171
134
        # everything already.
172
 
        a_tree = self.make_branch_and_tree('a')
173
 
        self.build_tree_contents([('a/foo', 'original\n')])
174
 
        a_tree.add('foo')
175
 
        a_tree.commit(message='initial commit')
176
 
 
177
 
        b_tree = a_tree.bzrdir.sprout('b').open_workingtree()
178
 
 
179
 
        self.build_tree_contents([('a/foo', 'changed\n')])
180
 
        a_tree.commit(message='later change')
181
 
 
182
 
        self.build_tree_contents([('a/foo', 'a third change')])
183
 
        a_tree.commit(message='a third change')
184
 
 
185
 
        self.assertEqual(a_tree.branch.last_revision_info()[0], 3)
186
 
 
187
 
        b_tree.merge_from_branch(a_tree.branch)
188
 
        b_tree.commit(message='merge')
189
 
 
190
 
        self.assertEqual(b_tree.branch.last_revision_info()[0], 2)
191
 
 
192
 
        os.chdir('b')
193
 
        self.run_bzr('pull --overwrite ../a')
194
 
        (last_revinfo_b) = b_tree.branch.last_revision_info()
195
 
        self.assertEqual(last_revinfo_b[0], 3)
196
 
        self.assertEqual(last_revinfo_b[1], a_tree.branch.last_revision())
 
135
        bzr = self.run_bzr
 
136
 
 
137
        def get_rh(expected_len):
 
138
            rh = self.capture('revision-history')
 
139
            # Make sure we don't have trailing empty revisions
 
140
            rh = rh.strip().split('\n')
 
141
            self.assertEqual(len(rh), expected_len)
 
142
            return rh
 
143
 
 
144
        os.mkdir('a')
 
145
        os.chdir('a')
 
146
        bzr('init')
 
147
        open('foo', 'wb').write('original\n')
 
148
        bzr('add', 'foo')
 
149
        bzr('commit', '-m', 'initial commit')
 
150
 
 
151
        os.chdir('..')
 
152
        bzr('branch', 'a', 'b')
 
153
 
 
154
        os.chdir('a')
 
155
        open('foo', 'wb').write('changed\n')
 
156
        bzr('commit', '-m', 'later change')
 
157
 
 
158
        open('foo', 'wb').write('another\n')
 
159
        bzr('commit', '-m', 'a third change')
 
160
 
 
161
        rev_history_a = get_rh(3)
 
162
 
 
163
        os.chdir('../b')
 
164
        bzr('merge', '../a')
 
165
        bzr('commit', '-m', 'merge')
 
166
 
 
167
        rev_history_b = get_rh(2)
 
168
 
 
169
        bzr('pull', '--overwrite', '../a')
 
170
        rev_history_b = get_rh(3)
 
171
 
 
172
        self.assertEqual(rev_history_b, rev_history_a)
197
173
 
198
174
    def test_overwrite_children(self):
199
175
        # Make sure pull --overwrite sets the revision-history
200
176
        # to be identical to the pull source, even if we have convergence
201
 
        a_tree = self.make_branch_and_tree('a')
202
 
        self.build_tree_contents([('a/foo', 'original\n')])
203
 
        a_tree.add('foo')
204
 
        a_tree.commit(message='initial commit')
205
 
 
206
 
        b_tree = a_tree.bzrdir.sprout('b').open_workingtree()
207
 
 
208
 
        self.build_tree_contents([('a/foo', 'changed\n')])
209
 
        a_tree.commit(message='later change')
210
 
 
211
 
        self.build_tree_contents([('a/foo', 'a third change')])
212
 
        a_tree.commit(message='a third change')
213
 
 
214
 
        self.assertEqual(a_tree.branch.last_revision_info()[0], 3)
215
 
 
216
 
        b_tree.merge_from_branch(a_tree.branch)
217
 
        b_tree.commit(message='merge')
218
 
 
219
 
        self.assertEqual(b_tree.branch.last_revision_info()[0], 2)
220
 
 
221
 
        self.build_tree_contents([('a/foo', 'a fourth change\n')])
222
 
        a_tree.commit(message='a fourth change')
223
 
 
224
 
        rev_info_a = a_tree.branch.last_revision_info()
225
 
        self.assertEqual(rev_info_a[0], 4)
 
177
        bzr = self.run_bzr
 
178
 
 
179
        def get_rh(expected_len):
 
180
            rh = self.capture('revision-history')
 
181
            # Make sure we don't have trailing empty revisions
 
182
            rh = rh.strip().split('\n')
 
183
            self.assertEqual(len(rh), expected_len)
 
184
            return rh
 
185
 
 
186
        os.mkdir('a')
 
187
        os.chdir('a')
 
188
        bzr('init')
 
189
        open('foo', 'wb').write('original\n')
 
190
        bzr('add', 'foo')
 
191
        bzr('commit', '-m', 'initial commit')
 
192
 
 
193
        os.chdir('..')
 
194
        bzr('branch', 'a', 'b')
 
195
 
 
196
        os.chdir('a')
 
197
        open('foo', 'wb').write('changed\n')
 
198
        bzr('commit', '-m', 'later change')
 
199
 
 
200
        open('foo', 'wb').write('another\n')
 
201
        bzr('commit', '-m', 'a third change')
 
202
 
 
203
        rev_history_a = get_rh(3)
 
204
 
 
205
        os.chdir('../b')
 
206
        bzr('merge', '../a')
 
207
        bzr('commit', '-m', 'merge')
 
208
 
 
209
        rev_history_b = get_rh(2)
 
210
 
 
211
        os.chdir('../a')
 
212
        open('foo', 'wb').write('a fourth change\n')
 
213
        bzr('commit', '-m', 'a fourth change')
 
214
 
 
215
        rev_history_a = get_rh(4)
226
216
 
227
217
        # With convergence, we could just pull over the
228
218
        # new change, but with --overwrite, we want to switch our history
229
 
        os.chdir('b')
230
 
        self.run_bzr('pull --overwrite ../a')
231
 
        rev_info_b = b_tree.branch.last_revision_info()
232
 
        self.assertEqual(rev_info_b[0], 4)
233
 
        self.assertEqual(rev_info_b, rev_info_a)
234
 
 
235
 
    def test_pull_remember(self):
236
 
        """Pull changes from one branch to another and test parent location."""
237
 
        transport = self.get_transport()
238
 
        tree_a = self.make_branch_and_tree('branch_a')
239
 
        branch_a = tree_a.branch
240
 
        self.build_tree(['branch_a/a'])
241
 
        tree_a.add('a')
242
 
        tree_a.commit('commit a')
243
 
        tree_b = branch_a.bzrdir.sprout('branch_b').open_workingtree()
244
 
        branch_b = tree_b.branch
245
 
        tree_c = branch_a.bzrdir.sprout('branch_c').open_workingtree()
246
 
        branch_c = tree_c.branch
247
 
        self.build_tree(['branch_a/b'])
248
 
        tree_a.add('b')
249
 
        tree_a.commit('commit b')
250
 
        # reset parent
251
 
        parent = branch_b.get_parent()
252
 
        branch_b.set_parent(None)
253
 
        self.assertEqual(None, branch_b.get_parent())
254
 
        # test pull for failure without parent set
255
 
        os.chdir('branch_b')
256
 
        out = self.run_bzr('pull', retcode=3)
257
 
        self.assertEqual(out,
258
 
                ('','bzr: ERROR: No pull location known or specified.\n'))
259
 
        # test implicit --remember when no parent set, this pull conflicts
260
 
        self.build_tree(['d'])
261
 
        tree_b.add('d')
262
 
        tree_b.commit('commit d')
263
 
        out = self.run_bzr('pull ../branch_a', retcode=3)
264
 
        self.assertEqual(out,
265
 
                ('','bzr: ERROR: These branches have diverged.'
266
 
                    ' Use the missing command to see how.\n'
267
 
                    'Use the merge command to reconcile them.\n'))
268
 
        self.assertEqual(branch_b.get_parent(), parent)
269
 
        # test implicit --remember after resolving previous failure
270
 
        uncommit(branch=branch_b, tree=tree_b)
271
 
        transport.delete('branch_b/d')
272
 
        self.run_bzr('pull')
273
 
        self.assertEqual(branch_b.get_parent(), parent)
274
 
        # test explicit --remember
275
 
        self.run_bzr('pull ../branch_c --remember')
276
 
        self.assertEqual(branch_b.get_parent(),
277
 
                          branch_c.bzrdir.root_transport.base)
278
 
 
279
 
    def test_pull_bundle(self):
280
 
        from bzrlib.testament import Testament
281
 
        # Build up 2 trees and prepare for a pull
282
 
        tree_a = self.make_branch_and_tree('branch_a')
283
 
        f = open('branch_a/a', 'wb')
284
 
        f.write('hello')
285
 
        f.close()
286
 
        tree_a.add('a')
287
 
        tree_a.commit('message')
288
 
 
289
 
        tree_b = tree_a.bzrdir.sprout('branch_b').open_workingtree()
290
 
 
291
 
        # Make a change to 'a' that 'b' can pull
292
 
        f = open('branch_a/a', 'wb')
293
 
        f.write('hey there')
294
 
        f.close()
295
 
        tree_a.commit('message')
296
 
 
297
 
        # Create the bundle for 'b' to pull
298
 
        os.chdir('branch_a')
299
 
        self.run_bzr('bundle ../branch_b -o ../bundle')
300
 
 
301
 
        os.chdir('../branch_b')
302
 
        out, err = self.run_bzr('pull ../bundle')
303
 
        self.assertEqual(out,
304
 
                         'Now on revision 2.\n')
305
 
        self.assertEqual(err,
306
 
                ' M  a\nAll changes applied successfully.\n')
307
 
 
308
 
        self.assertEqualDiff(tree_a.branch.last_revision(),
309
 
                             tree_b.branch.last_revision())
310
 
 
311
 
        testament_a = Testament.from_revision(tree_a.branch.repository,
312
 
                                              tree_a.get_parent_ids()[0])
313
 
        testament_b = Testament.from_revision(tree_b.branch.repository,
314
 
                                              tree_b.get_parent_ids()[0])
315
 
        self.assertEqualDiff(testament_a.as_text(),
316
 
                             testament_b.as_text())
317
 
 
318
 
        # it is legal to attempt to pull an already-merged bundle
319
 
        out, err = self.run_bzr('pull ../bundle')
320
 
        self.assertEqual(err, '')
321
 
        self.assertEqual(out, 'No revisions or tags to pull.\n')
322
 
 
323
 
    def test_pull_verbose_no_files(self):
324
 
        """Pull --verbose should not list modified files"""
325
 
        tree_a = self.make_branch_and_tree('tree_a')
326
 
        self.build_tree(['tree_a/foo'])
327
 
        tree_a.add('foo')
328
 
        tree_a.commit('bar')
329
 
        tree_b = self.make_branch_and_tree('tree_b')
330
 
        out = self.run_bzr('pull --verbose -d tree_b tree_a')[0]
331
 
        self.assertContainsRe(out, 'bar')
332
 
        self.assertNotContainsRe(out, 'added:')
333
 
        self.assertNotContainsRe(out, 'foo')
334
 
 
335
 
    def test_pull_quiet(self):
336
 
        """Check that bzr pull --quiet does not print anything"""
337
 
        tree_a = self.make_branch_and_tree('tree_a')
338
 
        self.build_tree(['tree_a/foo'])
339
 
        tree_a.add('foo')
340
 
        revision_id = tree_a.commit('bar')
341
 
        tree_b = tree_a.bzrdir.sprout('tree_b').open_workingtree()
342
 
        out, err = self.run_bzr('pull --quiet -d tree_b')
343
 
        self.assertEqual(out, '')
344
 
        self.assertEqual(err, '')
345
 
        self.assertEqual(tree_b.last_revision(), revision_id)
346
 
        self.build_tree(['tree_a/moo'])
347
 
        tree_a.add('moo')
348
 
        revision_id = tree_a.commit('quack')
349
 
        out, err = self.run_bzr('pull --quiet -d tree_b')
350
 
        self.assertEqual(out, '')
351
 
        self.assertEqual(err, '')
352
 
        self.assertEqual(tree_b.last_revision(), revision_id)
353
 
 
354
 
    def test_pull_from_directory_service(self):
355
 
        source = self.make_branch_and_tree('source')
356
 
        source.commit('commit 1')
357
 
        target = source.bzrdir.sprout('target').open_workingtree()
358
 
        source_last = source.commit('commit 2')
359
 
        class FooService(object):
360
 
            """A directory service that always returns source"""
361
 
 
362
 
            def look_up(self, name, url):
363
 
                return 'source'
364
 
        directories.register('foo:', FooService, 'Testing directory service')
365
 
        self.addCleanup(directories.remove, 'foo:')
366
 
        self.run_bzr('pull foo:bar -d target')
367
 
        self.assertEqual(source_last, target.last_revision())
368
 
 
369
 
    def test_pull_verbose_defaults_to_long(self):
370
 
        tree = self.example_branch('source')
371
 
        target = self.make_branch_and_tree('target')
372
 
        out = self.run_bzr('pull -v source -d target')[0]
373
 
        self.assertContainsRe(out,
374
 
                              r'revno: 1\ncommitter: .*\nbranch nick: source')
375
 
        self.assertNotContainsRe(out, r'\n {4}1 .*\n {6}setup\n')
376
 
 
377
 
    def test_pull_verbose_uses_default_log(self):
378
 
        tree = self.example_branch('source')
379
 
        target = self.make_branch_and_tree('target')
380
 
        target_config = target.branch.get_config()
381
 
        target_config.set_user_option('log_format', 'short')
382
 
        out = self.run_bzr('pull -v source -d target')[0]
383
 
        self.assertContainsRe(out, r'\n {4}1 .*\n {6}setup\n')
384
 
        self.assertNotContainsRe(
385
 
            out, r'revno: 1\ncommitter: .*\nbranch nick: source')
386
 
 
387
 
    def test_pull_smart_bound_branch(self):
388
 
        self.setup_smart_server_with_call_log()
389
 
        parent = self.make_branch_and_tree('parent')
390
 
        parent.commit(message='first commit')
391
 
        child = parent.bzrdir.sprout('child').open_workingtree()
392
 
        child.commit(message='second commit')
393
 
        checkout = parent.branch.create_checkout('checkout')
394
 
        self.run_bzr(['pull', self.get_url('child')], working_dir='checkout')
395
 
 
396
 
    def test_pull_smart_stacked_streaming_acceptance(self):
397
 
        """'bzr pull -r 123' works on stacked, smart branches, even when the
398
 
        revision specified by the revno is only present in the fallback
399
 
        repository.
400
 
 
401
 
        See <https://launchpad.net/bugs/380314>
402
 
        """
403
 
        self.setup_smart_server_with_call_log()
404
 
        # Make a stacked-on branch with two commits so that the
405
 
        # revision-history can't be determined just by looking at the parent
406
 
        # field in the revision in the stacked repo.
407
 
        parent = self.make_branch_and_tree('parent', format='1.9')
408
 
        parent.commit(message='first commit')
409
 
        parent.commit(message='second commit')
410
 
        local = parent.bzrdir.sprout('local').open_workingtree()
411
 
        local.commit(message='local commit')
412
 
        local.branch.create_clone_on_transport(
413
 
            self.get_transport('stacked'), stacked_on=self.get_url('parent'))
414
 
        empty = self.make_branch_and_tree('empty', format='1.9')
415
 
        self.reset_smart_call_log()
416
 
        self.run_bzr(['pull', '-r', '1', self.get_url('stacked')],
417
 
            working_dir='empty')
418
 
        # This figure represent the amount of work to perform this use case. It
419
 
        # is entirely ok to reduce this number if a test fails due to rpc_count
420
 
        # being too low. If rpc_count increases, more network roundtrips have
421
 
        # become necessary for this use case. Please do not adjust this number
422
 
        # upwards without agreement from bzr's network support maintainers.
423
 
        self.assertLength(19, self.hpss_calls)
424
 
        remote = Branch.open('stacked')
425
 
        self.assertEndsWith(remote.get_stacked_on_url(), '/parent')
426
 
 
427
 
    def test_pull_cross_format_warning(self):
428
 
        """You get a warning for probably slow cross-format pulls.
429
 
        """
430
 
        # this is assumed to be going through InterDifferingSerializer
431
 
        from_tree = self.make_branch_and_tree('from', format='2a')
432
 
        to_tree = self.make_branch_and_tree('to', format='1.14-rich-root')
433
 
        from_tree.commit(message='first commit')
434
 
        out, err = self.run_bzr(['pull', '-d', 'to', 'from'])
435
 
        self.assertContainsRe(err,
436
 
            "(?m)Doing on-the-fly conversion")
437
 
 
438
 
    def test_pull_cross_format_warning_no_IDS(self):
439
 
        """You get a warning for probably slow cross-format pulls.
440
 
        """
441
 
        # this simulates what would happen across the network, where
442
 
        # interdifferingserializer is not active
443
 
 
444
 
        debug.debug_flags.add('IDS_never')
445
 
        # TestCase take care of restoring them
446
 
 
447
 
        from_tree = self.make_branch_and_tree('from', format='2a')
448
 
        to_tree = self.make_branch_and_tree('to', format='1.14-rich-root')
449
 
        from_tree.commit(message='first commit')
450
 
        out, err = self.run_bzr(['pull', '-d', 'to', 'from'])
451
 
        self.assertContainsRe(err,
452
 
            "(?m)Doing on-the-fly conversion")
453
 
 
454
 
    def test_pull_cross_format_from_network(self):
455
 
        self.setup_smart_server_with_call_log()
456
 
        from_tree = self.make_branch_and_tree('from', format='2a')
457
 
        to_tree = self.make_branch_and_tree('to', format='1.14-rich-root')
458
 
        self.assertIsInstance(from_tree.branch, remote.RemoteBranch)
459
 
        from_tree.commit(message='first commit')
460
 
        out, err = self.run_bzr(['pull', '-d', 'to',
461
 
            from_tree.branch.bzrdir.root_transport.base])
462
 
        self.assertContainsRe(err,
463
 
            "(?m)Doing on-the-fly conversion")
464
 
 
465
 
    def test_pull_to_experimental_format_warning(self):
466
 
        """You get a warning for pulling into experimental formats.
467
 
        """
468
 
        from_tree = self.make_branch_and_tree('from', format='development-subtree')
469
 
        to_tree = self.make_branch_and_tree('to', format='development-subtree')
470
 
        from_tree.commit(message='first commit')
471
 
        out, err = self.run_bzr(['pull', '-d', 'to', 'from'])
472
 
        self.assertContainsRe(err,
473
 
            "(?m)Fetching into experimental format")
474
 
 
475
 
    def test_pull_cross_to_experimental_format_warning(self):
476
 
        """You get a warning for pulling into experimental formats.
477
 
        """
478
 
        from_tree = self.make_branch_and_tree('from', format='2a')
479
 
        to_tree = self.make_branch_and_tree('to', format='development-subtree')
480
 
        from_tree.commit(message='first commit')
481
 
        out, err = self.run_bzr(['pull', '-d', 'to', 'from'])
482
 
        self.assertContainsRe(err,
483
 
            "(?m)Fetching into experimental format")
484
 
 
485
 
    def test_pull_show_base(self):
486
 
        """bzr pull supports --show-base
487
 
 
488
 
        see https://bugs.launchpad.net/bzr/+bug/202374"""
489
 
        # create two trees with conflicts, setup conflict, check that
490
 
        # conflicted file looks correct
491
 
        a_tree = self.example_branch('a')
492
 
        b_tree = a_tree.bzrdir.sprout('b').open_workingtree()
493
 
 
494
 
        f = open(pathjoin('a', 'hello'),'wt')
495
 
        f.write('fee')
496
 
        f.close()
497
 
        a_tree.commit('fee')
498
 
 
499
 
        f = open(pathjoin('b', 'hello'),'wt')
500
 
        f.write('fie')
501
 
        f.close()
502
 
 
503
 
        out,err=self.run_bzr(['pull','-d','b','a','--show-base'])
504
 
 
505
 
        # check for message here
506
 
        self.assertEqual(err,
507
 
                         ' M  hello\nText conflict in hello\n1 conflicts encountered.\n')
508
 
 
509
 
        self.assertEqualDiff('<<<<<<< TREE\n'
510
 
                             'fie||||||| BASE-REVISION\n'
511
 
                             'foo=======\n'
512
 
                             'fee>>>>>>> MERGE-SOURCE\n',
513
 
                             open(pathjoin('b', 'hello')).read())
514
 
 
515
 
    def test_pull_show_base_working_tree_only(self):
516
 
        """--show-base only allowed if there's a working tree
517
 
 
518
 
        see https://bugs.launchpad.net/bzr/+bug/202374"""
519
 
        # create a branch, see that --show-base fails
520
 
        self.make_branch('from')
521
 
        self.make_branch('to')
522
 
        out=self.run_bzr(['pull','-d','to','from','--show-base'],retcode=3)
523
 
        self.assertEqual(out,
524
 
                         ('','bzr: ERROR: Need working tree for --show-base.\n'))
525
 
 
526
 
    def test_pull_tag_conflicts(self):
527
 
        """pulling tags with conflicts will change the exit code"""
528
 
        # create a branch, see that --show-base fails
529
 
        from_tree = self.make_branch_and_tree('from')
530
 
        from_tree.branch.tags.set_tag("mytag", "somerevid")
531
 
        to_tree = self.make_branch_and_tree('to')
532
 
        to_tree.branch.tags.set_tag("mytag", "anotherrevid")
533
 
        out = self.run_bzr(['pull','-d','to','from'],retcode=1)
534
 
        self.assertEqual(out,
535
 
            ('No revisions to pull.\nConflicting tags:\n    mytag\n', ''))
536
 
 
537
 
    def test_pull_tag_notification(self):
538
 
        """pulling tags with conflicts will change the exit code"""
539
 
        # create a branch, see that --show-base fails
540
 
        from_tree = self.make_branch_and_tree('from')
541
 
        from_tree.branch.tags.set_tag("mytag", "somerevid")
542
 
        to_tree = self.make_branch_and_tree('to')
543
 
        out = self.run_bzr(['pull', '-d', 'to', 'from'])
544
 
        self.assertEqual(out,
545
 
            ('1 tag(s) updated.\n', ''))
546
 
 
547
 
    def test_pull_tag_overwrite(self):
548
 
        """pulling tags with --overwrite only reports changed tags."""
549
 
        # create a branch, see that --show-base fails
550
 
        from_tree = self.make_branch_and_tree('from')
551
 
        from_tree.branch.tags.set_tag("mytag", "somerevid")
552
 
        to_tree = self.make_branch_and_tree('to')
553
 
        to_tree.branch.tags.set_tag("mytag", "somerevid")
554
 
        out = self.run_bzr(['pull', '--overwrite', '-d', 'to', 'from'])
555
 
        self.assertEqual(out,
556
 
            ('No revisions or tags to pull.\n', ''))
557
 
 
558
 
 
559
 
class TestPullOutput(script.TestCaseWithTransportAndScript):
560
 
 
561
 
    def test_pull_log_format(self):
562
 
        self.run_script("""
563
 
            $ bzr init trunk
564
 
            Created a standalone tree (format: 2a)
565
 
            $ cd trunk
566
 
            $ echo foo > file
567
 
            $ bzr add
568
 
            adding file
569
 
            $ bzr commit -m 'we need some foo'
570
 
            2>Committing to:...trunk/
571
 
            2>added file
572
 
            2>Committed revision 1.
573
 
            $ cd ..
574
 
            $ bzr init feature
575
 
            Created a standalone tree (format: 2a)
576
 
            $ cd feature
577
 
            $ bzr pull -v ../trunk -Olog_format=line
578
 
            Now on revision 1.
579
 
            Added Revisions:
580
 
            1: jrandom@example.com ...we need some foo
581
 
            2>+N  file
582
 
            2>All changes applied successfully.
583
 
            """)
 
219
        os.chdir('../b')
 
220
        bzr('pull', '--overwrite', '../a')
 
221
        rev_history_b = get_rh(4)
 
222
 
 
223
        self.assertEqual(rev_history_b, rev_history_a)
 
224
 
 
225