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
"""Black-box tests for bzr diff.
25
from bzrlib.branch import Branch
26
from bzrlib import workingtree
27
from bzrlib.tests.blackbox import ExternalBase
30
def subst_dates(string):
31
"""Replace date strings with constant values."""
32
return re.sub(r'\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2} [-\+]\d{4}',
33
'YYYY-MM-DD HH:MM:SS +ZZZZ', string)
36
class TestDiff(ExternalBase):
38
def make_example_branch(test):
39
# FIXME: copied from test_too_much -- share elsewhere?
41
file('hello', 'wb').write('foo\n')
42
test.runbzr('add hello')
43
test.runbzr('commit -m setup hello')
44
file('goodbye', 'wb').write('baz\n')
45
test.runbzr('add goodbye')
46
test.runbzr('commit -m setup goodbye')
49
self.make_example_branch()
50
file('hello', 'wt').write('hello world!')
51
self.runbzr('commit -m fixing hello')
52
output = self.runbzr('diff -r 2..3', backtick=1, retcode=1)
53
self.assert_('\n+hello world!' in output)
54
output = self.runbzr('diff -r last:3..last:1', backtick=1, retcode=1)
55
self.assert_('\n+baz' in output)
56
file('moo', 'wb').write('moo')
57
self.runbzr('add moo')
61
def test_diff_prefix(self):
62
"""diff --prefix appends to filenames in output"""
63
self.make_example_branch()
64
file('hello', 'wb').write('hello world!\n')
65
out, err = self.runbzr('diff --prefix old/:new/', retcode=1)
66
self.assertEquals(err, '')
67
self.assertEqualDiff(subst_dates(out), '''\
68
=== modified file 'hello'
69
--- old/hello\tYYYY-MM-DD HH:MM:SS +ZZZZ
70
+++ new/hello\tYYYY-MM-DD HH:MM:SS +ZZZZ
77
def test_diff_p1(self):
78
"""diff -p1 produces lkml-style diffs"""
79
self.make_example_branch()
80
file('hello', 'wb').write('hello world!\n')
81
out, err = self.runbzr('diff -p1', retcode=1)
82
self.assertEquals(err, '')
83
self.assertEqualDiff(subst_dates(out), '''\
84
=== modified file 'hello'
85
--- old/hello\tYYYY-MM-DD HH:MM:SS +ZZZZ
86
+++ new/hello\tYYYY-MM-DD HH:MM:SS +ZZZZ
93
def test_diff_p0(self):
94
"""diff -p0 produces diffs with no prefix"""
95
self.make_example_branch()
96
file('hello', 'wb').write('hello world!\n')
97
out, err = self.runbzr('diff -p0', retcode=1)
98
self.assertEquals(err, '')
99
self.assertEqualDiff(subst_dates(out), '''\
100
=== modified file 'hello'
101
--- hello\tYYYY-MM-DD HH:MM:SS +ZZZZ
102
+++ hello\tYYYY-MM-DD HH:MM:SS +ZZZZ
109
def test_diff_nonexistent(self):
110
# Get an error from a file that does not exist at all
112
self.make_example_branch()
113
out, err = self.runbzr('diff does-not-exist', retcode=3)
114
self.assertContainsRe(err, 'not versioned.*does-not-exist')
116
def test_diff_unversioned(self):
117
# Get an error when diffing a non-versioned file.
119
self.make_example_branch()
120
self.build_tree(['unversioned-file'])
121
out, err = self.runbzr('diff unversioned-file', retcode=3)
122
self.assertContainsRe(err, 'not versioned.*unversioned-file')
124
# TODO: What should diff say for a file deleted in working tree?
126
def example_branches(self):
127
self.build_tree(['branch1/', 'branch1/file'], line_endings='binary')
128
self.capture('init branch1')
129
self.capture('add branch1/file')
130
self.run_bzr_captured(['commit', '-m', 'add file', 'branch1'])
131
self.capture('branch branch1 branch2')
132
print >> open('branch2/file', 'wb'), 'new content'
133
self.run_bzr_captured(['commit', '-m', 'update file', 'branch2'])
135
def test_diff_branches(self):
136
self.example_branches()
137
# should open branch1 and diff against branch2,
138
out, err = self.run_bzr_captured(['diff', '-r', 'branch:branch2',
141
self.assertEquals('', err)
142
self.assertEquals("=== modified file 'file'\n"
143
"--- file\tYYYY-MM-DD HH:MM:SS +ZZZZ\n"
144
"+++ file\tYYYY-MM-DD HH:MM:SS +ZZZZ\n"
147
"+contents of branch1/file\n"
148
"\n", subst_dates(out))
149
out, err = self.run_bzr_captured(['diff', 'branch2', 'branch1'],
151
self.assertEquals('', err)
152
self.assertEqualDiff("=== modified file 'file'\n"
153
"--- file\tYYYY-MM-DD HH:MM:SS +ZZZZ\n"
154
"+++ file\tYYYY-MM-DD HH:MM:SS +ZZZZ\n"
157
"+contents of branch1/file\n"
158
"\n", subst_dates(out))
160
def test_diff_revno_branches(self):
161
self.example_branches()
162
print >> open('branch2/file', 'wb'), 'even newer content'
163
self.run_bzr_captured(['commit', '-m',
164
'update file once more', 'branch2'])
166
out, err = self.run_bzr_captured(['diff', '-r',
167
'revno:1:branch2..revno:1:branch1'],
169
self.assertEquals('', err)
170
self.assertEquals('', out)
171
out, err = self.run_bzr_captured(['diff', '-r',
172
'revno:2:branch2..revno:1:branch1'],
174
self.assertEquals('', err)
175
self.assertEqualDiff("=== modified file 'file'\n"
176
"--- file\tYYYY-MM-DD HH:MM:SS +ZZZZ\n"
177
"+++ file\tYYYY-MM-DD HH:MM:SS +ZZZZ\n"
180
"+contents of branch1/file\n"
181
"\n", subst_dates(out))
183
def example_branch2(self):
184
self.build_tree(['branch1/', 'branch1/file1'], line_endings='binary')
185
self.capture('init branch1')
186
self.capture('add branch1/file1')
187
print >> open('branch1/file1', 'wb'), 'original line'
188
self.run_bzr_captured(['commit', '-m', 'first commit', 'branch1'])
190
print >> open('branch1/file1', 'wb'), 'repo line'
191
self.run_bzr_captured(['commit', '-m', 'second commit', 'branch1'])
193
def test_diff_to_working_tree(self):
194
self.example_branch2()
196
print >> open('branch1/file1', 'wb'), 'new line'
197
output = self.run_bzr_captured(['diff', '-r', '1..', 'branch1'],
199
self.assertTrue('\n-original line\n+new line\n' in output[0])
201
def test_diff_across_rename(self):
202
"""The working tree path should always be considered for diffing"""
203
self.make_example_branch()
204
self.run_bzr('diff', '-r', '0..1', 'hello', retcode=1)
205
wt = workingtree.WorkingTree.open_containing('.')[0]
206
wt.rename_one('hello', 'hello1')
207
self.run_bzr('diff', 'hello1', retcode=1)
208
self.run_bzr('diff', '-r', '0..1', 'hello1', retcode=1)
211
class TestCheckoutDiff(TestDiff):
213
def make_example_branch(self):
214
super(TestCheckoutDiff, self).make_example_branch()
215
self.runbzr('checkout . checkout')
218
def example_branch2(self):
219
super(TestCheckoutDiff, self).example_branch2()
220
os.mkdir('checkouts')
221
self.runbzr('checkout branch1 checkouts/branch1')
222
os.chdir('checkouts')
224
def example_branches(self):
225
super(TestCheckoutDiff, self).example_branches()
226
os.mkdir('checkouts')
227
self.runbzr('checkout branch1 checkouts/branch1')
228
self.runbzr('checkout branch2 checkouts/branch2')
229
os.chdir('checkouts')
232
class TestDiffLabels(TestDiff):
234
def test_diff_label_removed(self):
235
super(TestDiffLabels, self).make_example_branch()
236
self.runbzr('remove hello')
237
diff = self.run_bzr_captured(['diff'], retcode=1)
238
self.assertTrue("=== removed file 'hello'" in diff[0])
240
def test_diff_label_added(self):
241
super(TestDiffLabels, self).make_example_branch()
242
file('barbar', 'wt').write('barbar')
243
self.runbzr('add barbar')
244
diff = self.run_bzr_captured(['diff'], retcode=1)
245
self.assertTrue("=== added file 'barbar'" in diff[0])
247
def test_diff_label_modified(self):
248
super(TestDiffLabels, self).make_example_branch()
249
file('hello', 'wt').write('barbar')
250
diff = self.run_bzr_captured(['diff'], retcode=1)
251
self.assertTrue("=== modified file 'hello'" in diff[0])
253
def test_diff_label_renamed(self):
254
super(TestDiffLabels, self).make_example_branch()
255
self.runbzr('rename hello gruezi')
256
diff = self.run_bzr_captured(['diff'], retcode=1)
257
self.assertTrue("=== renamed file 'hello' => 'gruezi'" in diff[0])