~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_merge.py

  • Committer: Tarmac
  • Author(s): Vincent Ladeuil
  • Date: 2017-01-30 14:42:05 UTC
  • mfrom: (6620.1.1 trunk)
  • Revision ID: tarmac-20170130144205-r8fh2xpmiuxyozpv
Merge  2.7 into trunk including fix for bug #1657238 [r=vila]

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
 
1
# Copyright (C) 2005-2012, 2016 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 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,
21
23
    conflicts,
22
24
    errors,
 
25
    inventory,
23
26
    knit,
 
27
    memorytree,
24
28
    merge as _mod_merge,
25
29
    option,
26
 
    progress,
 
30
    revision as _mod_revision,
 
31
    tests,
27
32
    transform,
28
33
    versionedfile,
29
34
    )
30
 
from bzrlib.branch import Branch
31
35
from bzrlib.conflicts import ConflictList, TextConflict
32
 
from bzrlib.errors import UnrelatedBranches, NoCommits, BzrCommandError
 
36
from bzrlib.errors import UnrelatedBranches, NoCommits
33
37
from bzrlib.merge import transform_tree, merge_inner, _PlanMerge
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)
 
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
    )
37
45
from bzrlib.workingtree import WorkingTree
38
46
 
39
47
 
75
83
        tip = wt1.commit('empty commit')
76
84
        wt2 = self.make_branch_and_tree('branch2')
77
85
        wt2.pull(wt1.branch)
78
 
        file('branch1/foo', 'wb').write('foo')
79
 
        file('branch1/bar', 'wb').write('bar')
 
86
        with file('branch1/foo', 'wb') as f:
 
87
            f.write('foo')
 
88
        with file('branch1/bar', 'wb') as f:
 
89
            f.write('bar')
80
90
        wt1.add('foo')
81
91
        wt1.add('bar')
82
92
        wt1.commit('add foobar')
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
 
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')
89
98
        self.assertEqual([tip], wt2.get_parent_ids())
90
 
        
 
99
 
91
100
    def test_pending_with_null(self):
92
101
        """When base is forced to revno 0, parent_ids are set"""
93
102
        wt2 = self.test_unrelated()
94
103
        wt1 = WorkingTree.open('.')
95
104
        br1 = wt1.branch
96
105
        br1.fetch(wt2.branch)
97
 
        # merge all of branch 2 into branch 1 even though they 
 
106
        # merge all of branch 2 into branch 1 even though they
98
107
        # are not related.
99
108
        wt1.merge_from_branch(wt2.branch, wt2.last_revision(), 'null:')
100
109
        self.assertEqual([br1.last_revision(), wt2.branch.last_revision()],
114
123
        finally:
115
124
            wt1.unlock()
116
125
 
 
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
 
117
152
    def test_create_rename(self):
118
153
        """Rename an inventory entry while creating the file"""
119
154
        tree =self.make_branch_and_tree('.')
120
 
        file('name1', 'wb').write('Hello')
 
155
        with file('name1', 'wb') as f: f.write('Hello')
121
156
        tree.add('name1')
122
157
        tree.commit(message="hello")
123
158
        tree.rename_one('name1', 'name2')
130
165
        os.mkdir('dirname1')
131
166
        tree.add('dirname1')
132
167
        filename = pathjoin('dirname1', 'name1')
133
 
        file(filename, 'wb').write('Hello')
 
168
        with file(filename, 'wb') as f: f.write('Hello')
134
169
        tree.add(filename)
135
170
        tree.commit(message="hello")
136
171
        filename2 = pathjoin('dirname1', 'name2')
148
183
        self.addCleanup(tree_b.unlock)
149
184
        tree_a.commit(message="hello again")
150
185
        log = StringIO()
151
 
        merge_inner(tree_b.branch, tree_a, tree_b.basis_tree(), 
 
186
        merge_inner(tree_b.branch, tree_a, tree_b.basis_tree(),
152
187
                    this_tree=tree_b, ignore_zero=True)
153
 
        log = self._get_log(keep_log_file=True)
154
 
        self.failUnless('All changes applied successfully.\n' not in log)
 
188
        self.assertTrue('All changes applied successfully.\n' not in
 
189
            self.get_log())
155
190
        tree_b.revert()
156
 
        merge_inner(tree_b.branch, tree_a, tree_b.basis_tree(), 
 
191
        merge_inner(tree_b.branch, tree_a, tree_b.basis_tree(),
157
192
                    this_tree=tree_b, ignore_zero=False)
158
 
        log = self._get_log(keep_log_file=True)
159
 
        self.failUnless('All changes applied successfully.\n' in log)
 
193
        self.assertTrue('All changes applied successfully.\n' in self.get_log())
160
194
 
161
195
    def test_merge_inner_conflicts(self):
162
196
        tree_a = self.make_branch_and_tree('a')
193
227
 
194
228
    def test_nested_merge(self):
195
229
        tree = self.make_branch_and_tree('tree',
196
 
            format='dirstate-with-subtree')
 
230
            format='development-subtree')
197
231
        sub_tree = self.make_branch_and_tree('tree/sub-tree',
198
 
            format='dirstate-with-subtree')
 
232
            format='development-subtree')
199
233
        sub_tree.set_root_id('sub-tree-root')
200
234
        self.build_tree_contents([('tree/sub-tree/file', 'text1')])
201
235
        sub_tree.add('file')
217
251
        tree_a.add('file')
218
252
        tree_a.commit('commit base')
219
253
        # basis_tree() is only guaranteed to be valid as long as it is actually
220
 
        # the basis tree. This mutates the tree after grabbing basis, so go to
221
 
        # the repository.
 
254
        # the basis tree. This test commits to the tree after grabbing basis,
 
255
        # so we go to the repository.
222
256
        base_tree = tree_a.branch.repository.revision_tree(tree_a.last_revision())
223
257
        tree_b = tree_a.bzrdir.sprout('tree_b').open_workingtree()
224
258
        self.build_tree_contents([('tree_a/file', 'content_2')])
225
259
        tree_a.commit('commit other')
226
260
        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.
227
263
        os.unlink('tree_b/file')
228
264
        merge_inner(tree_b.branch, other_tree, base_tree, this_tree=tree_b)
229
265
 
246
282
        self.assertEqual(tree_b.conflicts(),
247
283
                         [conflicts.ContentsConflict('file',
248
284
                          file_id='file-id')])
249
 
    
 
285
 
250
286
    def test_merge_type_registry(self):
251
287
        merge_type_option = option.Option.OPTIONS['merge-type']
252
 
        self.assertFalse('merge4' in [x[0] for x in 
 
288
        self.assertFalse('merge4' in [x[0] for x in
253
289
                        merge_type_option.iter_switches()])
254
290
        registry = _mod_merge.get_merge_type_registry()
255
291
        registry.register_lazy('merge4', 'bzrlib.merge', 'Merge4Merger',
256
292
                               'time-travelling merge')
257
 
        self.assertTrue('merge4' in [x[0] for x in 
 
293
        self.assertTrue('merge4' in [x[0] for x in
258
294
                        merge_type_option.iter_switches()])
259
295
        registry.remove('merge4')
260
 
        self.assertFalse('merge4' in [x[0] for x in 
 
296
        self.assertFalse('merge4' in [x[0] for x in
261
297
                        merge_type_option.iter_switches()])
262
298
 
263
299
    def test_merge_other_moves_we_deleted(self):
290
326
        tree_a.commit('commit 2')
291
327
        tree_b = tree_a.bzrdir.sprout('b').open_workingtree()
292
328
        tree_b.rename_one('file_1', 'renamed')
293
 
        merger = _mod_merge.Merger.from_uncommitted(tree_a, tree_b,
294
 
                                                    progress.DummyProgress())
 
329
        merger = _mod_merge.Merger.from_uncommitted(tree_a, tree_b)
295
330
        merger.merge_type = _mod_merge.Merge3Merger
296
331
        merger.do_merge()
297
332
        self.assertEqual(tree_a.get_parent_ids(), [tree_b.last_revision()])
305
340
        tree_a.commit('commit 2')
306
341
        tree_b = tree_a.bzrdir.sprout('b').open_workingtree()
307
342
        tree_b.rename_one('file_1', 'renamed')
308
 
        merger = _mod_merge.Merger.from_uncommitted(tree_a, tree_b,
309
 
                                                    progress.DummyProgress())
 
343
        merger = _mod_merge.Merger.from_uncommitted(tree_a, tree_b)
310
344
        merger.merge_type = _mod_merge.WeaveMerger
311
345
        merger.do_merge()
312
346
        self.assertEqual(tree_a.get_parent_ids(), [tree_b.last_revision()])
337
371
 
338
372
    def test_weave_cherrypick(self):
339
373
        this_tree, other_tree = self.prepare_cherrypick()
340
 
        merger = _mod_merge.Merger.from_revision_ids(progress.DummyProgress(),
 
374
        merger = _mod_merge.Merger.from_revision_ids(None,
341
375
            this_tree, 'rev3b', 'rev2b', other_tree.branch)
342
376
        merger.merge_type = _mod_merge.WeaveMerger
343
377
        merger.do_merge()
345
379
 
346
380
    def test_weave_cannot_reverse_cherrypick(self):
347
381
        this_tree, other_tree = self.prepare_cherrypick()
348
 
        merger = _mod_merge.Merger.from_revision_ids(progress.DummyProgress(),
 
382
        merger = _mod_merge.Merger.from_revision_ids(None,
349
383
            this_tree, 'rev2b', 'rev3b', other_tree.branch)
350
384
        merger.merge_type = _mod_merge.WeaveMerger
351
385
        self.assertRaises(errors.CannotReverseCherrypick, merger.do_merge)
352
386
 
353
387
    def test_merge3_can_reverse_cherrypick(self):
354
388
        this_tree, other_tree = self.prepare_cherrypick()
355
 
        merger = _mod_merge.Merger.from_revision_ids(progress.DummyProgress(),
 
389
        merger = _mod_merge.Merger.from_revision_ids(None,
356
390
            this_tree, 'rev2b', 'rev3b', other_tree.branch)
357
391
        merger.merge_type = _mod_merge.Merge3Merger
358
392
        merger.do_merge()
370
404
        this_tree.lock_write()
371
405
        self.addCleanup(this_tree.unlock)
372
406
 
373
 
        merger = _mod_merge.Merger.from_revision_ids(progress.DummyProgress(),
 
407
        merger = _mod_merge.Merger.from_revision_ids(None,
374
408
            this_tree, 'rev3b', 'rev2b', other_tree.branch)
375
409
        merger.merge_type = _mod_merge.Merge3Merger
376
410
        merger.do_merge()
381
415
                             '>>>>>>> MERGE-SOURCE\n',
382
416
                             'this/file')
383
417
 
 
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
 
384
437
    def test_make_merger(self):
385
438
        this_tree = self.make_branch_and_tree('this')
386
439
        this_tree.commit('rev1', rev_id='rev1')
389
442
        other_tree.commit('rev2', rev_id='rev2b')
390
443
        this_tree.lock_write()
391
444
        self.addCleanup(this_tree.unlock)
392
 
        merger = _mod_merge.Merger.from_revision_ids(progress.DummyProgress,
 
445
        merger = _mod_merge.Merger.from_revision_ids(None,
393
446
            this_tree, 'rev2b', other_branch=other_tree.branch)
394
447
        merger.merge_type = _mod_merge.Merge3Merger
395
448
        tree_merger = merger.make_merger()
396
449
        self.assertIs(_mod_merge.Merge3Merger, tree_merger.__class__)
397
 
        self.assertEqual('rev2b', tree_merger.other_tree.get_revision_id())
398
 
        self.assertEqual('rev1', tree_merger.base_tree.get_revision_id())
 
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)
399
455
 
400
456
    def test_make_preview_transform(self):
401
457
        this_tree = self.make_branch_and_tree('this')
409
465
        other_tree.commit('rev2', rev_id='rev2b')
410
466
        this_tree.lock_write()
411
467
        self.addCleanup(this_tree.unlock)
412
 
        merger = _mod_merge.Merger.from_revision_ids(progress.DummyProgress(),
 
468
        merger = _mod_merge.Merger.from_revision_ids(None,
413
469
            this_tree, 'rev2b', other_branch=other_tree.branch)
414
470
        merger.merge_type = _mod_merge.Merge3Merger
415
471
        tree_merger = merger.make_merger()
439
495
        other_tree.commit('rev2', rev_id='rev2b')
440
496
        this_tree.lock_write()
441
497
        self.addCleanup(this_tree.unlock)
442
 
        merger = _mod_merge.Merger.from_revision_ids(progress.DummyProgress(),
 
498
        merger = _mod_merge.Merger.from_revision_ids(None,
443
499
            this_tree, 'rev2b', other_branch=other_tree.branch)
444
500
        merger.merge_type = _mod_merge.Merge3Merger
445
501
        tree_merger = merger.make_merger()
450
506
        finally:
451
507
            tree_file.close()
452
508
 
 
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
 
453
526
    def test_merge_add_into_deleted_root(self):
454
527
        # Yes, people actually do this.  And report bugs if it breaks.
455
528
        source = self.make_branch_and_tree('source', format='rich-root-pack')
486
559
class TestPlanMerge(TestCaseWithMemoryTransport):
487
560
 
488
561
    def setUp(self):
489
 
        TestCaseWithMemoryTransport.setUp(self)
 
562
        super(TestPlanMerge, self).setUp()
490
563
        mapper = versionedfile.PrefixMapper()
491
564
        factory = knit.make_file_factory(True, mapper)
492
565
        self.vf = factory(self.get_transport())
516
589
        self.add_uncommitted_version(('root', 'C:'), [('root', 'A')], 'fabg')
517
590
        return _PlanMerge('B:', 'C:', self.plan_merge_vf, ('root',))
518
591
 
 
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
 
519
598
    def test_unique_lines(self):
520
599
        plan = self.setup_plan_merge()
521
600
        self.assertEqual(plan._unique_lines(
716
795
 
717
796
    def test_plan_merge_insert_order(self):
718
797
        """Weave merges are sensitive to the order of insertion.
719
 
        
 
798
 
720
799
        Specifically for overlapping regions, it effects which region gets put
721
800
        'first'. And when a user resolves an overlapping merge, if they use the
722
801
        same ordering, then the lines match the parents, if they don't only
819
898
                          ('unchanged', 'f\n'),
820
899
                          ('unchanged', 'g\n')],
821
900
                         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()))
822
1006
 
823
1007
    def assertRemoveExternalReferences(self, filtered_parent_map,
824
1008
                                       child_map, tails, parent_map):
1024
1208
                         ], list(plan))
1025
1209
 
1026
1210
 
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)
 
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)