1
# Copyright (C) 2005, 2006 by Canonical Ltd
1
# Copyright (C) 2005, 2006 Canonical Ltd
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
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
11
# GNU General Public License for more details.
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
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
25
from bzrlib import workingtree
24
26
from bzrlib.branch import Branch
27
from bzrlib.tests import TestSkipped
25
28
from bzrlib.tests.blackbox import ExternalBase
28
class TestDiff(ExternalBase):
30
def make_example_branch(test):
31
def subst_dates(string):
32
"""Replace date strings with constant values."""
33
return re.sub(r'\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2} [-\+]\d{4}',
34
'YYYY-MM-DD HH:MM:SS +ZZZZ', string)
37
class DiffBase(ExternalBase):
38
"""Base class with common setup method"""
40
def make_example_branch(self):
31
41
# FIXME: copied from test_too_much -- share elsewhere?
33
file('hello', 'wt').write('foo')
34
test.runbzr('add hello')
35
test.runbzr('commit -m setup hello')
36
file('goodbye', 'wt').write('baz')
37
test.runbzr('add goodbye')
38
test.runbzr('commit -m setup goodbye')
42
tree = self.make_branch_and_tree('.')
43
open('hello', 'wb').write('foo\n')
46
open('goodbye', 'wb').write('baz\n')
51
class TestDiff(DiffBase):
40
53
def test_diff(self):
41
54
self.make_example_branch()
51
64
self.runbzr('diff')
66
def test_diff_prefix(self):
67
"""diff --prefix appends to filenames in output"""
68
self.make_example_branch()
69
file('hello', 'wb').write('hello world!\n')
70
out, err = self.runbzr('diff --prefix old/:new/', retcode=1)
71
self.assertEquals(err, '')
72
self.assertEqualDiff(subst_dates(out), '''\
73
=== modified file 'hello'
74
--- old/hello\tYYYY-MM-DD HH:MM:SS +ZZZZ
75
+++ new/hello\tYYYY-MM-DD HH:MM:SS +ZZZZ
82
def test_diff_p1(self):
83
"""diff -p1 produces lkml-style diffs"""
84
self.make_example_branch()
85
file('hello', 'wb').write('hello world!\n')
86
out, err = self.runbzr('diff -p1', retcode=1)
87
self.assertEquals(err, '')
88
self.assertEqualDiff(subst_dates(out), '''\
89
=== modified file 'hello'
90
--- old/hello\tYYYY-MM-DD HH:MM:SS +ZZZZ
91
+++ new/hello\tYYYY-MM-DD HH:MM:SS +ZZZZ
98
def test_diff_p0(self):
99
"""diff -p0 produces diffs with no prefix"""
100
self.make_example_branch()
101
file('hello', 'wb').write('hello world!\n')
102
out, err = self.runbzr('diff -p0', retcode=1)
103
self.assertEquals(err, '')
104
self.assertEqualDiff(subst_dates(out), '''\
105
=== modified file 'hello'
106
--- hello\tYYYY-MM-DD HH:MM:SS +ZZZZ
107
+++ hello\tYYYY-MM-DD HH:MM:SS +ZZZZ
53
114
def test_diff_nonexistent(self):
54
115
# Get an error from a file that does not exist at all
57
118
out, err = self.runbzr('diff does-not-exist', retcode=3)
58
119
self.assertContainsRe(err, 'not versioned.*does-not-exist')
121
def test_diff_illegal_revision_specifiers(self):
122
out, err = self.runbzr('diff -r 1..23..123', retcode=3)
123
self.assertContainsRe(err, 'one or two revision specifiers')
60
125
def test_diff_unversioned(self):
61
126
# Get an error when diffing a non-versioned file.
79
144
def test_diff_branches(self):
80
145
self.example_branches()
81
146
# should open branch1 and diff against branch2,
82
output = self.run_bzr_captured(['diff', '-r', 'branch:branch2',
85
self.assertEquals(("=== modified file 'a/file'\n"
90
"+contents of branch1/file\n"
92
output = self.run_bzr_captured(['diff', 'branch2', 'branch1'],
94
self.assertEqualDiff(("=== modified file 'a/file'\n"
99
"+contents of branch1/file\n"
147
out, err = self.run_bzr_captured(['diff', '-r', 'branch:branch2',
150
self.assertEquals('', err)
151
self.assertEquals("=== modified file 'file'\n"
152
"--- file\tYYYY-MM-DD HH:MM:SS +ZZZZ\n"
153
"+++ file\tYYYY-MM-DD HH:MM:SS +ZZZZ\n"
156
"+contents of branch1/file\n"
157
"\n", subst_dates(out))
158
out, err = self.run_bzr_captured(['diff', 'branch2', 'branch1'],
160
self.assertEquals('', err)
161
self.assertEqualDiff("=== modified file 'file'\n"
162
"--- file\tYYYY-MM-DD HH:MM:SS +ZZZZ\n"
163
"+++ file\tYYYY-MM-DD HH:MM:SS +ZZZZ\n"
166
"+contents of branch1/file\n"
167
"\n", subst_dates(out))
169
def test_diff_revno_branches(self):
170
self.example_branches()
171
print >> open('branch2/file', 'wb'), 'even newer content'
172
self.run_bzr_captured(['commit', '-m',
173
'update file once more', 'branch2'])
175
out, err = self.run_bzr_captured(['diff', '-r',
176
'revno:1:branch2..revno:1:branch1'],
178
self.assertEquals('', err)
179
self.assertEquals('', out)
180
out, err = self.run_bzr_captured(['diff', '-r',
181
'revno:2:branch2..revno:1:branch1'],
183
self.assertEquals('', err)
184
self.assertEqualDiff("=== modified file 'file'\n"
185
"--- file\tYYYY-MM-DD HH:MM:SS +ZZZZ\n"
186
"+++ file\tYYYY-MM-DD HH:MM:SS +ZZZZ\n"
189
"+contents of branch1/file\n"
190
"\n", subst_dates(out))
102
192
def example_branch2(self):
103
193
self.build_tree(['branch1/', 'branch1/file1'], line_endings='binary')
113
203
self.example_branch2()
115
205
print >> open('branch1/file1', 'wb'), 'new line'
116
output = self.run_bzr_captured(['diff', '-r', '1..', 'branch1'], retcode=1)
206
output = self.run_bzr_captured(['diff', '-r', '1..', 'branch1'],
117
208
self.assertTrue('\n-original line\n+new line\n' in output[0])
210
def test_diff_across_rename(self):
211
"""The working tree path should always be considered for diffing"""
212
self.make_example_branch()
213
self.run_bzr('diff', '-r', '0..1', 'hello', retcode=1)
214
wt = workingtree.WorkingTree.open_containing('.')[0]
215
wt.rename_one('hello', 'hello1')
216
self.run_bzr('diff', 'hello1', retcode=1)
217
self.run_bzr('diff', '-r', '0..1', 'hello1', retcode=1)
120
220
class TestCheckoutDiff(TestDiff):
138
238
os.chdir('checkouts')
141
class TestDiffLabels(TestDiff):
241
class TestDiffLabels(DiffBase):
143
243
def test_diff_label_removed(self):
144
244
super(TestDiffLabels, self).make_example_branch()
145
245
self.runbzr('remove hello')
146
246
diff = self.run_bzr_captured(['diff'], retcode=1)
147
self.assertTrue("=== removed file 'a/hello'" in diff[0])
247
self.assertTrue("=== removed file 'hello'" in diff[0])
149
249
def test_diff_label_added(self):
150
250
super(TestDiffLabels, self).make_example_branch()
151
251
file('barbar', 'wt').write('barbar')
152
252
self.runbzr('add barbar')
153
253
diff = self.run_bzr_captured(['diff'], retcode=1)
154
self.assertTrue("=== added file 'b/barbar'" in diff[0])
254
self.assertTrue("=== added file 'barbar'" in diff[0])
156
256
def test_diff_label_modified(self):
157
257
super(TestDiffLabels, self).make_example_branch()
158
258
file('hello', 'wt').write('barbar')
159
259
diff = self.run_bzr_captured(['diff'], retcode=1)
160
self.assertTrue("=== modified file 'a/hello'" in diff[0])
260
self.assertTrue("=== modified file 'hello'" in diff[0])
162
262
def test_diff_label_renamed(self):
163
263
super(TestDiffLabels, self).make_example_branch()
164
264
self.runbzr('rename hello gruezi')
165
265
diff = self.run_bzr_captured(['diff'], retcode=1)
166
self.assertTrue("=== renamed file 'a/hello' => 'b/gruezi'" in diff[0])
266
self.assertTrue("=== renamed file 'hello' => 'gruezi'" in diff[0])
269
class TestExternalDiff(DiffBase):
271
def test_external_diff(self):
272
"""Test that we can spawn an external diff process"""
273
# We have to use run_bzr_subprocess, because we need to
274
# test writing directly to stdout, (there was a bug in
275
# subprocess.py that we had to workaround).
276
# However, if 'diff' may not be available
277
self.make_example_branch()
278
orig_progress = os.environ.get('BZR_PROGRESS_BAR')
280
os.environ['BZR_PROGRESS_BAR'] = 'none'
281
out, err = self.run_bzr_subprocess('diff', '-r', '1',
282
'--diff-options', '-ub',
283
universal_newlines=True,
286
if orig_progress is None:
287
del os.environ['BZR_PROGRESS_BAR']
289
os.environ['BZR_PROGRESS_BAR'] = orig_progress
291
if 'Diff is not installed on this machine' in err:
292
raise TestSkipped("No external 'diff' is available")
293
self.assertEqual('', err)
294
# We have to skip the stuff in the middle, because it depends
296
self.assertStartsWith(out, "=== added file 'goodbye'\n"
297
"--- goodbye\t1970-01-01 00:00:00 +0000\n"
299
self.assertEndsWith(out, "\n@@ -0,0 +1 @@\n"
303
class TestDiffOutput(DiffBase):
305
def test_diff_output(self):
306
# check that output doesn't mangle line-endings
307
self.make_example_branch()
308
file('hello', 'wb').write('hello world!\n')
309
output = self.run_bzr_subprocess('diff', retcode=1)[0]
310
self.assert_('\n+hello world!\n' in output)