1
# Copyright (C) 2005 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
20
from bzrlib.selftest import TestCaseInTempDir
21
from bzrlib.branch import Branch
22
from bzrlib.commit import Commit
23
from bzrlib.errors import PointlessCommit, BzrError
26
# TODO: Test commit with some added, and added-but-missing files
28
class TestCommit(TestCaseInTempDir):
30
def test_simple_commit(self):
31
"""Commit and check two versions of a single file."""
32
b = Branch.initialize('.')
33
file('hello', 'w').write('hello world')
35
b.commit(message='add hello')
36
file_id = b.working_tree().path2id('hello')
38
file('hello', 'w').write('version 2')
39
b.commit(message='commit 2')
41
eq = self.assertEquals
43
rh = b.revision_history()
44
rev = b.get_revision(rh[0])
45
eq(rev.message, 'add hello')
47
tree1 = b.revision_tree(rh[0])
48
text = tree1.get_file_text(file_id)
49
eq(text, 'hello world')
51
tree2 = b.revision_tree(rh[1])
52
eq(tree2.get_file_text(file_id), 'version 2')
55
def test_delete_commit(self):
56
"""Test a commit with a deleted file"""
57
b = Branch.initialize('.')
58
file('hello', 'w').write('hello world')
59
b.add(['hello'], ['hello-id'])
60
b.commit(message='add hello')
63
b.commit('removed hello', rev_id='rev2')
65
tree = b.revision_tree('rev2')
66
self.assertFalse(tree.has_id('hello-id'))
69
def test_pointless_commit(self):
70
"""Commit refuses unless there are changes or it's forced."""
71
b = Branch.initialize('.')
72
file('hello', 'w').write('hello')
74
b.commit(message='add hello')
75
self.assertEquals(b.revno(), 1)
76
self.assertRaises(PointlessCommit,
79
allow_pointless=False)
80
self.assertEquals(b.revno(), 1)
84
def test_commit_empty(self):
85
"""Commiting an empty tree works."""
86
b = Branch.initialize('.')
87
b.commit(message='empty tree', allow_pointless=True)
88
self.assertRaises(PointlessCommit,
91
allow_pointless=False)
92
b.commit(message='empty tree', allow_pointless=True)
93
self.assertEquals(b.revno(), 2)
96
def test_selective_delete(self):
97
"""Selective commit in tree with deletions"""
98
b = Branch.initialize('.')
99
file('hello', 'w').write('hello')
100
file('buongia', 'w').write('buongia')
101
b.add(['hello', 'buongia'],
102
['hello-id', 'buongia-id'])
103
b.commit(message='add files',
107
file('buongia', 'w').write('new text')
108
b.commit(message='update text',
109
specific_files=['buongia'],
110
allow_pointless=False,
113
b.commit(message='remove hello',
114
specific_files=['hello'],
115
allow_pointless=False,
118
eq = self.assertEquals
121
tree2 = b.revision_tree('test@rev-2')
122
self.assertTrue(tree2.has_filename('hello'))
123
self.assertEquals(tree2.get_file_text('hello-id'), 'hello')
124
self.assertEquals(tree2.get_file_text('buongia-id'), 'new text')
126
tree3 = b.revision_tree('test@rev-3')
127
self.assertFalse(tree3.has_filename('hello'))
128
self.assertEquals(tree3.get_file_text('buongia-id'), 'new text')
131
def test_commit_rename(self):
132
"""Test commit of a revision where a file is renamed."""
133
b = Branch.initialize('.')
134
self.build_tree(['hello'])
135
b.add(['hello'], ['hello-id'])
136
b.commit(message='one', rev_id='test@rev-1', allow_pointless=False)
138
b.rename_one('hello', 'fruity')
139
b.commit(message='renamed', rev_id='test@rev-2', allow_pointless=False)
141
eq = self.assertEquals
142
tree1 = b.revision_tree('test@rev-1')
143
eq(tree1.id2path('hello-id'), 'hello')
144
eq(tree1.get_file_text('hello-id'), 'contents of hello\n')
145
self.assertFalse(tree1.has_filename('fruity'))
146
self.check_inventory_shape(tree1.inventory, ['hello'])
147
ie = tree1.inventory['hello-id']
148
eq(ie.revision, 'test@rev-1')
150
tree2 = b.revision_tree('test@rev-2')
151
eq(tree2.id2path('hello-id'), 'fruity')
152
eq(tree2.get_file_text('hello-id'), 'contents of hello\n')
153
self.check_inventory_shape(tree2.inventory, ['fruity'])
154
ie = tree2.inventory['hello-id']
155
eq(ie.revision, 'test@rev-2')
158
def test_reused_rev_id(self):
159
"""Test that a revision id cannot be reused in a branch"""
160
b = Branch.initialize('.')
161
b.commit('initial', rev_id='test@rev-1', allow_pointless=True)
162
self.assertRaises(Exception,
166
allow_pointless=True)
170
def test_commit_move(self):
171
"""Test commit of revisions with moved files and directories"""
172
eq = self.assertEquals
173
b = Branch.initialize('.')
175
self.build_tree(['hello', 'a/', 'b/'])
176
b.add(['hello', 'a', 'b'], ['hello-id', 'a-id', 'b-id'])
177
b.commit('initial', rev_id=r1, allow_pointless=False)
179
b.move(['hello'], 'a')
181
b.commit('two', rev_id=r2, allow_pointless=False)
182
self.check_inventory_shape(b.inventory,
183
['a', 'a/hello', 'b'])
187
b.commit('three', rev_id=r3, allow_pointless=False)
188
self.check_inventory_shape(b.inventory,
189
['a', 'a/hello', 'a/b'])
190
self.check_inventory_shape(b.get_revision_inventory(r3),
191
['a', 'a/hello', 'a/b'])
193
b.move([os.sep.join(['a', 'hello'])],
194
os.sep.join(['a', 'b']))
196
b.commit('four', rev_id=r4, allow_pointless=False)
197
self.check_inventory_shape(b.inventory,
198
['a', 'a/b/hello', 'a/b'])
200
inv = b.get_revision_inventory(r4)
201
eq(inv['hello-id'].revision, r4)
202
eq(inv['a-id'].revision, r1)
203
eq(inv['b-id'].revision, r3)
206
def test_removed_commit(self):
207
"""Test a commit with a removed file"""
208
b = Branch.initialize('.')
209
file('hello', 'w').write('hello world')
210
b.add(['hello'], ['hello-id'])
211
b.commit(message='add hello')
214
b.commit('removed hello', rev_id='rev2')
216
tree = b.revision_tree('rev2')
217
self.assertFalse(tree.has_id('hello-id'))
220
def test_committed_ancestry(self):
221
"""Test commit appends revisions to ancestry."""
222
b = Branch.initialize('.')
225
file('hello', 'w').write((str(i) * 4) + '\n')
227
b.add(['hello'], ['hello-id'])
228
rev_id = 'test@rev-%d' % (i+1)
229
rev_ids.append(rev_id)
230
b.commit(message='rev %d' % (i+1),
232
eq = self.assertEquals
233
eq(b.revision_history(), rev_ids)
235
anc = b.get_ancestry(rev_ids[i])
236
eq(anc, [None] + rev_ids[:i+1])
238
def test_commit_new_subdir_child_selective(self):
239
b = Branch.initialize('.')
240
self.build_tree(['dir/', 'dir/file1', 'dir/file2'])
241
b.add(['dir', 'dir/file1', 'dir/file2'],
242
['dirid', 'file1id', 'file2id'])
243
b.commit('dir/file1', specific_files=['dir/file1'], rev_id='1')
244
inv = b.get_inventory('1')
245
self.assertEqual('1', inv['dirid'].revision)
246
self.assertEqual('1', inv['file1id'].revision)
247
# FIXME: This should raise a KeyError I think, rbc20051006
248
self.assertRaises(BzrError, inv.__getitem__, 'file2id')