~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_merge.py

  • Committer: John Arbash Meinel
  • Author(s): Mark Hammond
  • Date: 2008-09-09 17:02:21 UTC
  • mto: This revision was merged to the branch mainline in revision 3697.
  • Revision ID: john@arbash-meinel.com-20080909170221-svim3jw2mrz0amp3
An updated transparent icon for bzr.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2012, 2016 Canonical Ltd
 
1
# Copyright (C) 2005, 2006, 2007 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
20
from bzrlib import (
21
 
    branch as _mod_branch,
22
 
    cleanup,
23
21
    conflicts,
24
22
    errors,
25
 
    inventory,
26
23
    knit,
27
 
    memorytree,
28
24
    merge as _mod_merge,
29
25
    option,
30
 
    revision as _mod_revision,
31
 
    tests,
 
26
    progress,
32
27
    transform,
33
28
    versionedfile,
34
29
    )
 
30
from bzrlib.branch import Branch
35
31
from bzrlib.conflicts import ConflictList, TextConflict
36
 
from bzrlib.errors import UnrelatedBranches, NoCommits
 
32
from bzrlib.errors import UnrelatedBranches, NoCommits, BzrCommandError
37
33
from bzrlib.merge import transform_tree, merge_inner, _PlanMerge
38
 
from bzrlib.osutils import basename, pathjoin, file_kind
39
 
from bzrlib.tests import (
40
 
    features,
41
 
    TestCaseWithMemoryTransport,
42
 
    TestCaseWithTransport,
43
 
    test_merge_core,
44
 
    )
 
34
from bzrlib.osutils import pathjoin, file_kind
 
35
from bzrlib.tests import TestCaseWithTransport, TestCaseWithMemoryTransport
 
36
from bzrlib.trace import (enable_test_log, disable_test_log)
45
37
from bzrlib.workingtree import WorkingTree
46
38
 
47
39
 
83
75
        tip = wt1.commit('empty commit')
84
76
        wt2 = self.make_branch_and_tree('branch2')
85
77
        wt2.pull(wt1.branch)
86
 
        with file('branch1/foo', 'wb') as f:
87
 
            f.write('foo')
88
 
        with file('branch1/bar', 'wb') as f:
89
 
            f.write('bar')
 
78
        file('branch1/foo', 'wb').write('foo')
 
79
        file('branch1/bar', 'wb').write('bar')
90
80
        wt1.add('foo')
91
81
        wt1.add('bar')
92
82
        wt1.commit('add foobar')
93
 
        self.run_bzr('merge ../branch1/baz', retcode=3, working_dir='branch2')
94
 
        self.run_bzr('merge ../branch1/foo', working_dir='branch2')
95
 
        self.assertPathExists('branch2/foo')
96
 
        self.assertPathDoesNotExist('branch2/bar')
97
 
        wt2 = WorkingTree.open('branch2')
 
83
        os.chdir('branch2')
 
84
        self.run_bzr('merge ../branch1/baz', retcode=3)
 
85
        self.run_bzr('merge ../branch1/foo')
 
86
        self.failUnlessExists('foo')
 
87
        self.failIfExists('bar')
 
88
        wt2 = WorkingTree.open('.') # opens branch2
98
89
        self.assertEqual([tip], wt2.get_parent_ids())
99
 
 
 
90
        
100
91
    def test_pending_with_null(self):
101
92
        """When base is forced to revno 0, parent_ids are set"""
102
93
        wt2 = self.test_unrelated()
103
94
        wt1 = WorkingTree.open('.')
104
95
        br1 = wt1.branch
105
96
        br1.fetch(wt2.branch)
106
 
        # merge all of branch 2 into branch 1 even though they
 
97
        # merge all of branch 2 into branch 1 even though they 
107
98
        # are not related.
108
99
        wt1.merge_from_branch(wt2.branch, wt2.last_revision(), 'null:')
109
100
        self.assertEqual([br1.last_revision(), wt2.branch.last_revision()],
123
114
        finally:
124
115
            wt1.unlock()
125
116
 
126
 
    def test_merge_into_null_tree(self):
127
 
        wt = self.make_branch_and_tree('tree')
128
 
        null_tree = wt.basis_tree()
129
 
        self.build_tree(['tree/file'])
130
 
        wt.add('file')
131
 
        wt.commit('tree with root')
132
 
        merger = _mod_merge.Merge3Merger(null_tree, null_tree, null_tree, wt,
133
 
                                         this_branch=wt.branch,
134
 
                                         do_merge=False)
135
 
        with merger.make_preview_transform() as tt:
136
 
            self.assertEqual([], tt.find_conflicts())
137
 
            preview = tt.get_preview_tree()
138
 
            self.assertEqual(wt.get_root_id(), preview.get_root_id())
139
 
 
140
 
    def test_merge_unrelated_retains_root(self):
141
 
        wt = self.make_branch_and_tree('tree')
142
 
        other_tree = self.make_branch_and_tree('other')
143
 
        self.addCleanup(other_tree.lock_read().unlock)
144
 
        merger = _mod_merge.Merge3Merger(wt, wt, wt.basis_tree(), other_tree,
145
 
                                         this_branch=wt.branch,
146
 
                                         do_merge=False)
147
 
        with transform.TransformPreview(wt) as merger.tt:
148
 
            merger._compute_transform()
149
 
            new_root_id = merger.tt.final_file_id(merger.tt.root)
150
 
            self.assertEqual(wt.get_root_id(), new_root_id)
151
 
 
152
117
    def test_create_rename(self):
153
118
        """Rename an inventory entry while creating the file"""
154
119
        tree =self.make_branch_and_tree('.')
155
 
        with file('name1', 'wb') as f: f.write('Hello')
 
120
        file('name1', 'wb').write('Hello')
156
121
        tree.add('name1')
157
122
        tree.commit(message="hello")
158
123
        tree.rename_one('name1', 'name2')
165
130
        os.mkdir('dirname1')
166
131
        tree.add('dirname1')
167
132
        filename = pathjoin('dirname1', 'name1')
168
 
        with file(filename, 'wb') as f: f.write('Hello')
 
133
        file(filename, 'wb').write('Hello')
169
134
        tree.add(filename)
170
135
        tree.commit(message="hello")
171
136
        filename2 = pathjoin('dirname1', 'name2')
183
148
        self.addCleanup(tree_b.unlock)
184
149
        tree_a.commit(message="hello again")
185
150
        log = StringIO()
186
 
        merge_inner(tree_b.branch, tree_a, tree_b.basis_tree(),
 
151
        merge_inner(tree_b.branch, tree_a, tree_b.basis_tree(), 
187
152
                    this_tree=tree_b, ignore_zero=True)
188
 
        self.assertTrue('All changes applied successfully.\n' not in
189
 
            self.get_log())
 
153
        log = self._get_log(keep_log_file=True)
 
154
        self.failUnless('All changes applied successfully.\n' not in log)
190
155
        tree_b.revert()
191
 
        merge_inner(tree_b.branch, tree_a, tree_b.basis_tree(),
 
156
        merge_inner(tree_b.branch, tree_a, tree_b.basis_tree(), 
192
157
                    this_tree=tree_b, ignore_zero=False)
193
 
        self.assertTrue('All changes applied successfully.\n' in self.get_log())
 
158
        log = self._get_log(keep_log_file=True)
 
159
        self.failUnless('All changes applied successfully.\n' in log)
194
160
 
195
161
    def test_merge_inner_conflicts(self):
196
162
        tree_a = self.make_branch_and_tree('a')
227
193
 
228
194
    def test_nested_merge(self):
229
195
        tree = self.make_branch_and_tree('tree',
230
 
            format='development-subtree')
 
196
            format='dirstate-with-subtree')
231
197
        sub_tree = self.make_branch_and_tree('tree/sub-tree',
232
 
            format='development-subtree')
 
198
            format='dirstate-with-subtree')
233
199
        sub_tree.set_root_id('sub-tree-root')
234
200
        self.build_tree_contents([('tree/sub-tree/file', 'text1')])
235
201
        sub_tree.add('file')
251
217
        tree_a.add('file')
252
218
        tree_a.commit('commit base')
253
219
        # basis_tree() is only guaranteed to be valid as long as it is actually
254
 
        # the basis tree. This test commits to the tree after grabbing basis,
255
 
        # so we go to the repository.
 
220
        # the basis tree. This mutates the tree after grabbing basis, so go to
 
221
        # the repository.
256
222
        base_tree = tree_a.branch.repository.revision_tree(tree_a.last_revision())
257
223
        tree_b = tree_a.bzrdir.sprout('tree_b').open_workingtree()
258
224
        self.build_tree_contents([('tree_a/file', 'content_2')])
259
225
        tree_a.commit('commit other')
260
226
        other_tree = tree_a.basis_tree()
261
 
        # 'file' is now missing but isn't altered in any commit in b so no
262
 
        # change should be applied.
263
227
        os.unlink('tree_b/file')
264
228
        merge_inner(tree_b.branch, other_tree, base_tree, this_tree=tree_b)
265
229
 
282
246
        self.assertEqual(tree_b.conflicts(),
283
247
                         [conflicts.ContentsConflict('file',
284
248
                          file_id='file-id')])
285
 
 
 
249
    
286
250
    def test_merge_type_registry(self):
287
251
        merge_type_option = option.Option.OPTIONS['merge-type']
288
 
        self.assertFalse('merge4' in [x[0] for x in
 
252
        self.assertFalse('merge4' in [x[0] for x in 
289
253
                        merge_type_option.iter_switches()])
290
254
        registry = _mod_merge.get_merge_type_registry()
291
255
        registry.register_lazy('merge4', 'bzrlib.merge', 'Merge4Merger',
292
256
                               'time-travelling merge')
293
 
        self.assertTrue('merge4' in [x[0] for x in
 
257
        self.assertTrue('merge4' in [x[0] for x in 
294
258
                        merge_type_option.iter_switches()])
295
259
        registry.remove('merge4')
296
 
        self.assertFalse('merge4' in [x[0] for x in
 
260
        self.assertFalse('merge4' in [x[0] for x in 
297
261
                        merge_type_option.iter_switches()])
298
262
 
299
263
    def test_merge_other_moves_we_deleted(self):
326
290
        tree_a.commit('commit 2')
327
291
        tree_b = tree_a.bzrdir.sprout('b').open_workingtree()
328
292
        tree_b.rename_one('file_1', 'renamed')
329
 
        merger = _mod_merge.Merger.from_uncommitted(tree_a, tree_b)
 
293
        merger = _mod_merge.Merger.from_uncommitted(tree_a, tree_b,
 
294
                                                    progress.DummyProgress())
330
295
        merger.merge_type = _mod_merge.Merge3Merger
331
296
        merger.do_merge()
332
297
        self.assertEqual(tree_a.get_parent_ids(), [tree_b.last_revision()])
340
305
        tree_a.commit('commit 2')
341
306
        tree_b = tree_a.bzrdir.sprout('b').open_workingtree()
342
307
        tree_b.rename_one('file_1', 'renamed')
343
 
        merger = _mod_merge.Merger.from_uncommitted(tree_a, tree_b)
 
308
        merger = _mod_merge.Merger.from_uncommitted(tree_a, tree_b,
 
309
                                                    progress.DummyProgress())
344
310
        merger.merge_type = _mod_merge.WeaveMerger
345
311
        merger.do_merge()
346
312
        self.assertEqual(tree_a.get_parent_ids(), [tree_b.last_revision()])
371
337
 
372
338
    def test_weave_cherrypick(self):
373
339
        this_tree, other_tree = self.prepare_cherrypick()
374
 
        merger = _mod_merge.Merger.from_revision_ids(None,
 
340
        merger = _mod_merge.Merger.from_revision_ids(progress.DummyProgress(),
375
341
            this_tree, 'rev3b', 'rev2b', other_tree.branch)
376
342
        merger.merge_type = _mod_merge.WeaveMerger
377
343
        merger.do_merge()
379
345
 
380
346
    def test_weave_cannot_reverse_cherrypick(self):
381
347
        this_tree, other_tree = self.prepare_cherrypick()
382
 
        merger = _mod_merge.Merger.from_revision_ids(None,
 
348
        merger = _mod_merge.Merger.from_revision_ids(progress.DummyProgress(),
383
349
            this_tree, 'rev2b', 'rev3b', other_tree.branch)
384
350
        merger.merge_type = _mod_merge.WeaveMerger
385
351
        self.assertRaises(errors.CannotReverseCherrypick, merger.do_merge)
386
352
 
387
353
    def test_merge3_can_reverse_cherrypick(self):
388
354
        this_tree, other_tree = self.prepare_cherrypick()
389
 
        merger = _mod_merge.Merger.from_revision_ids(None,
 
355
        merger = _mod_merge.Merger.from_revision_ids(progress.DummyProgress(),
390
356
            this_tree, 'rev2b', 'rev3b', other_tree.branch)
391
357
        merger.merge_type = _mod_merge.Merge3Merger
392
358
        merger.do_merge()
404
370
        this_tree.lock_write()
405
371
        self.addCleanup(this_tree.unlock)
406
372
 
407
 
        merger = _mod_merge.Merger.from_revision_ids(None,
 
373
        merger = _mod_merge.Merger.from_revision_ids(progress.DummyProgress(),
408
374
            this_tree, 'rev3b', 'rev2b', other_tree.branch)
409
375
        merger.merge_type = _mod_merge.Merge3Merger
410
376
        merger.do_merge()
415
381
                             '>>>>>>> MERGE-SOURCE\n',
416
382
                             'this/file')
417
383
 
418
 
    def test_merge_reverse_revision_range(self):
419
 
        tree = self.make_branch_and_tree(".")
420
 
        tree.lock_write()
421
 
        self.addCleanup(tree.unlock)
422
 
        self.build_tree(['a'])
423
 
        tree.add('a')
424
 
        first_rev = tree.commit("added a")
425
 
        merger = _mod_merge.Merger.from_revision_ids(None, tree,
426
 
                                          _mod_revision.NULL_REVISION,
427
 
                                          first_rev)
428
 
        merger.merge_type = _mod_merge.Merge3Merger
429
 
        merger.interesting_files = 'a'
430
 
        conflict_count = merger.do_merge()
431
 
        self.assertEqual(0, conflict_count)
432
 
 
433
 
        self.assertPathDoesNotExist("a")
434
 
        tree.revert()
435
 
        self.assertPathExists("a")
436
 
 
437
384
    def test_make_merger(self):
438
385
        this_tree = self.make_branch_and_tree('this')
439
386
        this_tree.commit('rev1', rev_id='rev1')
442
389
        other_tree.commit('rev2', rev_id='rev2b')
443
390
        this_tree.lock_write()
444
391
        self.addCleanup(this_tree.unlock)
445
 
        merger = _mod_merge.Merger.from_revision_ids(None,
 
392
        merger = _mod_merge.Merger.from_revision_ids(progress.DummyProgress,
446
393
            this_tree, 'rev2b', other_branch=other_tree.branch)
447
394
        merger.merge_type = _mod_merge.Merge3Merger
448
395
        tree_merger = merger.make_merger()
449
396
        self.assertIs(_mod_merge.Merge3Merger, tree_merger.__class__)
450
 
        self.assertEqual('rev2b',
451
 
            tree_merger.other_tree.get_revision_id())
452
 
        self.assertEqual('rev1',
453
 
            tree_merger.base_tree.get_revision_id())
454
 
        self.assertEqual(other_tree.branch, tree_merger.other_branch)
 
397
        self.assertEqual('rev2b', tree_merger.other_tree.get_revision_id())
 
398
        self.assertEqual('rev1', tree_merger.base_tree.get_revision_id())
455
399
 
456
400
    def test_make_preview_transform(self):
457
401
        this_tree = self.make_branch_and_tree('this')
465
409
        other_tree.commit('rev2', rev_id='rev2b')
466
410
        this_tree.lock_write()
467
411
        self.addCleanup(this_tree.unlock)
468
 
        merger = _mod_merge.Merger.from_revision_ids(None,
 
412
        merger = _mod_merge.Merger.from_revision_ids(progress.DummyProgress(),
469
413
            this_tree, 'rev2b', other_branch=other_tree.branch)
470
414
        merger.merge_type = _mod_merge.Merge3Merger
471
415
        tree_merger = merger.make_merger()
495
439
        other_tree.commit('rev2', rev_id='rev2b')
496
440
        this_tree.lock_write()
497
441
        self.addCleanup(this_tree.unlock)
498
 
        merger = _mod_merge.Merger.from_revision_ids(None,
 
442
        merger = _mod_merge.Merger.from_revision_ids(progress.DummyProgress(),
499
443
            this_tree, 'rev2b', other_branch=other_tree.branch)
500
444
        merger.merge_type = _mod_merge.Merge3Merger
501
445
        tree_merger = merger.make_merger()
506
450
        finally:
507
451
            tree_file.close()
508
452
 
509
 
    def test_merge_require_tree_root(self):
510
 
        tree = self.make_branch_and_tree(".")
511
 
        tree.lock_write()
512
 
        self.addCleanup(tree.unlock)
513
 
        self.build_tree(['a'])
514
 
        tree.add('a')
515
 
        first_rev = tree.commit("added a")
516
 
        old_root_id = tree.get_root_id()
517
 
        merger = _mod_merge.Merger.from_revision_ids(None, tree,
518
 
                                          _mod_revision.NULL_REVISION,
519
 
                                          first_rev)
520
 
        merger.merge_type = _mod_merge.Merge3Merger
521
 
        conflict_count = merger.do_merge()
522
 
        self.assertEqual(0, conflict_count)
523
 
        self.assertEqual(set([old_root_id]), tree.all_file_ids())
524
 
        tree.set_parent_ids([])
525
 
 
526
453
    def test_merge_add_into_deleted_root(self):
527
454
        # Yes, people actually do this.  And report bugs if it breaks.
528
455
        source = self.make_branch_and_tree('source', format='rich-root-pack')
559
486
class TestPlanMerge(TestCaseWithMemoryTransport):
560
487
 
561
488
    def setUp(self):
562
 
        super(TestPlanMerge, self).setUp()
 
489
        TestCaseWithMemoryTransport.setUp(self)
563
490
        mapper = versionedfile.PrefixMapper()
564
491
        factory = knit.make_file_factory(True, mapper)
565
492
        self.vf = factory(self.get_transport())
589
516
        self.add_uncommitted_version(('root', 'C:'), [('root', 'A')], 'fabg')
590
517
        return _PlanMerge('B:', 'C:', self.plan_merge_vf, ('root',))
591
518
 
592
 
    def test_base_from_plan(self):
593
 
        self.setup_plan_merge()
594
 
        plan = self.plan_merge_vf.plan_merge('B', 'C')
595
 
        pwm = versionedfile.PlanWeaveMerge(plan)
596
 
        self.assertEqual(['a\n', 'b\n', 'c\n'], pwm.base_from_plan())
597
 
 
598
519
    def test_unique_lines(self):
599
520
        plan = self.setup_plan_merge()
600
521
        self.assertEqual(plan._unique_lines(
795
716
 
796
717
    def test_plan_merge_insert_order(self):
797
718
        """Weave merges are sensitive to the order of insertion.
798
 
 
 
719
        
799
720
        Specifically for overlapping regions, it effects which region gets put
800
721
        'first'. And when a user resolves an overlapping merge, if they use the
801
722
        same ordering, then the lines match the parents, if they don't only
898
819
                          ('unchanged', 'f\n'),
899
820
                          ('unchanged', 'g\n')],
900
821
                         list(plan))
901
 
        plan = self.plan_merge_vf.plan_lca_merge('F', 'G')
902
 
        # This is one of the main differences between plan_merge and
903
 
        # plan_lca_merge. plan_lca_merge generates a conflict for 'x => z',
904
 
        # because 'x' was not present in one of the bases. However, in this
905
 
        # case it is spurious because 'x' does not exist in the global base A.
906
 
        self.assertEqual([
907
 
                          ('unchanged', 'h\n'),
908
 
                          ('unchanged', 'a\n'),
909
 
                          ('conflicted-a', 'x\n'),
910
 
                          ('new-b', 'z\n'),
911
 
                          ('unchanged', 'c\n'),
912
 
                          ('unchanged', 'd\n'),
913
 
                          ('unchanged', 'y\n'),
914
 
                          ('unchanged', 'f\n'),
915
 
                          ('unchanged', 'g\n')],
916
 
                         list(plan))
917
 
 
918
 
    def test_criss_cross_flip_flop(self):
919
 
        # This is specificly trying to trigger problems when using limited
920
 
        # ancestry and weaves. The ancestry graph looks like:
921
 
        #       XX      unused ancestor, should not show up in the weave
922
 
        #       |
923
 
        #       A       Unique LCA
924
 
        #      / \  
925
 
        #     B   C     B & C both introduce a new line
926
 
        #     |\ /|  
927
 
        #     | X |  
928
 
        #     |/ \| 
929
 
        #     D   E     B & C are both merged, so both are common ancestors
930
 
        #               In the process of merging, both sides order the new
931
 
        #               lines differently
932
 
        #
933
 
        self.add_rev('root', 'XX', [], 'qrs')
934
 
        self.add_rev('root', 'A', ['XX'], 'abcdef')
935
 
        self.add_rev('root', 'B', ['A'], 'abcdgef')
936
 
        self.add_rev('root', 'C', ['A'], 'abcdhef')
937
 
        self.add_rev('root', 'D', ['B', 'C'], 'abcdghef')
938
 
        self.add_rev('root', 'E', ['C', 'B'], 'abcdhgef')
939
 
        plan = list(self.plan_merge_vf.plan_merge('D', 'E'))
940
 
        self.assertEqual([
941
 
                          ('unchanged', 'a\n'),
942
 
                          ('unchanged', 'b\n'),
943
 
                          ('unchanged', 'c\n'),
944
 
                          ('unchanged', 'd\n'),
945
 
                          ('new-b', 'h\n'),
946
 
                          ('unchanged', 'g\n'),
947
 
                          ('killed-b', 'h\n'),
948
 
                          ('unchanged', 'e\n'),
949
 
                          ('unchanged', 'f\n'),
950
 
                         ], plan)
951
 
        pwm = versionedfile.PlanWeaveMerge(plan)
952
 
        self.assertEqualDiff('\n'.join('abcdghef') + '\n',
953
 
                             ''.join(pwm.base_from_plan()))
954
 
        # Reversing the order reverses the merge plan, and final order of 'hg'
955
 
        # => 'gh'
956
 
        plan = list(self.plan_merge_vf.plan_merge('E', 'D'))
957
 
        self.assertEqual([
958
 
                          ('unchanged', 'a\n'),
959
 
                          ('unchanged', 'b\n'),
960
 
                          ('unchanged', 'c\n'),
961
 
                          ('unchanged', 'd\n'),
962
 
                          ('new-b', 'g\n'),
963
 
                          ('unchanged', 'h\n'),
964
 
                          ('killed-b', 'g\n'),
965
 
                          ('unchanged', 'e\n'),
966
 
                          ('unchanged', 'f\n'),
967
 
                         ], plan)
968
 
        pwm = versionedfile.PlanWeaveMerge(plan)
969
 
        self.assertEqualDiff('\n'.join('abcdhgef') + '\n',
970
 
                             ''.join(pwm.base_from_plan()))
971
 
        # This is where lca differs, in that it (fairly correctly) determines
972
 
        # that there is a conflict because both sides resolved the merge
973
 
        # differently
974
 
        plan = list(self.plan_merge_vf.plan_lca_merge('D', 'E'))
975
 
        self.assertEqual([
976
 
                          ('unchanged', 'a\n'),
977
 
                          ('unchanged', 'b\n'),
978
 
                          ('unchanged', 'c\n'),
979
 
                          ('unchanged', 'd\n'),
980
 
                          ('conflicted-b', 'h\n'),
981
 
                          ('unchanged', 'g\n'),
982
 
                          ('conflicted-a', 'h\n'),
983
 
                          ('unchanged', 'e\n'),
984
 
                          ('unchanged', 'f\n'),
985
 
                         ], plan)
986
 
        pwm = versionedfile.PlanWeaveMerge(plan)
987
 
        self.assertEqualDiff('\n'.join('abcdgef') + '\n',
988
 
                             ''.join(pwm.base_from_plan()))
989
 
        # Reversing it changes what line is doubled, but still gives a
990
 
        # double-conflict
991
 
        plan = list(self.plan_merge_vf.plan_lca_merge('E', 'D'))
992
 
        self.assertEqual([
993
 
                          ('unchanged', 'a\n'),
994
 
                          ('unchanged', 'b\n'),
995
 
                          ('unchanged', 'c\n'),
996
 
                          ('unchanged', 'd\n'),
997
 
                          ('conflicted-b', 'g\n'),
998
 
                          ('unchanged', 'h\n'),
999
 
                          ('conflicted-a', 'g\n'),
1000
 
                          ('unchanged', 'e\n'),
1001
 
                          ('unchanged', 'f\n'),
1002
 
                         ], plan)
1003
 
        pwm = versionedfile.PlanWeaveMerge(plan)
1004
 
        self.assertEqualDiff('\n'.join('abcdhef') + '\n',
1005
 
                             ''.join(pwm.base_from_plan()))
1006
822
 
1007
823
    def assertRemoveExternalReferences(self, filtered_parent_map,
1008
824
                                       child_map, tails, parent_map):
1208
1024
                         ], list(plan))
1209
1025
 
1210
1026
 
1211
 
class LoggingMerger(object):
1212
 
    # These seem to be the required attributes
1213
 
    requires_base = False
1214
 
    supports_reprocess = False
1215
 
    supports_show_base = False
1216
 
    supports_cherrypick = False
1217
 
    # We intentionally do not define supports_lca_trees
1218
 
 
1219
 
    def __init__(self, *args, **kwargs):
1220
 
        self.args = args
1221
 
        self.kwargs = kwargs
1222
 
 
1223
 
 
1224
 
class TestMergerBase(TestCaseWithMemoryTransport):
1225
 
    """Common functionality for Merger tests that don't write to disk."""
1226
 
 
1227
 
    def get_builder(self):
1228
 
        builder = self.make_branch_builder('path')
1229
 
        builder.start_series()
1230
 
        self.addCleanup(builder.finish_series)
1231
 
        return builder
1232
 
 
1233
 
    def setup_simple_graph(self):
1234
 
        """Create a simple 3-node graph.
1235
 
 
1236
 
        :return: A BranchBuilder
1237
 
        """
1238
 
        #
1239
 
        #  A
1240
 
        #  |\
1241
 
        #  B C
1242
 
        #
1243
 
        builder = self.get_builder()
1244
 
        builder.build_snapshot('A-id', None,
1245
 
            [('add', ('', None, 'directory', None))])
1246
 
        builder.build_snapshot('C-id', ['A-id'], [])
1247
 
        builder.build_snapshot('B-id', ['A-id'], [])
1248
 
        return builder
1249
 
 
1250
 
    def setup_criss_cross_graph(self):
1251
 
        """Create a 5-node graph with a criss-cross.
1252
 
 
1253
 
        :return: A BranchBuilder
1254
 
        """
1255
 
        # A
1256
 
        # |\
1257
 
        # B C
1258
 
        # |X|
1259
 
        # D E
1260
 
        builder = self.setup_simple_graph()
1261
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
1262
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1263
 
        return builder
1264
 
 
1265
 
    def make_Merger(self, builder, other_revision_id,
1266
 
                    interesting_files=None, interesting_ids=None):
1267
 
        """Make a Merger object from a branch builder"""
1268
 
        mem_tree = memorytree.MemoryTree.create_on_branch(builder.get_branch())
1269
 
        mem_tree.lock_write()
1270
 
        self.addCleanup(mem_tree.unlock)
1271
 
        merger = _mod_merge.Merger.from_revision_ids(None,
1272
 
            mem_tree, other_revision_id)
1273
 
        merger.set_interesting_files(interesting_files)
1274
 
        # It seems there is no matching function for set_interesting_ids
1275
 
        merger.interesting_ids = interesting_ids
1276
 
        merger.merge_type = _mod_merge.Merge3Merger
1277
 
        return merger
1278
 
 
1279
 
 
1280
 
class TestMergerInMemory(TestMergerBase):
1281
 
 
1282
 
    def test_cache_trees_with_revision_ids_None(self):
1283
 
        merger = self.make_Merger(self.setup_simple_graph(), 'C-id')
1284
 
        original_cache = dict(merger._cached_trees)
1285
 
        merger.cache_trees_with_revision_ids([None])
1286
 
        self.assertEqual(original_cache, merger._cached_trees)
1287
 
 
1288
 
    def test_cache_trees_with_revision_ids_no_revision_id(self):
1289
 
        merger = self.make_Merger(self.setup_simple_graph(), 'C-id')
1290
 
        original_cache = dict(merger._cached_trees)
1291
 
        tree = self.make_branch_and_memory_tree('tree')
1292
 
        merger.cache_trees_with_revision_ids([tree])
1293
 
        self.assertEqual(original_cache, merger._cached_trees)
1294
 
 
1295
 
    def test_cache_trees_with_revision_ids_having_revision_id(self):
1296
 
        merger = self.make_Merger(self.setup_simple_graph(), 'C-id')
1297
 
        original_cache = dict(merger._cached_trees)
1298
 
        tree = merger.this_branch.repository.revision_tree('B-id')
1299
 
        original_cache['B-id'] = tree
1300
 
        merger.cache_trees_with_revision_ids([tree])
1301
 
        self.assertEqual(original_cache, merger._cached_trees)
1302
 
 
1303
 
    def test_find_base(self):
1304
 
        merger = self.make_Merger(self.setup_simple_graph(), 'C-id')
1305
 
        self.assertEqual('A-id', merger.base_rev_id)
1306
 
        self.assertFalse(merger._is_criss_cross)
1307
 
        self.assertIs(None, merger._lca_trees)
1308
 
 
1309
 
    def test_find_base_criss_cross(self):
1310
 
        builder = self.setup_criss_cross_graph()
1311
 
        merger = self.make_Merger(builder, 'E-id')
1312
 
        self.assertEqual('A-id', merger.base_rev_id)
1313
 
        self.assertTrue(merger._is_criss_cross)
1314
 
        self.assertEqual(['B-id', 'C-id'], [t.get_revision_id()
1315
 
                                            for t in merger._lca_trees])
1316
 
        # If we swap the order, we should get a different lca order
1317
 
        builder.build_snapshot('F-id', ['E-id'], [])
1318
 
        merger = self.make_Merger(builder, 'D-id')
1319
 
        self.assertEqual(['C-id', 'B-id'], [t.get_revision_id()
1320
 
                                            for t in merger._lca_trees])
1321
 
 
1322
 
    def test_find_base_triple_criss_cross(self):
1323
 
        #       A-.
1324
 
        #      / \ \
1325
 
        #     B   C F # F is merged into both branches
1326
 
        #     |\ /| |
1327
 
        #     | X | |\
1328
 
        #     |/ \| | :
1329
 
        #   : D   E |
1330
 
        #    \|   |/
1331
 
        #     G   H
1332
 
        builder = self.setup_criss_cross_graph()
1333
 
        builder.build_snapshot('F-id', ['A-id'], [])
1334
 
        builder.build_snapshot('H-id', ['E-id', 'F-id'], [])
1335
 
        builder.build_snapshot('G-id', ['D-id', 'F-id'], [])
1336
 
        merger = self.make_Merger(builder, 'H-id')
1337
 
        self.assertEqual(['B-id', 'C-id', 'F-id'],
1338
 
                         [t.get_revision_id() for t in merger._lca_trees])
1339
 
 
1340
 
    def test_find_base_new_root_criss_cross(self):
1341
 
        # A   B
1342
 
        # |\ /|
1343
 
        # | X |
1344
 
        # |/ \|
1345
 
        # C   D
1346
 
        
1347
 
        builder = self.get_builder()
1348
 
        builder.build_snapshot('A-id', None,
1349
 
            [('add', ('', None, 'directory', None))])
1350
 
        builder.build_snapshot('B-id', [],
1351
 
            [('add', ('', None, 'directory', None))])
1352
 
        builder.build_snapshot('D-id', ['A-id', 'B-id'], [])
1353
 
        builder.build_snapshot('C-id', ['A-id', 'B-id'], [])
1354
 
        merger = self.make_Merger(builder, 'D-id')
1355
 
        self.assertEqual('A-id', merger.base_rev_id)
1356
 
        self.assertTrue(merger._is_criss_cross)
1357
 
        self.assertEqual(['A-id', 'B-id'], [t.get_revision_id()
1358
 
                                            for t in merger._lca_trees])
1359
 
 
1360
 
    def test_no_criss_cross_passed_to_merge_type(self):
1361
 
        class LCATreesMerger(LoggingMerger):
1362
 
            supports_lca_trees = True
1363
 
 
1364
 
        merger = self.make_Merger(self.setup_simple_graph(), 'C-id')
1365
 
        merger.merge_type = LCATreesMerger
1366
 
        merge_obj = merger.make_merger()
1367
 
        self.assertIsInstance(merge_obj, LCATreesMerger)
1368
 
        self.assertFalse('lca_trees' in merge_obj.kwargs)
1369
 
 
1370
 
    def test_criss_cross_passed_to_merge_type(self):
1371
 
        merger = self.make_Merger(self.setup_criss_cross_graph(), 'E-id')
1372
 
        merger.merge_type = _mod_merge.Merge3Merger
1373
 
        merge_obj = merger.make_merger()
1374
 
        self.assertEqual(['B-id', 'C-id'], [t.get_revision_id()
1375
 
                                            for t in merger._lca_trees])
1376
 
 
1377
 
    def test_criss_cross_not_supported_merge_type(self):
1378
 
        merger = self.make_Merger(self.setup_criss_cross_graph(), 'E-id')
1379
 
        # We explicitly do not define supports_lca_trees
1380
 
        merger.merge_type = LoggingMerger
1381
 
        merge_obj = merger.make_merger()
1382
 
        self.assertIsInstance(merge_obj, LoggingMerger)
1383
 
        self.assertFalse('lca_trees' in merge_obj.kwargs)
1384
 
 
1385
 
    def test_criss_cross_unsupported_merge_type(self):
1386
 
        class UnsupportedLCATreesMerger(LoggingMerger):
1387
 
            supports_lca_trees = False
1388
 
 
1389
 
        merger = self.make_Merger(self.setup_criss_cross_graph(), 'E-id')
1390
 
        merger.merge_type = UnsupportedLCATreesMerger
1391
 
        merge_obj = merger.make_merger()
1392
 
        self.assertIsInstance(merge_obj, UnsupportedLCATreesMerger)
1393
 
        self.assertFalse('lca_trees' in merge_obj.kwargs)
1394
 
 
1395
 
 
1396
 
class TestMergerEntriesLCA(TestMergerBase):
1397
 
 
1398
 
    def make_merge_obj(self, builder, other_revision_id,
1399
 
                       interesting_files=None, interesting_ids=None):
1400
 
        merger = self.make_Merger(builder, other_revision_id,
1401
 
            interesting_files=interesting_files,
1402
 
            interesting_ids=interesting_ids)
1403
 
        return merger.make_merger()
1404
 
 
1405
 
    def test_simple(self):
1406
 
        builder = self.get_builder()
1407
 
        builder.build_snapshot('A-id', None,
1408
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1409
 
             ('add', (u'a', 'a-id', 'file', 'a\nb\nc\n'))])
1410
 
        builder.build_snapshot('C-id', ['A-id'],
1411
 
            [('modify', ('a-id', 'a\nb\nC\nc\n'))])
1412
 
        builder.build_snapshot('B-id', ['A-id'],
1413
 
            [('modify', ('a-id', 'a\nB\nb\nc\n'))])
1414
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1415
 
            [('modify', ('a-id', 'a\nB\nb\nC\nc\nE\n'))])
1416
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'],
1417
 
            [('modify', ('a-id', 'a\nB\nb\nC\nc\n'))])
1418
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1419
 
 
1420
 
        self.assertEqual(['B-id', 'C-id'], [t.get_revision_id()
1421
 
                                            for t in merge_obj._lca_trees])
1422
 
        self.assertEqual('A-id', merge_obj.base_tree.get_revision_id())
1423
 
        entries = list(merge_obj._entries_lca())
1424
 
 
1425
 
        # (file_id, changed, parents, names, executable)
1426
 
        # BASE, lca1, lca2, OTHER, THIS
1427
 
        root_id = 'a-root-id'
1428
 
        self.assertEqual([('a-id', True,
1429
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
1430
 
                           ((u'a', [u'a', u'a']), u'a', u'a'),
1431
 
                           ((False, [False, False]), False, False)),
1432
 
                         ], entries)
1433
 
 
1434
 
    def test_not_in_base(self):
1435
 
        # LCAs all have the same last-modified revision for the file, as do
1436
 
        # the tips, but the base has something different
1437
 
        #       A    base, doesn't have the file
1438
 
        #       |\
1439
 
        #       B C  B introduces 'foo', C introduces 'bar'
1440
 
        #       |X|
1441
 
        #       D E  D and E now both have 'foo' and 'bar'
1442
 
        #       |X|
1443
 
        #       F G  the files are now in F, G, D and E, but not in A
1444
 
        #            G modifies 'bar'
1445
 
 
1446
 
        builder = self.get_builder()
1447
 
        builder.build_snapshot('A-id', None,
1448
 
            [('add', (u'', 'a-root-id', 'directory', None))])
1449
 
        builder.build_snapshot('B-id', ['A-id'],
1450
 
            [('add', (u'foo', 'foo-id', 'file', 'a\nb\nc\n'))])
1451
 
        builder.build_snapshot('C-id', ['A-id'],
1452
 
            [('add', (u'bar', 'bar-id', 'file', 'd\ne\nf\n'))])
1453
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'],
1454
 
            [('add', (u'bar', 'bar-id', 'file', 'd\ne\nf\n'))])
1455
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1456
 
            [('add', (u'foo', 'foo-id', 'file', 'a\nb\nc\n'))])
1457
 
        builder.build_snapshot('G-id', ['E-id', 'D-id'],
1458
 
            [('modify', (u'bar-id', 'd\ne\nf\nG\n'))])
1459
 
        builder.build_snapshot('F-id', ['D-id', 'E-id'], [])
1460
 
        merge_obj = self.make_merge_obj(builder, 'G-id')
1461
 
 
1462
 
        self.assertEqual(['D-id', 'E-id'], [t.get_revision_id()
1463
 
                                            for t in merge_obj._lca_trees])
1464
 
        self.assertEqual('A-id', merge_obj.base_tree.get_revision_id())
1465
 
        entries = list(merge_obj._entries_lca())
1466
 
        root_id = 'a-root-id'
1467
 
        self.assertEqual([('bar-id', True,
1468
 
                           ((None, [root_id, root_id]), root_id, root_id),
1469
 
                           ((None, [u'bar', u'bar']), u'bar', u'bar'),
1470
 
                           ((None, [False, False]), False, False)),
1471
 
                         ], entries)
1472
 
 
1473
 
    def test_not_in_this(self):
1474
 
        builder = self.get_builder()
1475
 
        builder.build_snapshot('A-id', None,
1476
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1477
 
             ('add', (u'a', 'a-id', 'file', 'a\nb\nc\n'))])
1478
 
        builder.build_snapshot('B-id', ['A-id'],
1479
 
            [('modify', ('a-id', 'a\nB\nb\nc\n'))])
1480
 
        builder.build_snapshot('C-id', ['A-id'],
1481
 
            [('modify', ('a-id', 'a\nb\nC\nc\n'))])
1482
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1483
 
            [('modify', ('a-id', 'a\nB\nb\nC\nc\nE\n'))])
1484
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'],
1485
 
            [('unversion', 'a-id')])
1486
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1487
 
 
1488
 
        self.assertEqual(['B-id', 'C-id'], [t.get_revision_id()
1489
 
                                            for t in merge_obj._lca_trees])
1490
 
        self.assertEqual('A-id', merge_obj.base_tree.get_revision_id())
1491
 
 
1492
 
        entries = list(merge_obj._entries_lca())
1493
 
        root_id = 'a-root-id'
1494
 
        self.assertEqual([('a-id', True,
1495
 
                           ((root_id, [root_id, root_id]), root_id, None),
1496
 
                           ((u'a', [u'a', u'a']), u'a', None),
1497
 
                           ((False, [False, False]), False, None)),
1498
 
                         ], entries)
1499
 
 
1500
 
    def test_file_not_in_one_lca(self):
1501
 
        #   A   # just root
1502
 
        #   |\
1503
 
        #   B C # B no file, C introduces a file
1504
 
        #   |X|
1505
 
        #   D E # D and E both have the file, unchanged from C
1506
 
        builder = self.get_builder()
1507
 
        builder.build_snapshot('A-id', None,
1508
 
            [('add', (u'', 'a-root-id', 'directory', None))])
1509
 
        builder.build_snapshot('B-id', ['A-id'], [])
1510
 
        builder.build_snapshot('C-id', ['A-id'],
1511
 
            [('add', (u'a', 'a-id', 'file', 'a\nb\nc\n'))])
1512
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], []) # Inherited from C
1513
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], # Merged from C
1514
 
            [('add', (u'a', 'a-id', 'file', 'a\nb\nc\n'))])
1515
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1516
 
 
1517
 
        self.assertEqual(['B-id', 'C-id'], [t.get_revision_id()
1518
 
                                            for t in merge_obj._lca_trees])
1519
 
        self.assertEqual('A-id', merge_obj.base_tree.get_revision_id())
1520
 
 
1521
 
        entries = list(merge_obj._entries_lca())
1522
 
        self.assertEqual([], entries)
1523
 
 
1524
 
    def test_not_in_other(self):
1525
 
        builder = self.get_builder()
1526
 
        builder.build_snapshot('A-id', None,
1527
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1528
 
             ('add', (u'a', 'a-id', 'file', 'a\nb\nc\n'))])
1529
 
        builder.build_snapshot('B-id', ['A-id'], [])
1530
 
        builder.build_snapshot('C-id', ['A-id'], [])
1531
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1532
 
            [('unversion', 'a-id')])
1533
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1534
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1535
 
 
1536
 
        entries = list(merge_obj._entries_lca())
1537
 
        root_id = 'a-root-id'
1538
 
        self.assertEqual([('a-id', True,
1539
 
                           ((root_id, [root_id, root_id]), None, root_id),
1540
 
                           ((u'a', [u'a', u'a']), None, u'a'),
1541
 
                           ((False, [False, False]), None, False)),
1542
 
                         ], entries)
1543
 
 
1544
 
    def test_not_in_other_or_lca(self):
1545
 
        #       A    base, introduces 'foo'
1546
 
        #       |\
1547
 
        #       B C  B nothing, C deletes foo
1548
 
        #       |X|
1549
 
        #       D E  D restores foo (same as B), E leaves it deleted
1550
 
        # Analysis:
1551
 
        #   A => B, no changes
1552
 
        #   A => C, delete foo (C should supersede B)
1553
 
        #   C => D, restore foo
1554
 
        #   C => E, no changes
1555
 
        # D would then win 'cleanly' and no record would be given
1556
 
        builder = self.get_builder()
1557
 
        builder.build_snapshot('A-id', None,
1558
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1559
 
             ('add', (u'foo', 'foo-id', 'file', 'content\n'))])
1560
 
        builder.build_snapshot('B-id', ['A-id'], [])
1561
 
        builder.build_snapshot('C-id', ['A-id'],
1562
 
            [('unversion', 'foo-id')])
1563
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
1564
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1565
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1566
 
 
1567
 
        entries = list(merge_obj._entries_lca())
1568
 
        self.assertEqual([], entries)
1569
 
 
1570
 
    def test_not_in_other_mod_in_lca1_not_in_lca2(self):
1571
 
        #       A    base, introduces 'foo'
1572
 
        #       |\
1573
 
        #       B C  B changes 'foo', C deletes foo
1574
 
        #       |X|
1575
 
        #       D E  D restores foo (same as B), E leaves it deleted (as C)
1576
 
        # Analysis:
1577
 
        #   A => B, modified foo
1578
 
        #   A => C, delete foo, C does not supersede B
1579
 
        #   B => D, no changes
1580
 
        #   C => D, resolve in favor of B
1581
 
        #   B => E, resolve in favor of E
1582
 
        #   C => E, no changes
1583
 
        # In this case, we have a conflict of how the changes were resolved. E
1584
 
        # picked C and D picked B, so we should issue a conflict
1585
 
        builder = self.get_builder()
1586
 
        builder.build_snapshot('A-id', None,
1587
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1588
 
             ('add', (u'foo', 'foo-id', 'file', 'content\n'))])
1589
 
        builder.build_snapshot('B-id', ['A-id'], [
1590
 
            ('modify', ('foo-id', 'new-content\n'))])
1591
 
        builder.build_snapshot('C-id', ['A-id'],
1592
 
            [('unversion', 'foo-id')])
1593
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
1594
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1595
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1596
 
 
1597
 
        entries = list(merge_obj._entries_lca())
1598
 
        root_id = 'a-root-id'
1599
 
        self.assertEqual([('foo-id', True,
1600
 
                           ((root_id, [root_id, None]), None, root_id),
1601
 
                           ((u'foo', [u'foo', None]), None, 'foo'),
1602
 
                           ((False, [False, None]), None, False)),
1603
 
                         ], entries)
1604
 
 
1605
 
    def test_only_in_one_lca(self):
1606
 
        #   A   add only root
1607
 
        #   |\
1608
 
        #   B C B nothing, C add file
1609
 
        #   |X|
1610
 
        #   D E D still has nothing, E removes file
1611
 
        # Analysis:
1612
 
        #   B => D, no change
1613
 
        #   C => D, removed the file
1614
 
        #   B => E, no change
1615
 
        #   C => E, removed the file
1616
 
        # Thus D & E have identical changes, and this is a no-op
1617
 
        # Alternatively:
1618
 
        #   A => B, no change
1619
 
        #   A => C, add file, thus C supersedes B
1620
 
        #   w/ C=BASE, D=THIS, E=OTHER we have 'happy convergence'
1621
 
        builder = self.get_builder()
1622
 
        builder.build_snapshot('A-id', None,
1623
 
            [('add', (u'', 'a-root-id', 'directory', None))])
1624
 
        builder.build_snapshot('B-id', ['A-id'], [])
1625
 
        builder.build_snapshot('C-id', ['A-id'],
1626
 
            [('add', (u'a', 'a-id', 'file', 'a\nb\nc\n'))])
1627
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1628
 
            [('unversion', 'a-id')])
1629
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1630
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1631
 
 
1632
 
        entries = list(merge_obj._entries_lca())
1633
 
        self.assertEqual([], entries)
1634
 
 
1635
 
    def test_only_in_other(self):
1636
 
        builder = self.get_builder()
1637
 
        builder.build_snapshot('A-id', None,
1638
 
            [('add', (u'', 'a-root-id', 'directory', None))])
1639
 
        builder.build_snapshot('B-id', ['A-id'], [])
1640
 
        builder.build_snapshot('C-id', ['A-id'], [])
1641
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1642
 
            [('add', (u'a', 'a-id', 'file', 'a\nb\nc\n'))])
1643
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1644
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1645
 
 
1646
 
        entries = list(merge_obj._entries_lca())
1647
 
        root_id = 'a-root-id'
1648
 
        self.assertEqual([('a-id', True,
1649
 
                           ((None, [None, None]), root_id, None),
1650
 
                           ((None, [None, None]), u'a', None),
1651
 
                           ((None, [None, None]), False, None)),
1652
 
                         ], entries)
1653
 
 
1654
 
    def test_one_lca_supersedes(self):
1655
 
        # One LCA supersedes the other LCAs last modified value, but the
1656
 
        # value is not the same as BASE.
1657
 
        #       A    base, introduces 'foo', last mod A
1658
 
        #       |\
1659
 
        #       B C  B modifies 'foo' (mod B), C does nothing (mod A)
1660
 
        #       |X|
1661
 
        #       D E  D does nothing (mod B), E updates 'foo' (mod E)
1662
 
        #       |X|
1663
 
        #       F G  F updates 'foo' (mod F). G does nothing (mod E)
1664
 
        #
1665
 
        #   At this point, G should not be considered to modify 'foo', even
1666
 
        #   though its LCAs disagree. This is because the modification in E
1667
 
        #   completely supersedes the value in D.
1668
 
        builder = self.get_builder()
1669
 
        builder.build_snapshot('A-id', None,
1670
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1671
 
             ('add', (u'foo', 'foo-id', 'file', 'A content\n'))])
1672
 
        builder.build_snapshot('C-id', ['A-id'], [])
1673
 
        builder.build_snapshot('B-id', ['A-id'],
1674
 
            [('modify', ('foo-id', 'B content\n'))])
1675
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1676
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1677
 
            [('modify', ('foo-id', 'E content\n'))])
1678
 
        builder.build_snapshot('G-id', ['E-id', 'D-id'], [])
1679
 
        builder.build_snapshot('F-id', ['D-id', 'E-id'],
1680
 
            [('modify', ('foo-id', 'F content\n'))])
1681
 
        merge_obj = self.make_merge_obj(builder, 'G-id')
1682
 
 
1683
 
        self.assertEqual([], list(merge_obj._entries_lca()))
1684
 
 
1685
 
    def test_one_lca_supersedes_path(self):
1686
 
        # Double-criss-cross merge, the ultimate base value is different from
1687
 
        # the intermediate.
1688
 
        #   A    value 'foo'
1689
 
        #   |\
1690
 
        #   B C  B value 'bar', C = 'foo'
1691
 
        #   |X|
1692
 
        #   D E  D = 'bar', E supersedes to 'bing'
1693
 
        #   |X|
1694
 
        #   F G  F = 'bing', G supersedes to 'barry'
1695
 
        #
1696
 
        # In this case, we technically should not care about the value 'bar' for
1697
 
        # D, because it was clearly superseded by E's 'bing'. The
1698
 
        # per-file/attribute graph would actually look like:
1699
 
        #   A
1700
 
        #   |
1701
 
        #   B
1702
 
        #   |
1703
 
        #   E
1704
 
        #   |
1705
 
        #   G
1706
 
        #
1707
 
        # Because the other side of the merge never modifies the value, it just
1708
 
        # takes the value from the merge.
1709
 
        #
1710
 
        # ATM this fails because we will prune 'foo' from the LCAs, but we
1711
 
        # won't prune 'bar'. This is getting far off into edge-case land, so we
1712
 
        # aren't supporting it yet.
1713
 
        #
1714
 
        builder = self.get_builder()
1715
 
        builder.build_snapshot('A-id', None,
1716
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1717
 
             ('add', (u'foo', 'foo-id', 'file', 'A content\n'))])
1718
 
        builder.build_snapshot('C-id', ['A-id'], [])
1719
 
        builder.build_snapshot('B-id', ['A-id'],
1720
 
            [('rename', ('foo', 'bar'))])
1721
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1722
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1723
 
            [('rename', ('foo', 'bing'))]) # override to bing
1724
 
        builder.build_snapshot('G-id', ['E-id', 'D-id'],
1725
 
            [('rename', ('bing', 'barry'))]) # override to barry
1726
 
        builder.build_snapshot('F-id', ['D-id', 'E-id'],
1727
 
            [('rename', ('bar', 'bing'))]) # Merge in E's change
1728
 
        merge_obj = self.make_merge_obj(builder, 'G-id')
1729
 
 
1730
 
        self.expectFailure("We don't do an actual heads() check on lca values,"
1731
 
            " or use the per-attribute graph",
1732
 
            self.assertEqual, [], list(merge_obj._entries_lca()))
1733
 
 
1734
 
    def test_one_lca_accidentally_pruned(self):
1735
 
        # Another incorrect resolution from the same basic flaw:
1736
 
        #   A    value 'foo'
1737
 
        #   |\
1738
 
        #   B C  B value 'bar', C = 'foo'
1739
 
        #   |X|
1740
 
        #   D E  D = 'bar', E reverts to 'foo'
1741
 
        #   |X|
1742
 
        #   F G  F = 'bing', G switches to 'bar'
1743
 
        #
1744
 
        # 'bar' will not be seen as an interesting change, because 'foo' will
1745
 
        # be pruned from the LCAs, even though it was newly introduced by E
1746
 
        # (superseding B).
1747
 
        builder = self.get_builder()
1748
 
        builder.build_snapshot('A-id', None,
1749
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1750
 
             ('add', (u'foo', 'foo-id', 'file', 'A content\n'))])
1751
 
        builder.build_snapshot('C-id', ['A-id'], [])
1752
 
        builder.build_snapshot('B-id', ['A-id'],
1753
 
            [('rename', ('foo', 'bar'))])
1754
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1755
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
1756
 
        builder.build_snapshot('G-id', ['E-id', 'D-id'],
1757
 
            [('rename', ('foo', 'bar'))])
1758
 
        builder.build_snapshot('F-id', ['D-id', 'E-id'],
1759
 
            [('rename', ('bar', 'bing'))]) # should end up conflicting
1760
 
        merge_obj = self.make_merge_obj(builder, 'G-id')
1761
 
 
1762
 
        entries = list(merge_obj._entries_lca())
1763
 
        root_id = 'a-root-id'
1764
 
        self.expectFailure("We prune values from BASE even when relevant.",
1765
 
            self.assertEqual,
1766
 
                [('foo-id', False,
1767
 
                  ((root_id, [root_id, root_id]), root_id, root_id),
1768
 
                  ((u'foo', [u'bar', u'foo']), u'bar', u'bing'),
1769
 
                  ((False, [False, False]), False, False)),
1770
 
                ], entries)
1771
 
 
1772
 
    def test_both_sides_revert(self):
1773
 
        # Both sides of a criss-cross revert the text to the lca
1774
 
        #       A    base, introduces 'foo'
1775
 
        #       |\
1776
 
        #       B C  B modifies 'foo', C modifies 'foo'
1777
 
        #       |X|
1778
 
        #       D E  D reverts to B, E reverts to C
1779
 
        # This should conflict
1780
 
        builder = self.get_builder()
1781
 
        builder.build_snapshot('A-id', None,
1782
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1783
 
             ('add', (u'foo', 'foo-id', 'file', 'A content\n'))])
1784
 
        builder.build_snapshot('B-id', ['A-id'],
1785
 
            [('modify', ('foo-id', 'B content\n'))])
1786
 
        builder.build_snapshot('C-id', ['A-id'],
1787
 
            [('modify', ('foo-id', 'C content\n'))])
1788
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
1789
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1790
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1791
 
 
1792
 
        entries = list(merge_obj._entries_lca())
1793
 
        root_id = 'a-root-id'
1794
 
        self.assertEqual([('foo-id', True,
1795
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
1796
 
                           ((u'foo', [u'foo', u'foo']), u'foo', u'foo'),
1797
 
                           ((False, [False, False]), False, False)),
1798
 
                         ], entries)
1799
 
 
1800
 
    def test_different_lca_resolve_one_side_updates_content(self):
1801
 
        # Both sides converge, but then one side updates the text.
1802
 
        #       A    base, introduces 'foo'
1803
 
        #       |\
1804
 
        #       B C  B modifies 'foo', C modifies 'foo'
1805
 
        #       |X|
1806
 
        #       D E  D reverts to B, E reverts to C
1807
 
        #       |
1808
 
        #       F    F updates to a new value
1809
 
        # We need to emit an entry for 'foo', because D & E differed on the
1810
 
        # merge resolution
1811
 
        builder = self.get_builder()
1812
 
        builder.build_snapshot('A-id', None,
1813
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1814
 
             ('add', (u'foo', 'foo-id', 'file', 'A content\n'))])
1815
 
        builder.build_snapshot('B-id', ['A-id'],
1816
 
            [('modify', ('foo-id', 'B content\n'))])
1817
 
        builder.build_snapshot('C-id', ['A-id'],
1818
 
            [('modify', ('foo-id', 'C content\n'))])
1819
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
1820
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1821
 
        builder.build_snapshot('F-id', ['D-id'],
1822
 
            [('modify', ('foo-id', 'F content\n'))])
1823
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1824
 
 
1825
 
        entries = list(merge_obj._entries_lca())
1826
 
        root_id = 'a-root-id'
1827
 
        self.assertEqual([('foo-id', True,
1828
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
1829
 
                           ((u'foo', [u'foo', u'foo']), u'foo', u'foo'),
1830
 
                           ((False, [False, False]), False, False)),
1831
 
                         ], entries)
1832
 
 
1833
 
    def test_same_lca_resolution_one_side_updates_content(self):
1834
 
        # Both sides converge, but then one side updates the text.
1835
 
        #       A    base, introduces 'foo'
1836
 
        #       |\
1837
 
        #       B C  B modifies 'foo', C modifies 'foo'
1838
 
        #       |X|
1839
 
        #       D E  D and E use C's value
1840
 
        #       |
1841
 
        #       F    F updates to a new value
1842
 
        # I think it is a bug that this conflicts, but we don't have a way to
1843
 
        # detect otherwise. And because of:
1844
 
        #   test_different_lca_resolve_one_side_updates_content
1845
 
        # We need to conflict.
1846
 
 
1847
 
        builder = self.get_builder()
1848
 
        builder.build_snapshot('A-id', None,
1849
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1850
 
             ('add', (u'foo', 'foo-id', 'file', 'A content\n'))])
1851
 
        builder.build_snapshot('B-id', ['A-id'],
1852
 
            [('modify', ('foo-id', 'B content\n'))])
1853
 
        builder.build_snapshot('C-id', ['A-id'],
1854
 
            [('modify', ('foo-id', 'C content\n'))])
1855
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
1856
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'],
1857
 
            [('modify', ('foo-id', 'C content\n'))]) # Same as E
1858
 
        builder.build_snapshot('F-id', ['D-id'],
1859
 
            [('modify', ('foo-id', 'F content\n'))])
1860
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1861
 
 
1862
 
        entries = list(merge_obj._entries_lca())
1863
 
        self.expectFailure("We don't detect that LCA resolution was the"
1864
 
                           " same on both sides",
1865
 
            self.assertEqual, [], entries)
1866
 
 
1867
 
    def test_only_path_changed(self):
1868
 
        builder = self.get_builder()
1869
 
        builder.build_snapshot('A-id', None,
1870
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1871
 
             ('add', (u'a', 'a-id', 'file', 'content\n'))])
1872
 
        builder.build_snapshot('B-id', ['A-id'], [])
1873
 
        builder.build_snapshot('C-id', ['A-id'], [])
1874
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1875
 
            [('rename', (u'a', u'b'))])
1876
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1877
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1878
 
        entries = list(merge_obj._entries_lca())
1879
 
        root_id = 'a-root-id'
1880
 
        # The content was not changed, only the path
1881
 
        self.assertEqual([('a-id', False,
1882
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
1883
 
                           ((u'a', [u'a', u'a']), u'b', u'a'),
1884
 
                           ((False, [False, False]), False, False)),
1885
 
                         ], entries)
1886
 
 
1887
 
    def test_kind_changed(self):
1888
 
        # Identical content, except 'D' changes a-id into a directory
1889
 
        builder = self.get_builder()
1890
 
        builder.build_snapshot('A-id', None,
1891
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1892
 
             ('add', (u'a', 'a-id', 'file', 'content\n'))])
1893
 
        builder.build_snapshot('B-id', ['A-id'], [])
1894
 
        builder.build_snapshot('C-id', ['A-id'], [])
1895
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1896
 
            [('unversion', 'a-id'),
1897
 
             ('flush', None),
1898
 
             ('add', (u'a', 'a-id', 'directory', None))])
1899
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1900
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1901
 
        entries = list(merge_obj._entries_lca())
1902
 
        root_id = 'a-root-id'
1903
 
        # Only the kind was changed (content)
1904
 
        self.assertEqual([('a-id', True,
1905
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
1906
 
                           ((u'a', [u'a', u'a']), u'a', u'a'),
1907
 
                           ((False, [False, False]), False, False)),
1908
 
                         ], entries)
1909
 
 
1910
 
    def test_this_changed_kind(self):
1911
 
        # Identical content, but THIS changes a file to a directory
1912
 
        builder = self.get_builder()
1913
 
        builder.build_snapshot('A-id', None,
1914
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1915
 
             ('add', (u'a', 'a-id', 'file', 'content\n'))])
1916
 
        builder.build_snapshot('B-id', ['A-id'], [])
1917
 
        builder.build_snapshot('C-id', ['A-id'], [])
1918
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
1919
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'],
1920
 
            [('unversion', 'a-id'),
1921
 
             ('flush', None),
1922
 
             ('add', (u'a', 'a-id', 'directory', None))])
1923
 
        merge_obj = self.make_merge_obj(builder, 'E-id')
1924
 
        entries = list(merge_obj._entries_lca())
1925
 
        # Only the kind was changed (content)
1926
 
        self.assertEqual([], entries)
1927
 
 
1928
 
    def test_interesting_files(self):
1929
 
        # Two files modified, but we should filter one of them
1930
 
        builder = self.get_builder()
1931
 
        builder.build_snapshot('A-id', None,
1932
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1933
 
             ('add', (u'a', 'a-id', 'file', 'content\n')),
1934
 
             ('add', (u'b', 'b-id', 'file', 'content\n'))])
1935
 
        builder.build_snapshot('B-id', ['A-id'], [])
1936
 
        builder.build_snapshot('C-id', ['A-id'], [])
1937
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1938
 
            [('modify', ('a-id', 'new-content\n')),
1939
 
             ('modify', ('b-id', 'new-content\n'))])
1940
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1941
 
        merge_obj = self.make_merge_obj(builder, 'E-id',
1942
 
                                        interesting_files=['b'])
1943
 
        entries = list(merge_obj._entries_lca())
1944
 
        root_id = 'a-root-id'
1945
 
        self.assertEqual([('b-id', True,
1946
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
1947
 
                           ((u'b', [u'b', u'b']), u'b', u'b'),
1948
 
                           ((False, [False, False]), False, False)),
1949
 
                         ], entries)
1950
 
 
1951
 
    def test_interesting_file_in_this(self):
1952
 
        # This renamed the file, but it should still match the entry in other
1953
 
        builder = self.get_builder()
1954
 
        builder.build_snapshot('A-id', None,
1955
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1956
 
             ('add', (u'a', 'a-id', 'file', 'content\n')),
1957
 
             ('add', (u'b', 'b-id', 'file', 'content\n'))])
1958
 
        builder.build_snapshot('B-id', ['A-id'], [])
1959
 
        builder.build_snapshot('C-id', ['A-id'], [])
1960
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1961
 
            [('modify', ('a-id', 'new-content\n')),
1962
 
             ('modify', ('b-id', 'new-content\n'))])
1963
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'],
1964
 
            [('rename', ('b', 'c'))])
1965
 
        merge_obj = self.make_merge_obj(builder, 'E-id',
1966
 
                                        interesting_files=['c'])
1967
 
        entries = list(merge_obj._entries_lca())
1968
 
        root_id = 'a-root-id'
1969
 
        self.assertEqual([('b-id', True,
1970
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
1971
 
                           ((u'b', [u'b', u'b']), u'b', u'c'),
1972
 
                           ((False, [False, False]), False, False)),
1973
 
                         ], entries)
1974
 
 
1975
 
    def test_interesting_file_in_base(self):
1976
 
        # This renamed the file, but it should still match the entry in BASE
1977
 
        builder = self.get_builder()
1978
 
        builder.build_snapshot('A-id', None,
1979
 
            [('add', (u'', 'a-root-id', 'directory', None)),
1980
 
             ('add', (u'a', 'a-id', 'file', 'content\n')),
1981
 
             ('add', (u'c', 'c-id', 'file', 'content\n'))])
1982
 
        builder.build_snapshot('B-id', ['A-id'],
1983
 
            [('rename', ('c', 'b'))])
1984
 
        builder.build_snapshot('C-id', ['A-id'],
1985
 
            [('rename', ('c', 'b'))])
1986
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
1987
 
            [('modify', ('a-id', 'new-content\n')),
1988
 
             ('modify', ('c-id', 'new-content\n'))])
1989
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
1990
 
        merge_obj = self.make_merge_obj(builder, 'E-id',
1991
 
                                        interesting_files=['c'])
1992
 
        entries = list(merge_obj._entries_lca())
1993
 
        root_id = 'a-root-id'
1994
 
        self.assertEqual([('c-id', True,
1995
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
1996
 
                           ((u'c', [u'b', u'b']), u'b', u'b'),
1997
 
                           ((False, [False, False]), False, False)),
1998
 
                         ], entries)
1999
 
 
2000
 
    def test_interesting_file_in_lca(self):
2001
 
        # This renamed the file, but it should still match the entry in LCA
2002
 
        builder = self.get_builder()
2003
 
        builder.build_snapshot('A-id', None,
2004
 
            [('add', (u'', 'a-root-id', 'directory', None)),
2005
 
             ('add', (u'a', 'a-id', 'file', 'content\n')),
2006
 
             ('add', (u'b', 'b-id', 'file', 'content\n'))])
2007
 
        builder.build_snapshot('B-id', ['A-id'],
2008
 
            [('rename', ('b', 'c'))])
2009
 
        builder.build_snapshot('C-id', ['A-id'], [])
2010
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
2011
 
            [('modify', ('a-id', 'new-content\n')),
2012
 
             ('modify', ('b-id', 'new-content\n'))])
2013
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'],
2014
 
            [('rename', ('c', 'b'))])
2015
 
        merge_obj = self.make_merge_obj(builder, 'E-id',
2016
 
                                        interesting_files=['c'])
2017
 
        entries = list(merge_obj._entries_lca())
2018
 
        root_id = 'a-root-id'
2019
 
        self.assertEqual([('b-id', True,
2020
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
2021
 
                           ((u'b', [u'c', u'b']), u'b', u'b'),
2022
 
                           ((False, [False, False]), False, False)),
2023
 
                         ], entries)
2024
 
 
2025
 
    def test_interesting_ids(self):
2026
 
        # Two files modified, but we should filter one of them
2027
 
        builder = self.get_builder()
2028
 
        builder.build_snapshot('A-id', None,
2029
 
            [('add', (u'', 'a-root-id', 'directory', None)),
2030
 
             ('add', (u'a', 'a-id', 'file', 'content\n')),
2031
 
             ('add', (u'b', 'b-id', 'file', 'content\n'))])
2032
 
        builder.build_snapshot('B-id', ['A-id'], [])
2033
 
        builder.build_snapshot('C-id', ['A-id'], [])
2034
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
2035
 
            [('modify', ('a-id', 'new-content\n')),
2036
 
             ('modify', ('b-id', 'new-content\n'))])
2037
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
2038
 
        merge_obj = self.make_merge_obj(builder, 'E-id',
2039
 
                                        interesting_ids=['b-id'])
2040
 
        entries = list(merge_obj._entries_lca())
2041
 
        root_id = 'a-root-id'
2042
 
        self.assertEqual([('b-id', True,
2043
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
2044
 
                           ((u'b', [u'b', u'b']), u'b', u'b'),
2045
 
                           ((False, [False, False]), False, False)),
2046
 
                         ], entries)
2047
 
 
2048
 
 
2049
 
 
2050
 
class TestMergerEntriesLCAOnDisk(tests.TestCaseWithTransport):
2051
 
 
2052
 
    def get_builder(self):
2053
 
        builder = self.make_branch_builder('path')
2054
 
        builder.start_series()
2055
 
        self.addCleanup(builder.finish_series)
2056
 
        return builder
2057
 
 
2058
 
    def get_wt_from_builder(self, builder):
2059
 
        """Get a real WorkingTree from the builder."""
2060
 
        the_branch = builder.get_branch()
2061
 
        wt = the_branch.bzrdir.create_workingtree()
2062
 
        # Note: This is a little bit ugly, but we are holding the branch
2063
 
        #       write-locked as part of the build process, and we would like to
2064
 
        #       maintain that. So we just force the WT to re-use the same
2065
 
        #       branch object.
2066
 
        wt._branch = the_branch
2067
 
        wt.lock_write()
2068
 
        self.addCleanup(wt.unlock)
2069
 
        return wt
2070
 
 
2071
 
    def do_merge(self, builder, other_revision_id):
2072
 
        wt = self.get_wt_from_builder(builder)
2073
 
        merger = _mod_merge.Merger.from_revision_ids(None,
2074
 
            wt, other_revision_id)
2075
 
        merger.merge_type = _mod_merge.Merge3Merger
2076
 
        return wt, merger.do_merge()
2077
 
 
2078
 
    def test_simple_lca(self):
2079
 
        builder = self.get_builder()
2080
 
        builder.build_snapshot('A-id', None,
2081
 
            [('add', (u'', 'a-root-id', 'directory', None)),
2082
 
             ('add', (u'a', 'a-id', 'file', 'a\nb\nc\n'))])
2083
 
        builder.build_snapshot('C-id', ['A-id'], [])
2084
 
        builder.build_snapshot('B-id', ['A-id'], [])
2085
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
2086
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'],
2087
 
            [('modify', ('a-id', 'a\nb\nc\nd\ne\nf\n'))])
2088
 
        wt, conflicts = self.do_merge(builder, 'E-id')
2089
 
        self.assertEqual(0, conflicts)
2090
 
        # The merge should have simply update the contents of 'a'
2091
 
        self.assertEqual('a\nb\nc\nd\ne\nf\n', wt.get_file_text('a-id'))
2092
 
 
2093
 
    def test_conflict_without_lca(self):
2094
 
        # This test would cause a merge conflict, unless we use the lca trees
2095
 
        # to determine the real ancestry
2096
 
        #   A       Path at 'foo'
2097
 
        #  / \
2098
 
        # B   C     Path renamed to 'bar' in B
2099
 
        # |\ /|
2100
 
        # | X |
2101
 
        # |/ \|
2102
 
        # D   E     Path at 'bar' in D and E
2103
 
        #     |
2104
 
        #     F     Path at 'baz' in F, which supersedes 'bar' and 'foo'
2105
 
        builder = self.get_builder()
2106
 
        builder.build_snapshot('A-id', None,
2107
 
            [('add', (u'', 'a-root-id', 'directory', None)),
2108
 
             ('add', (u'foo', 'foo-id', 'file', 'a\nb\nc\n'))])
2109
 
        builder.build_snapshot('C-id', ['A-id'], [])
2110
 
        builder.build_snapshot('B-id', ['A-id'],
2111
 
            [('rename', ('foo', 'bar'))])
2112
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], # merge the rename
2113
 
            [('rename', ('foo', 'bar'))])
2114
 
        builder.build_snapshot('F-id', ['E-id'],
2115
 
            [('rename', ('bar', 'baz'))])
2116
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
2117
 
        wt, conflicts = self.do_merge(builder, 'F-id')
2118
 
        self.assertEqual(0, conflicts)
2119
 
        # The merge should simply recognize that the final rename takes
2120
 
        # precedence
2121
 
        self.assertEqual('baz', wt.id2path('foo-id'))
2122
 
 
2123
 
    def test_other_deletes_lca_renames(self):
2124
 
        # This test would cause a merge conflict, unless we use the lca trees
2125
 
        # to determine the real ancestry
2126
 
        #   A       Path at 'foo'
2127
 
        #  / \
2128
 
        # B   C     Path renamed to 'bar' in B
2129
 
        # |\ /|
2130
 
        # | X |
2131
 
        # |/ \|
2132
 
        # D   E     Path at 'bar' in D and E
2133
 
        #     |
2134
 
        #     F     F deletes 'bar'
2135
 
        builder = self.get_builder()
2136
 
        builder.build_snapshot('A-id', None,
2137
 
            [('add', (u'', 'a-root-id', 'directory', None)),
2138
 
             ('add', (u'foo', 'foo-id', 'file', 'a\nb\nc\n'))])
2139
 
        builder.build_snapshot('C-id', ['A-id'], [])
2140
 
        builder.build_snapshot('B-id', ['A-id'],
2141
 
            [('rename', ('foo', 'bar'))])
2142
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], # merge the rename
2143
 
            [('rename', ('foo', 'bar'))])
2144
 
        builder.build_snapshot('F-id', ['E-id'],
2145
 
            [('unversion', 'foo-id')])
2146
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
2147
 
        wt, conflicts = self.do_merge(builder, 'F-id')
2148
 
        self.assertEqual(0, conflicts)
2149
 
        self.assertRaises(errors.NoSuchId, wt.id2path, 'foo-id')
2150
 
 
2151
 
    def test_executable_changes(self):
2152
 
        #   A       Path at 'foo'
2153
 
        #  / \
2154
 
        # B   C
2155
 
        # |\ /|
2156
 
        # | X |
2157
 
        # |/ \|
2158
 
        # D   E
2159
 
        #     |
2160
 
        #     F     Executable bit changed
2161
 
        builder = self.get_builder()
2162
 
        builder.build_snapshot('A-id', None,
2163
 
            [('add', (u'', 'a-root-id', 'directory', None)),
2164
 
             ('add', (u'foo', 'foo-id', 'file', 'a\nb\nc\n'))])
2165
 
        builder.build_snapshot('C-id', ['A-id'], [])
2166
 
        builder.build_snapshot('B-id', ['A-id'], [])
2167
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
2168
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
2169
 
        # Have to use a real WT, because BranchBuilder doesn't support exec bit
2170
 
        wt = self.get_wt_from_builder(builder)
2171
 
        tt = transform.TreeTransform(wt)
2172
 
        try:
2173
 
            tt.set_executability(True, tt.trans_id_tree_file_id('foo-id'))
2174
 
            tt.apply()
2175
 
        except:
2176
 
            tt.finalize()
2177
 
            raise
2178
 
        self.assertTrue(wt.is_executable('foo-id'))
2179
 
        wt.commit('F-id', rev_id='F-id')
2180
 
        # Reset to D, so that we can merge F
2181
 
        wt.set_parent_ids(['D-id'])
2182
 
        wt.branch.set_last_revision_info(3, 'D-id')
2183
 
        wt.revert()
2184
 
        self.assertFalse(wt.is_executable('foo-id'))
2185
 
        conflicts = wt.merge_from_branch(wt.branch, to_revision='F-id')
2186
 
        self.assertEqual(0, conflicts)
2187
 
        self.assertTrue(wt.is_executable('foo-id'))
2188
 
 
2189
 
    def test_create_symlink(self):
2190
 
        self.requireFeature(features.SymlinkFeature)
2191
 
        #   A
2192
 
        #  / \
2193
 
        # B   C
2194
 
        # |\ /|
2195
 
        # | X |
2196
 
        # |/ \|
2197
 
        # D   E
2198
 
        #     |
2199
 
        #     F     Add a symlink 'foo' => 'bar'
2200
 
        # Have to use a real WT, because BranchBuilder and MemoryTree don't
2201
 
        # have symlink support
2202
 
        builder = self.get_builder()
2203
 
        builder.build_snapshot('A-id', None,
2204
 
            [('add', (u'', 'a-root-id', 'directory', None))])
2205
 
        builder.build_snapshot('C-id', ['A-id'], [])
2206
 
        builder.build_snapshot('B-id', ['A-id'], [])
2207
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
2208
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
2209
 
        # Have to use a real WT, because BranchBuilder doesn't support exec bit
2210
 
        wt = self.get_wt_from_builder(builder)
2211
 
        os.symlink('bar', 'path/foo')
2212
 
        wt.add(['foo'], ['foo-id'])
2213
 
        self.assertEqual('bar', wt.get_symlink_target('foo-id'))
2214
 
        wt.commit('add symlink', rev_id='F-id')
2215
 
        # Reset to D, so that we can merge F
2216
 
        wt.set_parent_ids(['D-id'])
2217
 
        wt.branch.set_last_revision_info(3, 'D-id')
2218
 
        wt.revert()
2219
 
        self.assertIs(None, wt.path2id('foo'))
2220
 
        conflicts = wt.merge_from_branch(wt.branch, to_revision='F-id')
2221
 
        self.assertEqual(0, conflicts)
2222
 
        self.assertEqual('foo-id', wt.path2id('foo'))
2223
 
        self.assertEqual('bar', wt.get_symlink_target('foo-id'))
2224
 
 
2225
 
    def test_both_sides_revert(self):
2226
 
        # Both sides of a criss-cross revert the text to the lca
2227
 
        #       A    base, introduces 'foo'
2228
 
        #       |\
2229
 
        #       B C  B modifies 'foo', C modifies 'foo'
2230
 
        #       |X|
2231
 
        #       D E  D reverts to B, E reverts to C
2232
 
        # This should conflict
2233
 
        # This must be done with a real WorkingTree, because normally their
2234
 
        # inventory contains "None" rather than a real sha1
2235
 
        builder = self.get_builder()
2236
 
        builder.build_snapshot('A-id', None,
2237
 
            [('add', (u'', 'a-root-id', 'directory', None)),
2238
 
             ('add', (u'foo', 'foo-id', 'file', 'A content\n'))])
2239
 
        builder.build_snapshot('B-id', ['A-id'],
2240
 
            [('modify', ('foo-id', 'B content\n'))])
2241
 
        builder.build_snapshot('C-id', ['A-id'],
2242
 
            [('modify', ('foo-id', 'C content\n'))])
2243
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'], [])
2244
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
2245
 
        wt, conflicts = self.do_merge(builder, 'E-id')
2246
 
        self.assertEqual(1, conflicts)
2247
 
        self.assertEqualDiff('<<<<<<< TREE\n'
2248
 
                             'B content\n'
2249
 
                             '=======\n'
2250
 
                             'C content\n'
2251
 
                             '>>>>>>> MERGE-SOURCE\n',
2252
 
                             wt.get_file_text('foo-id'))
2253
 
 
2254
 
    def test_modified_symlink(self):
2255
 
        self.requireFeature(features.SymlinkFeature)
2256
 
        #   A       Create symlink foo => bar
2257
 
        #  / \
2258
 
        # B   C     B relinks foo => baz
2259
 
        # |\ /|
2260
 
        # | X |
2261
 
        # |/ \|
2262
 
        # D   E     D & E have foo => baz
2263
 
        #     |
2264
 
        #     F     F changes it to bing
2265
 
        #
2266
 
        # Merging D & F should result in F cleanly overriding D, because D's
2267
 
        # value actually comes from B
2268
 
 
2269
 
        # Have to use a real WT, because BranchBuilder and MemoryTree don't
2270
 
        # have symlink support
2271
 
        wt = self.make_branch_and_tree('path')
2272
 
        wt.lock_write()
2273
 
        self.addCleanup(wt.unlock)
2274
 
        os.symlink('bar', 'path/foo')
2275
 
        wt.add(['foo'], ['foo-id'])
2276
 
        wt.commit('add symlink', rev_id='A-id')
2277
 
        os.remove('path/foo')
2278
 
        os.symlink('baz', 'path/foo')
2279
 
        wt.commit('foo => baz', rev_id='B-id')
2280
 
        wt.set_last_revision('A-id')
2281
 
        wt.branch.set_last_revision_info(1, 'A-id')
2282
 
        wt.revert()
2283
 
        wt.commit('C', rev_id='C-id')
2284
 
        wt.merge_from_branch(wt.branch, 'B-id')
2285
 
        self.assertEqual('baz', wt.get_symlink_target('foo-id'))
2286
 
        wt.commit('E merges C & B', rev_id='E-id')
2287
 
        os.remove('path/foo')
2288
 
        os.symlink('bing', 'path/foo')
2289
 
        wt.commit('F foo => bing', rev_id='F-id')
2290
 
        wt.set_last_revision('B-id')
2291
 
        wt.branch.set_last_revision_info(2, 'B-id')
2292
 
        wt.revert()
2293
 
        wt.merge_from_branch(wt.branch, 'C-id')
2294
 
        wt.commit('D merges B & C', rev_id='D-id')
2295
 
        conflicts = wt.merge_from_branch(wt.branch, to_revision='F-id')
2296
 
        self.assertEqual(0, conflicts)
2297
 
        self.assertEqual('bing', wt.get_symlink_target('foo-id'))
2298
 
 
2299
 
    def test_renamed_symlink(self):
2300
 
        self.requireFeature(features.SymlinkFeature)
2301
 
        #   A       Create symlink foo => bar
2302
 
        #  / \
2303
 
        # B   C     B renames foo => barry
2304
 
        # |\ /|
2305
 
        # | X |
2306
 
        # |/ \|
2307
 
        # D   E     D & E have barry
2308
 
        #     |
2309
 
        #     F     F renames barry to blah
2310
 
        #
2311
 
        # Merging D & F should result in F cleanly overriding D, because D's
2312
 
        # value actually comes from B
2313
 
 
2314
 
        wt = self.make_branch_and_tree('path')
2315
 
        wt.lock_write()
2316
 
        self.addCleanup(wt.unlock)
2317
 
        os.symlink('bar', 'path/foo')
2318
 
        wt.add(['foo'], ['foo-id'])
2319
 
        wt.commit('A add symlink', rev_id='A-id')
2320
 
        wt.rename_one('foo', 'barry')
2321
 
        wt.commit('B foo => barry', rev_id='B-id')
2322
 
        wt.set_last_revision('A-id')
2323
 
        wt.branch.set_last_revision_info(1, 'A-id')
2324
 
        wt.revert()
2325
 
        wt.commit('C', rev_id='C-id')
2326
 
        wt.merge_from_branch(wt.branch, 'B-id')
2327
 
        self.assertEqual('barry', wt.id2path('foo-id'))
2328
 
        self.assertEqual('bar', wt.get_symlink_target('foo-id'))
2329
 
        wt.commit('E merges C & B', rev_id='E-id')
2330
 
        wt.rename_one('barry', 'blah')
2331
 
        wt.commit('F barry => blah', rev_id='F-id')
2332
 
        wt.set_last_revision('B-id')
2333
 
        wt.branch.set_last_revision_info(2, 'B-id')
2334
 
        wt.revert()
2335
 
        wt.merge_from_branch(wt.branch, 'C-id')
2336
 
        wt.commit('D merges B & C', rev_id='D-id')
2337
 
        self.assertEqual('barry', wt.id2path('foo-id'))
2338
 
        # Check the output of the Merger object directly
2339
 
        merger = _mod_merge.Merger.from_revision_ids(None,
2340
 
            wt, 'F-id')
2341
 
        merger.merge_type = _mod_merge.Merge3Merger
2342
 
        merge_obj = merger.make_merger()
2343
 
        root_id = wt.path2id('')
2344
 
        entries = list(merge_obj._entries_lca())
2345
 
        # No content change, just a path change
2346
 
        self.assertEqual([('foo-id', False,
2347
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
2348
 
                           ((u'foo', [u'barry', u'foo']), u'blah', u'barry'),
2349
 
                           ((False, [False, False]), False, False)),
2350
 
                         ], entries)
2351
 
        conflicts = wt.merge_from_branch(wt.branch, to_revision='F-id')
2352
 
        self.assertEqual(0, conflicts)
2353
 
        self.assertEqual('blah', wt.id2path('foo-id'))
2354
 
 
2355
 
    def test_symlink_no_content_change(self):
2356
 
        self.requireFeature(features.SymlinkFeature)
2357
 
        #   A       Create symlink foo => bar
2358
 
        #  / \
2359
 
        # B   C     B relinks foo => baz
2360
 
        # |\ /|
2361
 
        # | X |
2362
 
        # |/ \|
2363
 
        # D   E     D & E have foo => baz
2364
 
        # |
2365
 
        # F         F has foo => bing
2366
 
        #
2367
 
        # Merging E into F should not cause a conflict, because E doesn't have
2368
 
        # a content change relative to the LCAs (it does relative to A)
2369
 
        wt = self.make_branch_and_tree('path')
2370
 
        wt.lock_write()
2371
 
        self.addCleanup(wt.unlock)
2372
 
        os.symlink('bar', 'path/foo')
2373
 
        wt.add(['foo'], ['foo-id'])
2374
 
        wt.commit('add symlink', rev_id='A-id')
2375
 
        os.remove('path/foo')
2376
 
        os.symlink('baz', 'path/foo')
2377
 
        wt.commit('foo => baz', rev_id='B-id')
2378
 
        wt.set_last_revision('A-id')
2379
 
        wt.branch.set_last_revision_info(1, 'A-id')
2380
 
        wt.revert()
2381
 
        wt.commit('C', rev_id='C-id')
2382
 
        wt.merge_from_branch(wt.branch, 'B-id')
2383
 
        self.assertEqual('baz', wt.get_symlink_target('foo-id'))
2384
 
        wt.commit('E merges C & B', rev_id='E-id')
2385
 
        wt.set_last_revision('B-id')
2386
 
        wt.branch.set_last_revision_info(2, 'B-id')
2387
 
        wt.revert()
2388
 
        wt.merge_from_branch(wt.branch, 'C-id')
2389
 
        wt.commit('D merges B & C', rev_id='D-id')
2390
 
        os.remove('path/foo')
2391
 
        os.symlink('bing', 'path/foo')
2392
 
        wt.commit('F foo => bing', rev_id='F-id')
2393
 
 
2394
 
        # Check the output of the Merger object directly
2395
 
        merger = _mod_merge.Merger.from_revision_ids(None,
2396
 
            wt, 'E-id')
2397
 
        merger.merge_type = _mod_merge.Merge3Merger
2398
 
        merge_obj = merger.make_merger()
2399
 
        # Nothing interesting happened in OTHER relative to BASE
2400
 
        self.assertEqual([], list(merge_obj._entries_lca()))
2401
 
        # Now do a real merge, just to test the rest of the stack
2402
 
        conflicts = wt.merge_from_branch(wt.branch, to_revision='E-id')
2403
 
        self.assertEqual(0, conflicts)
2404
 
        self.assertEqual('bing', wt.get_symlink_target('foo-id'))
2405
 
 
2406
 
    def test_symlink_this_changed_kind(self):
2407
 
        self.requireFeature(features.SymlinkFeature)
2408
 
        #   A       Nothing
2409
 
        #  / \
2410
 
        # B   C     B creates symlink foo => bar
2411
 
        # |\ /|
2412
 
        # | X |
2413
 
        # |/ \|
2414
 
        # D   E     D changes foo into a file, E has foo => bing
2415
 
        #
2416
 
        # Mostly, this is trying to test that we don't try to os.readlink() on
2417
 
        # a file, or when there is nothing there
2418
 
        wt = self.make_branch_and_tree('path')
2419
 
        wt.lock_write()
2420
 
        self.addCleanup(wt.unlock)
2421
 
        wt.commit('base', rev_id='A-id')
2422
 
        os.symlink('bar', 'path/foo')
2423
 
        wt.add(['foo'], ['foo-id'])
2424
 
        wt.commit('add symlink foo => bar', rev_id='B-id')
2425
 
        wt.set_last_revision('A-id')
2426
 
        wt.branch.set_last_revision_info(1, 'A-id')
2427
 
        wt.revert()
2428
 
        wt.commit('C', rev_id='C-id')
2429
 
        wt.merge_from_branch(wt.branch, 'B-id')
2430
 
        self.assertEqual('bar', wt.get_symlink_target('foo-id'))
2431
 
        os.remove('path/foo')
2432
 
        # We have to change the link in E, or it won't try to do a comparison
2433
 
        os.symlink('bing', 'path/foo')
2434
 
        wt.commit('E merges C & B, overrides to bing', rev_id='E-id')
2435
 
        wt.set_last_revision('B-id')
2436
 
        wt.branch.set_last_revision_info(2, 'B-id')
2437
 
        wt.revert()
2438
 
        wt.merge_from_branch(wt.branch, 'C-id')
2439
 
        os.remove('path/foo')
2440
 
        self.build_tree_contents([('path/foo', 'file content\n')])
2441
 
        # XXX: workaround, WT doesn't detect kind changes unless you do
2442
 
        # iter_changes()
2443
 
        list(wt.iter_changes(wt.basis_tree()))
2444
 
        wt.commit('D merges B & C, makes it a file', rev_id='D-id')
2445
 
 
2446
 
        merger = _mod_merge.Merger.from_revision_ids(None,
2447
 
            wt, 'E-id')
2448
 
        merger.merge_type = _mod_merge.Merge3Merger
2449
 
        merge_obj = merger.make_merger()
2450
 
        entries = list(merge_obj._entries_lca())
2451
 
        root_id = wt.path2id('')
2452
 
        self.assertEqual([('foo-id', True,
2453
 
                           ((None, [root_id, None]), root_id, root_id),
2454
 
                           ((None, [u'foo', None]), u'foo', u'foo'),
2455
 
                           ((None, [False, None]), False, False)),
2456
 
                         ], entries)
2457
 
 
2458
 
    def test_symlink_all_wt(self):
2459
 
        """Check behavior if all trees are Working Trees."""
2460
 
        self.requireFeature(features.SymlinkFeature)
2461
 
        # The big issue is that entry.symlink_target is None for WorkingTrees.
2462
 
        # So we need to make sure we handle that case correctly.
2463
 
        #   A   foo => bar
2464
 
        #   |\
2465
 
        #   B C B relinks foo => baz
2466
 
        #   |X|
2467
 
        #   D E D & E have foo => baz
2468
 
        #     |
2469
 
        #     F F changes it to bing
2470
 
        # Merging D & F should result in F cleanly overriding D, because D's
2471
 
        # value actually comes from B
2472
 
 
2473
 
        wt = self.make_branch_and_tree('path')
2474
 
        wt.lock_write()
2475
 
        self.addCleanup(wt.unlock)
2476
 
        os.symlink('bar', 'path/foo')
2477
 
        wt.add(['foo'], ['foo-id'])
2478
 
        wt.commit('add symlink', rev_id='A-id')
2479
 
        os.remove('path/foo')
2480
 
        os.symlink('baz', 'path/foo')
2481
 
        wt.commit('foo => baz', rev_id='B-id')
2482
 
        wt.set_last_revision('A-id')
2483
 
        wt.branch.set_last_revision_info(1, 'A-id')
2484
 
        wt.revert()
2485
 
        wt.commit('C', rev_id='C-id')
2486
 
        wt.merge_from_branch(wt.branch, 'B-id')
2487
 
        self.assertEqual('baz', wt.get_symlink_target('foo-id'))
2488
 
        wt.commit('E merges C & B', rev_id='E-id')
2489
 
        os.remove('path/foo')
2490
 
        os.symlink('bing', 'path/foo')
2491
 
        wt.commit('F foo => bing', rev_id='F-id')
2492
 
        wt.set_last_revision('B-id')
2493
 
        wt.branch.set_last_revision_info(2, 'B-id')
2494
 
        wt.revert()
2495
 
        wt.merge_from_branch(wt.branch, 'C-id')
2496
 
        wt.commit('D merges B & C', rev_id='D-id')
2497
 
        wt_base = wt.bzrdir.sprout('base', 'A-id').open_workingtree()
2498
 
        wt_base.lock_read()
2499
 
        self.addCleanup(wt_base.unlock)
2500
 
        wt_lca1 = wt.bzrdir.sprout('b-tree', 'B-id').open_workingtree()
2501
 
        wt_lca1.lock_read()
2502
 
        self.addCleanup(wt_lca1.unlock)
2503
 
        wt_lca2 = wt.bzrdir.sprout('c-tree', 'C-id').open_workingtree()
2504
 
        wt_lca2.lock_read()
2505
 
        self.addCleanup(wt_lca2.unlock)
2506
 
        wt_other = wt.bzrdir.sprout('other', 'F-id').open_workingtree()
2507
 
        wt_other.lock_read()
2508
 
        self.addCleanup(wt_other.unlock)
2509
 
        merge_obj = _mod_merge.Merge3Merger(wt, wt, wt_base,
2510
 
            wt_other, lca_trees=[wt_lca1, wt_lca2], do_merge=False)
2511
 
        entries = list(merge_obj._entries_lca())
2512
 
        root_id = wt.path2id('')
2513
 
        self.assertEqual([('foo-id', True,
2514
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
2515
 
                           ((u'foo', [u'foo', u'foo']), u'foo', u'foo'),
2516
 
                           ((False, [False, False]), False, False)),
2517
 
                         ], entries)
2518
 
 
2519
 
    def test_other_reverted_path_to_base(self):
2520
 
        #   A       Path at 'foo'
2521
 
        #  / \
2522
 
        # B   C     Path at 'bar' in B
2523
 
        # |\ /|
2524
 
        # | X |
2525
 
        # |/ \|
2526
 
        # D   E     Path at 'bar'
2527
 
        #     |
2528
 
        #     F     Path at 'foo'
2529
 
        builder = self.get_builder()
2530
 
        builder.build_snapshot('A-id', None,
2531
 
            [('add', (u'', 'a-root-id', 'directory', None)),
2532
 
             ('add', (u'foo', 'foo-id', 'file', 'a\nb\nc\n'))])
2533
 
        builder.build_snapshot('C-id', ['A-id'], [])
2534
 
        builder.build_snapshot('B-id', ['A-id'],
2535
 
            [('rename', ('foo', 'bar'))])
2536
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
2537
 
            [('rename', ('foo', 'bar'))]) # merge the rename
2538
 
        builder.build_snapshot('F-id', ['E-id'],
2539
 
            [('rename', ('bar', 'foo'))]) # Rename back to BASE
2540
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
2541
 
        wt, conflicts = self.do_merge(builder, 'F-id')
2542
 
        self.assertEqual(0, conflicts)
2543
 
        self.assertEqual('foo', wt.id2path('foo-id'))
2544
 
 
2545
 
    def test_other_reverted_content_to_base(self):
2546
 
        builder = self.get_builder()
2547
 
        builder.build_snapshot('A-id', None,
2548
 
            [('add', (u'', 'a-root-id', 'directory', None)),
2549
 
             ('add', (u'foo', 'foo-id', 'file', 'base content\n'))])
2550
 
        builder.build_snapshot('C-id', ['A-id'], [])
2551
 
        builder.build_snapshot('B-id', ['A-id'],
2552
 
            [('modify', ('foo-id', 'B content\n'))])
2553
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
2554
 
            [('modify', ('foo-id', 'B content\n'))]) # merge the content
2555
 
        builder.build_snapshot('F-id', ['E-id'],
2556
 
            [('modify', ('foo-id', 'base content\n'))]) # Revert back to BASE
2557
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
2558
 
        wt, conflicts = self.do_merge(builder, 'F-id')
2559
 
        self.assertEqual(0, conflicts)
2560
 
        # TODO: We need to use the per-file graph to properly select a BASE
2561
 
        #       before this will work. Or at least use the LCA trees to find
2562
 
        #       the appropriate content base. (which is B, not A).
2563
 
        self.assertEqual('base content\n', wt.get_file_text('foo-id'))
2564
 
 
2565
 
    def test_other_modified_content(self):
2566
 
        builder = self.get_builder()
2567
 
        builder.build_snapshot('A-id', None,
2568
 
            [('add', (u'', 'a-root-id', 'directory', None)),
2569
 
             ('add', (u'foo', 'foo-id', 'file', 'base content\n'))])
2570
 
        builder.build_snapshot('C-id', ['A-id'], [])
2571
 
        builder.build_snapshot('B-id', ['A-id'],
2572
 
            [('modify', ('foo-id', 'B content\n'))])
2573
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
2574
 
            [('modify', ('foo-id', 'B content\n'))]) # merge the content
2575
 
        builder.build_snapshot('F-id', ['E-id'],
2576
 
            [('modify', ('foo-id', 'F content\n'))]) # Override B content
2577
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
2578
 
        wt, conflicts = self.do_merge(builder, 'F-id')
2579
 
        self.assertEqual(0, conflicts)
2580
 
        self.assertEqual('F content\n', wt.get_file_text('foo-id'))
2581
 
 
2582
 
    def test_all_wt(self):
2583
 
        """Check behavior if all trees are Working Trees."""
2584
 
        # The big issue is that entry.revision is None for WorkingTrees. (as is
2585
 
        # entry.text_sha1, etc. So we need to make sure we handle that case
2586
 
        # correctly.
2587
 
        #   A   Content of 'foo', path of 'a'
2588
 
        #   |\
2589
 
        #   B C B modifies content, C renames 'a' => 'b'
2590
 
        #   |X|
2591
 
        #   D E E updates content, renames 'b' => 'c'
2592
 
        builder = self.get_builder()
2593
 
        builder.build_snapshot('A-id', None,
2594
 
            [('add', (u'', 'a-root-id', 'directory', None)),
2595
 
             ('add', (u'a', 'a-id', 'file', 'base content\n')),
2596
 
             ('add', (u'foo', 'foo-id', 'file', 'base content\n'))])
2597
 
        builder.build_snapshot('B-id', ['A-id'],
2598
 
            [('modify', ('foo-id', 'B content\n'))])
2599
 
        builder.build_snapshot('C-id', ['A-id'],
2600
 
            [('rename', ('a', 'b'))])
2601
 
        builder.build_snapshot('E-id', ['C-id', 'B-id'],
2602
 
            [('rename', ('b', 'c')),
2603
 
             ('modify', ('foo-id', 'E content\n'))])
2604
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'],
2605
 
            [('rename', ('a', 'b'))]) # merged change
2606
 
        wt_this = self.get_wt_from_builder(builder)
2607
 
        wt_base = wt_this.bzrdir.sprout('base', 'A-id').open_workingtree()
2608
 
        wt_base.lock_read()
2609
 
        self.addCleanup(wt_base.unlock)
2610
 
        wt_lca1 = wt_this.bzrdir.sprout('b-tree', 'B-id').open_workingtree()
2611
 
        wt_lca1.lock_read()
2612
 
        self.addCleanup(wt_lca1.unlock)
2613
 
        wt_lca2 = wt_this.bzrdir.sprout('c-tree', 'C-id').open_workingtree()
2614
 
        wt_lca2.lock_read()
2615
 
        self.addCleanup(wt_lca2.unlock)
2616
 
        wt_other = wt_this.bzrdir.sprout('other', 'E-id').open_workingtree()
2617
 
        wt_other.lock_read()
2618
 
        self.addCleanup(wt_other.unlock)
2619
 
        merge_obj = _mod_merge.Merge3Merger(wt_this, wt_this, wt_base,
2620
 
            wt_other, lca_trees=[wt_lca1, wt_lca2], do_merge=False)
2621
 
        entries = list(merge_obj._entries_lca())
2622
 
        root_id = 'a-root-id'
2623
 
        self.assertEqual([('a-id', False,
2624
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
2625
 
                           ((u'a', [u'a', u'b']), u'c', u'b'),
2626
 
                           ((False, [False, False]), False, False)),
2627
 
                          ('foo-id', True,
2628
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
2629
 
                           ((u'foo', [u'foo', u'foo']), u'foo', u'foo'),
2630
 
                           ((False, [False, False]), False, False)),
2631
 
                         ], entries)
2632
 
 
2633
 
    def test_nested_tree_unmodified(self):
2634
 
        # Tested with a real WT, because BranchBuilder/MemoryTree don't handle
2635
 
        # 'tree-reference'
2636
 
        wt = self.make_branch_and_tree('tree',
2637
 
            format='development-subtree')
2638
 
        wt.lock_write()
2639
 
        self.addCleanup(wt.unlock)
2640
 
        sub_tree = self.make_branch_and_tree('tree/sub-tree',
2641
 
            format='development-subtree')
2642
 
        wt.set_root_id('a-root-id')
2643
 
        sub_tree.set_root_id('sub-tree-root')
2644
 
        self.build_tree_contents([('tree/sub-tree/file', 'text1')])
2645
 
        sub_tree.add('file')
2646
 
        sub_tree.commit('foo', rev_id='sub-A-id')
2647
 
        wt.add_reference(sub_tree)
2648
 
        wt.commit('set text to 1', rev_id='A-id', recursive=None)
2649
 
        # Now create a criss-cross merge in the parent, without modifying the
2650
 
        # subtree
2651
 
        wt.commit('B', rev_id='B-id', recursive=None)
2652
 
        wt.set_last_revision('A-id')
2653
 
        wt.branch.set_last_revision_info(1, 'A-id')
2654
 
        wt.commit('C', rev_id='C-id', recursive=None)
2655
 
        wt.merge_from_branch(wt.branch, to_revision='B-id')
2656
 
        wt.commit('E', rev_id='E-id', recursive=None)
2657
 
        wt.set_parent_ids(['B-id', 'C-id'])
2658
 
        wt.branch.set_last_revision_info(2, 'B-id')
2659
 
        wt.commit('D', rev_id='D-id', recursive=None)
2660
 
 
2661
 
        merger = _mod_merge.Merger.from_revision_ids(None,
2662
 
            wt, 'E-id')
2663
 
        merger.merge_type = _mod_merge.Merge3Merger
2664
 
        merge_obj = merger.make_merger()
2665
 
        entries = list(merge_obj._entries_lca())
2666
 
        self.assertEqual([], entries)
2667
 
 
2668
 
    def test_nested_tree_subtree_modified(self):
2669
 
        # Tested with a real WT, because BranchBuilder/MemoryTree don't handle
2670
 
        # 'tree-reference'
2671
 
        wt = self.make_branch_and_tree('tree',
2672
 
            format='development-subtree')
2673
 
        wt.lock_write()
2674
 
        self.addCleanup(wt.unlock)
2675
 
        sub_tree = self.make_branch_and_tree('tree/sub',
2676
 
            format='development-subtree')
2677
 
        wt.set_root_id('a-root-id')
2678
 
        sub_tree.set_root_id('sub-tree-root')
2679
 
        self.build_tree_contents([('tree/sub/file', 'text1')])
2680
 
        sub_tree.add('file')
2681
 
        sub_tree.commit('foo', rev_id='sub-A-id')
2682
 
        wt.add_reference(sub_tree)
2683
 
        wt.commit('set text to 1', rev_id='A-id', recursive=None)
2684
 
        # Now create a criss-cross merge in the parent, without modifying the
2685
 
        # subtree
2686
 
        wt.commit('B', rev_id='B-id', recursive=None)
2687
 
        wt.set_last_revision('A-id')
2688
 
        wt.branch.set_last_revision_info(1, 'A-id')
2689
 
        wt.commit('C', rev_id='C-id', recursive=None)
2690
 
        wt.merge_from_branch(wt.branch, to_revision='B-id')
2691
 
        self.build_tree_contents([('tree/sub/file', 'text2')])
2692
 
        sub_tree.commit('modify contents', rev_id='sub-B-id')
2693
 
        wt.commit('E', rev_id='E-id', recursive=None)
2694
 
        wt.set_parent_ids(['B-id', 'C-id'])
2695
 
        wt.branch.set_last_revision_info(2, 'B-id')
2696
 
        wt.commit('D', rev_id='D-id', recursive=None)
2697
 
 
2698
 
        merger = _mod_merge.Merger.from_revision_ids(None,
2699
 
            wt, 'E-id')
2700
 
        merger.merge_type = _mod_merge.Merge3Merger
2701
 
        merge_obj = merger.make_merger()
2702
 
        entries = list(merge_obj._entries_lca())
2703
 
        # Nothing interesting about this sub-tree, because content changes are
2704
 
        # computed at a higher level
2705
 
        self.assertEqual([], entries)
2706
 
 
2707
 
    def test_nested_tree_subtree_renamed(self):
2708
 
        # Tested with a real WT, because BranchBuilder/MemoryTree don't handle
2709
 
        # 'tree-reference'
2710
 
        wt = self.make_branch_and_tree('tree',
2711
 
            format='development-subtree')
2712
 
        wt.lock_write()
2713
 
        self.addCleanup(wt.unlock)
2714
 
        sub_tree = self.make_branch_and_tree('tree/sub',
2715
 
            format='development-subtree')
2716
 
        wt.set_root_id('a-root-id')
2717
 
        sub_tree.set_root_id('sub-tree-root')
2718
 
        self.build_tree_contents([('tree/sub/file', 'text1')])
2719
 
        sub_tree.add('file')
2720
 
        sub_tree.commit('foo', rev_id='sub-A-id')
2721
 
        wt.add_reference(sub_tree)
2722
 
        wt.commit('set text to 1', rev_id='A-id', recursive=None)
2723
 
        # Now create a criss-cross merge in the parent, without modifying the
2724
 
        # subtree
2725
 
        wt.commit('B', rev_id='B-id', recursive=None)
2726
 
        wt.set_last_revision('A-id')
2727
 
        wt.branch.set_last_revision_info(1, 'A-id')
2728
 
        wt.commit('C', rev_id='C-id', recursive=None)
2729
 
        wt.merge_from_branch(wt.branch, to_revision='B-id')
2730
 
        wt.rename_one('sub', 'alt_sub')
2731
 
        wt.commit('E', rev_id='E-id', recursive=None)
2732
 
        wt.set_last_revision('B-id')
2733
 
        wt.revert()
2734
 
        wt.set_parent_ids(['B-id', 'C-id'])
2735
 
        wt.branch.set_last_revision_info(2, 'B-id')
2736
 
        wt.commit('D', rev_id='D-id', recursive=None)
2737
 
 
2738
 
        merger = _mod_merge.Merger.from_revision_ids(None,
2739
 
            wt, 'E-id')
2740
 
        merger.merge_type = _mod_merge.Merge3Merger
2741
 
        merge_obj = merger.make_merger()
2742
 
        entries = list(merge_obj._entries_lca())
2743
 
        root_id = 'a-root-id'
2744
 
        self.assertEqual([('sub-tree-root', False,
2745
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
2746
 
                           ((u'sub', [u'sub', u'sub']), u'alt_sub', u'sub'),
2747
 
                           ((False, [False, False]), False, False)),
2748
 
                         ], entries)
2749
 
 
2750
 
    def test_nested_tree_subtree_renamed_and_modified(self):
2751
 
        # Tested with a real WT, because BranchBuilder/MemoryTree don't handle
2752
 
        # 'tree-reference'
2753
 
        wt = self.make_branch_and_tree('tree',
2754
 
            format='development-subtree')
2755
 
        wt.lock_write()
2756
 
        self.addCleanup(wt.unlock)
2757
 
        sub_tree = self.make_branch_and_tree('tree/sub',
2758
 
            format='development-subtree')
2759
 
        wt.set_root_id('a-root-id')
2760
 
        sub_tree.set_root_id('sub-tree-root')
2761
 
        self.build_tree_contents([('tree/sub/file', 'text1')])
2762
 
        sub_tree.add('file')
2763
 
        sub_tree.commit('foo', rev_id='sub-A-id')
2764
 
        wt.add_reference(sub_tree)
2765
 
        wt.commit('set text to 1', rev_id='A-id', recursive=None)
2766
 
        # Now create a criss-cross merge in the parent, without modifying the
2767
 
        # subtree
2768
 
        wt.commit('B', rev_id='B-id', recursive=None)
2769
 
        wt.set_last_revision('A-id')
2770
 
        wt.branch.set_last_revision_info(1, 'A-id')
2771
 
        wt.commit('C', rev_id='C-id', recursive=None)
2772
 
        wt.merge_from_branch(wt.branch, to_revision='B-id')
2773
 
        self.build_tree_contents([('tree/sub/file', 'text2')])
2774
 
        sub_tree.commit('modify contents', rev_id='sub-B-id')
2775
 
        wt.rename_one('sub', 'alt_sub')
2776
 
        wt.commit('E', rev_id='E-id', recursive=None)
2777
 
        wt.set_last_revision('B-id')
2778
 
        wt.revert()
2779
 
        wt.set_parent_ids(['B-id', 'C-id'])
2780
 
        wt.branch.set_last_revision_info(2, 'B-id')
2781
 
        wt.commit('D', rev_id='D-id', recursive=None)
2782
 
 
2783
 
        merger = _mod_merge.Merger.from_revision_ids(None,
2784
 
            wt, 'E-id')
2785
 
        merger.merge_type = _mod_merge.Merge3Merger
2786
 
        merge_obj = merger.make_merger()
2787
 
        entries = list(merge_obj._entries_lca())
2788
 
        root_id = 'a-root-id'
2789
 
        self.assertEqual([('sub-tree-root', False,
2790
 
                           ((root_id, [root_id, root_id]), root_id, root_id),
2791
 
                           ((u'sub', [u'sub', u'sub']), u'alt_sub', u'sub'),
2792
 
                           ((False, [False, False]), False, False)),
2793
 
                         ], entries)
2794
 
 
2795
 
 
2796
 
class TestLCAMultiWay(tests.TestCase):
2797
 
 
2798
 
    def assertLCAMultiWay(self, expected, base, lcas, other, this,
2799
 
                          allow_overriding_lca=True):
2800
 
        self.assertEqual(expected, _mod_merge.Merge3Merger._lca_multi_way(
2801
 
                                (base, lcas), other, this,
2802
 
                                allow_overriding_lca=allow_overriding_lca))
2803
 
 
2804
 
    def test_other_equal_equal_lcas(self):
2805
 
        """Test when OTHER=LCA and all LCAs are identical."""
2806
 
        self.assertLCAMultiWay('this',
2807
 
            'bval', ['bval', 'bval'], 'bval', 'bval')
2808
 
        self.assertLCAMultiWay('this',
2809
 
            'bval', ['lcaval', 'lcaval'], 'lcaval', 'bval')
2810
 
        self.assertLCAMultiWay('this',
2811
 
            'bval', ['lcaval', 'lcaval', 'lcaval'], 'lcaval', 'bval')
2812
 
        self.assertLCAMultiWay('this',
2813
 
            'bval', ['lcaval', 'lcaval', 'lcaval'], 'lcaval', 'tval')
2814
 
        self.assertLCAMultiWay('this',
2815
 
            'bval', ['lcaval', 'lcaval', 'lcaval'], 'lcaval', None)
2816
 
 
2817
 
    def test_other_equal_this(self):
2818
 
        """Test when other and this are identical."""
2819
 
        self.assertLCAMultiWay('this',
2820
 
            'bval', ['bval', 'bval'], 'oval', 'oval')
2821
 
        self.assertLCAMultiWay('this',
2822
 
            'bval', ['lcaval', 'lcaval'], 'oval', 'oval')
2823
 
        self.assertLCAMultiWay('this',
2824
 
            'bval', ['cval', 'dval'], 'oval', 'oval')
2825
 
        self.assertLCAMultiWay('this',
2826
 
            'bval', [None, 'lcaval'], 'oval', 'oval')
2827
 
        self.assertLCAMultiWay('this',
2828
 
            None, [None, 'lcaval'], 'oval', 'oval')
2829
 
        self.assertLCAMultiWay('this',
2830
 
            None, ['lcaval', 'lcaval'], 'oval', 'oval')
2831
 
        self.assertLCAMultiWay('this',
2832
 
            None, ['cval', 'dval'], 'oval', 'oval')
2833
 
        self.assertLCAMultiWay('this',
2834
 
            None, ['cval', 'dval'], None, None)
2835
 
        self.assertLCAMultiWay('this',
2836
 
            None, ['cval', 'dval', 'eval', 'fval'], 'oval', 'oval')
2837
 
 
2838
 
    def test_no_lcas(self):
2839
 
        self.assertLCAMultiWay('this',
2840
 
            'bval', [], 'bval', 'tval')
2841
 
        self.assertLCAMultiWay('other',
2842
 
            'bval', [], 'oval', 'bval')
2843
 
        self.assertLCAMultiWay('conflict',
2844
 
            'bval', [], 'oval', 'tval')
2845
 
        self.assertLCAMultiWay('this',
2846
 
            'bval', [], 'oval', 'oval')
2847
 
 
2848
 
    def test_lca_supersedes_other_lca(self):
2849
 
        """If one lca == base, the other lca takes precedence"""
2850
 
        self.assertLCAMultiWay('this',
2851
 
            'bval', ['bval', 'lcaval'], 'lcaval', 'tval')
2852
 
        self.assertLCAMultiWay('this',
2853
 
            'bval', ['bval', 'lcaval'], 'lcaval', 'bval')
2854
 
        # This is actually considered a 'revert' because the 'lcaval' in LCAS
2855
 
        # supersedes the BASE val (in the other LCA) but then OTHER reverts it
2856
 
        # back to bval.
2857
 
        self.assertLCAMultiWay('other',
2858
 
            'bval', ['bval', 'lcaval'], 'bval', 'lcaval')
2859
 
        self.assertLCAMultiWay('conflict',
2860
 
            'bval', ['bval', 'lcaval'], 'bval', 'tval')
2861
 
 
2862
 
    def test_other_and_this_pick_different_lca(self):
2863
 
        # OTHER and THIS resolve the lca conflict in different ways
2864
 
        self.assertLCAMultiWay('conflict',
2865
 
            'bval', ['lca1val', 'lca2val'], 'lca1val', 'lca2val')
2866
 
        self.assertLCAMultiWay('conflict',
2867
 
            'bval', ['lca1val', 'lca2val', 'lca3val'], 'lca1val', 'lca2val')
2868
 
        self.assertLCAMultiWay('conflict',
2869
 
            'bval', ['lca1val', 'lca2val', 'bval'], 'lca1val', 'lca2val')
2870
 
 
2871
 
    def test_other_in_lca(self):
2872
 
        # OTHER takes a value of one of the LCAs, THIS takes a new value, which
2873
 
        # theoretically supersedes both LCA values and 'wins'
2874
 
        self.assertLCAMultiWay('this',
2875
 
            'bval', ['lca1val', 'lca2val'], 'lca1val', 'newval')
2876
 
        self.assertLCAMultiWay('this',
2877
 
            'bval', ['lca1val', 'lca2val', 'lca3val'], 'lca1val', 'newval')
2878
 
        self.assertLCAMultiWay('conflict',
2879
 
            'bval', ['lca1val', 'lca2val'], 'lca1val', 'newval',
2880
 
            allow_overriding_lca=False)
2881
 
        self.assertLCAMultiWay('conflict',
2882
 
            'bval', ['lca1val', 'lca2val', 'lca3val'], 'lca1val', 'newval',
2883
 
            allow_overriding_lca=False)
2884
 
        # THIS reverted back to BASE, but that is an explicit supersede of all
2885
 
        # LCAs
2886
 
        self.assertLCAMultiWay('this',
2887
 
            'bval', ['lca1val', 'lca2val', 'lca3val'], 'lca1val', 'bval')
2888
 
        self.assertLCAMultiWay('this',
2889
 
            'bval', ['lca1val', 'lca2val', 'bval'], 'lca1val', 'bval')
2890
 
        self.assertLCAMultiWay('conflict',
2891
 
            'bval', ['lca1val', 'lca2val', 'lca3val'], 'lca1val', 'bval',
2892
 
            allow_overriding_lca=False)
2893
 
        self.assertLCAMultiWay('conflict',
2894
 
            'bval', ['lca1val', 'lca2val', 'bval'], 'lca1val', 'bval',
2895
 
            allow_overriding_lca=False)
2896
 
 
2897
 
    def test_this_in_lca(self):
2898
 
        # THIS takes a value of one of the LCAs, OTHER takes a new value, which
2899
 
        # theoretically supersedes both LCA values and 'wins'
2900
 
        self.assertLCAMultiWay('other',
2901
 
            'bval', ['lca1val', 'lca2val'], 'oval', 'lca1val')
2902
 
        self.assertLCAMultiWay('other',
2903
 
            'bval', ['lca1val', 'lca2val'], 'oval', 'lca2val')
2904
 
        self.assertLCAMultiWay('conflict',
2905
 
            'bval', ['lca1val', 'lca2val'], 'oval', 'lca1val',
2906
 
            allow_overriding_lca=False)
2907
 
        self.assertLCAMultiWay('conflict',
2908
 
            'bval', ['lca1val', 'lca2val'], 'oval', 'lca2val',
2909
 
            allow_overriding_lca=False)
2910
 
        # OTHER reverted back to BASE, but that is an explicit supersede of all
2911
 
        # LCAs
2912
 
        self.assertLCAMultiWay('other',
2913
 
            'bval', ['lca1val', 'lca2val', 'lca3val'], 'bval', 'lca3val')
2914
 
        self.assertLCAMultiWay('conflict',
2915
 
            'bval', ['lca1val', 'lca2val', 'lca3val'], 'bval', 'lca3val',
2916
 
            allow_overriding_lca=False)
2917
 
 
2918
 
    def test_all_differ(self):
2919
 
        self.assertLCAMultiWay('conflict',
2920
 
            'bval', ['lca1val', 'lca2val'], 'oval', 'tval')
2921
 
        self.assertLCAMultiWay('conflict',
2922
 
            'bval', ['lca1val', 'lca2val', 'lca2val'], 'oval', 'tval')
2923
 
        self.assertLCAMultiWay('conflict',
2924
 
            'bval', ['lca1val', 'lca2val', 'lca3val'], 'oval', 'tval')
2925
 
 
2926
 
 
2927
 
class TestConfigurableFileMerger(tests.TestCaseWithTransport):
2928
 
 
2929
 
    def setUp(self):
2930
 
        super(TestConfigurableFileMerger, self).setUp()
2931
 
        self.calls = []
2932
 
 
2933
 
    def get_merger_factory(self):
2934
 
        # Allows  the inner methods to access the test attributes
2935
 
        calls = self.calls
2936
 
 
2937
 
        class FooMerger(_mod_merge.ConfigurableFileMerger):
2938
 
            name_prefix = "foo"
2939
 
            default_files = ['bar']
2940
 
 
2941
 
            def merge_text(self, params):
2942
 
                calls.append('merge_text')
2943
 
                return ('not_applicable', None)
2944
 
 
2945
 
        def factory(merger):
2946
 
            result = FooMerger(merger)
2947
 
            # Make sure we start with a clean slate
2948
 
            self.assertEqual(None, result.affected_files)
2949
 
            # Track the original merger
2950
 
            self.merger = result
2951
 
            return result
2952
 
 
2953
 
        return factory
2954
 
 
2955
 
    def _install_hook(self, factory):
2956
 
        _mod_merge.Merger.hooks.install_named_hook('merge_file_content',
2957
 
                                                   factory, 'test factory')
2958
 
 
2959
 
    def make_builder(self):
2960
 
        builder = test_merge_core.MergeBuilder(self.test_base_dir)
2961
 
        self.addCleanup(builder.cleanup)
2962
 
        return builder
2963
 
 
2964
 
    def make_text_conflict(self, file_name='bar'):
2965
 
        factory = self.get_merger_factory()
2966
 
        self._install_hook(factory)
2967
 
        builder = self.make_builder()
2968
 
        builder.add_file('bar-id', builder.tree_root, file_name, 'text1', True)
2969
 
        builder.change_contents('bar-id', other='text4', this='text3')
2970
 
        return builder
2971
 
 
2972
 
    def make_kind_change(self):
2973
 
        factory = self.get_merger_factory()
2974
 
        self._install_hook(factory)
2975
 
        builder = self.make_builder()
2976
 
        builder.add_file('bar-id', builder.tree_root, 'bar', 'text1', True,
2977
 
                         this=False)
2978
 
        builder.add_dir('bar-dir', builder.tree_root, 'bar-id',
2979
 
                        base=False, other=False)
2980
 
        return builder
2981
 
 
2982
 
    def test_uses_this_branch(self):
2983
 
        builder = self.make_text_conflict()
2984
 
        tt = builder.make_preview_transform()
2985
 
        self.addCleanup(tt.finalize)
2986
 
 
2987
 
    def test_affected_files_cached(self):
2988
 
        """Ensures that the config variable is cached"""
2989
 
        builder = self.make_text_conflict()
2990
 
        conflicts = builder.merge()
2991
 
        # The hook should set the variable
2992
 
        self.assertEqual(['bar'], self.merger.affected_files)
2993
 
        self.assertEqual(1, len(conflicts))
2994
 
 
2995
 
    def test_hook_called_for_text_conflicts(self):
2996
 
        builder = self.make_text_conflict()
2997
 
        conflicts = builder.merge()
2998
 
        # The hook should call the merge_text() method
2999
 
        self.assertEqual(['merge_text'], self.calls)
3000
 
 
3001
 
    def test_hook_not_called_for_kind_change(self):
3002
 
        builder = self.make_kind_change()
3003
 
        conflicts = builder.merge()
3004
 
        # The hook should not call the merge_text() method
3005
 
        self.assertEqual([], self.calls)
3006
 
 
3007
 
    def test_hook_not_called_for_other_files(self):
3008
 
        builder = self.make_text_conflict('foobar')
3009
 
        conflicts = builder.merge()
3010
 
        # The hook should not call the merge_text() method
3011
 
        self.assertEqual([], self.calls)
3012
 
 
3013
 
 
3014
 
class TestMergeIntoBase(tests.TestCaseWithTransport):
3015
 
 
3016
 
    def setup_simple_branch(self, relpath, shape=None, root_id=None):
3017
 
        """One commit, containing tree specified by optional shape.
3018
 
        
3019
 
        Default is empty tree (just root entry).
3020
 
        """
3021
 
        if root_id is None:
3022
 
            root_id = '%s-root-id' % (relpath,)
3023
 
        wt = self.make_branch_and_tree(relpath)
3024
 
        wt.set_root_id(root_id)
3025
 
        if shape is not None:
3026
 
            adjusted_shape = [relpath + '/' + elem for elem in shape]
3027
 
            self.build_tree(adjusted_shape)
3028
 
            ids = ['%s-%s-id' % (relpath, basename(elem.rstrip('/')))
3029
 
                   for elem in shape]
3030
 
            wt.add(shape, ids=ids)
3031
 
        rev_id = 'r1-%s' % (relpath,)
3032
 
        wt.commit("Initial commit of %s" % (relpath,), rev_id=rev_id)
3033
 
        self.assertEqual(root_id, wt.path2id(''))
3034
 
        return wt
3035
 
 
3036
 
    def setup_two_branches(self, custom_root_ids=True):
3037
 
        """Setup 2 branches, one will be a library, the other a project."""
3038
 
        if custom_root_ids:
3039
 
            root_id = None
3040
 
        else:
3041
 
            root_id = inventory.ROOT_ID
3042
 
        project_wt = self.setup_simple_branch(
3043
 
            'project', ['README', 'dir/', 'dir/file.c'],
3044
 
            root_id)
3045
 
        lib_wt = self.setup_simple_branch(
3046
 
            'lib1', ['README', 'Makefile', 'foo.c'], root_id)
3047
 
 
3048
 
        return project_wt, lib_wt
3049
 
 
3050
 
    def do_merge_into(self, location, merge_as):
3051
 
        """Helper for using MergeIntoMerger.
3052
 
        
3053
 
        :param location: location of directory to merge from, either the
3054
 
            location of a branch or of a path inside a branch.
3055
 
        :param merge_as: the path in a tree to add the new directory as.
3056
 
        :returns: the conflicts from 'do_merge'.
3057
 
        """
3058
 
        operation = cleanup.OperationWithCleanups(self._merge_into)
3059
 
        return operation.run(location, merge_as)
3060
 
 
3061
 
    def _merge_into(self, op, location, merge_as):
3062
 
        # Open and lock the various tree and branch objects
3063
 
        wt, subdir_relpath = WorkingTree.open_containing(merge_as)
3064
 
        op.add_cleanup(wt.lock_write().unlock)
3065
 
        branch_to_merge, subdir_to_merge = _mod_branch.Branch.open_containing(
3066
 
            location)
3067
 
        op.add_cleanup(branch_to_merge.lock_read().unlock)
3068
 
        other_tree = branch_to_merge.basis_tree()
3069
 
        op.add_cleanup(other_tree.lock_read().unlock)
3070
 
        # Perform the merge
3071
 
        merger = _mod_merge.MergeIntoMerger(this_tree=wt, other_tree=other_tree,
3072
 
            other_branch=branch_to_merge, target_subdir=subdir_relpath,
3073
 
            source_subpath=subdir_to_merge)
3074
 
        merger.set_base_revision(_mod_revision.NULL_REVISION, branch_to_merge)
3075
 
        conflicts = merger.do_merge()
3076
 
        merger.set_pending()
3077
 
        return conflicts
3078
 
 
3079
 
    def assertTreeEntriesEqual(self, expected_entries, tree):
3080
 
        """Assert that 'tree' contains the expected inventory entries.
3081
 
 
3082
 
        :param expected_entries: sequence of (path, file-id) pairs.
3083
 
        """
3084
 
        files = [(path, ie.file_id) for path, ie in tree.iter_entries_by_dir()]
3085
 
        self.assertEqual(expected_entries, files)
3086
 
 
3087
 
 
3088
 
class TestMergeInto(TestMergeIntoBase):
3089
 
 
3090
 
    def test_newdir_with_unique_roots(self):
3091
 
        """Merge a branch with a unique root into a new directory."""
3092
 
        project_wt, lib_wt = self.setup_two_branches()
3093
 
        self.do_merge_into('lib1', 'project/lib1')
3094
 
        project_wt.lock_read()
3095
 
        self.addCleanup(project_wt.unlock)
3096
 
        # The r1-lib1 revision should be merged into this one
3097
 
        self.assertEqual(['r1-project', 'r1-lib1'], project_wt.get_parent_ids())
3098
 
        self.assertTreeEntriesEqual(
3099
 
            [('', 'project-root-id'),
3100
 
             ('README', 'project-README-id'),
3101
 
             ('dir', 'project-dir-id'),
3102
 
             ('lib1', 'lib1-root-id'),
3103
 
             ('dir/file.c', 'project-file.c-id'),
3104
 
             ('lib1/Makefile', 'lib1-Makefile-id'),
3105
 
             ('lib1/README', 'lib1-README-id'),
3106
 
             ('lib1/foo.c', 'lib1-foo.c-id'),
3107
 
            ], project_wt)
3108
 
 
3109
 
    def test_subdir(self):
3110
 
        """Merge a branch into a subdirectory of an existing directory."""
3111
 
        project_wt, lib_wt = self.setup_two_branches()
3112
 
        self.do_merge_into('lib1', 'project/dir/lib1')
3113
 
        project_wt.lock_read()
3114
 
        self.addCleanup(project_wt.unlock)
3115
 
        # The r1-lib1 revision should be merged into this one
3116
 
        self.assertEqual(['r1-project', 'r1-lib1'], project_wt.get_parent_ids())
3117
 
        self.assertTreeEntriesEqual(
3118
 
            [('', 'project-root-id'),
3119
 
             ('README', 'project-README-id'),
3120
 
             ('dir', 'project-dir-id'),
3121
 
             ('dir/file.c', 'project-file.c-id'),
3122
 
             ('dir/lib1', 'lib1-root-id'),
3123
 
             ('dir/lib1/Makefile', 'lib1-Makefile-id'),
3124
 
             ('dir/lib1/README', 'lib1-README-id'),
3125
 
             ('dir/lib1/foo.c', 'lib1-foo.c-id'),
3126
 
            ], project_wt)
3127
 
 
3128
 
    def test_newdir_with_repeat_roots(self):
3129
 
        """If the file-id of the dir to be merged already exists a new ID will
3130
 
        be allocated to let the merge happen.
3131
 
        """
3132
 
        project_wt, lib_wt = self.setup_two_branches(custom_root_ids=False)
3133
 
        root_id = project_wt.path2id('')
3134
 
        self.do_merge_into('lib1', 'project/lib1')
3135
 
        project_wt.lock_read()
3136
 
        self.addCleanup(project_wt.unlock)
3137
 
        # The r1-lib1 revision should be merged into this one
3138
 
        self.assertEqual(['r1-project', 'r1-lib1'], project_wt.get_parent_ids())
3139
 
        new_lib1_id = project_wt.path2id('lib1')
3140
 
        self.assertNotEqual(None, new_lib1_id)
3141
 
        self.assertTreeEntriesEqual(
3142
 
            [('', root_id),
3143
 
             ('README', 'project-README-id'),
3144
 
             ('dir', 'project-dir-id'),
3145
 
             ('lib1', new_lib1_id),
3146
 
             ('dir/file.c', 'project-file.c-id'),
3147
 
             ('lib1/Makefile', 'lib1-Makefile-id'),
3148
 
             ('lib1/README', 'lib1-README-id'),
3149
 
             ('lib1/foo.c', 'lib1-foo.c-id'),
3150
 
            ], project_wt)
3151
 
 
3152
 
    def test_name_conflict(self):
3153
 
        """When the target directory name already exists a conflict is
3154
 
        generated and the original directory is renamed to foo.moved.
3155
 
        """
3156
 
        dest_wt = self.setup_simple_branch('dest', ['dir/', 'dir/file.txt'])
3157
 
        src_wt = self.setup_simple_branch('src', ['README'])
3158
 
        conflicts = self.do_merge_into('src', 'dest/dir')
3159
 
        self.assertEqual(1, conflicts)
3160
 
        dest_wt.lock_read()
3161
 
        self.addCleanup(dest_wt.unlock)
3162
 
        # The r1-lib1 revision should be merged into this one
3163
 
        self.assertEqual(['r1-dest', 'r1-src'], dest_wt.get_parent_ids())
3164
 
        self.assertTreeEntriesEqual(
3165
 
            [('', 'dest-root-id'),
3166
 
             ('dir', 'src-root-id'),
3167
 
             ('dir.moved', 'dest-dir-id'),
3168
 
             ('dir/README', 'src-README-id'),
3169
 
             ('dir.moved/file.txt', 'dest-file.txt-id'),
3170
 
            ], dest_wt)
3171
 
 
3172
 
    def test_file_id_conflict(self):
3173
 
        """A conflict is generated if the merge-into adds a file (or other
3174
 
        inventory entry) with a file-id that already exists in the target tree.
3175
 
        """
3176
 
        dest_wt = self.setup_simple_branch('dest', ['file.txt'])
3177
 
        # Make a second tree with a file-id that will clash with file.txt in
3178
 
        # dest.
3179
 
        src_wt = self.make_branch_and_tree('src')
3180
 
        self.build_tree(['src/README'])
3181
 
        src_wt.add(['README'], ids=['dest-file.txt-id'])
3182
 
        src_wt.commit("Rev 1 of src.", rev_id='r1-src')
3183
 
        conflicts = self.do_merge_into('src', 'dest/dir')
3184
 
        # This is an edge case that shouldn't happen to users very often.  So
3185
 
        # we don't care really about the exact presentation of the conflict,
3186
 
        # just that there is one.
3187
 
        self.assertEqual(1, conflicts)
3188
 
 
3189
 
    def test_only_subdir(self):
3190
 
        """When the location points to just part of a tree, merge just that
3191
 
        subtree.
3192
 
        """
3193
 
        dest_wt = self.setup_simple_branch('dest')
3194
 
        src_wt = self.setup_simple_branch(
3195
 
            'src', ['hello.txt', 'dir/', 'dir/foo.c'])
3196
 
        conflicts = self.do_merge_into('src/dir', 'dest/dir')
3197
 
        dest_wt.lock_read()
3198
 
        self.addCleanup(dest_wt.unlock)
3199
 
        # The r1-lib1 revision should NOT be merged into this one (this is a
3200
 
        # partial merge).
3201
 
        self.assertEqual(['r1-dest'], dest_wt.get_parent_ids())
3202
 
        self.assertTreeEntriesEqual(
3203
 
            [('', 'dest-root-id'),
3204
 
             ('dir', 'src-dir-id'),
3205
 
             ('dir/foo.c', 'src-foo.c-id'),
3206
 
            ], dest_wt)
3207
 
 
3208
 
    def test_only_file(self):
3209
 
        """An edge case: merge just one file, not a whole dir."""
3210
 
        dest_wt = self.setup_simple_branch('dest')
3211
 
        two_file_wt = self.setup_simple_branch(
3212
 
            'two-file', ['file1.txt', 'file2.txt'])
3213
 
        conflicts = self.do_merge_into('two-file/file1.txt', 'dest/file1.txt')
3214
 
        dest_wt.lock_read()
3215
 
        self.addCleanup(dest_wt.unlock)
3216
 
        # The r1-lib1 revision should NOT be merged into this one
3217
 
        self.assertEqual(['r1-dest'], dest_wt.get_parent_ids())
3218
 
        self.assertTreeEntriesEqual(
3219
 
            [('', 'dest-root-id'), ('file1.txt', 'two-file-file1.txt-id')],
3220
 
            dest_wt)
3221
 
 
3222
 
    def test_no_such_source_path(self):
3223
 
        """PathNotInTree is raised if the specified path in the source tree
3224
 
        does not exist.
3225
 
        """
3226
 
        dest_wt = self.setup_simple_branch('dest')
3227
 
        two_file_wt = self.setup_simple_branch('src', ['dir/'])
3228
 
        self.assertRaises(_mod_merge.PathNotInTree, self.do_merge_into,
3229
 
            'src/no-such-dir', 'dest/foo')
3230
 
        dest_wt.lock_read()
3231
 
        self.addCleanup(dest_wt.unlock)
3232
 
        # The dest tree is unmodified.
3233
 
        self.assertEqual(['r1-dest'], dest_wt.get_parent_ids())
3234
 
        self.assertTreeEntriesEqual([('', 'dest-root-id')], dest_wt)
3235
 
 
3236
 
    def test_no_such_target_path(self):
3237
 
        """PathNotInTree is also raised if the specified path in the target
3238
 
        tree does not exist.
3239
 
        """
3240
 
        dest_wt = self.setup_simple_branch('dest')
3241
 
        two_file_wt = self.setup_simple_branch('src', ['file.txt'])
3242
 
        self.assertRaises(_mod_merge.PathNotInTree, self.do_merge_into,
3243
 
            'src', 'dest/no-such-dir/foo')
3244
 
        dest_wt.lock_read()
3245
 
        self.addCleanup(dest_wt.unlock)
3246
 
        # The dest tree is unmodified.
3247
 
        self.assertEqual(['r1-dest'], dest_wt.get_parent_ids())
3248
 
        self.assertTreeEntriesEqual([('', 'dest-root-id')], dest_wt)
3249
 
 
3250
 
 
3251
 
class TestMergeHooks(TestCaseWithTransport):
3252
 
 
3253
 
    def setUp(self):
3254
 
        super(TestMergeHooks, self).setUp()
3255
 
        self.tree_a = self.make_branch_and_tree('tree_a')
3256
 
        self.build_tree_contents([('tree_a/file', 'content_1')])
3257
 
        self.tree_a.add('file', 'file-id')
3258
 
        self.tree_a.commit('added file')
3259
 
 
3260
 
        self.tree_b = self.tree_a.bzrdir.sprout('tree_b').open_workingtree()
3261
 
        self.build_tree_contents([('tree_b/file', 'content_2')])
3262
 
        self.tree_b.commit('modify file')
3263
 
 
3264
 
    def test_pre_merge_hook_inject_different_tree(self):
3265
 
        tree_c = self.tree_b.bzrdir.sprout('tree_c').open_workingtree()
3266
 
        self.build_tree_contents([('tree_c/file', 'content_3')])
3267
 
        tree_c.commit("more content")
3268
 
        calls = []
3269
 
        def factory(merger):
3270
 
            self.assertIsInstance(merger, _mod_merge.Merge3Merger)
3271
 
            merger.other_tree = tree_c
3272
 
            calls.append(merger)
3273
 
        _mod_merge.Merger.hooks.install_named_hook('pre_merge',
3274
 
                                                   factory, 'test factory')
3275
 
        self.tree_a.merge_from_branch(self.tree_b.branch)
3276
 
 
3277
 
        self.assertFileEqual("content_3", 'tree_a/file')
3278
 
        self.assertLength(1, calls)
3279
 
 
3280
 
    def test_post_merge_hook_called(self):
3281
 
        calls = []
3282
 
        def factory(merger):
3283
 
            self.assertIsInstance(merger, _mod_merge.Merge3Merger)
3284
 
            calls.append(merger)
3285
 
        _mod_merge.Merger.hooks.install_named_hook('post_merge',
3286
 
                                                   factory, 'test factory')
3287
 
 
3288
 
        self.tree_a.merge_from_branch(self.tree_b.branch)
3289
 
 
3290
 
        self.assertFileEqual("content_2", 'tree_a/file')
3291
 
        self.assertLength(1, calls)
 
1027
class TestMergeImplementation(object):
 
1028
 
 
1029
    def do_merge(self, target_tree, source_tree, **kwargs):
 
1030
        merger = _mod_merge.Merger.from_revision_ids(progress.DummyProgress(),
 
1031
            target_tree, source_tree.last_revision(),
 
1032
            other_branch=source_tree.branch)
 
1033
        merger.merge_type=self.merge_type
 
1034
        for name, value in kwargs.items():
 
1035
            setattr(merger, name, value)
 
1036
        merger.do_merge()
 
1037
 
 
1038
    def test_merge_specific_file(self):
 
1039
        this_tree = self.make_branch_and_tree('this')
 
1040
        this_tree.lock_write()
 
1041
        self.addCleanup(this_tree.unlock)
 
1042
        self.build_tree_contents([
 
1043
            ('this/file1', 'a\nb\n'),
 
1044
            ('this/file2', 'a\nb\n')
 
1045
        ])
 
1046
        this_tree.add(['file1', 'file2'])
 
1047
        this_tree.commit('Added files')
 
1048
        other_tree = this_tree.bzrdir.sprout('other').open_workingtree()
 
1049
        self.build_tree_contents([
 
1050
            ('other/file1', 'a\nb\nc\n'),
 
1051
            ('other/file2', 'a\nb\nc\n')
 
1052
        ])
 
1053
        other_tree.commit('modified both')
 
1054
        self.build_tree_contents([
 
1055
            ('this/file1', 'd\na\nb\n'),
 
1056
            ('this/file2', 'd\na\nb\n')
 
1057
        ])
 
1058
        this_tree.commit('modified both')
 
1059
        self.do_merge(this_tree, other_tree, interesting_files=['file1'])
 
1060
        self.assertFileEqual('d\na\nb\nc\n', 'this/file1')
 
1061
        self.assertFileEqual('d\na\nb\n', 'this/file2')
 
1062
 
 
1063
    def test_merge_move_and_change(self):
 
1064
        this_tree = self.make_branch_and_tree('this')
 
1065
        this_tree.lock_write()
 
1066
        self.addCleanup(this_tree.unlock)
 
1067
        self.build_tree_contents([
 
1068
            ('this/file1', 'line 1\nline 2\nline 3\nline 4\n'),
 
1069
        ])
 
1070
        this_tree.add('file1',)
 
1071
        this_tree.commit('Added file')
 
1072
        other_tree = this_tree.bzrdir.sprout('other').open_workingtree()
 
1073
        self.build_tree_contents([
 
1074
            ('other/file1', 'line 1\nline 2 to 2.1\nline 3\nline 4\n'),
 
1075
        ])
 
1076
        other_tree.commit('Changed 2 to 2.1')
 
1077
        self.build_tree_contents([
 
1078
            ('this/file1', 'line 1\nline 3\nline 2\nline 4\n'),
 
1079
        ])
 
1080
        this_tree.commit('Swapped 2 & 3')
 
1081
        self.do_merge(this_tree, other_tree)
 
1082
        self.assertFileEqual('line 1\n'
 
1083
            '<<<<<<< TREE\n'
 
1084
            'line 3\n'
 
1085
            'line 2\n'
 
1086
            '=======\n'
 
1087
            'line 2 to 2.1\n'
 
1088
            'line 3\n'
 
1089
            '>>>>>>> MERGE-SOURCE\n'
 
1090
            'line 4\n', 'this/file1')
 
1091
 
 
1092
 
 
1093
class TestMerge3Merge(TestCaseWithTransport, TestMergeImplementation):
 
1094
 
 
1095
    merge_type = _mod_merge.Merge3Merger
 
1096
 
 
1097
 
 
1098
class TestWeaveMerge(TestCaseWithTransport, TestMergeImplementation):
 
1099
 
 
1100
    merge_type = _mod_merge.WeaveMerger
 
1101
 
 
1102
 
 
1103
class TestLCAMerge(TestCaseWithTransport, TestMergeImplementation):
 
1104
 
 
1105
    merge_type = _mod_merge.LCAMerger
 
1106
 
 
1107
    def test_merge_move_and_change(self):
 
1108
        self.expectFailure("lca merge doesn't conflict for move and change",
 
1109
            super(TestLCAMerge, self).test_merge_move_and_change)