1
# Copyright (C) 2006, 2007 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
# Author: Aaron Bentley <aaron.bentley@utoronto.ca>
19
"""Black-box tests for bzr merge.
24
from bzrlib import merge_directive
25
from bzrlib.branch import Branch
26
from bzrlib.bzrdir import BzrDir
27
from bzrlib.conflicts import ConflictList, ContentsConflict
28
from bzrlib.osutils import abspath, file_kind, pathjoin
29
from bzrlib.tests.blackbox import ExternalBase
30
import bzrlib.urlutils as urlutils
31
from bzrlib.workingtree import WorkingTree
34
class TestMerge(ExternalBase):
36
def example_branch(self, path='.'):
37
tree = self.make_branch_and_tree(path)
38
self.build_tree_contents([
39
(pathjoin(path, 'hello'), 'foo'),
40
(pathjoin(path, 'goodbye'), 'baz')])
42
tree.commit(message='setup')
44
tree.commit(message='setup')
47
def test_merge_reprocess(self):
48
d = BzrDir.create_standalone_workingtree('.')
50
self.run_bzr('merge . --reprocess --merge-type weave')
53
from bzrlib.branch import Branch
55
a_tree = self.example_branch('a')
56
ancestor = a_tree.branch.revno()
57
b_tree = a_tree.bzrdir.sprout('b').open_workingtree()
58
self.build_tree_contents([('b/goodbye', 'quux')])
59
b_tree.commit(message="more u's are always good")
61
self.build_tree_contents([('a/hello', 'quuux')])
62
# We can't merge when there are in-tree changes
64
self.run_bzr('merge ../b', retcode=3)
65
a = WorkingTree.open('.')
66
a_tip = a.commit("Like an epidemic of u's")
67
self.run_bzr('merge ../b -r last:1..last:1 --merge-type blooof',
69
self.run_bzr('merge ../b -r last:1..last:1 --merge-type merge3')
70
a_tree.revert(backups=False)
71
self.run_bzr('merge ../b -r last:1..last:1 --merge-type weave')
72
a_tree.revert(backups=False)
73
self.run_bzr_error(['Show-base is not supported for this merge type'],
74
'merge ../b -r last:1..last:1 --merge-type weave'
76
a_tree.revert(backups=False)
77
self.run_bzr('merge ../b -r last:1..last:1 --reprocess')
78
a_tree.revert(backups=False)
79
self.run_bzr('merge ../b -r last:1')
80
self.check_file_contents('goodbye', 'quux')
81
# Merging a branch pulls its revision into the tree
82
b = Branch.open('../b')
83
b_tip = b.last_revision()
84
self.failUnless(a.branch.repository.has_revision(b_tip))
85
self.assertEqual([a_tip, b_tip], a.get_parent_ids())
86
a_tree.revert(backups=False)
87
out, err = self.run_bzr('merge -r revno:1:./hello', retcode=3)
88
self.assertTrue("Not a branch" in err)
89
self.run_bzr('merge -r revno:%d:./..revno:%d:../b'
90
%(ancestor,b.revno()))
91
self.assertEquals(a.get_parent_ids(),
92
[a.branch.last_revision(), b.last_revision()])
93
self.check_file_contents('goodbye', 'quux')
94
a_tree.revert(backups=False)
95
self.run_bzr('merge -r revno:%d:../b'%b.revno())
96
self.assertEquals(a.get_parent_ids(),
97
[a.branch.last_revision(), b.last_revision()])
98
a_tip = a.commit('merged')
99
self.run_bzr('merge ../b -r last:1')
100
self.assertEqual([a_tip], a.get_parent_ids())
102
def test_merge_with_missing_file(self):
103
"""Merge handles missing file conflicts"""
104
self.build_tree_contents([
107
('a/sub/a.txt', 'hello\n'),
108
('a/b.txt', 'hello\n'),
109
('a/sub/c.txt', 'hello\n')])
110
a_tree = self.make_branch_and_tree('a')
111
a_tree.add(['sub', 'b.txt', 'sub/c.txt', 'sub/a.txt'])
112
a_tree.commit(message='added a')
113
b_tree = a_tree.bzrdir.sprout('b').open_workingtree()
114
self.build_tree_contents([
115
('a/sub/a.txt', 'hello\nthere\n'),
116
('a/b.txt', 'hello\nthere\n'),
117
('a/sub/c.txt', 'hello\nthere\n')])
118
a_tree.commit(message='Added there')
119
os.remove('a/sub/a.txt')
120
os.remove('a/sub/c.txt')
123
a_tree.commit(message='Removed a.txt')
124
self.build_tree_contents([
125
('b/sub/a.txt', 'hello\nsomething\n'),
126
('b/b.txt', 'hello\nsomething\n'),
127
('b/sub/c.txt', 'hello\nsomething\n')])
128
b_tree.commit(message='Modified a.txt')
130
self.run_bzr('merge ../a/', retcode=1)
131
self.failUnlessExists('sub/a.txt.THIS')
132
self.failUnlessExists('sub/a.txt.BASE')
134
self.run_bzr('merge ../b/', retcode=1)
135
self.failUnlessExists('sub/a.txt.OTHER')
136
self.failUnlessExists('sub/a.txt.BASE')
138
def test_merge_remember(self):
139
"""Merge changes from one branch to another and test parent location."""
140
tree_a = self.make_branch_and_tree('branch_a')
141
branch_a = tree_a.branch
142
self.build_tree(['branch_a/a'])
144
tree_a.commit('commit a')
145
branch_b = branch_a.bzrdir.sprout('branch_b').open_branch()
146
tree_b = branch_b.bzrdir.open_workingtree()
147
branch_c = branch_a.bzrdir.sprout('branch_c').open_branch()
148
tree_c = branch_c.bzrdir.open_workingtree()
149
self.build_tree(['branch_a/b'])
151
tree_a.commit('commit b')
152
self.build_tree(['branch_c/c'])
154
tree_c.commit('commit c')
156
parent = branch_b.get_parent()
157
branch_b.set_parent(None)
158
self.assertEqual(None, branch_b.get_parent())
159
# test merge for failure without parent set
161
out = self.run_bzr('merge', retcode=3)
162
self.assertEquals(out,
163
('','bzr: ERROR: No location specified or remembered\n'))
164
# test implicit --remember when no parent set, this merge conflicts
165
self.build_tree(['d'])
167
self.run_bzr_error(['Working tree ".*" has uncommitted changes'],
169
self.assertEquals(abspath(branch_b.get_parent()), abspath(parent))
170
# test implicit --remember after resolving conflict
171
tree_b.commit('commit d')
172
out, err = self.run_bzr('merge')
174
base = urlutils.local_path_from_url(branch_a.base)
175
self.assertEquals(out, 'Merging from remembered location %s\n' % (base,))
176
self.assertEquals(err, '+N b\nAll changes applied successfully.\n')
177
self.assertEquals(abspath(branch_b.get_parent()), abspath(parent))
178
# re-open tree as external run_bzr modified it
179
tree_b = branch_b.bzrdir.open_workingtree()
180
tree_b.commit('merge branch_a')
181
# test explicit --remember
182
out, err = self.run_bzr('merge ../branch_c --remember')
183
self.assertEquals(out, '')
184
self.assertEquals(err, '+N c\nAll changes applied successfully.\n')
185
self.assertEquals(abspath(branch_b.get_parent()),
186
abspath(branch_c.bzrdir.root_transport.base))
187
# re-open tree as external run_bzr modified it
188
tree_b = branch_b.bzrdir.open_workingtree()
189
tree_b.commit('merge branch_c')
191
def test_merge_bundle(self):
192
from bzrlib.testament import Testament
193
tree_a = self.make_branch_and_tree('branch_a')
194
self.build_tree_contents([('branch_a/a', 'hello')])
196
tree_a.commit('message')
198
tree_b = tree_a.bzrdir.sprout('branch_b').open_workingtree()
199
self.build_tree_contents([('branch_a/a', 'hey there')])
200
tree_a.commit('message')
202
self.build_tree_contents([('branch_b/a', 'goodbye')])
203
tree_b.commit('message')
205
self.run_bzr('bundle ../branch_a -o ../bundle')
206
os.chdir('../branch_a')
207
self.run_bzr('merge ../bundle', retcode=1)
208
testament_a = Testament.from_revision(tree_a.branch.repository,
209
tree_b.get_parent_ids()[0])
210
testament_b = Testament.from_revision(tree_b.branch.repository,
211
tree_b.get_parent_ids()[0])
212
self.assertEqualDiff(testament_a.as_text(),
213
testament_b.as_text())
214
tree_a.set_conflicts(ConflictList())
215
tree_a.commit('message')
216
# it is legal to attempt to merge an already-merged bundle
217
output = self.run_bzr('merge ../bundle')[1]
218
# but it does nothing
219
self.assertFalse(tree_a.changes_from(tree_a.basis_tree()).has_changed())
220
self.assertEqual('Nothing to do.\n', output)
222
def test_merge_uncommitted(self):
223
"""Check that merge --uncommitted behaves properly"""
224
tree_a = self.make_branch_and_tree('a')
225
self.build_tree(['a/file_1', 'a/file_2'])
226
tree_a.add(['file_1', 'file_2'])
227
tree_a.commit('commit 1')
228
tree_b = tree_a.bzrdir.sprout('b').open_workingtree()
229
self.failUnlessExists('b/file_1')
230
tree_a.rename_one('file_1', 'file_i')
231
tree_a.commit('commit 2')
232
tree_a.rename_one('file_2', 'file_ii')
234
self.run_bzr('merge a --uncommitted -d b')
235
self.failUnlessExists('b/file_1')
236
self.failUnlessExists('b/file_ii')
238
self.run_bzr_error(('Cannot use --uncommitted and --revision',),
239
'merge /a --uncommitted -r1 -d b')
241
def test_merge_uncommitted_file(self):
242
"""It should be possible to merge changes from a single file."""
243
tree_a = self.make_branch_and_tree('tree_a')
244
tree_a.commit('initial commit')
245
tree_a.bzrdir.sprout('tree_b')
246
self.build_tree(['tree_a/file1', 'tree_a/file2'])
247
tree_a.add(['file1', 'file2'])
249
self.run_bzr(['merge', '--uncommitted', '../tree_a/file1'])
250
self.failUnlessExists('file1')
251
self.failIfExists('file2')
253
def pullable_branch(self):
254
tree_a = self.make_branch_and_tree('a')
255
self.build_tree(['a/file'])
257
self.id1 = tree_a.commit('commit 1')
259
tree_b = self.make_branch_and_tree('b')
260
tree_b.pull(tree_a.branch)
261
file('b/file', 'wb').write('foo')
262
self.id2 = tree_b.commit('commit 2')
264
def test_merge_pull(self):
265
self.pullable_branch()
267
(out, err) = self.run_bzr('merge --pull ../b')
268
self.assertContainsRe(out, 'Now on revision 2\\.')
269
tree_a = WorkingTree.open('.')
270
self.assertEqual([self.id2], tree_a.get_parent_ids())
272
def test_merge_kind_change(self):
273
tree_a = self.make_branch_and_tree('tree_a')
274
self.build_tree_contents([('tree_a/file', 'content_1')])
275
tree_a.add('file', 'file-id')
276
tree_a.commit('added file')
277
tree_b = tree_a.bzrdir.sprout('tree_b').open_workingtree()
278
os.unlink('tree_a/file')
279
self.build_tree(['tree_a/file/'])
280
tree_a.commit('changed file to directory')
282
self.run_bzr('merge ../tree_a')
283
self.assertEqual('directory', file_kind('file'))
285
self.assertEqual('file', file_kind('file'))
286
self.build_tree_contents([('file', 'content_2')])
287
tree_b.commit('content change')
288
self.run_bzr('merge ../tree_a', retcode=1)
289
self.assertEqual(tree_b.conflicts(),
290
[ContentsConflict('file', file_id='file-id')])
292
def test_directive_cherrypick(self):
293
source = self.make_branch_and_tree('source')
294
self.build_tree(['source/a'])
296
source.commit('Added a', rev_id='rev1')
297
self.build_tree(['source/b'])
299
source.commit('Added b', rev_id='rev2')
300
target = self.make_branch_and_tree('target')
301
target.commit('empty commit')
302
self.write_directive('directive', source.branch, 'target', 'rev2',
304
self.run_bzr('merge -d target directive')
305
self.failIfExists('target/a')
306
self.failUnlessExists('target/b')
308
def write_directive(self, filename, source, target, revision_id,
309
base_revision_id=None, mangle_patch=False):
310
md = merge_directive.MergeDirective2.from_objects(
311
source.repository, revision_id, 0, 0, target,
312
base_revision_id=base_revision_id)
315
self.build_tree_contents([(filename, ''.join(md.to_lines()))])
317
def test_directive_verify_warning(self):
318
source = self.make_branch_and_tree('source')
319
self.build_tree(['source/a'])
321
source.commit('Added a', rev_id='rev1')
322
target = self.make_branch_and_tree('target')
323
target.commit('empty commit')
324
self.write_directive('directive', source.branch, 'target', 'rev1')
325
err = self.run_bzr('merge -d target directive')[1]
326
self.assertNotContainsRe(err, 'Preview patch does not match changes')
328
self.write_directive('directive', source.branch, 'target', 'rev1',
330
err = self.run_bzr('merge -d target directive')[1]
331
self.assertContainsRe(err, 'Preview patch does not match changes')
333
def test_merge_arbitrary(self):
334
target = self.make_branch_and_tree('target')
335
target.commit('empty')
336
# We need a revision that has no integer revno
337
branch_a = target.bzrdir.sprout('branch_a').open_workingtree()
338
self.build_tree(['branch_a/file1'])
339
branch_a.add('file1')
340
branch_a.commit('added file1', rev_id='rev2a')
341
branch_b = target.bzrdir.sprout('branch_b').open_workingtree()
342
self.build_tree(['branch_b/file2'])
343
branch_b.add('file2')
344
branch_b.commit('added file2', rev_id='rev2b')
345
branch_b.merge_from_branch(branch_a.branch)
346
self.failUnlessExists('branch_b/file1')
347
branch_b.commit('merged branch_a', rev_id='rev3b')
349
# It works if the revid has an interger revno
350
self.run_bzr('merge -d target -r revid:rev2a branch_a')
351
self.failUnlessExists('target/file1')
352
self.failIfExists('target/file2')
355
# It should work if the revid has no integer revno
356
self.run_bzr('merge -d target -r revid:rev2a branch_b')
357
self.failUnlessExists('target/file1')
358
self.failIfExists('target/file2')
360
def assertDirectoryContent(self, directory, entries, message=''):
361
"""Assert whether entries (file or directories) exist in a directory.
363
It also checks that there are no extra entries.
365
ondisk = os.listdir(directory)
366
if set(ondisk) == set(entries):
370
raise AssertionError(
371
'%s"%s" directory content is different:\na = %s\nb = %s\n'
372
% (message, directory, sorted(entries), sorted(ondisk)))
374
def test_cherrypicking_merge(self):
376
source = self.make_branch_and_tree('source')
377
for f in ('a', 'b', 'c', 'd'):
378
self.build_tree(['source/'+f])
380
source.commit('added '+f, rev_id='rev_'+f)
382
target = source.bzrdir.sprout('target', 'rev_a').open_workingtree()
383
self.assertDirectoryContent('target', ['.bzr', 'a'])
385
self.run_bzr('merge -d target -r revid:rev_b..revid:rev_c source')
386
self.assertDirectoryContent('target', ['.bzr', 'a', 'c'])
389
self.run_bzr('merge -d target -r revid:rev_b..revid:rev_d source')
390
self.assertDirectoryContent('target', ['.bzr', 'a', 'c', 'd'])
392
# pick 1 revision with option --changes
393
self.run_bzr('merge -d target -c revid:rev_d source')
394
self.assertDirectoryContent('target', ['.bzr', 'a', 'd'])
added
removed
bzrlib/tests/blackbox/test_merge.py
