14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
"""Implementation tests for bzrlib.merge.Merger."""
17
"""Implementation tests for breezy.merge.Merger."""
21
from bzrlib.conflicts import TextConflict
21
from ..conflicts import TextConflict
24
24
merge as _mod_merge,
28
from bzrlib.tests import (
30
28
TestCaseWithTransport,
32
from bzrlib.tests.test_merge_core import MergeBuilder
33
from bzrlib.transform import TreeTransform
37
def load_tests(standard_tests, module, loader):
30
from .test_merge_core import MergeBuilder
31
from ..transform import TreeTransform
35
def load_tests(loader, standard_tests, pattern):
38
36
"""Multiply tests for tranport implementations."""
39
37
result = loader.suiteClass()
41
39
(name, {'merge_type': merger})
42
for name, merger in option._merge_type_registry.items()]
40
for name, merger in _mod_merge.merge_type_registry.items()]
43
41
return multiply_tests(standard_tests, scenarios, result)
46
44
class TestMergeImplementation(TestCaseWithTransport):
48
46
def do_merge(self, target_tree, source_tree, **kwargs):
49
merger = _mod_merge.Merger.from_revision_ids(None,
47
merger = _mod_merge.Merger.from_revision_ids(
50
48
target_tree, source_tree.last_revision(),
51
49
other_branch=source_tree.branch)
52
50
merger.merge_type=self.merge_type
59
57
this_tree.lock_write()
60
58
self.addCleanup(this_tree.unlock)
61
59
self.build_tree_contents([
62
('this/file1', 'a\nb\n'),
63
('this/file2', 'a\nb\n')
60
('this/file1', b'a\nb\n'),
61
('this/file2', b'a\nb\n')
65
63
this_tree.add(['file1', 'file2'])
66
64
this_tree.commit('Added files')
67
other_tree = this_tree.bzrdir.sprout('other').open_workingtree()
65
other_tree = this_tree.controldir.sprout('other').open_workingtree()
68
66
self.build_tree_contents([
69
('other/file1', 'a\nb\nc\n'),
70
('other/file2', 'a\nb\nc\n')
67
('other/file1', b'a\nb\nc\n'),
68
('other/file2', b'a\nb\nc\n')
72
70
other_tree.commit('modified both')
73
71
self.build_tree_contents([
74
('this/file1', 'd\na\nb\n'),
75
('this/file2', 'd\na\nb\n')
72
('this/file1', b'd\na\nb\n'),
73
('this/file2', b'd\na\nb\n')
77
75
this_tree.commit('modified both')
78
76
self.do_merge(this_tree, other_tree, interesting_files=['file1'])
79
self.assertFileEqual('d\na\nb\nc\n', 'this/file1')
80
self.assertFileEqual('d\na\nb\n', 'this/file2')
77
self.assertFileEqual(b'd\na\nb\nc\n', 'this/file1')
78
self.assertFileEqual(b'd\na\nb\n', 'this/file2')
82
80
def test_merge_move_and_change(self):
83
81
this_tree = self.make_branch_and_tree('this')
84
82
this_tree.lock_write()
85
83
self.addCleanup(this_tree.unlock)
86
84
self.build_tree_contents([
87
('this/file1', 'line 1\nline 2\nline 3\nline 4\n'),
85
('this/file1', b'line 1\nline 2\nline 3\nline 4\n'),
89
87
this_tree.add('file1',)
90
88
this_tree.commit('Added file')
91
other_tree = this_tree.bzrdir.sprout('other').open_workingtree()
89
other_tree = this_tree.controldir.sprout('other').open_workingtree()
92
90
self.build_tree_contents([
93
('other/file1', 'line 1\nline 2 to 2.1\nline 3\nline 4\n'),
91
('other/file1', b'line 1\nline 2 to 2.1\nline 3\nline 4\n'),
95
93
other_tree.commit('Changed 2 to 2.1')
96
94
self.build_tree_contents([
97
('this/file1', 'line 1\nline 3\nline 2\nline 4\n'),
95
('this/file1', b'line 1\nline 3\nline 2\nline 4\n'),
99
97
this_tree.commit('Swapped 2 & 3')
100
98
self.do_merge(this_tree, other_tree)
102
100
self.expectFailure(
103
101
"lca merge doesn't conflict for move and change",
104
102
self.assertFileEqual,
112
'>>>>>>> MERGE-SOURCE\n'
113
'line 4\n', 'this/file1')
110
b'>>>>>>> MERGE-SOURCE\n'
111
b'line 4\n', 'this/file1')
115
self.assertFileEqual('line 1\n'
122
'>>>>>>> MERGE-SOURCE\n'
123
'line 4\n', 'this/file1')
113
self.assertFileEqual(b'line 1\n'
120
b'>>>>>>> MERGE-SOURCE\n'
121
b'line 4\n', 'this/file1')
125
123
def test_modify_conflicts_with_delete(self):
126
124
# If one side deletes a line, and the other modifies that line, then
127
125
# the modification should be considered a conflict
128
126
builder = self.make_branch_builder('test')
129
127
builder.start_series()
130
builder.build_snapshot('BASE-id', None,
128
builder.build_snapshot(None,
131
129
[('add', ('', None, 'directory', None)),
132
('add', ('foo', 'foo-id', 'file', 'a\nb\nc\nd\ne\n')),
130
('add', ('foo', b'foo-id', 'file', b'a\nb\nc\nd\ne\n')),
131
], revision_id=b'BASE-id')
135
builder.build_snapshot('OTHER-id', ['BASE-id'],
136
[('modify', ('foo-id', 'a\nc\nd\ne\n'))])
133
builder.build_snapshot([b'BASE-id'],
134
[('modify', ('foo', b'a\nc\nd\ne\n'))],
135
revision_id=b'OTHER-id')
137
136
# Modify 'b\n', add 'X\n'
138
builder.build_snapshot('THIS-id', ['BASE-id'],
139
[('modify', ('foo-id', 'a\nb2\nc\nd\nX\ne\n'))])
137
builder.build_snapshot([b'BASE-id'],
138
[('modify', ('foo', b'a\nb2\nc\nd\nX\ne\n'))],
139
revision_id=b'THIS-id')
140
140
builder.finish_series()
141
141
branch = builder.get_branch()
142
this_tree = branch.bzrdir.create_workingtree()
142
this_tree = branch.controldir.create_workingtree()
143
143
this_tree.lock_write()
144
144
self.addCleanup(this_tree.unlock)
145
other_tree = this_tree.bzrdir.sprout('other',
146
'OTHER-id').open_workingtree()
145
other_tree = this_tree.controldir.sprout('other', b'OTHER-id').open_workingtree()
147
146
self.do_merge(this_tree, other_tree)
148
147
if self.merge_type is _mod_merge.LCAMerger:
149
148
self.expectFailure("lca merge doesn't track deleted lines",
150
149
self.assertFileEqual,
155
'>>>>>>> MERGE-SOURCE\n'
154
b'>>>>>>> MERGE-SOURCE\n'
161
160
self.assertFileEqual(
166
'>>>>>>> MERGE-SOURCE\n'
165
b'>>>>>>> MERGE-SOURCE\n'
172
171
def get_limbodir_deletiondir(self, wt):
173
172
transform = TreeTransform(wt)
176
175
transform.finalize()
177
176
return (limbodir, deletiondir)
179
def test_merge_with_existing_limbo(self):
180
wt = self.make_branch_and_tree('this')
181
(limbodir, deletiondir) = self.get_limbodir_deletiondir(wt)
178
def test_merge_with_existing_limbo_empty(self):
179
"""Empty limbo dir is just cleaned up - see bug 427773"""
180
wt = self.make_branch_and_tree('this')
181
(limbodir, deletiondir) = self.get_limbodir_deletiondir(wt)
183
self.do_merge(wt, wt)
185
def test_merge_with_existing_limbo_non_empty(self):
186
wt = self.make_branch_and_tree('this')
187
(limbodir, deletiondir) = self.get_limbodir_deletiondir(wt)
189
os.mkdir(os.path.join(limbodir, 'something'))
183
190
self.assertRaises(errors.ExistingLimbo, self.do_merge, wt, wt)
184
191
self.assertRaises(errors.LockError, wt.unlock)
186
def test_merge_with_pending_deletion(self):
187
wt = self.make_branch_and_tree('this')
188
(limbodir, deletiondir) = self.get_limbodir_deletiondir(wt)
189
os.mkdir(deletiondir)
193
def test_merge_with_pending_deletion_empty(self):
194
wt = self.make_branch_and_tree('this')
195
(limbodir, deletiondir) = self.get_limbodir_deletiondir(wt)
196
os.mkdir(deletiondir)
197
self.do_merge(wt, wt)
199
def test_merge_with_pending_deletion_non_empty(self):
200
"""Also see bug 427773"""
201
wt = self.make_branch_and_tree('this')
202
(limbodir, deletiondir) = self.get_limbodir_deletiondir(wt)
203
os.mkdir(deletiondir)
204
os.mkdir(os.path.join(deletiondir, 'something'))
190
205
self.assertRaises(errors.ExistingPendingDeletion, self.do_merge, wt, wt)
191
206
self.assertRaises(errors.LockError, wt.unlock)
286
301
def create_file_needing_contents_merge(self, builder, file_id):
287
builder.add_file(file_id, builder.tree_root, "name1", "text1", True)
288
builder.change_contents(file_id, other="text4", this="text3")
302
builder.add_file(file_id, builder.tree_root, "name1", b"text1", True)
303
builder.change_contents(file_id, other=b"text4", this=b"text3")
290
305
def test_change_vs_change(self):
291
306
"""Hook is used for (changed, changed)"""
292
307
self.install_hook_success()
293
308
builder = self.make_merge_builder()
294
builder.add_file("1", builder.tree_root, "name1", "text1", True)
295
builder.change_contents("1", other="text4", this="text3")
309
builder.add_file(b"1", builder.tree_root, "name1", b"text1", True)
310
builder.change_contents(b"1", other=b"text4", this=b"text3")
296
311
conflicts = builder.merge(self.merge_type)
297
312
self.assertEqual(conflicts, [])
298
313
self.assertEqual(
299
builder.this.get_file('1').read(), 'text-merged-by-hook')
314
builder.this.get_file('name1').read(), b'text-merged-by-hook')
301
316
def test_change_vs_deleted(self):
302
317
"""Hook is used for (changed, deleted)"""
303
318
self.install_hook_success()
304
319
builder = self.make_merge_builder()
305
builder.add_file("1", builder.tree_root, "name1", "text1", True)
306
builder.change_contents("1", this="text2")
307
builder.remove_file("1", other=True)
320
builder.add_file(b"1", builder.tree_root, "name1", b"text1", True)
321
builder.change_contents(b"1", this=b"text2")
322
builder.remove_file(b"1", other=True)
308
323
conflicts = builder.merge(self.merge_type)
309
324
self.assertEqual(conflicts, [])
310
325
self.assertEqual(
311
builder.this.get_file('1').read(), 'text-merged-by-hook')
326
builder.this.get_file('name1').read(), b'text-merged-by-hook')
313
328
def test_result_can_be_delete(self):
314
329
"""A hook's result can be the deletion of a file."""
315
330
self.install_hook_delete()
316
331
builder = self.make_merge_builder()
317
self.create_file_needing_contents_merge(builder, "1")
332
self.create_file_needing_contents_merge(builder, b"1")
318
333
conflicts = builder.merge(self.merge_type)
319
334
self.assertEqual(conflicts, [])
320
self.assertRaises(errors.NoSuchId, builder.this.id2path, '1')
335
self.assertRaises(errors.NoSuchId, builder.this.id2path, b'1')
321
336
self.assertEqual([], list(builder.this.list_files()))
323
338
def test_result_can_be_conflict(self):
324
339
"""A hook's result can be a conflict."""
325
340
self.install_hook_conflict()
326
341
builder = self.make_merge_builder()
327
self.create_file_needing_contents_merge(builder, "1")
342
self.create_file_needing_contents_merge(builder, b"1")
328
343
conflicts = builder.merge(self.merge_type)
329
self.assertEqual(conflicts, [TextConflict('name1', file_id='1')])
344
self.assertEqual(conflicts, [TextConflict('name1', file_id=b'1')])
330
345
# The hook still gets to set the file contents in this case, so that it
331
346
# can insert custom conflict markers.
332
347
self.assertEqual(
333
builder.this.get_file('1').read(),
334
'text-with-conflict-markers-from-hook')
348
builder.this.get_file('name1').read(),
349
b'text-with-conflict-markers-from-hook')
336
351
def test_can_access_this_other_and_base_versions(self):
337
352
"""The hook function can call params.merger.get_lines to access the
340
355
self.install_hook_log_lines()
341
356
builder = self.make_merge_builder()
342
builder.add_file("1", builder.tree_root, "name1", "text1", True)
343
builder.change_contents("1", this="text2", other="text3")
357
builder.add_file(b"1", builder.tree_root, "name1", b"text1", True)
358
builder.change_contents(b"1", this=b"text2", other=b"text3")
344
359
conflicts = builder.merge(self.merge_type)
345
360
self.assertEqual(
346
[('log_lines', ['text2'], ['text3'], ['text1'])], self.hook_log)
361
[('log_lines', [b'text2'], [b'text3'], [b'text1'])], self.hook_log)
348
363
def test_chain_when_not_active(self):
349
364
"""When a hook function returns None, merging still works."""
350
365
self.install_hook_inactive()
351
366
self.install_hook_success()
352
367
builder = self.make_merge_builder()
353
self.create_file_needing_contents_merge(builder, "1")
368
self.create_file_needing_contents_merge(builder, b"1")
354
369
conflicts = builder.merge(self.merge_type)
355
370
self.assertEqual(conflicts, [])
356
371
self.assertEqual(
357
builder.this.get_file('1').read(), 'text-merged-by-hook')
372
builder.this.get_file('name1').read(), b'text-merged-by-hook')
358
373
self.assertEqual([('inactive',), ('success',)], self.hook_log)
360
375
def test_chain_when_not_applicable(self):