~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_merge.py

  • Committer: Aaron Bentley
  • Date: 2007-01-20 01:44:46 UTC
  • mto: This revision was merged to the branch mainline in revision 2241.
  • Revision ID: aaron.bentley@utoronto.ca-20070120014446-2znre037lrj823cd
Cleanup formatting and error handling

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2005, 2006 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
import os
18
18
from StringIO import StringIO
19
19
 
20
 
from bzrlib import (
21
 
    conflicts,
22
 
    errors,
23
 
    knit,
24
 
    memorytree,
25
 
    merge as _mod_merge,
26
 
    option,
27
 
    progress,
28
 
    tests,
29
 
    transform,
30
 
    versionedfile,
31
 
    )
 
20
from bzrlib import conflicts
 
21
from bzrlib.branch import Branch
 
22
from bzrlib.builtins import merge
32
23
from bzrlib.conflicts import ConflictList, TextConflict
33
 
from bzrlib.errors import UnrelatedBranches, NoCommits
34
 
from bzrlib.merge import transform_tree, merge_inner, _PlanMerge
35
 
from bzrlib.osutils import pathjoin, file_kind
36
 
from bzrlib.tests import (
37
 
    TestCaseWithMemoryTransport,
38
 
    TestCaseWithTransport,
39
 
    test_merge_core,
40
 
    )
 
24
from bzrlib.errors import UnrelatedBranches, NoCommits, BzrCommandError
 
25
from bzrlib.merge import transform_tree, merge_inner
 
26
from bzrlib.osutils import pathjoin
 
27
from bzrlib.revision import common_ancestor
 
28
from bzrlib.tests import TestCaseWithTransport
 
29
from bzrlib.trace import (enable_test_log, disable_test_log)
41
30
from bzrlib.workingtree import WorkingTree
42
31
 
43
32
 
48
37
        wt = self.make_branch_and_tree('.')
49
38
        rev_a = wt.commit("lala!")
50
39
        self.assertEqual([rev_a], wt.get_parent_ids())
51
 
        self.assertRaises(errors.PointlessMerge, wt.merge_from_branch,
52
 
                          wt.branch)
 
40
        merge([u'.', -1], [None, None])
53
41
        self.assertEqual([rev_a], wt.get_parent_ids())
54
 
        return wt
55
42
 
56
43
    def test_undo(self):
57
44
        wt = self.make_branch_and_tree('.')
58
45
        wt.commit("lala!")
59
46
        wt.commit("haha!")
60
47
        wt.commit("blabla!")
61
 
        wt.merge_from_branch(wt.branch, wt.branch.get_rev_id(2),
62
 
                             wt.branch.get_rev_id(1))
 
48
        merge([u'.', 2], [u'.', 1])
63
49
 
64
50
    def test_nocommits(self):
65
 
        wt = self.test_pending()
 
51
        self.test_pending()
66
52
        wt2 = self.make_branch_and_tree('branch2')
67
 
        self.assertRaises(NoCommits, wt.merge_from_branch, wt2.branch)
68
 
        return wt, wt2
 
53
        self.assertRaises(NoCommits, merge, ['branch2', -1], 
 
54
                          [None, None])
 
55
        return wt2
69
56
 
70
57
    def test_unrelated(self):
71
 
        wt, wt2 = self.test_nocommits()
 
58
        wt2 = self.test_nocommits()
72
59
        wt2.commit("blah")
73
 
        self.assertRaises(UnrelatedBranches, wt.merge_from_branch, wt2.branch)
 
60
        self.assertRaises(UnrelatedBranches, merge, ['branch2', -1], 
 
61
                          [None, None])
74
62
        return wt2
75
63
 
76
64
    def test_merge_one_file(self):
85
73
        wt1.add('bar')
86
74
        wt1.commit('add foobar')
87
75
        os.chdir('branch2')
88
 
        self.run_bzr('merge ../branch1/baz', retcode=3)
89
 
        self.run_bzr('merge ../branch1/foo')
 
76
        self.run_bzr('merge', '../branch1/baz', retcode=3)
 
77
        self.run_bzr('merge', '../branch1/foo')
90
78
        self.failUnlessExists('foo')
91
79
        self.failIfExists('bar')
92
80
        wt2 = WorkingTree.open('.') # opens branch2
93
81
        self.assertEqual([tip], wt2.get_parent_ids())
94
 
 
 
82
        
95
83
    def test_pending_with_null(self):
96
84
        """When base is forced to revno 0, parent_ids are set"""
97
85
        wt2 = self.test_unrelated()
98
86
        wt1 = WorkingTree.open('.')
99
87
        br1 = wt1.branch
100
88
        br1.fetch(wt2.branch)
101
 
        # merge all of branch 2 into branch 1 even though they
 
89
        # merge all of branch 2 into branch 1 even though they 
102
90
        # are not related.
103
 
        wt1.merge_from_branch(wt2.branch, wt2.last_revision(), 'null:')
 
91
        self.assertRaises(BzrCommandError, merge, ['branch2', -1],
 
92
                          ['branch2', 0], reprocess=True, show_base=True)
 
93
        merge(['branch2', -1], ['branch2', 0], reprocess=True)
104
94
        self.assertEqual([br1.last_revision(), wt2.branch.last_revision()],
105
95
            wt1.get_parent_ids())
106
96
        return (wt1, wt2.branch)
109
99
        """Merge base is sane when two unrelated branches are merged"""
110
100
        wt1, br2 = self.test_pending_with_null()
111
101
        wt1.commit("blah")
112
 
        wt1.lock_read()
113
 
        try:
114
 
            last = wt1.branch.last_revision()
115
 
            last2 = br2.last_revision()
116
 
            graph = wt1.branch.repository.get_graph()
117
 
            self.assertEqual(last2, graph.find_unique_lca(last, last2))
118
 
        finally:
119
 
            wt1.unlock()
 
102
        last = wt1.branch.last_revision()
 
103
        self.assertEqual(common_ancestor(last, last, wt1.branch.repository), last)
120
104
 
121
105
    def test_create_rename(self):
122
106
        """Rename an inventory entry while creating the file"""
148
132
        tree_a.commit(message="hello")
149
133
        dir_b = tree_a.bzrdir.sprout('b')
150
134
        tree_b = dir_b.open_workingtree()
151
 
        tree_b.lock_write()
152
 
        self.addCleanup(tree_b.unlock)
153
135
        tree_a.commit(message="hello again")
154
136
        log = StringIO()
155
 
        merge_inner(tree_b.branch, tree_a, tree_b.basis_tree(),
 
137
        merge_inner(tree_b.branch, tree_a, tree_b.basis_tree(), 
156
138
                    this_tree=tree_b, ignore_zero=True)
157
 
        self.failUnless('All changes applied successfully.\n' not in
158
 
            self.get_log())
159
 
        tree_b.revert()
160
 
        merge_inner(tree_b.branch, tree_a, tree_b.basis_tree(),
 
139
        log = self._get_log(keep_log_file=True)
 
140
        self.failUnless('All changes applied successfully.\n' not in log)
 
141
        tree_b.revert([])
 
142
        merge_inner(tree_b.branch, tree_a, tree_b.basis_tree(), 
161
143
                    this_tree=tree_b, ignore_zero=False)
162
 
        self.failUnless('All changes applied successfully.\n' in self.get_log())
 
144
        log = self._get_log(keep_log_file=True)
 
145
        self.failUnless('All changes applied successfully.\n' in log)
163
146
 
164
147
    def test_merge_inner_conflicts(self):
165
148
        tree_a = self.make_branch_and_tree('a')
172
155
        self.build_tree(['a/b/'])
173
156
        tree_a.add('b', 'b-id')
174
157
        tree_a.commit('added b')
175
 
        # basis_tree() is only guaranteed to be valid as long as it is actually
176
 
        # the basis tree. This mutates the tree after grabbing basis, so go to
177
 
        # the repository.
178
 
        base_tree = tree_a.branch.repository.revision_tree(tree_a.last_revision())
 
158
        base_tree = tree_a.basis_tree()
179
159
        tree_z = tree_a.bzrdir.sprout('z').open_workingtree()
180
160
        self.build_tree(['a/b/c'])
181
161
        tree_a.add('b/c')
187
167
            conflicts.MissingParent('Created directory', 'b', 'b-id'),
188
168
            conflicts.UnversionedParent('Versioned directory', 'b', 'b-id')],
189
169
            tree_z.conflicts())
190
 
        merge_inner(tree_a.branch, tree_z.basis_tree(), base_tree,
 
170
        merge_inner(tree_a.branch, tree_z.basis_tree(), base_tree, 
191
171
                    this_tree=tree_a)
192
172
        self.assertEqual([
193
173
            conflicts.DeletingParent('Not deleting', 'b', 'b-id'),
194
174
            conflicts.UnversionedParent('Versioned directory', 'b', 'b-id')],
195
175
            tree_a.conflicts())
196
 
 
197
 
    def test_nested_merge(self):
198
 
        tree = self.make_branch_and_tree('tree',
199
 
            format='dirstate-with-subtree')
200
 
        sub_tree = self.make_branch_and_tree('tree/sub-tree',
201
 
            format='dirstate-with-subtree')
202
 
        sub_tree.set_root_id('sub-tree-root')
203
 
        self.build_tree_contents([('tree/sub-tree/file', 'text1')])
204
 
        sub_tree.add('file')
205
 
        sub_tree.commit('foo')
206
 
        tree.add_reference(sub_tree)
207
 
        tree.commit('set text to 1')
208
 
        tree2 = tree.bzrdir.sprout('tree2').open_workingtree()
209
 
        # modify the file in the subtree
210
 
        self.build_tree_contents([('tree2/sub-tree/file', 'text2')])
211
 
        # and merge the changes from the diverged subtree into the containing
212
 
        # tree
213
 
        tree2.commit('changed file text')
214
 
        tree.merge_from_branch(tree2.branch)
215
 
        self.assertFileEqual('text2', 'tree/sub-tree/file')
216
 
 
217
 
    def test_merge_with_missing(self):
218
 
        tree_a = self.make_branch_and_tree('tree_a')
219
 
        self.build_tree_contents([('tree_a/file', 'content_1')])
220
 
        tree_a.add('file')
221
 
        tree_a.commit('commit base')
222
 
        # basis_tree() is only guaranteed to be valid as long as it is actually
223
 
        # the basis tree. This test commits to the tree after grabbing basis,
224
 
        # so we go to the repository.
225
 
        base_tree = tree_a.branch.repository.revision_tree(tree_a.last_revision())
226
 
        tree_b = tree_a.bzrdir.sprout('tree_b').open_workingtree()
227
 
        self.build_tree_contents([('tree_a/file', 'content_2')])
228
 
        tree_a.commit('commit other')
229
 
        other_tree = tree_a.basis_tree()
230
 
        # 'file' is now missing but isn't altered in any commit in b so no
231
 
        # change should be applied.
232
 
        os.unlink('tree_b/file')
233
 
        merge_inner(tree_b.branch, other_tree, base_tree, this_tree=tree_b)
234
 
 
235
 
    def test_merge_kind_change(self):
236
 
        tree_a = self.make_branch_and_tree('tree_a')
237
 
        self.build_tree_contents([('tree_a/file', 'content_1')])
238
 
        tree_a.add('file', 'file-id')
239
 
        tree_a.commit('added file')
240
 
        tree_b = tree_a.bzrdir.sprout('tree_b').open_workingtree()
241
 
        os.unlink('tree_a/file')
242
 
        self.build_tree(['tree_a/file/'])
243
 
        tree_a.commit('changed file to directory')
244
 
        tree_b.merge_from_branch(tree_a.branch)
245
 
        self.assertEqual('directory', file_kind('tree_b/file'))
246
 
        tree_b.revert()
247
 
        self.assertEqual('file', file_kind('tree_b/file'))
248
 
        self.build_tree_contents([('tree_b/file', 'content_2')])
249
 
        tree_b.commit('content change')
250
 
        tree_b.merge_from_branch(tree_a.branch)
251
 
        self.assertEqual(tree_b.conflicts(),
252
 
                         [conflicts.ContentsConflict('file',
253
 
                          file_id='file-id')])
254
 
 
255
 
    def test_merge_type_registry(self):
256
 
        merge_type_option = option.Option.OPTIONS['merge-type']
257
 
        self.assertFalse('merge4' in [x[0] for x in
258
 
                        merge_type_option.iter_switches()])
259
 
        registry = _mod_merge.get_merge_type_registry()
260
 
        registry.register_lazy('merge4', 'bzrlib.merge', 'Merge4Merger',
261
 
                               'time-travelling merge')
262
 
        self.assertTrue('merge4' in [x[0] for x in
263
 
                        merge_type_option.iter_switches()])
264
 
        registry.remove('merge4')
265
 
        self.assertFalse('merge4' in [x[0] for x in
266
 
                        merge_type_option.iter_switches()])
267
 
 
268
 
    def test_merge_other_moves_we_deleted(self):
269
 
        tree_a = self.make_branch_and_tree('A')
270
 
        tree_a.lock_write()
271
 
        self.addCleanup(tree_a.unlock)
272
 
        self.build_tree(['A/a'])
273
 
        tree_a.add('a')
274
 
        tree_a.commit('1', rev_id='rev-1')
275
 
        tree_a.flush()
276
 
        tree_a.rename_one('a', 'b')
277
 
        tree_a.commit('2')
278
 
        bzrdir_b = tree_a.bzrdir.sprout('B', revision_id='rev-1')
279
 
        tree_b = bzrdir_b.open_workingtree()
280
 
        tree_b.lock_write()
281
 
        self.addCleanup(tree_b.unlock)
282
 
        os.unlink('B/a')
283
 
        tree_b.commit('3')
284
 
        try:
285
 
            tree_b.merge_from_branch(tree_a.branch)
286
 
        except AttributeError:
287
 
            self.fail('tried to join a path when name was None')
288
 
 
289
 
    def test_merge_uncommitted_otherbasis_ancestor_of_thisbasis(self):
290
 
        tree_a = self.make_branch_and_tree('a')
291
 
        self.build_tree(['a/file_1', 'a/file_2'])
292
 
        tree_a.add(['file_1'])
293
 
        tree_a.commit('commit 1')
294
 
        tree_a.add(['file_2'])
295
 
        tree_a.commit('commit 2')
296
 
        tree_b = tree_a.bzrdir.sprout('b').open_workingtree()
297
 
        tree_b.rename_one('file_1', 'renamed')
298
 
        merger = _mod_merge.Merger.from_uncommitted(tree_a, tree_b)
299
 
        merger.merge_type = _mod_merge.Merge3Merger
300
 
        merger.do_merge()
301
 
        self.assertEqual(tree_a.get_parent_ids(), [tree_b.last_revision()])
302
 
 
303
 
    def test_merge_uncommitted_otherbasis_ancestor_of_thisbasis_weave(self):
304
 
        tree_a = self.make_branch_and_tree('a')
305
 
        self.build_tree(['a/file_1', 'a/file_2'])
306
 
        tree_a.add(['file_1'])
307
 
        tree_a.commit('commit 1')
308
 
        tree_a.add(['file_2'])
309
 
        tree_a.commit('commit 2')
310
 
        tree_b = tree_a.bzrdir.sprout('b').open_workingtree()
311
 
        tree_b.rename_one('file_1', 'renamed')
312
 
        merger = _mod_merge.Merger.from_uncommitted(tree_a, tree_b)
313
 
        merger.merge_type = _mod_merge.WeaveMerger
314
 
        merger.do_merge()
315
 
        self.assertEqual(tree_a.get_parent_ids(), [tree_b.last_revision()])
316
 
 
317
 
    def prepare_cherrypick(self):
318
 
        """Prepare a pair of trees for cherrypicking tests.
319
 
 
320
 
        Both trees have a file, 'file'.
321
 
        rev1 sets content to 'a'.
322
 
        rev2b adds 'b'.
323
 
        rev3b adds 'c'.
324
 
        A full merge of rev2b and rev3b into this_tree would add both 'b' and
325
 
        'c'.  A successful cherrypick of rev2b-rev3b into this_tree will add
326
 
        'c', but not 'b'.
327
 
        """
328
 
        this_tree = self.make_branch_and_tree('this')
329
 
        self.build_tree_contents([('this/file', "a\n")])
330
 
        this_tree.add('file')
331
 
        this_tree.commit('rev1')
332
 
        other_tree = this_tree.bzrdir.sprout('other').open_workingtree()
333
 
        self.build_tree_contents([('other/file', "a\nb\n")])
334
 
        other_tree.commit('rev2b', rev_id='rev2b')
335
 
        self.build_tree_contents([('other/file', "c\na\nb\n")])
336
 
        other_tree.commit('rev3b', rev_id='rev3b')
337
 
        this_tree.lock_write()
338
 
        self.addCleanup(this_tree.unlock)
339
 
        return this_tree, other_tree
340
 
 
341
 
    def test_weave_cherrypick(self):
342
 
        this_tree, other_tree = self.prepare_cherrypick()
343
 
        merger = _mod_merge.Merger.from_revision_ids(None,
344
 
            this_tree, 'rev3b', 'rev2b', other_tree.branch)
345
 
        merger.merge_type = _mod_merge.WeaveMerger
346
 
        merger.do_merge()
347
 
        self.assertFileEqual('c\na\n', 'this/file')
348
 
 
349
 
    def test_weave_cannot_reverse_cherrypick(self):
350
 
        this_tree, other_tree = self.prepare_cherrypick()
351
 
        merger = _mod_merge.Merger.from_revision_ids(None,
352
 
            this_tree, 'rev2b', 'rev3b', other_tree.branch)
353
 
        merger.merge_type = _mod_merge.WeaveMerger
354
 
        self.assertRaises(errors.CannotReverseCherrypick, merger.do_merge)
355
 
 
356
 
    def test_merge3_can_reverse_cherrypick(self):
357
 
        this_tree, other_tree = self.prepare_cherrypick()
358
 
        merger = _mod_merge.Merger.from_revision_ids(None,
359
 
            this_tree, 'rev2b', 'rev3b', other_tree.branch)
360
 
        merger.merge_type = _mod_merge.Merge3Merger
361
 
        merger.do_merge()
362
 
 
363
 
    def test_merge3_will_detect_cherrypick(self):
364
 
        this_tree = self.make_branch_and_tree('this')
365
 
        self.build_tree_contents([('this/file', "a\n")])
366
 
        this_tree.add('file')
367
 
        this_tree.commit('rev1')
368
 
        other_tree = this_tree.bzrdir.sprout('other').open_workingtree()
369
 
        self.build_tree_contents([('other/file', "a\nb\n")])
370
 
        other_tree.commit('rev2b', rev_id='rev2b')
371
 
        self.build_tree_contents([('other/file', "a\nb\nc\n")])
372
 
        other_tree.commit('rev3b', rev_id='rev3b')
373
 
        this_tree.lock_write()
374
 
        self.addCleanup(this_tree.unlock)
375
 
 
376
 
        merger = _mod_merge.Merger.from_revision_ids(None,
377
 
            this_tree, 'rev3b', 'rev2b', other_tree.branch)
378
 
        merger.merge_type = _mod_merge.Merge3Merger
379
 
        merger.do_merge()
380
 
        self.assertFileEqual('a\n'
381
 
                             '<<<<<<< TREE\n'
382
 
                             '=======\n'
383
 
                             'c\n'
384
 
                             '>>>>>>> MERGE-SOURCE\n',
385
 
                             'this/file')
386
 
 
387
 
    def test_make_merger(self):
388
 
        this_tree = self.make_branch_and_tree('this')
389
 
        this_tree.commit('rev1', rev_id='rev1')
390
 
        other_tree = this_tree.bzrdir.sprout('other').open_workingtree()
391
 
        this_tree.commit('rev2', rev_id='rev2a')
392
 
        other_tree.commit('rev2', rev_id='rev2b')
393
 
        this_tree.lock_write()
394
 
        self.addCleanup(this_tree.unlock)
395
 
        merger = _mod_merge.Merger.from_revision_ids(None,
396
 
            this_tree, 'rev2b', other_branch=other_tree.branch)
397
 
        merger.merge_type = _mod_merge.Merge3Merger
398
 
        tree_merger = merger.make_merger()
399
 
        self.assertIs(_mod_merge.Merge3Merger, tree_merger.__class__)
400
 
        self.assertEqual('rev2b', tree_merger.other_tree.get_revision_id())
401
 
        self.assertEqual('rev1', tree_merger.base_tree.get_revision_id())
402
 
 
403
 
    def test_make_preview_transform(self):
404
 
        this_tree = self.make_branch_and_tree('this')
405
 
        self.build_tree_contents([('this/file', '1\n')])
406
 
        this_tree.add('file', 'file-id')
407
 
        this_tree.commit('rev1', rev_id='rev1')
408
 
        other_tree = this_tree.bzrdir.sprout('other').open_workingtree()
409
 
        self.build_tree_contents([('this/file', '1\n2a\n')])
410
 
        this_tree.commit('rev2', rev_id='rev2a')
411
 
        self.build_tree_contents([('other/file', '2b\n1\n')])
412
 
        other_tree.commit('rev2', rev_id='rev2b')
413
 
        this_tree.lock_write()
414
 
        self.addCleanup(this_tree.unlock)
415
 
        merger = _mod_merge.Merger.from_revision_ids(None,
416
 
            this_tree, 'rev2b', other_branch=other_tree.branch)
417
 
        merger.merge_type = _mod_merge.Merge3Merger
418
 
        tree_merger = merger.make_merger()
419
 
        tt = tree_merger.make_preview_transform()
420
 
        self.addCleanup(tt.finalize)
421
 
        preview_tree = tt.get_preview_tree()
422
 
        tree_file = this_tree.get_file('file-id')
423
 
        try:
424
 
            self.assertEqual('1\n2a\n', tree_file.read())
425
 
        finally:
426
 
            tree_file.close()
427
 
        preview_file = preview_tree.get_file('file-id')
428
 
        try:
429
 
            self.assertEqual('2b\n1\n2a\n', preview_file.read())
430
 
        finally:
431
 
            preview_file.close()
432
 
 
433
 
    def test_do_merge(self):
434
 
        this_tree = self.make_branch_and_tree('this')
435
 
        self.build_tree_contents([('this/file', '1\n')])
436
 
        this_tree.add('file', 'file-id')
437
 
        this_tree.commit('rev1', rev_id='rev1')
438
 
        other_tree = this_tree.bzrdir.sprout('other').open_workingtree()
439
 
        self.build_tree_contents([('this/file', '1\n2a\n')])
440
 
        this_tree.commit('rev2', rev_id='rev2a')
441
 
        self.build_tree_contents([('other/file', '2b\n1\n')])
442
 
        other_tree.commit('rev2', rev_id='rev2b')
443
 
        this_tree.lock_write()
444
 
        self.addCleanup(this_tree.unlock)
445
 
        merger = _mod_merge.Merger.from_revision_ids(None,
446
 
            this_tree, 'rev2b', other_branch=other_tree.branch)
447
 
        merger.merge_type = _mod_merge.Merge3Merger
448
 
        tree_merger = merger.make_merger()
449
 
        tt = tree_merger.do_merge()
450
 
        tree_file = this_tree.get_file('file-id')
451
 
        try:
452
 
            self.assertEqual('2b\n1\n2a\n', tree_file.read())
453
 
        finally:
454
 
            tree_file.close()
455
 
 
456
 
    def test_merge_add_into_deleted_root(self):
457
 
        # Yes, people actually do this.  And report bugs if it breaks.
458
 
        source = self.make_branch_and_tree('source', format='rich-root-pack')
459
 
        self.build_tree(['source/foo/'])
460
 
        source.add('foo', 'foo-id')
461
 
        source.commit('Add foo')
462
 
        target = source.bzrdir.sprout('target').open_workingtree()
463
 
        subtree = target.extract('foo-id')
464
 
        subtree.commit('Delete root')
465
 
        self.build_tree(['source/bar'])
466
 
        source.add('bar', 'bar-id')
467
 
        source.commit('Add bar')
468
 
        subtree.merge_from_branch(source.branch)
469
 
 
470
 
    def test_merge_joined_branch(self):
471
 
        source = self.make_branch_and_tree('source', format='rich-root-pack')
472
 
        self.build_tree(['source/foo'])
473
 
        source.add('foo')
474
 
        source.commit('Add foo')
475
 
        target = self.make_branch_and_tree('target', format='rich-root-pack')
476
 
        self.build_tree(['target/bla'])
477
 
        target.add('bla')
478
 
        target.commit('Add bla')
479
 
        nested = source.bzrdir.sprout('target/subtree').open_workingtree()
480
 
        target.subsume(nested)
481
 
        target.commit('Join nested')
482
 
        self.build_tree(['source/bar'])
483
 
        source.add('bar')
484
 
        source.commit('Add bar')
485
 
        target.merge_from_branch(source.branch)
486
 
        target.commit('Merge source')
487
 
 
488
 
 
489
 
class TestPlanMerge(TestCaseWithMemoryTransport):
490
 
 
491
 
    def setUp(self):
492
 
        TestCaseWithMemoryTransport.setUp(self)
493
 
        mapper = versionedfile.PrefixMapper()
494
 
        factory = knit.make_file_factory(True, mapper)
495
 
        self.vf = factory(self.get_transport())
496
 
        self.plan_merge_vf = versionedfile._PlanMergeVersionedFile('root')
497
 
        self.plan_merge_vf.fallback_versionedfiles.append(self.vf)
498
 
 
499
 
    def add_version(self, key, parents, text):
500
 
        self.vf.add_lines(key, parents, [c+'\n' for c in text])
501
 
 
502
 
    def add_rev(self, prefix, revision_id, parents, text):
503
 
        self.add_version((prefix, revision_id), [(prefix, p) for p in parents],
504
 
                         text)
505
 
 
506
 
    def add_uncommitted_version(self, key, parents, text):
507
 
        self.plan_merge_vf.add_lines(key, parents,
508
 
                                     [c+'\n' for c in text])
509
 
 
510
 
    def setup_plan_merge(self):
511
 
        self.add_rev('root', 'A', [], 'abc')
512
 
        self.add_rev('root', 'B', ['A'], 'acehg')
513
 
        self.add_rev('root', 'C', ['A'], 'fabg')
514
 
        return _PlanMerge('B', 'C', self.plan_merge_vf, ('root',))
515
 
 
516
 
    def setup_plan_merge_uncommitted(self):
517
 
        self.add_version(('root', 'A'), [], 'abc')
518
 
        self.add_uncommitted_version(('root', 'B:'), [('root', 'A')], 'acehg')
519
 
        self.add_uncommitted_version(('root', 'C:'), [('root', 'A')], 'fabg')
520
 
        return _PlanMerge('B:', 'C:', self.plan_merge_vf, ('root',))
521
 
 
522
 
    def test_base_from_plan(self):
523
 
        self.setup_plan_merge()
524
 
        plan = self.plan_merge_vf.plan_merge('B', 'C')
525
 
        pwm = versionedfile.PlanWeaveMerge(plan)
526
 
        self.assertEqual(['a\n', 'b\n', 'c\n'], pwm.base_from_plan())
527
 
 
528
 
    def test_unique_lines(self):
529
 
        plan = self.setup_plan_merge()
530
 
        self.assertEqual(plan._unique_lines(
531
 
            plan._get_matching_blocks('B', 'C')),
532
 
            ([1, 2, 3], [0, 2]))
533
 
 
534
 
    def test_plan_merge(self):
535
 
        self.setup_plan_merge()
536
 
        plan = self.plan_merge_vf.plan_merge('B', 'C')
537
 
        self.assertEqual([
538
 
                          ('new-b', 'f\n'),
539
 
                          ('unchanged', 'a\n'),
540
 
                          ('killed-a', 'b\n'),
541
 
                          ('killed-b', 'c\n'),
542
 
                          ('new-a', 'e\n'),
543
 
                          ('new-a', 'h\n'),
544
 
                          ('new-a', 'g\n'),
545
 
                          ('new-b', 'g\n')],
546
 
                         list(plan))
547
 
 
548
 
    def test_plan_merge_cherrypick(self):
549
 
        self.add_rev('root', 'A', [], 'abc')
550
 
        self.add_rev('root', 'B', ['A'], 'abcde')
551
 
        self.add_rev('root', 'C', ['A'], 'abcefg')
552
 
        self.add_rev('root', 'D', ['A', 'B', 'C'], 'abcdegh')
553
 
        my_plan = _PlanMerge('B', 'D', self.plan_merge_vf, ('root',))
554
 
        # We shortcut when one text supersedes the other in the per-file graph.
555
 
        # We don't actually need to compare the texts at this point.
556
 
        self.assertEqual([
557
 
                          ('new-b', 'a\n'),
558
 
                          ('new-b', 'b\n'),
559
 
                          ('new-b', 'c\n'),
560
 
                          ('new-b', 'd\n'),
561
 
                          ('new-b', 'e\n'),
562
 
                          ('new-b', 'g\n'),
563
 
                          ('new-b', 'h\n')],
564
 
                          list(my_plan.plan_merge()))
565
 
 
566
 
    def test_plan_merge_no_common_ancestor(self):
567
 
        self.add_rev('root', 'A', [], 'abc')
568
 
        self.add_rev('root', 'B', [], 'xyz')
569
 
        my_plan = _PlanMerge('A', 'B', self.plan_merge_vf, ('root',))
570
 
        self.assertEqual([
571
 
                          ('new-a', 'a\n'),
572
 
                          ('new-a', 'b\n'),
573
 
                          ('new-a', 'c\n'),
574
 
                          ('new-b', 'x\n'),
575
 
                          ('new-b', 'y\n'),
576
 
                          ('new-b', 'z\n')],
577
 
                          list(my_plan.plan_merge()))
578
 
 
579
 
    def test_plan_merge_tail_ancestors(self):
580
 
        # The graph looks like this:
581
 
        #       A       # Common to all ancestors
582
 
        #      / \
583
 
        #     B   C     # Ancestors of E, only common to one side
584
 
        #     |\ /|
585
 
        #     D E F     # D, F are unique to G, H respectively
586
 
        #     |/ \|     # E is the LCA for G & H, and the unique LCA for
587
 
        #     G   H     # I, J
588
 
        #     |\ /|
589
 
        #     | X |
590
 
        #     |/ \|
591
 
        #     I   J     # criss-cross merge of G, H
592
 
        #
593
 
        # In this situation, a simple pruning of ancestors of E will leave D &
594
 
        # F "dangling", which looks like they introduce lines different from
595
 
        # the ones in E, but in actuality C&B introduced the lines, and they
596
 
        # are already present in E
597
 
 
598
 
        # Introduce the base text
599
 
        self.add_rev('root', 'A', [], 'abc')
600
 
        # Introduces a new line B
601
 
        self.add_rev('root', 'B', ['A'], 'aBbc')
602
 
        # Introduces a new line C
603
 
        self.add_rev('root', 'C', ['A'], 'abCc')
604
 
        # Introduce new line D
605
 
        self.add_rev('root', 'D', ['B'], 'DaBbc')
606
 
        # Merges B and C by just incorporating both
607
 
        self.add_rev('root', 'E', ['B', 'C'], 'aBbCc')
608
 
        # Introduce new line F
609
 
        self.add_rev('root', 'F', ['C'], 'abCcF')
610
 
        # Merge D & E by just combining the texts
611
 
        self.add_rev('root', 'G', ['D', 'E'], 'DaBbCc')
612
 
        # Merge F & E by just combining the texts
613
 
        self.add_rev('root', 'H', ['F', 'E'], 'aBbCcF')
614
 
        # Merge G & H by just combining texts
615
 
        self.add_rev('root', 'I', ['G', 'H'], 'DaBbCcF')
616
 
        # Merge G & H but supersede an old line in B
617
 
        self.add_rev('root', 'J', ['H', 'G'], 'DaJbCcF')
618
 
        plan = self.plan_merge_vf.plan_merge('I', 'J')
619
 
        self.assertEqual([
620
 
                          ('unchanged', 'D\n'),
621
 
                          ('unchanged', 'a\n'),
622
 
                          ('killed-b', 'B\n'),
623
 
                          ('new-b', 'J\n'),
624
 
                          ('unchanged', 'b\n'),
625
 
                          ('unchanged', 'C\n'),
626
 
                          ('unchanged', 'c\n'),
627
 
                          ('unchanged', 'F\n')],
628
 
                         list(plan))
629
 
 
630
 
    def test_plan_merge_tail_triple_ancestors(self):
631
 
        # The graph looks like this:
632
 
        #       A       # Common to all ancestors
633
 
        #      / \
634
 
        #     B   C     # Ancestors of E, only common to one side
635
 
        #     |\ /|
636
 
        #     D E F     # D, F are unique to G, H respectively
637
 
        #     |/|\|     # E is the LCA for G & H, and the unique LCA for
638
 
        #     G Q H     # I, J
639
 
        #     |\ /|     # Q is just an extra node which is merged into both
640
 
        #     | X |     # I and J
641
 
        #     |/ \|
642
 
        #     I   J     # criss-cross merge of G, H
643
 
        #
644
 
        # This is the same as the test_plan_merge_tail_ancestors, except we add
645
 
        # a third LCA that doesn't add new lines, but will trigger our more
646
 
        # involved ancestry logic
647
 
 
648
 
        self.add_rev('root', 'A', [], 'abc')
649
 
        self.add_rev('root', 'B', ['A'], 'aBbc')
650
 
        self.add_rev('root', 'C', ['A'], 'abCc')
651
 
        self.add_rev('root', 'D', ['B'], 'DaBbc')
652
 
        self.add_rev('root', 'E', ['B', 'C'], 'aBbCc')
653
 
        self.add_rev('root', 'F', ['C'], 'abCcF')
654
 
        self.add_rev('root', 'G', ['D', 'E'], 'DaBbCc')
655
 
        self.add_rev('root', 'H', ['F', 'E'], 'aBbCcF')
656
 
        self.add_rev('root', 'Q', ['E'], 'aBbCc')
657
 
        self.add_rev('root', 'I', ['G', 'Q', 'H'], 'DaBbCcF')
658
 
        # Merge G & H but supersede an old line in B
659
 
        self.add_rev('root', 'J', ['H', 'Q', 'G'], 'DaJbCcF')
660
 
        plan = self.plan_merge_vf.plan_merge('I', 'J')
661
 
        self.assertEqual([
662
 
                          ('unchanged', 'D\n'),
663
 
                          ('unchanged', 'a\n'),
664
 
                          ('killed-b', 'B\n'),
665
 
                          ('new-b', 'J\n'),
666
 
                          ('unchanged', 'b\n'),
667
 
                          ('unchanged', 'C\n'),
668
 
                          ('unchanged', 'c\n'),
669
 
                          ('unchanged', 'F\n')],
670
 
                         list(plan))
671
 
 
672
 
    def test_plan_merge_2_tail_triple_ancestors(self):
673
 
        # The graph looks like this:
674
 
        #     A   B     # 2 tails going back to NULL
675
 
        #     |\ /|
676
 
        #     D E F     # D, is unique to G, F to H
677
 
        #     |/|\|     # E is the LCA for G & H, and the unique LCA for
678
 
        #     G Q H     # I, J
679
 
        #     |\ /|     # Q is just an extra node which is merged into both
680
 
        #     | X |     # I and J
681
 
        #     |/ \|
682
 
        #     I   J     # criss-cross merge of G, H (and Q)
683
 
        #
684
 
 
685
 
        # This is meant to test after hitting a 3-way LCA, and multiple tail
686
 
        # ancestors (only have NULL_REVISION in common)
687
 
 
688
 
        self.add_rev('root', 'A', [], 'abc')
689
 
        self.add_rev('root', 'B', [], 'def')
690
 
        self.add_rev('root', 'D', ['A'], 'Dabc')
691
 
        self.add_rev('root', 'E', ['A', 'B'], 'abcdef')
692
 
        self.add_rev('root', 'F', ['B'], 'defF')
693
 
        self.add_rev('root', 'G', ['D', 'E'], 'Dabcdef')
694
 
        self.add_rev('root', 'H', ['F', 'E'], 'abcdefF')
695
 
        self.add_rev('root', 'Q', ['E'], 'abcdef')
696
 
        self.add_rev('root', 'I', ['G', 'Q', 'H'], 'DabcdefF')
697
 
        # Merge G & H but supersede an old line in B
698
 
        self.add_rev('root', 'J', ['H', 'Q', 'G'], 'DabcdJfF')
699
 
        plan = self.plan_merge_vf.plan_merge('I', 'J')
700
 
        self.assertEqual([
701
 
                          ('unchanged', 'D\n'),
702
 
                          ('unchanged', 'a\n'),
703
 
                          ('unchanged', 'b\n'),
704
 
                          ('unchanged', 'c\n'),
705
 
                          ('unchanged', 'd\n'),
706
 
                          ('killed-b', 'e\n'),
707
 
                          ('new-b', 'J\n'),
708
 
                          ('unchanged', 'f\n'),
709
 
                          ('unchanged', 'F\n')],
710
 
                         list(plan))
711
 
 
712
 
    def test_plan_merge_uncommitted_files(self):
713
 
        self.setup_plan_merge_uncommitted()
714
 
        plan = self.plan_merge_vf.plan_merge('B:', 'C:')
715
 
        self.assertEqual([
716
 
                          ('new-b', 'f\n'),
717
 
                          ('unchanged', 'a\n'),
718
 
                          ('killed-a', 'b\n'),
719
 
                          ('killed-b', 'c\n'),
720
 
                          ('new-a', 'e\n'),
721
 
                          ('new-a', 'h\n'),
722
 
                          ('new-a', 'g\n'),
723
 
                          ('new-b', 'g\n')],
724
 
                         list(plan))
725
 
 
726
 
    def test_plan_merge_insert_order(self):
727
 
        """Weave merges are sensitive to the order of insertion.
728
 
 
729
 
        Specifically for overlapping regions, it effects which region gets put
730
 
        'first'. And when a user resolves an overlapping merge, if they use the
731
 
        same ordering, then the lines match the parents, if they don't only
732
 
        *some* of the lines match.
733
 
        """
734
 
        self.add_rev('root', 'A', [], 'abcdef')
735
 
        self.add_rev('root', 'B', ['A'], 'abwxcdef')
736
 
        self.add_rev('root', 'C', ['A'], 'abyzcdef')
737
 
        # Merge, and resolve the conflict by adding *both* sets of lines
738
 
        # If we get the ordering wrong, these will look like new lines in D,
739
 
        # rather than carried over from B, C
740
 
        self.add_rev('root', 'D', ['B', 'C'],
741
 
                         'abwxyzcdef')
742
 
        # Supersede the lines in B and delete the lines in C, which will
743
 
        # conflict if they are treated as being in D
744
 
        self.add_rev('root', 'E', ['C', 'B'],
745
 
                         'abnocdef')
746
 
        # Same thing for the lines in C
747
 
        self.add_rev('root', 'F', ['C'], 'abpqcdef')
748
 
        plan = self.plan_merge_vf.plan_merge('D', 'E')
749
 
        self.assertEqual([
750
 
                          ('unchanged', 'a\n'),
751
 
                          ('unchanged', 'b\n'),
752
 
                          ('killed-b', 'w\n'),
753
 
                          ('killed-b', 'x\n'),
754
 
                          ('killed-b', 'y\n'),
755
 
                          ('killed-b', 'z\n'),
756
 
                          ('new-b', 'n\n'),
757
 
                          ('new-b', 'o\n'),
758
 
                          ('unchanged', 'c\n'),
759
 
                          ('unchanged', 'd\n'),
760
 
                          ('unchanged', 'e\n'),
761
 
                          ('unchanged', 'f\n')],
762
 
                         list(plan))
763
 
        plan = self.plan_merge_vf.plan_merge('E', 'D')
764
 
        # Going in the opposite direction shows the effect of the opposite plan
765
 
        self.assertEqual([
766
 
                          ('unchanged', 'a\n'),
767
 
                          ('unchanged', 'b\n'),
768
 
                          ('new-b', 'w\n'),
769
 
                          ('new-b', 'x\n'),
770
 
                          ('killed-a', 'y\n'),
771
 
                          ('killed-a', 'z\n'),
772
 
                          ('killed-both', 'w\n'),
773
 
                          ('killed-both', 'x\n'),
774
 
                          ('new-a', 'n\n'),
775
 
                          ('new-a', 'o\n'),
776
 
                          ('unchanged', 'c\n'),
777
 
                          ('unchanged', 'd\n'),
778
 
                          ('unchanged', 'e\n'),
779
 
                          ('unchanged', 'f\n')],
780
 
                         list(plan))
781
 
 
782
 
    def test_plan_merge_criss_cross(self):
783
 
        # This is specificly trying to trigger problems when using limited
784
 
        # ancestry and weaves. The ancestry graph looks like:
785
 
        #       XX      unused ancestor, should not show up in the weave
786
 
        #       |
787
 
        #       A       Unique LCA
788
 
        #       |\
789
 
        #       B \     Introduces a line 'foo'
790
 
        #      / \ \
791
 
        #     C   D E   C & D both have 'foo', E has different changes
792
 
        #     |\ /| |
793
 
        #     | X | |
794
 
        #     |/ \|/
795
 
        #     F   G      All of C, D, E are merged into F and G, so they are
796
 
        #                all common ancestors.
797
 
        #
798
 
        # The specific issue with weaves:
799
 
        #   B introduced a text ('foo') that is present in both C and D.
800
 
        #   If we do not include B (because it isn't an ancestor of E), then
801
 
        #   the A=>C and A=>D look like both sides independently introduce the
802
 
        #   text ('foo'). If F does not modify the text, it would still appear
803
 
        #   to have deleted on of the versions from C or D. If G then modifies
804
 
        #   'foo', it should appear as superseding the value in F (since it
805
 
        #   came from B), rather than conflict because of the resolution during
806
 
        #   C & D.
807
 
        self.add_rev('root', 'XX', [], 'qrs')
808
 
        self.add_rev('root', 'A', ['XX'], 'abcdef')
809
 
        self.add_rev('root', 'B', ['A'], 'axcdef')
810
 
        self.add_rev('root', 'C', ['B'], 'axcdefg')
811
 
        self.add_rev('root', 'D', ['B'], 'haxcdef')
812
 
        self.add_rev('root', 'E', ['A'], 'abcdyf')
813
 
        # Simple combining of all texts
814
 
        self.add_rev('root', 'F', ['C', 'D', 'E'], 'haxcdyfg')
815
 
        # combine and supersede 'x'
816
 
        self.add_rev('root', 'G', ['C', 'D', 'E'], 'hazcdyfg')
817
 
        plan = self.plan_merge_vf.plan_merge('F', 'G')
818
 
        self.assertEqual([
819
 
                          ('unchanged', 'h\n'),
820
 
                          ('unchanged', 'a\n'),
821
 
                          ('killed-base', 'b\n'),
822
 
                          ('killed-b', 'x\n'),
823
 
                          ('new-b', 'z\n'),
824
 
                          ('unchanged', 'c\n'),
825
 
                          ('unchanged', 'd\n'),
826
 
                          ('killed-base', 'e\n'),
827
 
                          ('unchanged', 'y\n'),
828
 
                          ('unchanged', 'f\n'),
829
 
                          ('unchanged', 'g\n')],
830
 
                         list(plan))
831
 
        plan = self.plan_merge_vf.plan_lca_merge('F', 'G')
832
 
        # This is one of the main differences between plan_merge and
833
 
        # plan_lca_merge. plan_lca_merge generates a conflict for 'x => z',
834
 
        # because 'x' was not present in one of the bases. However, in this
835
 
        # case it is spurious because 'x' does not exist in the global base A.
836
 
        self.assertEqual([
837
 
                          ('unchanged', 'h\n'),
838
 
                          ('unchanged', 'a\n'),
839
 
                          ('conflicted-a', 'x\n'),
840
 
                          ('new-b', 'z\n'),
841
 
                          ('unchanged', 'c\n'),
842
 
                          ('unchanged', 'd\n'),
843
 
                          ('unchanged', 'y\n'),
844
 
                          ('unchanged', 'f\n'),
845
 
                          ('unchanged', 'g\n')],
846
 
                         list(plan))
847
 
 
848
 
    def test_criss_cross_flip_flop(self):
849
 
        # This is specificly trying to trigger problems when using limited
850
 
        # ancestry and weaves. The ancestry graph looks like:
851
 
        #       XX      unused ancestor, should not show up in the weave
852
 
        #       |
853
 
        #       A       Unique LCA
854
 
        #      / \  
855
 
        #     B   C     B & C both introduce a new line
856
 
        #     |\ /|  
857
 
        #     | X |  
858
 
        #     |/ \| 
859
 
        #     D   E     B & C are both merged, so both are common ancestors
860
 
        #               In the process of merging, both sides order the new
861
 
        #               lines differently
862
 
        #
863
 
        self.add_rev('root', 'XX', [], 'qrs')
864
 
        self.add_rev('root', 'A', ['XX'], 'abcdef')
865
 
        self.add_rev('root', 'B', ['A'], 'abcdgef')
866
 
        self.add_rev('root', 'C', ['A'], 'abcdhef')
867
 
        self.add_rev('root', 'D', ['B', 'C'], 'abcdghef')
868
 
        self.add_rev('root', 'E', ['C', 'B'], 'abcdhgef')
869
 
        plan = list(self.plan_merge_vf.plan_merge('D', 'E'))
870
 
        self.assertEqual([
871
 
                          ('unchanged', 'a\n'),
872
 
                          ('unchanged', 'b\n'),
873
 
                          ('unchanged', 'c\n'),
874
 
                          ('unchanged', 'd\n'),
875
 
                          ('new-b', 'h\n'),
876
 
                          ('unchanged', 'g\n'),
877
 
                          ('killed-b', 'h\n'),
878
 
                          ('unchanged', 'e\n'),
879
 
                          ('unchanged', 'f\n'),
880
 
                         ], plan)
881
 
        pwm = versionedfile.PlanWeaveMerge(plan)
882
 
        self.assertEqualDiff('\n'.join('abcdghef') + '\n',
883
 
                             ''.join(pwm.base_from_plan()))
884
 
        # Reversing the order reverses the merge plan, and final order of 'hg'
885
 
        # => 'gh'
886
 
        plan = list(self.plan_merge_vf.plan_merge('E', 'D'))
887
 
        self.assertEqual([
888
 
                          ('unchanged', 'a\n'),
889
 
                          ('unchanged', 'b\n'),
890
 
                          ('unchanged', 'c\n'),
891
 
                          ('unchanged', 'd\n'),
892
 
                          ('new-b', 'g\n'),
893
 
                          ('unchanged', 'h\n'),
894
 
                          ('killed-b', 'g\n'),
895
 
                          ('unchanged', 'e\n'),
896
 
                          ('unchanged', 'f\n'),
897
 
                         ], plan)
898
 
        pwm = versionedfile.PlanWeaveMerge(plan)
899
 
        self.assertEqualDiff('\n'.join('abcdhgef') + '\n',
900
 
                             ''.join(pwm.base_from_plan()))
901
 
        # This is where lca differs, in that it (fairly correctly) determines
902
 
        # that there is a conflict because both sides resolved the merge
903
 
        # differently
904
 
        plan = list(self.plan_merge_vf.plan_lca_merge('D', 'E'))
905
 
        self.assertEqual([
906
 
                          ('unchanged', 'a\n'),
907
 
                          ('unchanged', 'b\n'),
908
 
                          ('unchanged', 'c\n'),
909
 
                          ('unchanged', 'd\n'),
910
 
                          ('conflicted-b', 'h\n'),
911
 
                          ('unchanged', 'g\n'),
912
 
                          ('conflicted-a', 'h\n'),
913
 
                          ('unchanged', 'e\n'),
914
 
                          ('unchanged', 'f\n'),
915
 
                         ], plan)
916
 
        pwm = versionedfile.PlanWeaveMerge(plan)
917
 
        self.assertEqualDiff('\n'.join('abcdgef') + '\n',
918
 
                             ''.join(pwm.base_from_plan()))
919
 
        # Reversing it changes what line is doubled, but still gives a
920
 
        # double-conflict
921
 
        plan = list(self.plan_merge_vf.plan_lca_merge('E', 'D'))
922
 
        self.assertEqual([
923
 
                          ('unchanged', 'a\n'),
924
 
                          ('unchanged', 'b\n'),
925
 
                          ('unchanged', 'c\n'),
926
 
                          ('unchanged', 'd\n'),
927
 
                          ('conflicted-b', 'g\n'),
928
 
                          ('unchanged', 'h\n'),
929
 
                          ('conflicted-a', 'g\n'),
930
 
                          ('unchanged', 'e\n'),
931
 
                          ('unchanged', 'f\n'),
932
 
                         ], plan)
933
 
        pwm = versionedfile.PlanWeaveMerge(plan)
934
 
        self.assertEqualDiff('\n'.join('abcdhef') + '\n',
935
 
                             ''.join(pwm.base_from_plan()))
936
 
 
937
 
    def assertRemoveExternalReferences(self, filtered_parent_map,
938
 
                                       child_map, tails, parent_map):
939
 
        """Assert results for _PlanMerge._remove_external_references."""
940
 
        (act_filtered_parent_map, act_child_map,
941
 
         act_tails) = _PlanMerge._remove_external_references(parent_map)
942
 
 
943
 
        # The parent map *should* preserve ordering, but the ordering of
944
 
        # children is not strictly defined
945
 
        # child_map = dict((k, sorted(children))
946
 
        #                  for k, children in child_map.iteritems())
947
 
        # act_child_map = dict(k, sorted(children)
948
 
        #                      for k, children in act_child_map.iteritems())
949
 
        self.assertEqual(filtered_parent_map, act_filtered_parent_map)
950
 
        self.assertEqual(child_map, act_child_map)
951
 
        self.assertEqual(sorted(tails), sorted(act_tails))
952
 
 
953
 
    def test__remove_external_references(self):
954
 
        # First, nothing to remove
955
 
        self.assertRemoveExternalReferences({3: [2], 2: [1], 1: []},
956
 
            {1: [2], 2: [3], 3: []}, [1], {3: [2], 2: [1], 1: []})
957
 
        # The reverse direction
958
 
        self.assertRemoveExternalReferences({1: [2], 2: [3], 3: []},
959
 
            {3: [2], 2: [1], 1: []}, [3], {1: [2], 2: [3], 3: []})
960
 
        # Extra references
961
 
        self.assertRemoveExternalReferences({3: [2], 2: [1], 1: []},
962
 
            {1: [2], 2: [3], 3: []}, [1], {3: [2, 4], 2: [1, 5], 1: [6]})
963
 
        # Multiple tails
964
 
        self.assertRemoveExternalReferences(
965
 
            {4: [2, 3], 3: [], 2: [1], 1: []},
966
 
            {1: [2], 2: [4], 3: [4], 4: []},
967
 
            [1, 3],
968
 
            {4: [2, 3], 3: [5], 2: [1], 1: [6]})
969
 
        # Multiple children
970
 
        self.assertRemoveExternalReferences(
971
 
            {1: [3], 2: [3, 4], 3: [], 4: []},
972
 
            {1: [], 2: [], 3: [1, 2], 4: [2]},
973
 
            [3, 4],
974
 
            {1: [3], 2: [3, 4], 3: [5], 4: []})
975
 
 
976
 
    def assertPruneTails(self, pruned_map, tails, parent_map):
977
 
        child_map = {}
978
 
        for key, parent_keys in parent_map.iteritems():
979
 
            child_map.setdefault(key, [])
980
 
            for pkey in parent_keys:
981
 
                child_map.setdefault(pkey, []).append(key)
982
 
        _PlanMerge._prune_tails(parent_map, child_map, tails)
983
 
        self.assertEqual(pruned_map, parent_map)
984
 
 
985
 
    def test__prune_tails(self):
986
 
        # Nothing requested to prune
987
 
        self.assertPruneTails({1: [], 2: [], 3: []}, [],
988
 
                              {1: [], 2: [], 3: []})
989
 
        # Prune a single entry
990
 
        self.assertPruneTails({1: [], 3: []}, [2],
991
 
                              {1: [], 2: [], 3: []})
992
 
        # Prune a chain
993
 
        self.assertPruneTails({1: []}, [3],
994
 
                              {1: [], 2: [3], 3: []})
995
 
        # Prune a chain with a diamond
996
 
        self.assertPruneTails({1: []}, [5],
997
 
                              {1: [], 2: [3, 4], 3: [5], 4: [5], 5: []})
998
 
        # Prune a partial chain
999
 
        self.assertPruneTails({1: [6], 6:[]}, [5],
1000
 
                              {1: [2, 6], 2: [3, 4], 3: [5], 4: [5], 5: [],
1001
 
                               6: []})
1002
 
        # Prune a chain with multiple tips, that pulls out intermediates
1003
 
        self.assertPruneTails({1:[3], 3:[]}, [4, 5],
1004
 
                              {1: [2, 3], 2: [4, 5], 3: [], 4:[], 5:[]})
1005
 
        self.assertPruneTails({1:[3], 3:[]}, [5, 4],
1006
 
                              {1: [2, 3], 2: [4, 5], 3: [], 4:[], 5:[]})
1007
 
 
1008
 
    def test_subtract_plans(self):
1009
 
        old_plan = [
1010
 
        ('unchanged', 'a\n'),
1011
 
        ('new-a', 'b\n'),
1012
 
        ('killed-a', 'c\n'),
1013
 
        ('new-b', 'd\n'),
1014
 
        ('new-b', 'e\n'),
1015
 
        ('killed-b', 'f\n'),
1016
 
        ('killed-b', 'g\n'),
1017
 
        ]
1018
 
        new_plan = [
1019
 
        ('unchanged', 'a\n'),
1020
 
        ('new-a', 'b\n'),
1021
 
        ('killed-a', 'c\n'),
1022
 
        ('new-b', 'd\n'),
1023
 
        ('new-b', 'h\n'),
1024
 
        ('killed-b', 'f\n'),
1025
 
        ('killed-b', 'i\n'),
1026
 
        ]
1027
 
        subtracted_plan = [
1028
 
        ('unchanged', 'a\n'),
1029
 
        ('new-a', 'b\n'),
1030
 
        ('killed-a', 'c\n'),
1031
 
        ('new-b', 'h\n'),
1032
 
        ('unchanged', 'f\n'),
1033
 
        ('killed-b', 'i\n'),
1034
 
        ]
1035
 
        self.assertEqual(subtracted_plan,
1036
 
            list(_PlanMerge._subtract_plans(old_plan, new_plan)))
1037
 
 
1038
 
    def setup_merge_with_base(self):
1039
 
        self.add_rev('root', 'COMMON', [], 'abc')
1040
 
        self.add_rev('root', 'THIS', ['COMMON'], 'abcd')
1041
 
        self.add_rev('root', 'BASE', ['COMMON'], 'eabc')
1042
 
        self.add_rev('root', 'OTHER', ['BASE'], 'eafb')
1043
 
 
1044
 
    def test_plan_merge_with_base(self):
1045
 
        self.setup_merge_with_base()
1046
 
        plan = self.plan_merge_vf.plan_merge('THIS', 'OTHER', 'BASE')
1047
 
        self.assertEqual([('unchanged', 'a\n'),
1048
 
                          ('new-b', 'f\n'),
1049
 
                          ('unchanged', 'b\n'),
1050
 
                          ('killed-b', 'c\n'),
1051
 
                          ('new-a', 'd\n')
1052
 
                         ], list(plan))
1053
 
 
1054
 
    def test_plan_lca_merge(self):
1055
 
        self.setup_plan_merge()
1056
 
        plan = self.plan_merge_vf.plan_lca_merge('B', 'C')
1057
 
        self.assertEqual([
1058
 
                          ('new-b', 'f\n'),
1059
 
                          ('unchanged', 'a\n'),
1060
 
                          ('killed-b', 'c\n'),
1061
 
                          ('new-a', 'e\n'),
1062
 
                          ('new-a', 'h\n'),
1063
 
                          ('killed-a', 'b\n'),
1064
 
                          ('unchanged', 'g\n')],
1065
 
                         list(plan))
1066
 
 
1067
 
    def test_plan_lca_merge_uncommitted_files(self):
1068
 
        self.setup_plan_merge_uncommitted()
1069
 
        plan = self.plan_merge_vf.plan_lca_merge('B:', 'C:')
1070
 
        self.assertEqual([
1071
 
                          ('new-b', 'f\n'),
1072
 
                          ('unchanged', 'a\n'),
1073
 
                          ('killed-b', 'c\n'),
1074
 
                          ('new-a', 'e\n'),
1075
 
                          ('new-a', 'h\n'),
1076
 
                          ('killed-a', 'b\n'),
1077
 
                          ('unchanged', 'g\n')],
1078
 
                         list(plan))
1079
 
 
1080
 
    def test_plan_lca_merge_with_base(self):
1081
 
        self.setup_merge_with_base()
1082
 
        plan = self.plan_merge_vf.plan_lca_merge('THIS', 'OTHER', 'BASE')
1083
 
        self.assertEqual([('unchanged', 'a\n'),
1084
 
                          ('new-b', 'f\n'),
1085
 
                          ('unchanged', 'b\n'),
1086
 
                          ('killed-b', 'c\n'),
1087
 
                          ('new-a', 'd\n')
1088
 
                         ], list(plan))
1089
 
 
1090
 
    def test_plan_lca_merge_with_criss_cross(self):
1091
 
        self.add_version(('root', 'ROOT'), [], 'abc')
1092
 
        # each side makes a change
1093
 
        self.add_version(('root', 'REV1'), [('root', 'ROOT')], 'abcd')
1094
 
        self.add_version(('root', 'REV2'), [('root', 'ROOT')], 'abce')
1095
 
        # both sides merge, discarding others' changes
1096
 
        self.add_version(('root', 'LCA1'),
1097
 
            [('root', 'REV1'), ('root', 'REV2')], 'abcd')
1098
 
        self.add_version(('root', 'LCA2'),
1099
 
            [('root', 'REV1'), ('root', 'REV2')], 'fabce')
1100
 
        plan = self.plan_merge_vf.plan_lca_merge('LCA1', 'LCA2')
1101
 
        self.assertEqual([('new-b', 'f\n'),
1102
 
                          ('unchanged', 'a\n'),
1103
 
                          ('unchanged', 'b\n'),
1104
 
                          ('unchanged', 'c\n'),
1105
 
                          ('conflicted-a', 'd\n'),
1106
 
                          ('conflicted-b', 'e\n'),
1107
 
                         ], list(plan))
1108
 
 
1109
 
    def test_plan_lca_merge_with_null(self):
1110
 
        self.add_version(('root', 'A'), [], 'ab')
1111
 
        self.add_version(('root', 'B'), [], 'bc')
1112
 
        plan = self.plan_merge_vf.plan_lca_merge('A', 'B')
1113
 
        self.assertEqual([('new-a', 'a\n'),
1114
 
                          ('unchanged', 'b\n'),
1115
 
                          ('new-b', 'c\n'),
1116
 
                         ], list(plan))
1117
 
 
1118
 
    def test_plan_merge_with_delete_and_change(self):
1119
 
        self.add_rev('root', 'C', [], 'a')
1120
 
        self.add_rev('root', 'A', ['C'], 'b')
1121
 
        self.add_rev('root', 'B', ['C'], '')
1122
 
        plan = self.plan_merge_vf.plan_merge('A', 'B')
1123
 
        self.assertEqual([('killed-both', 'a\n'),
1124
 
                          ('new-a', 'b\n'),
1125
 
                         ], list(plan))
1126
 
 
1127
 
    def test_plan_merge_with_move_and_change(self):
1128
 
        self.add_rev('root', 'C', [], 'abcd')
1129
 
        self.add_rev('root', 'A', ['C'], 'acbd')
1130
 
        self.add_rev('root', 'B', ['C'], 'aBcd')
1131
 
        plan = self.plan_merge_vf.plan_merge('A', 'B')
1132
 
        self.assertEqual([('unchanged', 'a\n'),
1133
 
                          ('new-a', 'c\n'),
1134
 
                          ('killed-b', 'b\n'),
1135
 
                          ('new-b', 'B\n'),
1136
 
                          ('killed-a', 'c\n'),
1137
 
                          ('unchanged', 'd\n'),
1138
 
                         ], list(plan))
1139
 
 
1140
 
 
1141
 
class LoggingMerger(object):
1142
 
    # These seem to be the required attributes
1143
 
    requires_base = False
1144
 
    supports_reprocess = False
1145
 
    supports_show_base = False
1146
 
    supports_cherrypick = False
1147
 
    # We intentionally do not define supports_lca_trees
1148
 
 
1149
 
    def __init__(self, *args, **kwargs):
1150
 
        self.args = args
1151
 
        self.kwargs = kwargs
1152
 
 
1153
 
 
1154
 
class TestMergerBase(TestCaseWithMemoryTransport):
1155
 
    """Common functionality for Merger tests that don't write to disk."""
1156
 
 
1157
 
    def get_builder(self):
1158
 
        builder = self.make_branch_builder('path')
1159
 
        builder.start_series()
1160
 
        self.addCleanup(builder.finish_series)
1161
 
        return builder
1162
 
 
1163
 
    def setup_simple_graph(self):
1164
 
        """Create a simple 3-node graph.
1165
 
 
1166
 
        :return: A BranchBuilder
1167
 
        """
1168
 
        #
1169
 
        #  A
1170
 
        #  |\
1171
 
        #  B C
1172
 
        #
1173
 
        builder = self.get_builder()
1174
 
        builder.build_snapshot('A-id', None,
1175
 
            [('add', ('', None, 'directory', None))])
1176
 
        builder.build_snapshot('C-id', ['A-id'], [])
1177
 
        builder.build_snapshot('B-id', ['A-id'], [])
1178
 
        return builder
1179
 
 
1180
 
    def setup_criss_cross_graph(self):
1181
 
        """Create a 5-node graph with a criss-cross.
1182
 
 
1183
 
        :return: A BranchBuilder
1184
 
        """
1185
 
        # A
1186
 
        # |\
1187
 
        # B C
1188
 
        # |X|
1189
 
        # D E
1190
 
        builder = self.setup_simple_graph()
1191
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
1192
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1193
 
        return builder
1194
 
 
1195
 
    def make_Merger(self, builder, other_revision_id,
1196
 
                    interesting_files=None, interesting_ids=None):
1197
 
        """Make a Merger object from a branch builder"""
1198
 
        mem_tree = memorytree.MemoryTree.create_on_branch(builder.get_branch())
1199
 
        mem_tree.lock_write()
1200
 
        self.addCleanup(mem_tree.unlock)
1201
 
        merger = _mod_merge.Merger.from_revision_ids(None,
1202
 
            mem_tree, other_revision_id)
1203
 
        merger.set_interesting_files(interesting_files)
1204
 
        # It seems there is no matching function for set_interesting_ids
1205
 
        merger.interesting_ids = interesting_ids
1206
 
        merger.merge_type = _mod_merge.Merge3Merger
1207
 
        return merger
1208
 
 
1209
 
 
1210
 
class TestMergerInMemory(TestMergerBase):
1211
 
 
1212
 
    def test_cache_trees_with_revision_ids_None(self):
1213
 
        merger = self.make_Merger(self.setup_simple_graph(), 'C-id')
1214
 
        original_cache = dict(merger._cached_trees)
1215
 
        merger.cache_trees_with_revision_ids([None])
1216
 
        self.assertEqual(original_cache, merger._cached_trees)
1217
 
 
1218
 
    def test_cache_trees_with_revision_ids_no_revision_id(self):
1219
 
        merger = self.make_Merger(self.setup_simple_graph(), 'C-id')
1220
 
        original_cache = dict(merger._cached_trees)
1221
 
        tree = self.make_branch_and_memory_tree('tree')
1222
 
        merger.cache_trees_with_revision_ids([tree])
1223
 
        self.assertEqual(original_cache, merger._cached_trees)
1224
 
 
1225
 
    def test_cache_trees_with_revision_ids_having_revision_id(self):
1226
 
        merger = self.make_Merger(self.setup_simple_graph(), 'C-id')
1227
 
        original_cache = dict(merger._cached_trees)
1228
 
        tree = merger.this_branch.repository.revision_tree('B-id')
1229
 
        original_cache['B-id'] = tree
1230
 
        merger.cache_trees_with_revision_ids([tree])
1231
 
        self.assertEqual(original_cache, merger._cached_trees)
1232
 
 
1233
 
    def test_find_base(self):
1234
 
        merger = self.make_Merger(self.setup_simple_graph(), 'C-id')
1235
 
        self.assertEqual('A-id', merger.base_rev_id)
1236
 
        self.assertFalse(merger._is_criss_cross)
1237
 
        self.assertIs(None, merger._lca_trees)
1238
 
 
1239
 
    def test_find_base_criss_cross(self):
1240
 
        builder = self.setup_criss_cross_graph()
1241
 
        merger = self.make_Merger(builder, 'E-id')
1242
 
        self.assertEqual('A-id', merger.base_rev_id)
1243
 
        self.assertTrue(merger._is_criss_cross)
1244
 
        self.assertEqual(['B-id', 'C-id'], [t.get_revision_id()
1245
 
                                            for t in merger._lca_trees])
1246
 
        # If we swap the order, we should get a different lca order
1247
 
        builder.build_snapshot('F-id', ['E-id'], [])
1248
 
        merger = self.make_Merger(builder, 'D-id')
1249
 
        self.assertEqual(['C-id', 'B-id'], [t.get_revision_id()
1250
 
                                            for t in merger._lca_trees])
1251
 
 
1252
 
    def test_find_base_triple_criss_cross(self):
1253
 
        #       A-.
1254
 
        #      / \ \
1255
 
        #     B   C F # F is merged into both branches
1256
 
        #     |\ /| |
1257
 
        #     | X | |\
1258
 
        #     |/ \| | :
1259
 
        #   : D   E |
1260
 
        #    \|   |/
1261
 
        #     G   H
1262
 
        builder = self.setup_criss_cross_graph()
1263
 
        builder.build_snapshot('F-id', ['A-id'], [])
1264
 
        builder.build_snapshot('H-id', ['E-id', 'F-id'], [])
1265
 
        builder.build_snapshot('G-id', ['D-id', 'F-id'], [])
1266
 
        merger = self.make_Merger(builder, 'H-id')
1267
 
        self.assertEqual(['B-id', 'C-id', 'F-id'],
1268
 
                         [t.get_revision_id() for t in merger._lca_trees])
1269
 
 
1270
 
    def test_no_criss_cross_passed_to_merge_type(self):
1271
 
        class LCATreesMerger(LoggingMerger):
1272
 
            supports_lca_trees = True
1273
 
 
1274
 
        merger = self.make_Merger(self.setup_simple_graph(), 'C-id')
1275
 
        merger.merge_type = LCATreesMerger
1276
 
        merge_obj = merger.make_merger()
1277
 
        self.assertIsInstance(merge_obj, LCATreesMerger)
1278
 
        self.assertFalse('lca_trees' in merge_obj.kwargs)
1279
 
 
1280
 
    def test_criss_cross_passed_to_merge_type(self):
1281
 
        merger = self.make_Merger(self.setup_criss_cross_graph(), 'E-id')
1282
 
        merger.merge_type = _mod_merge.Merge3Merger
1283
 
        merge_obj = merger.make_merger()
1284
 
        self.assertEqual(['B-id', 'C-id'], [t.get_revision_id()
1285
 
                                            for t in merger._lca_trees])
1286
 
 
1287
 
    def test_criss_cross_not_supported_merge_type(self):
1288
 
        merger = self.make_Merger(self.setup_criss_cross_graph(), 'E-id')
1289
 
        # We explicitly do not define supports_lca_trees
1290
 
        merger.merge_type = LoggingMerger
1291
 
        merge_obj = merger.make_merger()
1292
 
        self.assertIsInstance(merge_obj, LoggingMerger)
1293
 
        self.assertFalse('lca_trees' in merge_obj.kwargs)
1294
 
 
1295
 
    def test_criss_cross_unsupported_merge_type(self):
1296
 
        class UnsupportedLCATreesMerger(LoggingMerger):
1297
 
            supports_lca_trees = False
1298
 
 
1299
 
        merger = self.make_Merger(self.setup_criss_cross_graph(), 'E-id')
1300
 
        merger.merge_type = UnsupportedLCATreesMerger
1301
 
        merge_obj = merger.make_merger()
1302
 
        self.assertIsInstance(merge_obj, UnsupportedLCATreesMerger)
1303
 
        self.assertFalse('lca_trees' in merge_obj.kwargs)
1304
 
 
1305
 
 
1306
 
class TestMergerEntriesLCA(TestMergerBase):
1307
 
 
1308
 
    def make_merge_obj(self, builder, other_revision_id,
1309
 
                       interesting_files=None, interesting_ids=None):
1310
 
        merger = self.make_Merger(builder, other_revision_id,
1311
 
            interesting_files=interesting_files,
1312
 
            interesting_ids=interesting_ids)
1313
 
        return merger.make_merger()
1314
 
 
1315
 
    def test_simple(self):
1316
 
        builder = self.get_builder()
1317
 
        builder.build_snapshot('A-id', None,
1318
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1319
 
             ('add', (u'a', 'a-id', 'file', 'a\nb\nc\n'))])
1320
 
        builder.build_snapshot('C-id', ['A-id'],
1321
 
            [('modify', ('a-id', 'a\nb\nC\nc\n'))])
1322
 
        builder.build_snapshot('B-id', ['A-id'],
1323
 
            [('modify', ('a-id', 'a\nB\nb\nc\n'))])
1324
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1325
 
            [('modify', ('a-id', 'a\nB\nb\nC\nc\nE\n'))])
1326
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'],
1327
 
            [('modify', ('a-id', 'a\nB\nb\nC\nc\n'))])
1328
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1329
 
 
1330
 
        self.assertEqual(['B-id', 'C-id'], [t.get_revision_id()
1331
 
                                            for t in merge_obj._lca_trees])
1332
 
        self.assertEqual('A-id', merge_obj.base_tree.get_revision_id())
1333
 
        entries = list(merge_obj._entries_lca())
1334
 
 
1335
 
        # (file_id, changed, parents, names, executable)
1336
 
        # BASE, lca1, lca2, OTHER, THIS
1337
 
        root_id = 'a-root-id'
1338
 
        self.assertEqual([('a-id', True,
1339
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
1340
 
                           ((u'a', [u'a', u'a']), u'a', u'a'),
1341
 
                           ((False, [False, False]), False, False)),
1342
 
                         ], entries)
1343
 
 
1344
 
    def test_not_in_base(self):
1345
 
        # LCAs all have the same last-modified revision for the file, as do
1346
 
        # the tips, but the base has something different
1347
 
        #       A    base, doesn't have the file
1348
 
        #       |\
1349
 
        #       B C  B introduces 'foo', C introduces 'bar'
1350
 
        #       |X|
1351
 
        #       D E  D and E now both have 'foo' and 'bar'
1352
 
        #       |X|
1353
 
        #       F G  the files are now in F, G, D and E, but not in A
1354
 
        #            G modifies 'bar'
1355
 
 
1356
 
        builder = self.get_builder()
1357
 
        builder.build_snapshot('A-id', None,
1358
 
            [('add', (u'', 'a-root-id', 'directory', None))])
1359
 
        builder.build_snapshot('B-id', ['A-id'],
1360
 
            [('add', (u'foo', 'foo-id', 'file', 'a\nb\nc\n'))])
1361
 
        builder.build_snapshot('C-id', ['A-id'],
1362
 
            [('add', (u'bar', 'bar-id', 'file', 'd\ne\nf\n'))])
1363
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'],
1364
 
            [('add', (u'bar', 'bar-id', 'file', 'd\ne\nf\n'))])
1365
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1366
 
            [('add', (u'foo', 'foo-id', 'file', 'a\nb\nc\n'))])
1367
 
        builder.build_snapshot('G-id', ['E-id', 'D-id'],
1368
 
            [('modify', (u'bar-id', 'd\ne\nf\nG\n'))])
1369
 
        builder.build_snapshot('F-id', ['D-id', 'E-id'], [])
1370
 
        merge_obj = self.make_merge_obj(builder, 'G-id')
1371
 
 
1372
 
        self.assertEqual(['D-id', 'E-id'], [t.get_revision_id()
1373
 
                                            for t in merge_obj._lca_trees])
1374
 
        self.assertEqual('A-id', merge_obj.base_tree.get_revision_id())
1375
 
        entries = list(merge_obj._entries_lca())
1376
 
        root_id = 'a-root-id'
1377
 
        self.assertEqual([('bar-id', True,
1378
 
                           ((None, [root_id, root_id]), root_id, root_id),
1379
 
                           ((None, [u'bar', u'bar']), u'bar', u'bar'),
1380
 
                           ((None, [False, False]), False, False)),
1381
 
                         ], entries)
1382
 
 
1383
 
    def test_not_in_this(self):
1384
 
        builder = self.get_builder()
1385
 
        builder.build_snapshot('A-id', None,
1386
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1387
 
             ('add', (u'a', 'a-id', 'file', 'a\nb\nc\n'))])
1388
 
        builder.build_snapshot('B-id', ['A-id'],
1389
 
            [('modify', ('a-id', 'a\nB\nb\nc\n'))])
1390
 
        builder.build_snapshot('C-id', ['A-id'],
1391
 
            [('modify', ('a-id', 'a\nb\nC\nc\n'))])
1392
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1393
 
            [('modify', ('a-id', 'a\nB\nb\nC\nc\nE\n'))])
1394
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'],
1395
 
            [('unversion', 'a-id')])
1396
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1397
 
 
1398
 
        self.assertEqual(['B-id', 'C-id'], [t.get_revision_id()
1399
 
                                            for t in merge_obj._lca_trees])
1400
 
        self.assertEqual('A-id', merge_obj.base_tree.get_revision_id())
1401
 
 
1402
 
        entries = list(merge_obj._entries_lca())
1403
 
        root_id = 'a-root-id'
1404
 
        self.assertEqual([('a-id', True,
1405
 
                           ((root_id, [root_id, root_id]), root_id, None),
1406
 
                           ((u'a', [u'a', u'a']), u'a', None),
1407
 
                           ((False, [False, False]), False, None)),
1408
 
                         ], entries)
1409
 
 
1410
 
    def test_file_not_in_one_lca(self):
1411
 
        #   A   # just root
1412
 
        #   |\
1413
 
        #   B C # B no file, C introduces a file
1414
 
        #   |X|
1415
 
        #   D E # D and E both have the file, unchanged from C
1416
 
        builder = self.get_builder()
1417
 
        builder.build_snapshot('A-id', None,
1418
 
            [('add', (u'', 'a-root-id', 'directory', None))])
1419
 
        builder.build_snapshot('B-id', ['A-id'], [])
1420
 
        builder.build_snapshot('C-id', ['A-id'],
1421
 
            [('add', (u'a', 'a-id', 'file', 'a\nb\nc\n'))])
1422
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], []) # Inherited from C
1423
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], # Merged from C
1424
 
            [('add', (u'a', 'a-id', 'file', 'a\nb\nc\n'))])
1425
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1426
 
 
1427
 
        self.assertEqual(['B-id', 'C-id'], [t.get_revision_id()
1428
 
                                            for t in merge_obj._lca_trees])
1429
 
        self.assertEqual('A-id', merge_obj.base_tree.get_revision_id())
1430
 
 
1431
 
        entries = list(merge_obj._entries_lca())
1432
 
        self.assertEqual([], entries)
1433
 
 
1434
 
    def test_not_in_other(self):
1435
 
        builder = self.get_builder()
1436
 
        builder.build_snapshot('A-id', None,
1437
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1438
 
             ('add', (u'a', 'a-id', 'file', 'a\nb\nc\n'))])
1439
 
        builder.build_snapshot('B-id', ['A-id'], [])
1440
 
        builder.build_snapshot('C-id', ['A-id'], [])
1441
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1442
 
            [('unversion', 'a-id')])
1443
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1444
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1445
 
 
1446
 
        entries = list(merge_obj._entries_lca())
1447
 
        root_id = 'a-root-id'
1448
 
        self.assertEqual([('a-id', True,
1449
 
                           ((root_id, [root_id, root_id]), None, root_id),
1450
 
                           ((u'a', [u'a', u'a']), None, u'a'),
1451
 
                           ((False, [False, False]), None, False)),
1452
 
                         ], entries)
1453
 
 
1454
 
    def test_not_in_other_or_lca(self):
1455
 
        #       A    base, introduces 'foo'
1456
 
        #       |\
1457
 
        #       B C  B nothing, C deletes foo
1458
 
        #       |X|
1459
 
        #       D E  D restores foo (same as B), E leaves it deleted
1460
 
        # Analysis:
1461
 
        #   A => B, no changes
1462
 
        #   A => C, delete foo (C should supersede B)
1463
 
        #   C => D, restore foo
1464
 
        #   C => E, no changes
1465
 
        # D would then win 'cleanly' and no record would be given
1466
 
        builder = self.get_builder()
1467
 
        builder.build_snapshot('A-id', None,
1468
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1469
 
             ('add', (u'foo', 'foo-id', 'file', 'content\n'))])
1470
 
        builder.build_snapshot('B-id', ['A-id'], [])
1471
 
        builder.build_snapshot('C-id', ['A-id'],
1472
 
            [('unversion', 'foo-id')])
1473
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
1474
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1475
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1476
 
 
1477
 
        entries = list(merge_obj._entries_lca())
1478
 
        self.assertEqual([], entries)
1479
 
 
1480
 
    def test_not_in_other_mod_in_lca1_not_in_lca2(self):
1481
 
        #       A    base, introduces 'foo'
1482
 
        #       |\
1483
 
        #       B C  B changes 'foo', C deletes foo
1484
 
        #       |X|
1485
 
        #       D E  D restores foo (same as B), E leaves it deleted (as C)
1486
 
        # Analysis:
1487
 
        #   A => B, modified foo
1488
 
        #   A => C, delete foo, C does not supersede B
1489
 
        #   B => D, no changes
1490
 
        #   C => D, resolve in favor of B
1491
 
        #   B => E, resolve in favor of E
1492
 
        #   C => E, no changes
1493
 
        # In this case, we have a conflict of how the changes were resolved. E
1494
 
        # picked C and D picked B, so we should issue a conflict
1495
 
        builder = self.get_builder()
1496
 
        builder.build_snapshot('A-id', None,
1497
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1498
 
             ('add', (u'foo', 'foo-id', 'file', 'content\n'))])
1499
 
        builder.build_snapshot('B-id', ['A-id'], [
1500
 
            ('modify', ('foo-id', 'new-content\n'))])
1501
 
        builder.build_snapshot('C-id', ['A-id'],
1502
 
            [('unversion', 'foo-id')])
1503
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
1504
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1505
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1506
 
 
1507
 
        entries = list(merge_obj._entries_lca())
1508
 
        root_id = 'a-root-id'
1509
 
        self.assertEqual([('foo-id', True,
1510
 
                           ((root_id, [root_id, None]), None, root_id),
1511
 
                           ((u'foo', [u'foo', None]), None, 'foo'),
1512
 
                           ((False, [False, None]), None, False)),
1513
 
                         ], entries)
1514
 
 
1515
 
    def test_only_in_one_lca(self):
1516
 
        #   A   add only root
1517
 
        #   |\
1518
 
        #   B C B nothing, C add file
1519
 
        #   |X|
1520
 
        #   D E D still has nothing, E removes file
1521
 
        # Analysis:
1522
 
        #   B => D, no change
1523
 
        #   C => D, removed the file
1524
 
        #   B => E, no change
1525
 
        #   C => E, removed the file
1526
 
        # Thus D & E have identical changes, and this is a no-op
1527
 
        # Alternatively:
1528
 
        #   A => B, no change
1529
 
        #   A => C, add file, thus C supersedes B
1530
 
        #   w/ C=BASE, D=THIS, E=OTHER we have 'happy convergence'
1531
 
        builder = self.get_builder()
1532
 
        builder.build_snapshot('A-id', None,
1533
 
            [('add', (u'', 'a-root-id', 'directory', None))])
1534
 
        builder.build_snapshot('B-id', ['A-id'], [])
1535
 
        builder.build_snapshot('C-id', ['A-id'],
1536
 
            [('add', (u'a', 'a-id', 'file', 'a\nb\nc\n'))])
1537
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1538
 
            [('unversion', 'a-id')])
1539
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1540
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1541
 
 
1542
 
        entries = list(merge_obj._entries_lca())
1543
 
        self.assertEqual([], entries)
1544
 
 
1545
 
    def test_only_in_other(self):
1546
 
        builder = self.get_builder()
1547
 
        builder.build_snapshot('A-id', None,
1548
 
            [('add', (u'', 'a-root-id', 'directory', None))])
1549
 
        builder.build_snapshot('B-id', ['A-id'], [])
1550
 
        builder.build_snapshot('C-id', ['A-id'], [])
1551
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1552
 
            [('add', (u'a', 'a-id', 'file', 'a\nb\nc\n'))])
1553
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1554
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1555
 
 
1556
 
        entries = list(merge_obj._entries_lca())
1557
 
        root_id = 'a-root-id'
1558
 
        self.assertEqual([('a-id', True,
1559
 
                           ((None, [None, None]), root_id, None),
1560
 
                           ((None, [None, None]), u'a', None),
1561
 
                           ((None, [None, None]), False, None)),
1562
 
                         ], entries)
1563
 
 
1564
 
    def test_one_lca_supersedes(self):
1565
 
        # One LCA supersedes the other LCAs last modified value, but the
1566
 
        # value is not the same as BASE.
1567
 
        #       A    base, introduces 'foo', last mod A
1568
 
        #       |\
1569
 
        #       B C  B modifies 'foo' (mod B), C does nothing (mod A)
1570
 
        #       |X|
1571
 
        #       D E  D does nothing (mod B), E updates 'foo' (mod E)
1572
 
        #       |X|
1573
 
        #       F G  F updates 'foo' (mod F). G does nothing (mod E)
1574
 
        #
1575
 
        #   At this point, G should not be considered to modify 'foo', even
1576
 
        #   though its LCAs disagree. This is because the modification in E
1577
 
        #   completely supersedes the value in D.
1578
 
        builder = self.get_builder()
1579
 
        builder.build_snapshot('A-id', None,
1580
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1581
 
             ('add', (u'foo', 'foo-id', 'file', 'A content\n'))])
1582
 
        builder.build_snapshot('C-id', ['A-id'], [])
1583
 
        builder.build_snapshot('B-id', ['A-id'],
1584
 
            [('modify', ('foo-id', 'B content\n'))])
1585
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1586
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1587
 
            [('modify', ('foo-id', 'E content\n'))])
1588
 
        builder.build_snapshot('G-id', ['E-id', 'D-id'], [])
1589
 
        builder.build_snapshot('F-id', ['D-id', 'E-id'],
1590
 
            [('modify', ('foo-id', 'F content\n'))])
1591
 
        merge_obj = self.make_merge_obj(builder, 'G-id')
1592
 
 
1593
 
        self.assertEqual([], list(merge_obj._entries_lca()))
1594
 
 
1595
 
    def test_one_lca_supersedes_path(self):
1596
 
        # Double-criss-cross merge, the ultimate base value is different from
1597
 
        # the intermediate.
1598
 
        #   A    value 'foo'
1599
 
        #   |\
1600
 
        #   B C  B value 'bar', C = 'foo'
1601
 
        #   |X|
1602
 
        #   D E  D = 'bar', E supersedes to 'bing'
1603
 
        #   |X|
1604
 
        #   F G  F = 'bing', G supersedes to 'barry'
1605
 
        #
1606
 
        # In this case, we technically should not care about the value 'bar' for
1607
 
        # D, because it was clearly superseded by E's 'bing'. The
1608
 
        # per-file/attribute graph would actually look like:
1609
 
        #   A
1610
 
        #   |
1611
 
        #   B
1612
 
        #   |
1613
 
        #   E
1614
 
        #   |
1615
 
        #   G
1616
 
        #
1617
 
        # Because the other side of the merge never modifies the value, it just
1618
 
        # takes the value from the merge.
1619
 
        #
1620
 
        # ATM this fails because we will prune 'foo' from the LCAs, but we
1621
 
        # won't prune 'bar'. This is getting far off into edge-case land, so we
1622
 
        # aren't supporting it yet.
1623
 
        #
1624
 
        builder = self.get_builder()
1625
 
        builder.build_snapshot('A-id', None,
1626
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1627
 
             ('add', (u'foo', 'foo-id', 'file', 'A content\n'))])
1628
 
        builder.build_snapshot('C-id', ['A-id'], [])
1629
 
        builder.build_snapshot('B-id', ['A-id'],
1630
 
            [('rename', ('foo', 'bar'))])
1631
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1632
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1633
 
            [('rename', ('foo', 'bing'))]) # override to bing
1634
 
        builder.build_snapshot('G-id', ['E-id', 'D-id'],
1635
 
            [('rename', ('bing', 'barry'))]) # override to barry
1636
 
        builder.build_snapshot('F-id', ['D-id', 'E-id'],
1637
 
            [('rename', ('bar', 'bing'))]) # Merge in E's change
1638
 
        merge_obj = self.make_merge_obj(builder, 'G-id')
1639
 
 
1640
 
        self.expectFailure("We don't do an actual heads() check on lca values,"
1641
 
            " or use the per-attribute graph",
1642
 
            self.assertEqual, [], list(merge_obj._entries_lca()))
1643
 
 
1644
 
    def test_one_lca_accidentally_pruned(self):
1645
 
        # Another incorrect resolution from the same basic flaw:
1646
 
        #   A    value 'foo'
1647
 
        #   |\
1648
 
        #   B C  B value 'bar', C = 'foo'
1649
 
        #   |X|
1650
 
        #   D E  D = 'bar', E reverts to 'foo'
1651
 
        #   |X|
1652
 
        #   F G  F = 'bing', G switches to 'bar'
1653
 
        #
1654
 
        # 'bar' will not be seen as an interesting change, because 'foo' will
1655
 
        # be pruned from the LCAs, even though it was newly introduced by E
1656
 
        # (superseding B).
1657
 
        builder = self.get_builder()
1658
 
        builder.build_snapshot('A-id', None,
1659
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1660
 
             ('add', (u'foo', 'foo-id', 'file', 'A content\n'))])
1661
 
        builder.build_snapshot('C-id', ['A-id'], [])
1662
 
        builder.build_snapshot('B-id', ['A-id'],
1663
 
            [('rename', ('foo', 'bar'))])
1664
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1665
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
1666
 
        builder.build_snapshot('G-id', ['E-id', 'D-id'],
1667
 
            [('rename', ('foo', 'bar'))])
1668
 
        builder.build_snapshot('F-id', ['D-id', 'E-id'],
1669
 
            [('rename', ('bar', 'bing'))]) # should end up conflicting
1670
 
        merge_obj = self.make_merge_obj(builder, 'G-id')
1671
 
 
1672
 
        entries = list(merge_obj._entries_lca())
1673
 
        root_id = 'a-root-id'
1674
 
        self.expectFailure("We prune values from BASE even when relevant.",
1675
 
            self.assertEqual,
1676
 
                [('foo-id', False,
1677
 
                  ((root_id, [root_id, root_id]), root_id, root_id),
1678
 
                  ((u'foo', [u'bar', u'foo']), u'bar', u'bing'),
1679
 
                  ((False, [False, False]), False, False)),
1680
 
                ], entries)
1681
 
 
1682
 
    def test_both_sides_revert(self):
1683
 
        # Both sides of a criss-cross revert the text to the lca
1684
 
        #       A    base, introduces 'foo'
1685
 
        #       |\
1686
 
        #       B C  B modifies 'foo', C modifies 'foo'
1687
 
        #       |X|
1688
 
        #       D E  D reverts to B, E reverts to C
1689
 
        # This should conflict
1690
 
        builder = self.get_builder()
1691
 
        builder.build_snapshot('A-id', None,
1692
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1693
 
             ('add', (u'foo', 'foo-id', 'file', 'A content\n'))])
1694
 
        builder.build_snapshot('B-id', ['A-id'],
1695
 
            [('modify', ('foo-id', 'B content\n'))])
1696
 
        builder.build_snapshot('C-id', ['A-id'],
1697
 
            [('modify', ('foo-id', 'C content\n'))])
1698
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
1699
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1700
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1701
 
 
1702
 
        entries = list(merge_obj._entries_lca())
1703
 
        root_id = 'a-root-id'
1704
 
        self.assertEqual([('foo-id', True,
1705
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
1706
 
                           ((u'foo', [u'foo', u'foo']), u'foo', u'foo'),
1707
 
                           ((False, [False, False]), False, False)),
1708
 
                         ], entries)
1709
 
 
1710
 
    def test_different_lca_resolve_one_side_updates_content(self):
1711
 
        # Both sides converge, but then one side updates the text.
1712
 
        #       A    base, introduces 'foo'
1713
 
        #       |\
1714
 
        #       B C  B modifies 'foo', C modifies 'foo'
1715
 
        #       |X|
1716
 
        #       D E  D reverts to B, E reverts to C
1717
 
        #       |
1718
 
        #       F    F updates to a new value
1719
 
        # We need to emit an entry for 'foo', because D & E differed on the
1720
 
        # merge resolution
1721
 
        builder = self.get_builder()
1722
 
        builder.build_snapshot('A-id', None,
1723
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1724
 
             ('add', (u'foo', 'foo-id', 'file', 'A content\n'))])
1725
 
        builder.build_snapshot('B-id', ['A-id'],
1726
 
            [('modify', ('foo-id', 'B content\n'))])
1727
 
        builder.build_snapshot('C-id', ['A-id'],
1728
 
            [('modify', ('foo-id', 'C content\n'))])
1729
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
1730
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1731
 
        builder.build_snapshot('F-id', ['D-id'],
1732
 
            [('modify', ('foo-id', 'F content\n'))])
1733
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1734
 
 
1735
 
        entries = list(merge_obj._entries_lca())
1736
 
        root_id = 'a-root-id'
1737
 
        self.assertEqual([('foo-id', True,
1738
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
1739
 
                           ((u'foo', [u'foo', u'foo']), u'foo', u'foo'),
1740
 
                           ((False, [False, False]), False, False)),
1741
 
                         ], entries)
1742
 
 
1743
 
    def test_same_lca_resolution_one_side_updates_content(self):
1744
 
        # Both sides converge, but then one side updates the text.
1745
 
        #       A    base, introduces 'foo'
1746
 
        #       |\
1747
 
        #       B C  B modifies 'foo', C modifies 'foo'
1748
 
        #       |X|
1749
 
        #       D E  D and E use C's value
1750
 
        #       |
1751
 
        #       F    F updates to a new value
1752
 
        # I think it is a bug that this conflicts, but we don't have a way to
1753
 
        # detect otherwise. And because of:
1754
 
        #   test_different_lca_resolve_one_side_updates_content
1755
 
        # We need to conflict.
1756
 
 
1757
 
        builder = self.get_builder()
1758
 
        builder.build_snapshot('A-id', None,
1759
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1760
 
             ('add', (u'foo', 'foo-id', 'file', 'A content\n'))])
1761
 
        builder.build_snapshot('B-id', ['A-id'],
1762
 
            [('modify', ('foo-id', 'B content\n'))])
1763
 
        builder.build_snapshot('C-id', ['A-id'],
1764
 
            [('modify', ('foo-id', 'C content\n'))])
1765
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
1766
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'],
1767
 
            [('modify', ('foo-id', 'C content\n'))]) # Same as E
1768
 
        builder.build_snapshot('F-id', ['D-id'],
1769
 
            [('modify', ('foo-id', 'F content\n'))])
1770
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1771
 
 
1772
 
        entries = list(merge_obj._entries_lca())
1773
 
        self.expectFailure("We don't detect that LCA resolution was the"
1774
 
                           " same on both sides",
1775
 
            self.assertEqual, [], entries)
1776
 
 
1777
 
    def test_only_path_changed(self):
1778
 
        builder = self.get_builder()
1779
 
        builder.build_snapshot('A-id', None,
1780
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1781
 
             ('add', (u'a', 'a-id', 'file', 'content\n'))])
1782
 
        builder.build_snapshot('B-id', ['A-id'], [])
1783
 
        builder.build_snapshot('C-id', ['A-id'], [])
1784
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1785
 
            [('rename', (u'a', u'b'))])
1786
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1787
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1788
 
        entries = list(merge_obj._entries_lca())
1789
 
        root_id = 'a-root-id'
1790
 
        # The content was not changed, only the path
1791
 
        self.assertEqual([('a-id', False,
1792
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
1793
 
                           ((u'a', [u'a', u'a']), u'b', u'a'),
1794
 
                           ((False, [False, False]), False, False)),
1795
 
                         ], entries)
1796
 
 
1797
 
    def test_kind_changed(self):
1798
 
        # Identical content, except 'D' changes a-id into a directory
1799
 
        builder = self.get_builder()
1800
 
        builder.build_snapshot('A-id', None,
1801
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1802
 
             ('add', (u'a', 'a-id', 'file', 'content\n'))])
1803
 
        builder.build_snapshot('B-id', ['A-id'], [])
1804
 
        builder.build_snapshot('C-id', ['A-id'], [])
1805
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1806
 
            [('unversion', 'a-id'),
1807
 
             ('add', (u'a', 'a-id', 'directory', None))])
1808
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1809
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1810
 
        entries = list(merge_obj._entries_lca())
1811
 
        root_id = 'a-root-id'
1812
 
        # Only the kind was changed (content)
1813
 
        self.assertEqual([('a-id', True,
1814
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
1815
 
                           ((u'a', [u'a', u'a']), u'a', u'a'),
1816
 
                           ((False, [False, False]), False, False)),
1817
 
                         ], entries)
1818
 
 
1819
 
    def test_this_changed_kind(self):
1820
 
        # Identical content, but THIS changes a file to a directory
1821
 
        builder = self.get_builder()
1822
 
        builder.build_snapshot('A-id', None,
1823
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1824
 
             ('add', (u'a', 'a-id', 'file', 'content\n'))])
1825
 
        builder.build_snapshot('B-id', ['A-id'], [])
1826
 
        builder.build_snapshot('C-id', ['A-id'], [])
1827
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
1828
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'],
1829
 
            [('unversion', 'a-id'),
1830
 
             ('add', (u'a', 'a-id', 'directory', None))])
1831
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1832
 
        entries = list(merge_obj._entries_lca())
1833
 
        # Only the kind was changed (content)
1834
 
        self.assertEqual([], entries)
1835
 
 
1836
 
    def test_interesting_files(self):
1837
 
        # Two files modified, but we should filter one of them
1838
 
        builder = self.get_builder()
1839
 
        builder.build_snapshot('A-id', None,
1840
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1841
 
             ('add', (u'a', 'a-id', 'file', 'content\n')),
1842
 
             ('add', (u'b', 'b-id', 'file', 'content\n'))])
1843
 
        builder.build_snapshot('B-id', ['A-id'], [])
1844
 
        builder.build_snapshot('C-id', ['A-id'], [])
1845
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1846
 
            [('modify', ('a-id', 'new-content\n')),
1847
 
             ('modify', ('b-id', 'new-content\n'))])
1848
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1849
 
        merge_obj = self.make_merge_obj(builder, 'E-id',
1850
 
                                        interesting_files=['b'])
1851
 
        entries = list(merge_obj._entries_lca())
1852
 
        root_id = 'a-root-id'
1853
 
        self.assertEqual([('b-id', True,
1854
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
1855
 
                           ((u'b', [u'b', u'b']), u'b', u'b'),
1856
 
                           ((False, [False, False]), False, False)),
1857
 
                         ], entries)
1858
 
 
1859
 
    def test_interesting_file_in_this(self):
1860
 
        # This renamed the file, but it should still match the entry in other
1861
 
        builder = self.get_builder()
1862
 
        builder.build_snapshot('A-id', None,
1863
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1864
 
             ('add', (u'a', 'a-id', 'file', 'content\n')),
1865
 
             ('add', (u'b', 'b-id', 'file', 'content\n'))])
1866
 
        builder.build_snapshot('B-id', ['A-id'], [])
1867
 
        builder.build_snapshot('C-id', ['A-id'], [])
1868
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1869
 
            [('modify', ('a-id', 'new-content\n')),
1870
 
             ('modify', ('b-id', 'new-content\n'))])
1871
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'],
1872
 
            [('rename', ('b', 'c'))])
1873
 
        merge_obj = self.make_merge_obj(builder, 'E-id',
1874
 
                                        interesting_files=['c'])
1875
 
        entries = list(merge_obj._entries_lca())
1876
 
        root_id = 'a-root-id'
1877
 
        self.assertEqual([('b-id', True,
1878
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
1879
 
                           ((u'b', [u'b', u'b']), u'b', u'c'),
1880
 
                           ((False, [False, False]), False, False)),
1881
 
                         ], entries)
1882
 
 
1883
 
    def test_interesting_file_in_base(self):
1884
 
        # This renamed the file, but it should still match the entry in BASE
1885
 
        builder = self.get_builder()
1886
 
        builder.build_snapshot('A-id', None,
1887
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1888
 
             ('add', (u'a', 'a-id', 'file', 'content\n')),
1889
 
             ('add', (u'c', 'c-id', 'file', 'content\n'))])
1890
 
        builder.build_snapshot('B-id', ['A-id'],
1891
 
            [('rename', ('c', 'b'))])
1892
 
        builder.build_snapshot('C-id', ['A-id'],
1893
 
            [('rename', ('c', 'b'))])
1894
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1895
 
            [('modify', ('a-id', 'new-content\n')),
1896
 
             ('modify', ('c-id', 'new-content\n'))])
1897
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1898
 
        merge_obj = self.make_merge_obj(builder, 'E-id',
1899
 
                                        interesting_files=['c'])
1900
 
        entries = list(merge_obj._entries_lca())
1901
 
        root_id = 'a-root-id'
1902
 
        self.assertEqual([('c-id', True,
1903
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
1904
 
                           ((u'c', [u'b', u'b']), u'b', u'b'),
1905
 
                           ((False, [False, False]), False, False)),
1906
 
                         ], entries)
1907
 
 
1908
 
    def test_interesting_file_in_lca(self):
1909
 
        # This renamed the file, but it should still match the entry in LCA
1910
 
        builder = self.get_builder()
1911
 
        builder.build_snapshot('A-id', None,
1912
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1913
 
             ('add', (u'a', 'a-id', 'file', 'content\n')),
1914
 
             ('add', (u'b', 'b-id', 'file', 'content\n'))])
1915
 
        builder.build_snapshot('B-id', ['A-id'],
1916
 
            [('rename', ('b', 'c'))])
1917
 
        builder.build_snapshot('C-id', ['A-id'], [])
1918
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1919
 
            [('modify', ('a-id', 'new-content\n')),
1920
 
             ('modify', ('b-id', 'new-content\n'))])
1921
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'],
1922
 
            [('rename', ('c', 'b'))])
1923
 
        merge_obj = self.make_merge_obj(builder, 'E-id',
1924
 
                                        interesting_files=['c'])
1925
 
        entries = list(merge_obj._entries_lca())
1926
 
        root_id = 'a-root-id'
1927
 
        self.assertEqual([('b-id', True,
1928
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
1929
 
                           ((u'b', [u'c', u'b']), u'b', u'b'),
1930
 
                           ((False, [False, False]), False, False)),
1931
 
                         ], entries)
1932
 
 
1933
 
    def test_interesting_ids(self):
1934
 
        # Two files modified, but we should filter one of them
1935
 
        builder = self.get_builder()
1936
 
        builder.build_snapshot('A-id', None,
1937
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1938
 
             ('add', (u'a', 'a-id', 'file', 'content\n')),
1939
 
             ('add', (u'b', 'b-id', 'file', 'content\n'))])
1940
 
        builder.build_snapshot('B-id', ['A-id'], [])
1941
 
        builder.build_snapshot('C-id', ['A-id'], [])
1942
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1943
 
            [('modify', ('a-id', 'new-content\n')),
1944
 
             ('modify', ('b-id', 'new-content\n'))])
1945
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1946
 
        merge_obj = self.make_merge_obj(builder, 'E-id',
1947
 
                                        interesting_ids=['b-id'])
1948
 
        entries = list(merge_obj._entries_lca())
1949
 
        root_id = 'a-root-id'
1950
 
        self.assertEqual([('b-id', True,
1951
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
1952
 
                           ((u'b', [u'b', u'b']), u'b', u'b'),
1953
 
                           ((False, [False, False]), False, False)),
1954
 
                         ], entries)
1955
 
 
1956
 
 
1957
 
 
1958
 
class TestMergerEntriesLCAOnDisk(tests.TestCaseWithTransport):
1959
 
 
1960
 
    def get_builder(self):
1961
 
        builder = self.make_branch_builder('path')
1962
 
        builder.start_series()
1963
 
        self.addCleanup(builder.finish_series)
1964
 
        return builder
1965
 
 
1966
 
    def get_wt_from_builder(self, builder):
1967
 
        """Get a real WorkingTree from the builder."""
1968
 
        the_branch = builder.get_branch()
1969
 
        wt = the_branch.bzrdir.create_workingtree()
1970
 
        # Note: This is a little bit ugly, but we are holding the branch
1971
 
        #       write-locked as part of the build process, and we would like to
1972
 
        #       maintain that. So we just force the WT to re-use the same
1973
 
        #       branch object.
1974
 
        wt._branch = the_branch
1975
 
        wt.lock_write()
1976
 
        self.addCleanup(wt.unlock)
1977
 
        return wt
1978
 
 
1979
 
    def do_merge(self, builder, other_revision_id):
1980
 
        wt = self.get_wt_from_builder(builder)
1981
 
        merger = _mod_merge.Merger.from_revision_ids(None,
1982
 
            wt, other_revision_id)
1983
 
        merger.merge_type = _mod_merge.Merge3Merger
1984
 
        return wt, merger.do_merge()
1985
 
 
1986
 
    def test_simple_lca(self):
1987
 
        builder = self.get_builder()
1988
 
        builder.build_snapshot('A-id', None,
1989
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1990
 
             ('add', (u'a', 'a-id', 'file', 'a\nb\nc\n'))])
1991
 
        builder.build_snapshot('C-id', ['A-id'], [])
1992
 
        builder.build_snapshot('B-id', ['A-id'], [])
1993
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
1994
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'],
1995
 
            [('modify', ('a-id', 'a\nb\nc\nd\ne\nf\n'))])
1996
 
        wt, conflicts = self.do_merge(builder, 'E-id')
1997
 
        self.assertEqual(0, conflicts)
1998
 
        # The merge should have simply update the contents of 'a'
1999
 
        self.assertEqual('a\nb\nc\nd\ne\nf\n', wt.get_file_text('a-id'))
2000
 
 
2001
 
    def test_conflict_without_lca(self):
2002
 
        # This test would cause a merge conflict, unless we use the lca trees
2003
 
        # to determine the real ancestry
2004
 
        #   A       Path at 'foo'
2005
 
        #  / \
2006
 
        # B   C     Path renamed to 'bar' in B
2007
 
        # |\ /|
2008
 
        # | X |
2009
 
        # |/ \|
2010
 
        # D   E     Path at 'bar' in D and E
2011
 
        #     |
2012
 
        #     F     Path at 'baz' in F, which supersedes 'bar' and 'foo'
2013
 
        builder = self.get_builder()
2014
 
        builder.build_snapshot('A-id', None,
2015
 
            [('add', (u'', 'a-root-id', 'directory', None)),
2016
 
             ('add', (u'foo', 'foo-id', 'file', 'a\nb\nc\n'))])
2017
 
        builder.build_snapshot('C-id', ['A-id'], [])
2018
 
        builder.build_snapshot('B-id', ['A-id'],
2019
 
            [('rename', ('foo', 'bar'))])
2020
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], # merge the rename
2021
 
            [('rename', ('foo', 'bar'))])
2022
 
        builder.build_snapshot('F-id', ['E-id'],
2023
 
            [('rename', ('bar', 'baz'))])
2024
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
2025
 
        wt, conflicts = self.do_merge(builder, 'F-id')
2026
 
        self.assertEqual(0, conflicts)
2027
 
        # The merge should simply recognize that the final rename takes
2028
 
        # precedence
2029
 
        self.assertEqual('baz', wt.id2path('foo-id'))
2030
 
 
2031
 
    def test_other_deletes_lca_renames(self):
2032
 
        # This test would cause a merge conflict, unless we use the lca trees
2033
 
        # to determine the real ancestry
2034
 
        #   A       Path at 'foo'
2035
 
        #  / \
2036
 
        # B   C     Path renamed to 'bar' in B
2037
 
        # |\ /|
2038
 
        # | X |
2039
 
        # |/ \|
2040
 
        # D   E     Path at 'bar' in D and E
2041
 
        #     |
2042
 
        #     F     F deletes 'bar'
2043
 
        builder = self.get_builder()
2044
 
        builder.build_snapshot('A-id', None,
2045
 
            [('add', (u'', 'a-root-id', 'directory', None)),
2046
 
             ('add', (u'foo', 'foo-id', 'file', 'a\nb\nc\n'))])
2047
 
        builder.build_snapshot('C-id', ['A-id'], [])
2048
 
        builder.build_snapshot('B-id', ['A-id'],
2049
 
            [('rename', ('foo', 'bar'))])
2050
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], # merge the rename
2051
 
            [('rename', ('foo', 'bar'))])
2052
 
        builder.build_snapshot('F-id', ['E-id'],
2053
 
            [('unversion', 'foo-id')])
2054
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
2055
 
        wt, conflicts = self.do_merge(builder, 'F-id')
2056
 
        self.assertEqual(0, conflicts)
2057
 
        self.assertRaises(errors.NoSuchId, wt.id2path, 'foo-id')
2058
 
 
2059
 
    def test_executable_changes(self):
2060
 
        #   A       Path at 'foo'
2061
 
        #  / \
2062
 
        # B   C
2063
 
        # |\ /|
2064
 
        # | X |
2065
 
        # |/ \|
2066
 
        # D   E
2067
 
        #     |
2068
 
        #     F     Executable bit changed
2069
 
        builder = self.get_builder()
2070
 
        builder.build_snapshot('A-id', None,
2071
 
            [('add', (u'', 'a-root-id', 'directory', None)),
2072
 
             ('add', (u'foo', 'foo-id', 'file', 'a\nb\nc\n'))])
2073
 
        builder.build_snapshot('C-id', ['A-id'], [])
2074
 
        builder.build_snapshot('B-id', ['A-id'], [])
2075
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
2076
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
2077
 
        # Have to use a real WT, because BranchBuilder doesn't support exec bit
2078
 
        wt = self.get_wt_from_builder(builder)
2079
 
        tt = transform.TreeTransform(wt)
2080
 
        try:
2081
 
            tt.set_executability(True, tt.trans_id_tree_file_id('foo-id'))
2082
 
            tt.apply()
2083
 
        except:
2084
 
            tt.finalize()
2085
 
            raise
2086
 
        self.assertTrue(wt.is_executable('foo-id'))
2087
 
        wt.commit('F-id', rev_id='F-id')
2088
 
        # Reset to D, so that we can merge F
2089
 
        wt.set_parent_ids(['D-id'])
2090
 
        wt.branch.set_last_revision_info(3, 'D-id')
2091
 
        wt.revert()
2092
 
        self.assertFalse(wt.is_executable('foo-id'))
2093
 
        conflicts = wt.merge_from_branch(wt.branch, to_revision='F-id')
2094
 
        self.assertEqual(0, conflicts)
2095
 
        self.assertTrue(wt.is_executable('foo-id'))
2096
 
 
2097
 
    def test_create_symlink(self):
2098
 
        self.requireFeature(tests.SymlinkFeature)
2099
 
        #   A
2100
 
        #  / \
2101
 
        # B   C
2102
 
        # |\ /|
2103
 
        # | X |
2104
 
        # |/ \|
2105
 
        # D   E
2106
 
        #     |
2107
 
        #     F     Add a symlink 'foo' => 'bar'
2108
 
        # Have to use a real WT, because BranchBuilder and MemoryTree don't
2109
 
        # have symlink support
2110
 
        builder = self.get_builder()
2111
 
        builder.build_snapshot('A-id', None,
2112
 
            [('add', (u'', 'a-root-id', 'directory', None))])
2113
 
        builder.build_snapshot('C-id', ['A-id'], [])
2114
 
        builder.build_snapshot('B-id', ['A-id'], [])
2115
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
2116
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
2117
 
        # Have to use a real WT, because BranchBuilder doesn't support exec bit
2118
 
        wt = self.get_wt_from_builder(builder)
2119
 
        os.symlink('bar', 'path/foo')
2120
 
        wt.add(['foo'], ['foo-id'])
2121
 
        self.assertEqual('bar', wt.get_symlink_target('foo-id'))
2122
 
        wt.commit('add symlink', rev_id='F-id')
2123
 
        # Reset to D, so that we can merge F
2124
 
        wt.set_parent_ids(['D-id'])
2125
 
        wt.branch.set_last_revision_info(3, 'D-id')
2126
 
        wt.revert()
2127
 
        self.assertIs(None, wt.path2id('foo'))
2128
 
        conflicts = wt.merge_from_branch(wt.branch, to_revision='F-id')
2129
 
        self.assertEqual(0, conflicts)
2130
 
        self.assertEqual('foo-id', wt.path2id('foo'))
2131
 
        self.assertEqual('bar', wt.get_symlink_target('foo-id'))
2132
 
 
2133
 
    def test_both_sides_revert(self):
2134
 
        # Both sides of a criss-cross revert the text to the lca
2135
 
        #       A    base, introduces 'foo'
2136
 
        #       |\
2137
 
        #       B C  B modifies 'foo', C modifies 'foo'
2138
 
        #       |X|
2139
 
        #       D E  D reverts to B, E reverts to C
2140
 
        # This should conflict
2141
 
        # This must be done with a real WorkingTree, because normally their
2142
 
        # inventory contains "None" rather than a real sha1
2143
 
        builder = self.get_builder()
2144
 
        builder.build_snapshot('A-id', None,
2145
 
            [('add', (u'', 'a-root-id', 'directory', None)),
2146
 
             ('add', (u'foo', 'foo-id', 'file', 'A content\n'))])
2147
 
        builder.build_snapshot('B-id', ['A-id'],
2148
 
            [('modify', ('foo-id', 'B content\n'))])
2149
 
        builder.build_snapshot('C-id', ['A-id'],
2150
 
            [('modify', ('foo-id', 'C content\n'))])
2151
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
2152
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
2153
 
        wt, conflicts = self.do_merge(builder, 'E-id')
2154
 
        self.assertEqual(1, conflicts)
2155
 
        self.assertEqualDiff('<<<<<<< TREE\n'
2156
 
                             'B content\n'
2157
 
                             '=======\n'
2158
 
                             'C content\n'
2159
 
                             '>>>>>>> MERGE-SOURCE\n',
2160
 
                             wt.get_file_text('foo-id'))
2161
 
 
2162
 
    def test_modified_symlink(self):
2163
 
        self.requireFeature(tests.SymlinkFeature)
2164
 
        #   A       Create symlink foo => bar
2165
 
        #  / \
2166
 
        # B   C     B relinks foo => baz
2167
 
        # |\ /|
2168
 
        # | X |
2169
 
        # |/ \|
2170
 
        # D   E     D & E have foo => baz
2171
 
        #     |
2172
 
        #     F     F changes it to bing
2173
 
        #
2174
 
        # Merging D & F should result in F cleanly overriding D, because D's
2175
 
        # value actually comes from B
2176
 
 
2177
 
        # Have to use a real WT, because BranchBuilder and MemoryTree don't
2178
 
        # have symlink support
2179
 
        wt = self.make_branch_and_tree('path')
2180
 
        wt.lock_write()
2181
 
        self.addCleanup(wt.unlock)
2182
 
        os.symlink('bar', 'path/foo')
2183
 
        wt.add(['foo'], ['foo-id'])
2184
 
        wt.commit('add symlink', rev_id='A-id')
2185
 
        os.remove('path/foo')
2186
 
        os.symlink('baz', 'path/foo')
2187
 
        wt.commit('foo => baz', rev_id='B-id')
2188
 
        wt.set_last_revision('A-id')
2189
 
        wt.branch.set_last_revision_info(1, 'A-id')
2190
 
        wt.revert()
2191
 
        wt.commit('C', rev_id='C-id')
2192
 
        wt.merge_from_branch(wt.branch, 'B-id')
2193
 
        self.assertEqual('baz', wt.get_symlink_target('foo-id'))
2194
 
        wt.commit('E merges C & B', rev_id='E-id')
2195
 
        os.remove('path/foo')
2196
 
        os.symlink('bing', 'path/foo')
2197
 
        wt.commit('F foo => bing', rev_id='F-id')
2198
 
        wt.set_last_revision('B-id')
2199
 
        wt.branch.set_last_revision_info(2, 'B-id')
2200
 
        wt.revert()
2201
 
        wt.merge_from_branch(wt.branch, 'C-id')
2202
 
        wt.commit('D merges B & C', rev_id='D-id')
2203
 
        conflicts = wt.merge_from_branch(wt.branch, to_revision='F-id')
2204
 
        self.assertEqual(0, conflicts)
2205
 
        self.assertEqual('bing', wt.get_symlink_target('foo-id'))
2206
 
 
2207
 
    def test_renamed_symlink(self):
2208
 
        self.requireFeature(tests.SymlinkFeature)
2209
 
        #   A       Create symlink foo => bar
2210
 
        #  / \
2211
 
        # B   C     B renames foo => barry
2212
 
        # |\ /|
2213
 
        # | X |
2214
 
        # |/ \|
2215
 
        # D   E     D & E have barry
2216
 
        #     |
2217
 
        #     F     F renames barry to blah
2218
 
        #
2219
 
        # Merging D & F should result in F cleanly overriding D, because D's
2220
 
        # value actually comes from B
2221
 
 
2222
 
        wt = self.make_branch_and_tree('path')
2223
 
        wt.lock_write()
2224
 
        self.addCleanup(wt.unlock)
2225
 
        os.symlink('bar', 'path/foo')
2226
 
        wt.add(['foo'], ['foo-id'])
2227
 
        wt.commit('A add symlink', rev_id='A-id')
2228
 
        wt.rename_one('foo', 'barry')
2229
 
        wt.commit('B foo => barry', rev_id='B-id')
2230
 
        wt.set_last_revision('A-id')
2231
 
        wt.branch.set_last_revision_info(1, 'A-id')
2232
 
        wt.revert()
2233
 
        wt.commit('C', rev_id='C-id')
2234
 
        wt.merge_from_branch(wt.branch, 'B-id')
2235
 
        self.assertEqual('barry', wt.id2path('foo-id'))
2236
 
        self.assertEqual('bar', wt.get_symlink_target('foo-id'))
2237
 
        wt.commit('E merges C & B', rev_id='E-id')
2238
 
        wt.rename_one('barry', 'blah')
2239
 
        wt.commit('F barry => blah', rev_id='F-id')
2240
 
        wt.set_last_revision('B-id')
2241
 
        wt.branch.set_last_revision_info(2, 'B-id')
2242
 
        wt.revert()
2243
 
        wt.merge_from_branch(wt.branch, 'C-id')
2244
 
        wt.commit('D merges B & C', rev_id='D-id')
2245
 
        self.assertEqual('barry', wt.id2path('foo-id'))
2246
 
        # Check the output of the Merger object directly
2247
 
        merger = _mod_merge.Merger.from_revision_ids(None,
2248
 
            wt, 'F-id')
2249
 
        merger.merge_type = _mod_merge.Merge3Merger
2250
 
        merge_obj = merger.make_merger()
2251
 
        root_id = wt.path2id('')
2252
 
        entries = list(merge_obj._entries_lca())
2253
 
        # No content change, just a path change
2254
 
        self.assertEqual([('foo-id', False,
2255
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
2256
 
                           ((u'foo', [u'barry', u'foo']), u'blah', u'barry'),
2257
 
                           ((False, [False, False]), False, False)),
2258
 
                         ], entries)
2259
 
        conflicts = wt.merge_from_branch(wt.branch, to_revision='F-id')
2260
 
        self.assertEqual(0, conflicts)
2261
 
        self.assertEqual('blah', wt.id2path('foo-id'))
2262
 
 
2263
 
    def test_symlink_no_content_change(self):
2264
 
        self.requireFeature(tests.SymlinkFeature)
2265
 
        #   A       Create symlink foo => bar
2266
 
        #  / \
2267
 
        # B   C     B relinks foo => baz
2268
 
        # |\ /|
2269
 
        # | X |
2270
 
        # |/ \|
2271
 
        # D   E     D & E have foo => baz
2272
 
        # |
2273
 
        # F         F has foo => bing
2274
 
        #
2275
 
        # Merging E into F should not cause a conflict, because E doesn't have
2276
 
        # a content change relative to the LCAs (it does relative to A)
2277
 
        wt = self.make_branch_and_tree('path')
2278
 
        wt.lock_write()
2279
 
        self.addCleanup(wt.unlock)
2280
 
        os.symlink('bar', 'path/foo')
2281
 
        wt.add(['foo'], ['foo-id'])
2282
 
        wt.commit('add symlink', rev_id='A-id')
2283
 
        os.remove('path/foo')
2284
 
        os.symlink('baz', 'path/foo')
2285
 
        wt.commit('foo => baz', rev_id='B-id')
2286
 
        wt.set_last_revision('A-id')
2287
 
        wt.branch.set_last_revision_info(1, 'A-id')
2288
 
        wt.revert()
2289
 
        wt.commit('C', rev_id='C-id')
2290
 
        wt.merge_from_branch(wt.branch, 'B-id')
2291
 
        self.assertEqual('baz', wt.get_symlink_target('foo-id'))
2292
 
        wt.commit('E merges C & B', rev_id='E-id')
2293
 
        wt.set_last_revision('B-id')
2294
 
        wt.branch.set_last_revision_info(2, 'B-id')
2295
 
        wt.revert()
2296
 
        wt.merge_from_branch(wt.branch, 'C-id')
2297
 
        wt.commit('D merges B & C', rev_id='D-id')
2298
 
        os.remove('path/foo')
2299
 
        os.symlink('bing', 'path/foo')
2300
 
        wt.commit('F foo => bing', rev_id='F-id')
2301
 
 
2302
 
        # Check the output of the Merger object directly
2303
 
        merger = _mod_merge.Merger.from_revision_ids(None,
2304
 
            wt, 'E-id')
2305
 
        merger.merge_type = _mod_merge.Merge3Merger
2306
 
        merge_obj = merger.make_merger()
2307
 
        # Nothing interesting happened in OTHER relative to BASE
2308
 
        self.assertEqual([], list(merge_obj._entries_lca()))
2309
 
        # Now do a real merge, just to test the rest of the stack
2310
 
        conflicts = wt.merge_from_branch(wt.branch, to_revision='E-id')
2311
 
        self.assertEqual(0, conflicts)
2312
 
        self.assertEqual('bing', wt.get_symlink_target('foo-id'))
2313
 
 
2314
 
    def test_symlink_this_changed_kind(self):
2315
 
        self.requireFeature(tests.SymlinkFeature)
2316
 
        #   A       Nothing
2317
 
        #  / \
2318
 
        # B   C     B creates symlink foo => bar
2319
 
        # |\ /|
2320
 
        # | X |
2321
 
        # |/ \|
2322
 
        # D   E     D changes foo into a file, E has foo => bing
2323
 
        #
2324
 
        # Mostly, this is trying to test that we don't try to os.readlink() on
2325
 
        # a file, or when there is nothing there
2326
 
        wt = self.make_branch_and_tree('path')
2327
 
        wt.lock_write()
2328
 
        self.addCleanup(wt.unlock)
2329
 
        wt.commit('base', rev_id='A-id')
2330
 
        os.symlink('bar', 'path/foo')
2331
 
        wt.add(['foo'], ['foo-id'])
2332
 
        wt.commit('add symlink foo => bar', rev_id='B-id')
2333
 
        wt.set_last_revision('A-id')
2334
 
        wt.branch.set_last_revision_info(1, 'A-id')
2335
 
        wt.revert()
2336
 
        wt.commit('C', rev_id='C-id')
2337
 
        wt.merge_from_branch(wt.branch, 'B-id')
2338
 
        self.assertEqual('bar', wt.get_symlink_target('foo-id'))
2339
 
        os.remove('path/foo')
2340
 
        # We have to change the link in E, or it won't try to do a comparison
2341
 
        os.symlink('bing', 'path/foo')
2342
 
        wt.commit('E merges C & B, overrides to bing', rev_id='E-id')
2343
 
        wt.set_last_revision('B-id')
2344
 
        wt.branch.set_last_revision_info(2, 'B-id')
2345
 
        wt.revert()
2346
 
        wt.merge_from_branch(wt.branch, 'C-id')
2347
 
        os.remove('path/foo')
2348
 
        self.build_tree_contents([('path/foo', 'file content\n')])
2349
 
        # XXX: workaround, WT doesn't detect kind changes unless you do
2350
 
        # iter_changes()
2351
 
        list(wt.iter_changes(wt.basis_tree()))
2352
 
        wt.commit('D merges B & C, makes it a file', rev_id='D-id')
2353
 
 
2354
 
        merger = _mod_merge.Merger.from_revision_ids(None,
2355
 
            wt, 'E-id')
2356
 
        merger.merge_type = _mod_merge.Merge3Merger
2357
 
        merge_obj = merger.make_merger()
2358
 
        entries = list(merge_obj._entries_lca())
2359
 
        root_id = wt.path2id('')
2360
 
        self.assertEqual([('foo-id', True,
2361
 
                           ((None, [root_id, None]), root_id, root_id),
2362
 
                           ((None, [u'foo', None]), u'foo', u'foo'),
2363
 
                           ((None, [False, None]), False, False)),
2364
 
                         ], entries)
2365
 
 
2366
 
    def test_symlink_all_wt(self):
2367
 
        """Check behavior if all trees are Working Trees."""
2368
 
        self.requireFeature(tests.SymlinkFeature)
2369
 
        # The big issue is that entry.symlink_target is None for WorkingTrees.
2370
 
        # So we need to make sure we handle that case correctly.
2371
 
        #   A   foo => bar
2372
 
        #   |\
2373
 
        #   B C B relinks foo => baz
2374
 
        #   |X|
2375
 
        #   D E D & E have foo => baz
2376
 
        #     |
2377
 
        #     F F changes it to bing
2378
 
        # Merging D & F should result in F cleanly overriding D, because D's
2379
 
        # value actually comes from B
2380
 
 
2381
 
        wt = self.make_branch_and_tree('path')
2382
 
        wt.lock_write()
2383
 
        self.addCleanup(wt.unlock)
2384
 
        os.symlink('bar', 'path/foo')
2385
 
        wt.add(['foo'], ['foo-id'])
2386
 
        wt.commit('add symlink', rev_id='A-id')
2387
 
        os.remove('path/foo')
2388
 
        os.symlink('baz', 'path/foo')
2389
 
        wt.commit('foo => baz', rev_id='B-id')
2390
 
        wt.set_last_revision('A-id')
2391
 
        wt.branch.set_last_revision_info(1, 'A-id')
2392
 
        wt.revert()
2393
 
        wt.commit('C', rev_id='C-id')
2394
 
        wt.merge_from_branch(wt.branch, 'B-id')
2395
 
        self.assertEqual('baz', wt.get_symlink_target('foo-id'))
2396
 
        wt.commit('E merges C & B', rev_id='E-id')
2397
 
        os.remove('path/foo')
2398
 
        os.symlink('bing', 'path/foo')
2399
 
        wt.commit('F foo => bing', rev_id='F-id')
2400
 
        wt.set_last_revision('B-id')
2401
 
        wt.branch.set_last_revision_info(2, 'B-id')
2402
 
        wt.revert()
2403
 
        wt.merge_from_branch(wt.branch, 'C-id')
2404
 
        wt.commit('D merges B & C', rev_id='D-id')
2405
 
        wt_base = wt.bzrdir.sprout('base', 'A-id').open_workingtree()
2406
 
        wt_base.lock_read()
2407
 
        self.addCleanup(wt_base.unlock)
2408
 
        wt_lca1 = wt.bzrdir.sprout('b-tree', 'B-id').open_workingtree()
2409
 
        wt_lca1.lock_read()
2410
 
        self.addCleanup(wt_lca1.unlock)
2411
 
        wt_lca2 = wt.bzrdir.sprout('c-tree', 'C-id').open_workingtree()
2412
 
        wt_lca2.lock_read()
2413
 
        self.addCleanup(wt_lca2.unlock)
2414
 
        wt_other = wt.bzrdir.sprout('other', 'F-id').open_workingtree()
2415
 
        wt_other.lock_read()
2416
 
        self.addCleanup(wt_other.unlock)
2417
 
        merge_obj = _mod_merge.Merge3Merger(wt, wt, wt_base,
2418
 
            wt_other, lca_trees=[wt_lca1, wt_lca2], do_merge=False)
2419
 
        entries = list(merge_obj._entries_lca())
2420
 
        root_id = wt.path2id('')
2421
 
        self.assertEqual([('foo-id', True,
2422
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
2423
 
                           ((u'foo', [u'foo', u'foo']), u'foo', u'foo'),
2424
 
                           ((False, [False, False]), False, False)),
2425
 
                         ], entries)
2426
 
 
2427
 
    def test_other_reverted_path_to_base(self):
2428
 
        #   A       Path at 'foo'
2429
 
        #  / \
2430
 
        # B   C     Path at 'bar' in B
2431
 
        # |\ /|
2432
 
        # | X |
2433
 
        # |/ \|
2434
 
        # D   E     Path at 'bar'
2435
 
        #     |
2436
 
        #     F     Path at 'foo'
2437
 
        builder = self.get_builder()
2438
 
        builder.build_snapshot('A-id', None,
2439
 
            [('add', (u'', 'a-root-id', 'directory', None)),
2440
 
             ('add', (u'foo', 'foo-id', 'file', 'a\nb\nc\n'))])
2441
 
        builder.build_snapshot('C-id', ['A-id'], [])
2442
 
        builder.build_snapshot('B-id', ['A-id'],
2443
 
            [('rename', ('foo', 'bar'))])
2444
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
2445
 
            [('rename', ('foo', 'bar'))]) # merge the rename
2446
 
        builder.build_snapshot('F-id', ['E-id'],
2447
 
            [('rename', ('bar', 'foo'))]) # Rename back to BASE
2448
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
2449
 
        wt, conflicts = self.do_merge(builder, 'F-id')
2450
 
        self.assertEqual(0, conflicts)
2451
 
        self.assertEqual('foo', wt.id2path('foo-id'))
2452
 
 
2453
 
    def test_other_reverted_content_to_base(self):
2454
 
        builder = self.get_builder()
2455
 
        builder.build_snapshot('A-id', None,
2456
 
            [('add', (u'', 'a-root-id', 'directory', None)),
2457
 
             ('add', (u'foo', 'foo-id', 'file', 'base content\n'))])
2458
 
        builder.build_snapshot('C-id', ['A-id'], [])
2459
 
        builder.build_snapshot('B-id', ['A-id'],
2460
 
            [('modify', ('foo-id', 'B content\n'))])
2461
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
2462
 
            [('modify', ('foo-id', 'B content\n'))]) # merge the content
2463
 
        builder.build_snapshot('F-id', ['E-id'],
2464
 
            [('modify', ('foo-id', 'base content\n'))]) # Revert back to BASE
2465
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
2466
 
        wt, conflicts = self.do_merge(builder, 'F-id')
2467
 
        self.assertEqual(0, conflicts)
2468
 
        # TODO: We need to use the per-file graph to properly select a BASE
2469
 
        #       before this will work. Or at least use the LCA trees to find
2470
 
        #       the appropriate content base. (which is B, not A).
2471
 
        self.assertEqual('base content\n', wt.get_file_text('foo-id'))
2472
 
 
2473
 
    def test_other_modified_content(self):
2474
 
        builder = self.get_builder()
2475
 
        builder.build_snapshot('A-id', None,
2476
 
            [('add', (u'', 'a-root-id', 'directory', None)),
2477
 
             ('add', (u'foo', 'foo-id', 'file', 'base content\n'))])
2478
 
        builder.build_snapshot('C-id', ['A-id'], [])
2479
 
        builder.build_snapshot('B-id', ['A-id'],
2480
 
            [('modify', ('foo-id', 'B content\n'))])
2481
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
2482
 
            [('modify', ('foo-id', 'B content\n'))]) # merge the content
2483
 
        builder.build_snapshot('F-id', ['E-id'],
2484
 
            [('modify', ('foo-id', 'F content\n'))]) # Override B content
2485
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
2486
 
        wt, conflicts = self.do_merge(builder, 'F-id')
2487
 
        self.assertEqual(0, conflicts)
2488
 
        self.assertEqual('F content\n', wt.get_file_text('foo-id'))
2489
 
 
2490
 
    def test_all_wt(self):
2491
 
        """Check behavior if all trees are Working Trees."""
2492
 
        # The big issue is that entry.revision is None for WorkingTrees. (as is
2493
 
        # entry.text_sha1, etc. So we need to make sure we handle that case
2494
 
        # correctly.
2495
 
        #   A   Content of 'foo', path of 'a'
2496
 
        #   |\
2497
 
        #   B C B modifies content, C renames 'a' => 'b'
2498
 
        #   |X|
2499
 
        #   D E E updates content, renames 'b' => 'c'
2500
 
        builder = self.get_builder()
2501
 
        builder.build_snapshot('A-id', None,
2502
 
            [('add', (u'', 'a-root-id', 'directory', None)),
2503
 
             ('add', (u'a', 'a-id', 'file', 'base content\n')),
2504
 
             ('add', (u'foo', 'foo-id', 'file', 'base content\n'))])
2505
 
        builder.build_snapshot('B-id', ['A-id'],
2506
 
            [('modify', ('foo-id', 'B content\n'))])
2507
 
        builder.build_snapshot('C-id', ['A-id'],
2508
 
            [('rename', ('a', 'b'))])
2509
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
2510
 
            [('rename', ('b', 'c')),
2511
 
             ('modify', ('foo-id', 'E content\n'))])
2512
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'],
2513
 
            [('rename', ('a', 'b'))]) # merged change
2514
 
        wt_this = self.get_wt_from_builder(builder)
2515
 
        wt_base = wt_this.bzrdir.sprout('base', 'A-id').open_workingtree()
2516
 
        wt_base.lock_read()
2517
 
        self.addCleanup(wt_base.unlock)
2518
 
        wt_lca1 = wt_this.bzrdir.sprout('b-tree', 'B-id').open_workingtree()
2519
 
        wt_lca1.lock_read()
2520
 
        self.addCleanup(wt_lca1.unlock)
2521
 
        wt_lca2 = wt_this.bzrdir.sprout('c-tree', 'C-id').open_workingtree()
2522
 
        wt_lca2.lock_read()
2523
 
        self.addCleanup(wt_lca2.unlock)
2524
 
        wt_other = wt_this.bzrdir.sprout('other', 'E-id').open_workingtree()
2525
 
        wt_other.lock_read()
2526
 
        self.addCleanup(wt_other.unlock)
2527
 
        merge_obj = _mod_merge.Merge3Merger(wt_this, wt_this, wt_base,
2528
 
            wt_other, lca_trees=[wt_lca1, wt_lca2], do_merge=False)
2529
 
        entries = list(merge_obj._entries_lca())
2530
 
        root_id = 'a-root-id'
2531
 
        self.assertEqual([('a-id', False,
2532
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
2533
 
                           ((u'a', [u'a', u'b']), u'c', u'b'),
2534
 
                           ((False, [False, False]), False, False)),
2535
 
                          ('foo-id', True,
2536
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
2537
 
                           ((u'foo', [u'foo', u'foo']), u'foo', u'foo'),
2538
 
                           ((False, [False, False]), False, False)),
2539
 
                         ], entries)
2540
 
 
2541
 
    def test_nested_tree_unmodified(self):
2542
 
        # Tested with a real WT, because BranchBuilder/MemoryTree don't handle
2543
 
        # 'tree-reference'
2544
 
        wt = self.make_branch_and_tree('tree',
2545
 
            format='dirstate-with-subtree')
2546
 
        wt.lock_write()
2547
 
        self.addCleanup(wt.unlock)
2548
 
        sub_tree = self.make_branch_and_tree('tree/sub-tree',
2549
 
            format='dirstate-with-subtree')
2550
 
        wt.set_root_id('a-root-id')
2551
 
        sub_tree.set_root_id('sub-tree-root')
2552
 
        self.build_tree_contents([('tree/sub-tree/file', 'text1')])
2553
 
        sub_tree.add('file')
2554
 
        sub_tree.commit('foo', rev_id='sub-A-id')
2555
 
        wt.add_reference(sub_tree)
2556
 
        wt.commit('set text to 1', rev_id='A-id', recursive=None)
2557
 
        # Now create a criss-cross merge in the parent, without modifying the
2558
 
        # subtree
2559
 
        wt.commit('B', rev_id='B-id', recursive=None)
2560
 
        wt.set_last_revision('A-id')
2561
 
        wt.branch.set_last_revision_info(1, 'A-id')
2562
 
        wt.commit('C', rev_id='C-id', recursive=None)
2563
 
        wt.merge_from_branch(wt.branch, to_revision='B-id')
2564
 
        wt.commit('E', rev_id='E-id', recursive=None)
2565
 
        wt.set_parent_ids(['B-id', 'C-id'])
2566
 
        wt.branch.set_last_revision_info(2, 'B-id')
2567
 
        wt.commit('D', rev_id='D-id', recursive=None)
2568
 
 
2569
 
        merger = _mod_merge.Merger.from_revision_ids(None,
2570
 
            wt, 'E-id')
2571
 
        merger.merge_type = _mod_merge.Merge3Merger
2572
 
        merge_obj = merger.make_merger()
2573
 
        entries = list(merge_obj._entries_lca())
2574
 
        self.assertEqual([], entries)
2575
 
 
2576
 
    def test_nested_tree_subtree_modified(self):
2577
 
        # Tested with a real WT, because BranchBuilder/MemoryTree don't handle
2578
 
        # 'tree-reference'
2579
 
        wt = self.make_branch_and_tree('tree',
2580
 
            format='dirstate-with-subtree')
2581
 
        wt.lock_write()
2582
 
        self.addCleanup(wt.unlock)
2583
 
        sub_tree = self.make_branch_and_tree('tree/sub',
2584
 
            format='dirstate-with-subtree')
2585
 
        wt.set_root_id('a-root-id')
2586
 
        sub_tree.set_root_id('sub-tree-root')
2587
 
        self.build_tree_contents([('tree/sub/file', 'text1')])
2588
 
        sub_tree.add('file')
2589
 
        sub_tree.commit('foo', rev_id='sub-A-id')
2590
 
        wt.add_reference(sub_tree)
2591
 
        wt.commit('set text to 1', rev_id='A-id', recursive=None)
2592
 
        # Now create a criss-cross merge in the parent, without modifying the
2593
 
        # subtree
2594
 
        wt.commit('B', rev_id='B-id', recursive=None)
2595
 
        wt.set_last_revision('A-id')
2596
 
        wt.branch.set_last_revision_info(1, 'A-id')
2597
 
        wt.commit('C', rev_id='C-id', recursive=None)
2598
 
        wt.merge_from_branch(wt.branch, to_revision='B-id')
2599
 
        self.build_tree_contents([('tree/sub/file', 'text2')])
2600
 
        sub_tree.commit('modify contents', rev_id='sub-B-id')
2601
 
        wt.commit('E', rev_id='E-id', recursive=None)
2602
 
        wt.set_parent_ids(['B-id', 'C-id'])
2603
 
        wt.branch.set_last_revision_info(2, 'B-id')
2604
 
        wt.commit('D', rev_id='D-id', recursive=None)
2605
 
 
2606
 
        merger = _mod_merge.Merger.from_revision_ids(None,
2607
 
            wt, 'E-id')
2608
 
        merger.merge_type = _mod_merge.Merge3Merger
2609
 
        merge_obj = merger.make_merger()
2610
 
        entries = list(merge_obj._entries_lca())
2611
 
        # Nothing interesting about this sub-tree, because content changes are
2612
 
        # computed at a higher level
2613
 
        self.assertEqual([], entries)
2614
 
 
2615
 
    def test_nested_tree_subtree_renamed(self):
2616
 
        # Tested with a real WT, because BranchBuilder/MemoryTree don't handle
2617
 
        # 'tree-reference'
2618
 
        wt = self.make_branch_and_tree('tree',
2619
 
            format='dirstate-with-subtree')
2620
 
        wt.lock_write()
2621
 
        self.addCleanup(wt.unlock)
2622
 
        sub_tree = self.make_branch_and_tree('tree/sub',
2623
 
            format='dirstate-with-subtree')
2624
 
        wt.set_root_id('a-root-id')
2625
 
        sub_tree.set_root_id('sub-tree-root')
2626
 
        self.build_tree_contents([('tree/sub/file', 'text1')])
2627
 
        sub_tree.add('file')
2628
 
        sub_tree.commit('foo', rev_id='sub-A-id')
2629
 
        wt.add_reference(sub_tree)
2630
 
        wt.commit('set text to 1', rev_id='A-id', recursive=None)
2631
 
        # Now create a criss-cross merge in the parent, without modifying the
2632
 
        # subtree
2633
 
        wt.commit('B', rev_id='B-id', recursive=None)
2634
 
        wt.set_last_revision('A-id')
2635
 
        wt.branch.set_last_revision_info(1, 'A-id')
2636
 
        wt.commit('C', rev_id='C-id', recursive=None)
2637
 
        wt.merge_from_branch(wt.branch, to_revision='B-id')
2638
 
        wt.rename_one('sub', 'alt_sub')
2639
 
        wt.commit('E', rev_id='E-id', recursive=None)
2640
 
        wt.set_last_revision('B-id')
2641
 
        wt.revert()
2642
 
        wt.set_parent_ids(['B-id', 'C-id'])
2643
 
        wt.branch.set_last_revision_info(2, 'B-id')
2644
 
        wt.commit('D', rev_id='D-id', recursive=None)
2645
 
 
2646
 
        merger = _mod_merge.Merger.from_revision_ids(None,
2647
 
            wt, 'E-id')
2648
 
        merger.merge_type = _mod_merge.Merge3Merger
2649
 
        merge_obj = merger.make_merger()
2650
 
        entries = list(merge_obj._entries_lca())
2651
 
        root_id = 'a-root-id'
2652
 
        self.assertEqual([('sub-tree-root', False,
2653
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
2654
 
                           ((u'sub', [u'sub', u'sub']), u'alt_sub', u'sub'),
2655
 
                           ((False, [False, False]), False, False)),
2656
 
                         ], entries)
2657
 
 
2658
 
    def test_nested_tree_subtree_renamed_and_modified(self):
2659
 
        # Tested with a real WT, because BranchBuilder/MemoryTree don't handle
2660
 
        # 'tree-reference'
2661
 
        wt = self.make_branch_and_tree('tree',
2662
 
            format='dirstate-with-subtree')
2663
 
        wt.lock_write()
2664
 
        self.addCleanup(wt.unlock)
2665
 
        sub_tree = self.make_branch_and_tree('tree/sub',
2666
 
            format='dirstate-with-subtree')
2667
 
        wt.set_root_id('a-root-id')
2668
 
        sub_tree.set_root_id('sub-tree-root')
2669
 
        self.build_tree_contents([('tree/sub/file', 'text1')])
2670
 
        sub_tree.add('file')
2671
 
        sub_tree.commit('foo', rev_id='sub-A-id')
2672
 
        wt.add_reference(sub_tree)
2673
 
        wt.commit('set text to 1', rev_id='A-id', recursive=None)
2674
 
        # Now create a criss-cross merge in the parent, without modifying the
2675
 
        # subtree
2676
 
        wt.commit('B', rev_id='B-id', recursive=None)
2677
 
        wt.set_last_revision('A-id')
2678
 
        wt.branch.set_last_revision_info(1, 'A-id')
2679
 
        wt.commit('C', rev_id='C-id', recursive=None)
2680
 
        wt.merge_from_branch(wt.branch, to_revision='B-id')
2681
 
        self.build_tree_contents([('tree/sub/file', 'text2')])
2682
 
        sub_tree.commit('modify contents', rev_id='sub-B-id')
2683
 
        wt.rename_one('sub', 'alt_sub')
2684
 
        wt.commit('E', rev_id='E-id', recursive=None)
2685
 
        wt.set_last_revision('B-id')
2686
 
        wt.revert()
2687
 
        wt.set_parent_ids(['B-id', 'C-id'])
2688
 
        wt.branch.set_last_revision_info(2, 'B-id')
2689
 
        wt.commit('D', rev_id='D-id', recursive=None)
2690
 
 
2691
 
        merger = _mod_merge.Merger.from_revision_ids(None,
2692
 
            wt, 'E-id')
2693
 
        merger.merge_type = _mod_merge.Merge3Merger
2694
 
        merge_obj = merger.make_merger()
2695
 
        entries = list(merge_obj._entries_lca())
2696
 
        root_id = 'a-root-id'
2697
 
        self.assertEqual([('sub-tree-root', False,
2698
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
2699
 
                           ((u'sub', [u'sub', u'sub']), u'alt_sub', u'sub'),
2700
 
                           ((False, [False, False]), False, False)),
2701
 
                         ], entries)
2702
 
 
2703
 
 
2704
 
class TestLCAMultiWay(tests.TestCase):
2705
 
 
2706
 
    def assertLCAMultiWay(self, expected, base, lcas, other, this,
2707
 
                          allow_overriding_lca=True):
2708
 
        self.assertEqual(expected, _mod_merge.Merge3Merger._lca_multi_way(
2709
 
                                (base, lcas), other, this,
2710
 
                                allow_overriding_lca=allow_overriding_lca))
2711
 
 
2712
 
    def test_other_equal_equal_lcas(self):
2713
 
        """Test when OTHER=LCA and all LCAs are identical."""
2714
 
        self.assertLCAMultiWay('this',
2715
 
            'bval', ['bval', 'bval'], 'bval', 'bval')
2716
 
        self.assertLCAMultiWay('this',
2717
 
            'bval', ['lcaval', 'lcaval'], 'lcaval', 'bval')
2718
 
        self.assertLCAMultiWay('this',
2719
 
            'bval', ['lcaval', 'lcaval', 'lcaval'], 'lcaval', 'bval')
2720
 
        self.assertLCAMultiWay('this',
2721
 
            'bval', ['lcaval', 'lcaval', 'lcaval'], 'lcaval', 'tval')
2722
 
        self.assertLCAMultiWay('this',
2723
 
            'bval', ['lcaval', 'lcaval', 'lcaval'], 'lcaval', None)
2724
 
 
2725
 
    def test_other_equal_this(self):
2726
 
        """Test when other and this are identical."""
2727
 
        self.assertLCAMultiWay('this',
2728
 
            'bval', ['bval', 'bval'], 'oval', 'oval')
2729
 
        self.assertLCAMultiWay('this',
2730
 
            'bval', ['lcaval', 'lcaval'], 'oval', 'oval')
2731
 
        self.assertLCAMultiWay('this',
2732
 
            'bval', ['cval', 'dval'], 'oval', 'oval')
2733
 
        self.assertLCAMultiWay('this',
2734
 
            'bval', [None, 'lcaval'], 'oval', 'oval')
2735
 
        self.assertLCAMultiWay('this',
2736
 
            None, [None, 'lcaval'], 'oval', 'oval')
2737
 
        self.assertLCAMultiWay('this',
2738
 
            None, ['lcaval', 'lcaval'], 'oval', 'oval')
2739
 
        self.assertLCAMultiWay('this',
2740
 
            None, ['cval', 'dval'], 'oval', 'oval')
2741
 
        self.assertLCAMultiWay('this',
2742
 
            None, ['cval', 'dval'], None, None)
2743
 
        self.assertLCAMultiWay('this',
2744
 
            None, ['cval', 'dval', 'eval', 'fval'], 'oval', 'oval')
2745
 
 
2746
 
    def test_no_lcas(self):
2747
 
        self.assertLCAMultiWay('this',
2748
 
            'bval', [], 'bval', 'tval')
2749
 
        self.assertLCAMultiWay('other',
2750
 
            'bval', [], 'oval', 'bval')
2751
 
        self.assertLCAMultiWay('conflict',
2752
 
            'bval', [], 'oval', 'tval')
2753
 
        self.assertLCAMultiWay('this',
2754
 
            'bval', [], 'oval', 'oval')
2755
 
 
2756
 
    def test_lca_supersedes_other_lca(self):
2757
 
        """If one lca == base, the other lca takes precedence"""
2758
 
        self.assertLCAMultiWay('this',
2759
 
            'bval', ['bval', 'lcaval'], 'lcaval', 'tval')
2760
 
        self.assertLCAMultiWay('this',
2761
 
            'bval', ['bval', 'lcaval'], 'lcaval', 'bval')
2762
 
        # This is actually considered a 'revert' because the 'lcaval' in LCAS
2763
 
        # supersedes the BASE val (in the other LCA) but then OTHER reverts it
2764
 
        # back to bval.
2765
 
        self.assertLCAMultiWay('other',
2766
 
            'bval', ['bval', 'lcaval'], 'bval', 'lcaval')
2767
 
        self.assertLCAMultiWay('conflict',
2768
 
            'bval', ['bval', 'lcaval'], 'bval', 'tval')
2769
 
 
2770
 
    def test_other_and_this_pick_different_lca(self):
2771
 
        # OTHER and THIS resolve the lca conflict in different ways
2772
 
        self.assertLCAMultiWay('conflict',
2773
 
            'bval', ['lca1val', 'lca2val'], 'lca1val', 'lca2val')
2774
 
        self.assertLCAMultiWay('conflict',
2775
 
            'bval', ['lca1val', 'lca2val', 'lca3val'], 'lca1val', 'lca2val')
2776
 
        self.assertLCAMultiWay('conflict',
2777
 
            'bval', ['lca1val', 'lca2val', 'bval'], 'lca1val', 'lca2val')
2778
 
 
2779
 
    def test_other_in_lca(self):
2780
 
        # OTHER takes a value of one of the LCAs, THIS takes a new value, which
2781
 
        # theoretically supersedes both LCA values and 'wins'
2782
 
        self.assertLCAMultiWay('this',
2783
 
            'bval', ['lca1val', 'lca2val'], 'lca1val', 'newval')
2784
 
        self.assertLCAMultiWay('this',
2785
 
            'bval', ['lca1val', 'lca2val', 'lca3val'], 'lca1val', 'newval')
2786
 
        self.assertLCAMultiWay('conflict',
2787
 
            'bval', ['lca1val', 'lca2val'], 'lca1val', 'newval',
2788
 
            allow_overriding_lca=False)
2789
 
        self.assertLCAMultiWay('conflict',
2790
 
            'bval', ['lca1val', 'lca2val', 'lca3val'], 'lca1val', 'newval',
2791
 
            allow_overriding_lca=False)
2792
 
        # THIS reverted back to BASE, but that is an explicit supersede of all
2793
 
        # LCAs
2794
 
        self.assertLCAMultiWay('this',
2795
 
            'bval', ['lca1val', 'lca2val', 'lca3val'], 'lca1val', 'bval')
2796
 
        self.assertLCAMultiWay('this',
2797
 
            'bval', ['lca1val', 'lca2val', 'bval'], 'lca1val', 'bval')
2798
 
        self.assertLCAMultiWay('conflict',
2799
 
            'bval', ['lca1val', 'lca2val', 'lca3val'], 'lca1val', 'bval',
2800
 
            allow_overriding_lca=False)
2801
 
        self.assertLCAMultiWay('conflict',
2802
 
            'bval', ['lca1val', 'lca2val', 'bval'], 'lca1val', 'bval',
2803
 
            allow_overriding_lca=False)
2804
 
 
2805
 
    def test_this_in_lca(self):
2806
 
        # THIS takes a value of one of the LCAs, OTHER takes a new value, which
2807
 
        # theoretically supersedes both LCA values and 'wins'
2808
 
        self.assertLCAMultiWay('other',
2809
 
            'bval', ['lca1val', 'lca2val'], 'oval', 'lca1val')
2810
 
        self.assertLCAMultiWay('other',
2811
 
            'bval', ['lca1val', 'lca2val'], 'oval', 'lca2val')
2812
 
        self.assertLCAMultiWay('conflict',
2813
 
            'bval', ['lca1val', 'lca2val'], 'oval', 'lca1val',
2814
 
            allow_overriding_lca=False)
2815
 
        self.assertLCAMultiWay('conflict',
2816
 
            'bval', ['lca1val', 'lca2val'], 'oval', 'lca2val',
2817
 
            allow_overriding_lca=False)
2818
 
        # OTHER reverted back to BASE, but that is an explicit supersede of all
2819
 
        # LCAs
2820
 
        self.assertLCAMultiWay('other',
2821
 
            'bval', ['lca1val', 'lca2val', 'lca3val'], 'bval', 'lca3val')
2822
 
        self.assertLCAMultiWay('conflict',
2823
 
            'bval', ['lca1val', 'lca2val', 'lca3val'], 'bval', 'lca3val',
2824
 
            allow_overriding_lca=False)
2825
 
 
2826
 
    def test_all_differ(self):
2827
 
        self.assertLCAMultiWay('conflict',
2828
 
            'bval', ['lca1val', 'lca2val'], 'oval', 'tval')
2829
 
        self.assertLCAMultiWay('conflict',
2830
 
            'bval', ['lca1val', 'lca2val', 'lca2val'], 'oval', 'tval')
2831
 
        self.assertLCAMultiWay('conflict',
2832
 
            'bval', ['lca1val', 'lca2val', 'lca3val'], 'oval', 'tval')
2833
 
 
2834
 
 
2835
 
class TestConfigurableFileMerger(tests.TestCaseWithTransport):
2836
 
 
2837
 
    def setUp(self):
2838
 
        super(TestConfigurableFileMerger, self).setUp()
2839
 
        self.calls = []
2840
 
 
2841
 
    def get_merger_factory(self):
2842
 
        # Allows  the inner methods to access the test attributes
2843
 
        test = self
2844
 
 
2845
 
        class FooMerger(_mod_merge.ConfigurableFileMerger):
2846
 
            name_prefix = "foo"
2847
 
            default_files = ['bar']
2848
 
 
2849
 
            def merge_text(self, params):
2850
 
                test.calls.append('merge_text')
2851
 
                return ('not_applicable', None)
2852
 
 
2853
 
        def factory(merger):
2854
 
            result = FooMerger(merger)
2855
 
            # Make sure we start with a clean slate
2856
 
            self.assertEqual(None, result.affected_files)
2857
 
            # Track the original merger
2858
 
            self.merger = result
2859
 
            return result
2860
 
 
2861
 
        return factory
2862
 
 
2863
 
    def _install_hook(self, factory):
2864
 
        _mod_merge.Merger.hooks.install_named_hook('merge_file_content',
2865
 
                                                   factory, 'test factory')
2866
 
 
2867
 
    def make_builder(self):
2868
 
        builder = test_merge_core.MergeBuilder(self.test_base_dir)
2869
 
        self.addCleanup(builder.cleanup)
2870
 
        return builder
2871
 
 
2872
 
    def make_text_conflict(self, file_name='bar'):
2873
 
        factory = self.get_merger_factory()
2874
 
        self._install_hook(factory)
2875
 
        builder = self.make_builder()
2876
 
        builder.add_file('bar-id', builder.tree_root, file_name, 'text1', True)
2877
 
        builder.change_contents('bar-id', other='text4', this='text3')
2878
 
        return builder
2879
 
 
2880
 
    def make_kind_change(self):
2881
 
        factory = self.get_merger_factory()
2882
 
        self._install_hook(factory)
2883
 
        builder = self.make_builder()
2884
 
        builder.add_file('bar-id', builder.tree_root, 'bar', 'text1', True,
2885
 
                         this=False)
2886
 
        builder.add_dir('bar-dir', builder.tree_root, 'bar-id',
2887
 
                        base=False, other=False)
2888
 
        return builder
2889
 
 
2890
 
    def test_uses_this_branch(self):
2891
 
        builder = self.make_text_conflict()
2892
 
        tt = builder.make_preview_transform()
2893
 
        self.addCleanup(tt.finalize)
2894
 
 
2895
 
    def test_affected_files_cached(self):
2896
 
        """Ensures that the config variable is cached"""
2897
 
        builder = self.make_text_conflict()
2898
 
        conflicts = builder.merge()
2899
 
        # The hook should set the variable
2900
 
        self.assertEqual(['bar'], self.merger.affected_files)
2901
 
        self.assertEqual(1, len(conflicts))
2902
 
 
2903
 
    def test_hook_called_for_text_conflicts(self):
2904
 
        builder = self.make_text_conflict()
2905
 
        conflicts = builder.merge()
2906
 
        # The hook should call the merge_text() method
2907
 
        self.assertEqual(['merge_text'], self.calls)
2908
 
 
2909
 
    def test_hook_not_called_for_kind_change(self):
2910
 
        builder = self.make_kind_change()
2911
 
        conflicts = builder.merge()
2912
 
        # The hook should not call the merge_text() method
2913
 
        self.assertEqual([], self.calls)
2914
 
 
2915
 
    def test_hook_not_called_for_other_files(self):
2916
 
        builder = self.make_text_conflict('foobar')
2917
 
        conflicts = builder.merge()
2918
 
        # The hook should not call the merge_text() method
2919
 
        self.assertEqual([], self.calls)