1
# Copyright (C) 2005, 2006 by Canonical Ltd
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.
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.
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
18
from StringIO import StringIO
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
32
class TestMerge(TestCaseWithTransport):
33
"""Test appending more than one revision"""
35
def test_pending(self):
36
wt = self.make_branch_and_tree('.')
37
rev_a = wt.commit("lala!")
38
self.assertEqual([rev_a], wt.get_parent_ids())
39
merge([u'.', -1], [None, None])
40
self.assertEqual([rev_a], wt.get_parent_ids())
43
wt = self.make_branch_and_tree('.')
47
merge([u'.', 2], [u'.', 1])
49
def test_nocommits(self):
51
wt2 = self.make_branch_and_tree('branch2')
52
self.assertRaises(NoCommits, merge, ['branch2', -1],
56
def test_unrelated(self):
57
wt2 = self.test_nocommits()
59
self.assertRaises(UnrelatedBranches, merge, ['branch2', -1],
63
def test_merge_one_file(self):
64
"""Do a partial merge of a tree which should not affect tree parents."""
65
wt1 = self.make_branch_and_tree('branch1')
66
tip = wt1.commit('empty commit')
67
wt2 = self.make_branch_and_tree('branch2')
69
file('branch1/foo', 'wb').write('foo')
70
file('branch1/bar', 'wb').write('bar')
73
wt1.commit('add foobar')
75
self.run_bzr('merge', '../branch1/baz', retcode=3)
76
self.run_bzr('merge', '../branch1/foo')
77
self.failUnlessExists('foo')
78
self.failIfExists('bar')
79
wt2 = WorkingTree.open('.') # opens branch2
80
self.assertEqual([tip], wt2.get_parent_ids())
82
def test_pending_with_null(self):
83
"""When base is forced to revno 0, pending_merges is set"""
84
wt2 = self.test_unrelated()
85
wt1 = WorkingTree.open('.')
88
# merge all of branch 2 into branch 1 even though they
90
self.assertRaises(BzrCommandError, merge, ['branch2', -1],
91
['branch2', 0], reprocess=True, show_base=True)
92
merge(['branch2', -1], ['branch2', 0], reprocess=True)
93
self.assertEqual([br1.last_revision(), wt2.branch.last_revision()],
95
return (wt1, wt2.branch)
97
def test_two_roots(self):
98
"""Merge base is sane when two unrelated branches are merged"""
99
wt1, br2 = self.test_pending_with_null()
101
last = wt1.branch.last_revision()
102
self.assertEqual(common_ancestor(last, last, wt1.branch.repository), last)
104
def test_create_rename(self):
105
"""Rename an inventory entry while creating the file"""
106
tree =self.make_branch_and_tree('.')
107
file('name1', 'wb').write('Hello')
109
tree.commit(message="hello")
110
tree.rename_one('name1', 'name2')
112
transform_tree(tree, tree.branch.basis_tree())
114
def test_layered_rename(self):
115
"""Rename both child and parent at same time"""
116
tree =self.make_branch_and_tree('.')
119
filename = pathjoin('dirname1', 'name1')
120
file(filename, 'wb').write('Hello')
122
tree.commit(message="hello")
123
filename2 = pathjoin('dirname1', 'name2')
124
tree.rename_one(filename, filename2)
125
tree.rename_one('dirname1', 'dirname2')
126
transform_tree(tree, tree.branch.basis_tree())
128
def test_ignore_zero_merge_inner(self):
129
# Test that merge_inner's ignore zero paramter is effective
130
tree_a =self.make_branch_and_tree('a')
131
tree_a.commit(message="hello")
132
dir_b = tree_a.bzrdir.sprout('b')
133
tree_b = dir_b.open_workingtree()
134
tree_a.commit(message="hello again")
136
merge_inner(tree_b.branch, tree_a, tree_b.basis_tree(),
137
this_tree=tree_b, ignore_zero=True)
138
log = self._get_log()
139
self.failUnless('All changes applied successfully.\n' not in log)
141
merge_inner(tree_b.branch, tree_a, tree_b.basis_tree(),
142
this_tree=tree_b, ignore_zero=False)
143
log = self._get_log()
144
self.failUnless('All changes applied successfully.\n' in log)
146
def test_merge_inner_conflicts(self):
147
tree_a = self.make_branch_and_tree('a')
148
tree_a.set_conflicts(ConflictList([TextConflict('patha')]))
149
merge_inner(tree_a.branch, tree_a, tree_a, this_tree=tree_a)
150
self.assertEqual(1, len(tree_a.conflicts()))