~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_merge.py

Deprecate compare_trees and move its body to InterTree.changes_from.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2005, 2006 by Canonical Ltd
 
2
#
 
3
# This program is free software; you can redistribute it and/or modify
 
4
# it under the terms of the GNU General Public License as published by
 
5
# the Free Software Foundation; either version 2 of the License, or
 
6
# (at your option) any later version.
 
7
#
 
8
# This program is distributed in the hope that it will be useful,
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
# GNU General Public License for more details.
 
12
#
 
13
# You should have received a copy of the GNU General Public License
 
14
# along with this program; if not, write to the Free Software
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
import os
 
18
from StringIO import StringIO
 
19
 
 
20
from bzrlib.branch import Branch
 
21
from bzrlib.builtins import merge
 
22
from bzrlib.conflicts import ConflictList, TextConflict
 
23
from bzrlib.errors import UnrelatedBranches, NoCommits, BzrCommandError
 
24
from bzrlib.merge import transform_tree, merge_inner
 
25
from bzrlib.osutils import pathjoin
 
26
from bzrlib.revision import common_ancestor
 
27
from bzrlib.tests import TestCaseWithTransport
 
28
from bzrlib.trace import (enable_test_log, disable_test_log)
 
29
from bzrlib.workingtree import WorkingTree
 
30
 
 
31
 
 
32
class TestMerge(TestCaseWithTransport):
 
33
    """Test appending more than one revision"""
 
34
 
 
35
    def test_pending(self):
 
36
        wt = self.make_branch_and_tree('.')
 
37
        wt.commit("lala!")
 
38
        self.assertEquals(len(wt.pending_merges()), 0)
 
39
        merge([u'.', -1], [None, None])
 
40
        self.assertEquals(len(wt.pending_merges()), 0)
 
41
 
 
42
    def test_undo(self):
 
43
        wt = self.make_branch_and_tree('.')
 
44
        wt.commit("lala!")
 
45
        wt.commit("haha!")
 
46
        wt.commit("blabla!")
 
47
        merge([u'.', 2], [u'.', 1])
 
48
 
 
49
    def test_nocommits(self):
 
50
        self.test_pending()
 
51
        wt2 = self.make_branch_and_tree('branch2')
 
52
        self.assertRaises(NoCommits, merge, ['branch2', -1], 
 
53
                          [None, None])
 
54
        return wt2
 
55
 
 
56
    def test_unrelated(self):
 
57
        wt2 = self.test_nocommits()
 
58
        wt2.commit("blah")
 
59
        self.assertRaises(UnrelatedBranches, merge, ['branch2', -1], 
 
60
                          [None, None])
 
61
        return wt2
 
62
 
 
63
    def test_merge_one(self):
 
64
        wt1 = self.make_branch_and_tree('branch1')
 
65
        wt1.commit('empty commit')
 
66
        wt2 = self.make_branch_and_tree('branch2')
 
67
        wt2.pull(wt1.branch)
 
68
        file('branch1/foo', 'wb').write('foo')
 
69
        file('branch1/bar', 'wb').write('bar')
 
70
        wt1.add('foo')
 
71
        wt1.add('bar')
 
72
        wt1.commit('add foobar')
 
73
        os.chdir('branch2')
 
74
        self.run_bzr('merge', '../branch1/baz', retcode=3)
 
75
        self.run_bzr('merge', '../branch1/foo')
 
76
        self.failUnlessExists('foo')
 
77
        self.failIfExists('bar')
 
78
        wt2 = WorkingTree.open_containing('branch2')[0]
 
79
        self.assertEqual(wt2.pending_merges(), [])
 
80
 
 
81
    def test_pending_with_null(self):
 
82
        """When base is forced to revno 0, pending_merges is set"""
 
83
        wt2 = self.test_unrelated()
 
84
        wt1 = WorkingTree.open('.')
 
85
        br1 = wt1.branch
 
86
        br1.fetch(wt2.branch)
 
87
        # merge all of branch 2 into branch 1 even though they 
 
88
        # are not related.
 
89
        self.assertRaises(BzrCommandError, merge, ['branch2', -1], 
 
90
                          ['branch2', 0], reprocess=True, show_base=True)
 
91
        merge(['branch2', -1], ['branch2', 0], reprocess=True)
 
92
        self.assertEquals(len(wt1.pending_merges()), 1)
 
93
        return (wt1, wt2.branch)
 
94
 
 
95
    def test_two_roots(self):
 
96
        """Merge base is sane when two unrelated branches are merged"""
 
97
        wt1, br2 = self.test_pending_with_null()
 
98
        wt1.commit("blah")
 
99
        last = wt1.branch.last_revision()
 
100
        self.assertEquals(common_ancestor(last, last, wt1.branch.repository), last)
 
101
 
 
102
    def test_create_rename(self):
 
103
        """Rename an inventory entry while creating the file"""
 
104
        tree =self.make_branch_and_tree('.')
 
105
        file('name1', 'wb').write('Hello')
 
106
        tree.add('name1')
 
107
        tree.commit(message="hello")
 
108
        tree.rename_one('name1', 'name2')
 
109
        os.unlink('name2')
 
110
        transform_tree(tree, tree.branch.basis_tree())
 
111
 
 
112
    def test_layered_rename(self):
 
113
        """Rename both child and parent at same time"""
 
114
        tree =self.make_branch_and_tree('.')
 
115
        os.mkdir('dirname1')
 
116
        tree.add('dirname1')
 
117
        filename = pathjoin('dirname1', 'name1')
 
118
        file(filename, 'wb').write('Hello')
 
119
        tree.add(filename)
 
120
        tree.commit(message="hello")
 
121
        filename2 = pathjoin('dirname1', 'name2')
 
122
        tree.rename_one(filename, filename2)
 
123
        tree.rename_one('dirname1', 'dirname2')
 
124
        transform_tree(tree, tree.branch.basis_tree())
 
125
 
 
126
    def test_ignore_zero_merge_inner(self):
 
127
        # Test that merge_inner's ignore zero paramter is effective
 
128
        tree_a =self.make_branch_and_tree('a')
 
129
        tree_a.commit(message="hello")
 
130
        dir_b = tree_a.bzrdir.sprout('b')
 
131
        tree_b = dir_b.open_workingtree()
 
132
        tree_a.commit(message="hello again")
 
133
        log = StringIO()
 
134
        merge_inner(tree_b.branch, tree_a, tree_b.basis_tree(), 
 
135
                    this_tree=tree_b, ignore_zero=True)
 
136
        log = self._get_log()
 
137
        self.failUnless('All changes applied successfully.\n' not in log)
 
138
        tree_b.revert([])
 
139
        merge_inner(tree_b.branch, tree_a, tree_b.basis_tree(), 
 
140
                    this_tree=tree_b, ignore_zero=False)
 
141
        log = self._get_log()
 
142
        self.failUnless('All changes applied successfully.\n' in log)
 
143
 
 
144
    def test_merge_inner_conflicts(self):
 
145
        tree_a = self.make_branch_and_tree('a')
 
146
        tree_a.set_conflicts(ConflictList([TextConflict('patha')]))
 
147
        merge_inner(tree_a.branch, tree_a, tree_a, this_tree=tree_a)
 
148
        self.assertEqual(1, len(tree_a.conflicts()))