1
# Copyright (C) 2004 - 2006 Aaron Bentley, Canonical Ltd
2
# <aaron.bentley@utoronto.ca>
4
# This program is free software; you can redistribute it and/or modify
5
# it under the terms of the GNU General Public License as published by
6
# the Free Software Foundation; either version 2 of the License, or
7
# (at your option) any later version.
9
# This program is distributed in the hope that it will be useful,
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12
# GNU General Public License for more details.
14
# You should have received a copy of the GNU General Public License
15
# along with this program; if not, write to the Free Software
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22
from bzrlib.iterablefile import IterableFile
23
from bzrlib.patches import (MalformedLine,
38
class PatchesTester(unittest.TestCase):
39
def datafile(self, filename):
40
data_path = os.path.join(os.path.dirname(__file__),
41
"test_patches_data", filename)
42
return file(data_path, "rb")
44
def testValidPatchHeader(self):
45
"""Parse a valid patch header"""
46
lines = "--- orig/commands.py\n+++ mod/dommands.py\n".split('\n')
47
(orig, mod) = get_patch_names(lines.__iter__())
48
assert(orig == "orig/commands.py")
49
assert(mod == "mod/dommands.py")
51
def testInvalidPatchHeader(self):
52
"""Parse an invalid patch header"""
53
lines = "-- orig/commands.py\n+++ mod/dommands.py".split('\n')
54
self.assertRaises(MalformedPatchHeader, get_patch_names,
57
def testValidHunkHeader(self):
58
"""Parse a valid hunk header"""
59
header = "@@ -34,11 +50,6 @@\n"
60
hunk = hunk_from_header(header);
61
assert (hunk.orig_pos == 34)
62
assert (hunk.orig_range == 11)
63
assert (hunk.mod_pos == 50)
64
assert (hunk.mod_range == 6)
65
assert (str(hunk) == header)
67
def testValidHunkHeader2(self):
68
"""Parse a tricky, valid hunk header"""
69
header = "@@ -1 +0,0 @@\n"
70
hunk = hunk_from_header(header);
71
assert (hunk.orig_pos == 1)
72
assert (hunk.orig_range == 1)
73
assert (hunk.mod_pos == 0)
74
assert (hunk.mod_range == 0)
75
assert (str(hunk) == header)
77
def makeMalformed(self, header):
78
self.assertRaises(MalformedHunkHeader, hunk_from_header, header)
80
def testInvalidHeader(self):
81
"""Parse an invalid hunk header"""
82
self.makeMalformed(" -34,11 +50,6 \n")
83
self.makeMalformed("@@ +50,6 -34,11 @@\n")
84
self.makeMalformed("@@ -34,11 +50,6 @@")
85
self.makeMalformed("@@ -34.5,11 +50,6 @@\n")
86
self.makeMalformed("@@-34,11 +50,6@@\n")
87
self.makeMalformed("@@ 34,11 50,6 @@\n")
88
self.makeMalformed("@@ -34,11 @@\n")
89
self.makeMalformed("@@ -34,11 +50,6.5 @@\n")
90
self.makeMalformed("@@ -34,11 +50,-6 @@\n")
92
def lineThing(self,text, type):
93
line = parse_line(text)
94
assert(isinstance(line, type))
95
assert(str(line)==text)
97
def makeMalformedLine(self, text):
98
self.assertRaises(MalformedLine, parse_line, text)
100
def testValidLine(self):
101
"""Parse a valid hunk line"""
102
self.lineThing(" hello\n", ContextLine)
103
self.lineThing("+hello\n", InsertLine)
104
self.lineThing("-hello\n", RemoveLine)
106
def testMalformedLine(self):
107
"""Parse invalid valid hunk lines"""
108
self.makeMalformedLine("hello\n")
110
def compare_parsed(self, patchtext):
111
lines = patchtext.splitlines(True)
112
patch = parse_patch(lines.__iter__())
114
i = difference_index(patchtext, pstr)
116
print "%i: \"%s\" != \"%s\"" % (i, patchtext[i], pstr[i])
117
self.assertEqual (patchtext, str(patch))
120
"""Test parsing a whole patch"""
121
patchtext = self.datafile("patchtext.patch").read()
122
self.compare_parsed(patchtext)
125
"""Handle patches missing half the position, range tuple"""
127
"""--- orig/__vavg__.cl
130
__qbpsbezng__ = "erfgehpgherqgrkg ra"
131
+__qbp__ = Na nygreangr Nepu pbzznaqyvar vagresnpr
133
self.compare_parsed(patchtext)
135
def testLineLookup(self):
137
"""Make sure we can accurately look up mod line from orig"""
138
patch = parse_patch(self.datafile("diff"))
139
orig = list(self.datafile("orig"))
140
mod = list(self.datafile("mod"))
142
for i in range(len(orig)):
143
mod_pos = patch.pos_in_mod(i)
145
removals.append(orig[i])
147
assert(mod[mod_pos]==orig[i])
148
rem_iter = removals.__iter__()
149
for hunk in patch.hunks:
150
for line in hunk.lines:
151
if isinstance(line, RemoveLine):
152
next = rem_iter.next()
153
if line.contents != next:
154
sys.stdout.write(" orig:%spatch:%s" % (next,
156
assert(line.contents == next)
157
self.assertRaises(StopIteration, rem_iter.next)
159
def testPatching(self):
160
"""Test a few patch files, and make sure they work."""
162
('diff-2', 'orig-2', 'mod-2'),
163
('diff-3', 'orig-3', 'mod-3'),
164
('diff-4', 'orig-4', 'mod-4'),
165
('diff-5', 'orig-5', 'mod-5'),
166
('diff-6', 'orig-6', 'mod-6'),
168
for diff, orig, mod in files:
169
patch = self.datafile(diff)
170
orig_lines = list(self.datafile(orig))
171
mod_lines = list(self.datafile(mod))
173
patched_file = IterableFile(iter_patched(orig_lines, patch))
176
for patch_line in patched_file:
177
self.assertEqual(patch_line, mod_lines[count])
179
self.assertEqual(count, len(mod_lines))
181
def testFirstLineRenumber(self):
182
"""Make sure we handle lines at the beginning of the hunk"""
183
patch = parse_patch(self.datafile("insert_top.patch"))
184
assert (patch.pos_in_mod(0)==1)
186
def testParsePatches(self):
187
"""Make sure file names can be extracted from tricky unified diffs"""
214
filenames = [('orig-7', 'mod-7'),
216
patches = parse_patches(patchtext.splitlines(True))
218
for patch in patches:
219
patch_files.append((patch.oldname, patch.newname))
220
assert (patch_files == filenames)
223
patchesTestSuite = unittest.makeSuite(PatchesTester,'test')
224
runner = unittest.TextTestRunner(verbosity=0)
225
return runner.run(patchesTestSuite)
228
if __name__ == "__main__":
added
removed
bzrlib/tests/test_patches.py
