1
# Copyright (C) 2004 - 2008 Aaron Bentley, Canonical Ltd
1
# Copyright (C) 2004 - 2006 Aaron Bentley, Canonical Ltd
2
2
# <aaron.bentley@utoronto.ca>
4
4
# This program is free software; you can redistribute it and/or modify
16
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21
from bzrlib.tests import TestCase
23
22
from bzrlib.iterablefile import IterableFile
24
23
from bzrlib.patches import (MalformedLine,
25
24
MalformedHunkHeader,
39
class PatchesTester(TestCase):
38
class PatchesTester(unittest.TestCase):
41
39
def datafile(self, filename):
42
40
data_path = os.path.join(os.path.dirname(__file__),
43
41
"test_patches_data", filename)
47
45
"""Parse a valid patch header"""
48
46
lines = "--- orig/commands.py\n+++ mod/dommands.py\n".split('\n')
49
47
(orig, mod) = get_patch_names(lines.__iter__())
50
self.assertEqual(orig, "orig/commands.py")
51
self.assertEqual(mod, "mod/dommands.py")
48
assert(orig == "orig/commands.py")
49
assert(mod == "mod/dommands.py")
53
51
def testInvalidPatchHeader(self):
54
52
"""Parse an invalid patch header"""
60
58
"""Parse a valid hunk header"""
61
59
header = "@@ -34,11 +50,6 @@\n"
62
60
hunk = hunk_from_header(header);
63
self.assertEqual(hunk.orig_pos, 34)
64
self.assertEqual(hunk.orig_range, 11)
65
self.assertEqual(hunk.mod_pos, 50)
66
self.assertEqual(hunk.mod_range, 6)
67
self.assertEqual(str(hunk), 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)
69
67
def testValidHunkHeader2(self):
70
68
"""Parse a tricky, valid hunk header"""
71
69
header = "@@ -1 +0,0 @@\n"
72
70
hunk = hunk_from_header(header);
73
self.assertEqual(hunk.orig_pos, 1)
74
self.assertEqual(hunk.orig_range, 1)
75
self.assertEqual(hunk.mod_pos, 0)
76
self.assertEqual(hunk.mod_range, 0)
77
self.assertEqual(str(hunk), 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)
79
77
def testPDiff(self):
80
78
"""Parse a hunk header produced by diff -p"""
101
99
def lineThing(self,text, type):
102
100
line = parse_line(text)
103
self.assertIsInstance(line, type)
104
self.assertEqual(str(line), text)
101
assert(isinstance(line, type))
102
assert(str(line)==text)
106
104
def makeMalformedLine(self, text):
107
105
self.assertRaises(MalformedLine, parse_line, text)
162
160
if line.contents != next:
163
161
sys.stdout.write(" orig:%spatch:%s" % (next,
165
self.assertEqual(line.contents, next)
163
assert(line.contents == next)
166
164
self.assertRaises(StopIteration, rem_iter.next)
168
166
def testPatching(self):
190
188
def testFirstLineRenumber(self):
191
189
"""Make sure we handle lines at the beginning of the hunk"""
192
190
patch = parse_patch(self.datafile("insert_top.patch"))
193
self.assertEqual(patch.pos_in_mod(0), 1)
191
assert (patch.pos_in_mod(0)==1)
195
193
def testParsePatches(self):
196
194
"""Make sure file names can be extracted from tricky unified diffs"""
227
225
for patch in patches:
228
226
patch_files.append((patch.oldname, patch.newname))
229
self.assertEqual(patch_files, filenames)
227
assert (patch_files == filenames)
230
patchesTestSuite = unittest.makeSuite(PatchesTester,'test')
231
runner = unittest.TextTestRunner(verbosity=0)
232
return runner.run(patchesTestSuite)
235
if __name__ == "__main__":