/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_patches.py

  • Committer: Robert Collins
  • Date: 2008-08-20 02:07:36 UTC
  • mfrom: (3640 +trunk)
  • mto: This revision was merged to the branch mainline in revision 3682.
  • Revision ID: robertc@robertcollins.net-20080820020736-g2xe4921zzxtymle
Merge bzr.dev

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_patches.py
1
 
# Copyright (C) 2004 - 2006 Aaron Bentley
 
1
# Copyright (C) 2004 - 2008 Aaron Bentley, Canonical Ltd
2
2
# <aaron.bentley@utoronto.ca>
3
3
#
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.
8
 
#
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.
13
 
#
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
17
 
 
18
 
 
19
 
import unittest
 
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.
 
8
#
 
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.
 
13
#
 
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
 
17
 
 
18
 
20
19
import os.path
21
20
 
 
21
from bzrlib.tests import TestCase
 
22
 
22
23
from bzrlib.iterablefile import IterableFile
23
24
from bzrlib.patches import (MalformedLine, 
24
25
                            MalformedHunkHeader, 
31
32
                            hunk_from_header, 
32
33
                            iter_patched, 
33
34
                            parse_line,
34
 
                            parse_patch)
35
 
 
36
 
 
37
 
class PatchesTester(unittest.TestCase):
 
35
                            parse_patch,
 
36
                            parse_patches)
 
37
 
 
38
 
 
39
class PatchesTester(TestCase):
 
40
 
38
41
    def datafile(self, filename):
39
42
        data_path = os.path.join(os.path.dirname(__file__), 
40
43
                                 "test_patches_data", filename)
44
47
        """Parse a valid patch header"""
45
48
        lines = "--- orig/commands.py\n+++ mod/dommands.py\n".split('\n')
46
49
        (orig, mod) = get_patch_names(lines.__iter__())
47
 
        assert(orig == "orig/commands.py")
48
 
        assert(mod == "mod/dommands.py")
 
50
        self.assertEqual(orig, "orig/commands.py")
 
51
        self.assertEqual(mod, "mod/dommands.py")
49
52
 
50
53
    def testInvalidPatchHeader(self):
51
54
        """Parse an invalid patch header"""
57
60
        """Parse a valid hunk header"""
58
61
        header = "@@ -34,11 +50,6 @@\n"
59
62
        hunk = hunk_from_header(header);
60
 
        assert (hunk.orig_pos == 34)
61
 
        assert (hunk.orig_range == 11)
62
 
        assert (hunk.mod_pos == 50)
63
 
        assert (hunk.mod_range == 6)
64
 
        assert (str(hunk) == 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)
65
68
 
66
69
    def testValidHunkHeader2(self):
67
70
        """Parse a tricky, valid hunk header"""
68
71
        header = "@@ -1 +0,0 @@\n"
69
72
        hunk = hunk_from_header(header);
70
 
        assert (hunk.orig_pos == 1)
71
 
        assert (hunk.orig_range == 1)
72
 
        assert (hunk.mod_pos == 0)
73
 
        assert (hunk.mod_range == 0)
74
 
        assert (str(hunk) == 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)
 
78
 
 
79
    def testPDiff(self):
 
80
        """Parse a hunk header produced by diff -p"""
 
81
        header = "@@ -407,7 +292,7 @@ bzr 0.18rc1  2007-07-10\n"
 
82
        hunk = hunk_from_header(header)
 
83
        self.assertEqual('bzr 0.18rc1  2007-07-10', hunk.tail)
 
84
        self.assertEqual(header, str(hunk))
75
85
 
76
86
    def makeMalformed(self, header):
77
87
        self.assertRaises(MalformedHunkHeader, hunk_from_header, header)
90
100
 
91
101
    def lineThing(self,text, type):
92
102
        line = parse_line(text)
93
 
        assert(isinstance(line, type))
94
 
        assert(str(line)==text)
 
103
        self.assertIsInstance(line, type)
 
104
        self.assertEqual(str(line), text)
95
105
 
96
106
    def makeMalformedLine(self, text):
97
107
        self.assertRaises(MalformedLine, parse_line, text)
143
153
            if mod_pos is None:
144
154
                removals.append(orig[i])
145
155
                continue
146
 
            assert(mod[mod_pos]==orig[i])
 
156
            self.assertEqual(mod[mod_pos], orig[i])
147
157
        rem_iter = removals.__iter__()
148
158
        for hunk in patch.hunks:
149
159
            for line in hunk.lines:
152
162
                    if line.contents != next:
153
163
                        sys.stdout.write(" orig:%spatch:%s" % (next,
154
164
                                         line.contents))
155
 
                    assert(line.contents == next)
 
165
                    self.assertEqual(line.contents, next)
156
166
        self.assertRaises(StopIteration, rem_iter.next)
157
167
 
158
168
    def testPatching(self):
180
190
    def testFirstLineRenumber(self):
181
191
        """Make sure we handle lines at the beginning of the hunk"""
182
192
        patch = parse_patch(self.datafile("insert_top.patch"))
183
 
        assert (patch.pos_in_mod(0)==1)
184
 
 
185
 
def test():
186
 
    patchesTestSuite = unittest.makeSuite(PatchesTester,'test')
187
 
    runner = unittest.TextTestRunner(verbosity=0)
188
 
    return runner.run(patchesTestSuite)
189
 
 
190
 
 
191
 
if __name__ == "__main__":
192
 
    test()
 
193
        self.assertEqual(patch.pos_in_mod(0), 1)
 
194
 
 
195
    def testParsePatches(self):
 
196
        """Make sure file names can be extracted from tricky unified diffs"""
 
197
        patchtext = \
 
198
"""--- orig-7
 
199
+++ mod-7
 
200
@@ -1,10 +1,10 @@
 
201
 -- a
 
202
--- b
 
203
+++ c
 
204
 xx d
 
205
 xx e
 
206
 ++ f
 
207
-++ g
 
208
+-- h
 
209
 xx i
 
210
 xx j
 
211
 -- k
 
212
--- l
 
213
+++ m
 
214
--- orig-8
 
215
+++ mod-8
 
216
@@ -1 +1 @@
 
217
--- A
 
218
+++ B
 
219
@@ -1 +1 @@
 
220
--- C
 
221
+++ D
 
222
"""
 
223
        filenames = [('orig-7', 'mod-7'),
 
224
                     ('orig-8', 'mod-8')]
 
225
        patches = parse_patches(patchtext.splitlines(True))
 
226
        patch_files = []
 
227
        for patch in patches:
 
228
            patch_files.append((patch.oldname, patch.newname))
 
229
        self.assertEqual(patch_files, filenames)