~bzr-pqm/bzr/bzr.dev

# Copyright (C) 2004 - 2006 Aaron Bentley

# <aaron.bentley@utoronto.ca>

#

#    This program is free software; you can redistribute it and/or modify

#    it under the terms of the GNU General Public License as published by

#    the Free Software Foundation; either version 2 of the License, or

#    (at your option) any later version.

#

#    This program is distributed in the hope that it will be useful,

#    but WITHOUT ANY WARRANTY; without even the implied warranty of

#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

#    GNU General Public License for more details.

#

#    You should have received a copy of the GNU General Public License

#    along with this program; if not, write to the Free Software

#    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

import unittest

import os.path

from bzrlib.iterablefile import IterableFile

from bzrlib.patches import (MalformedLine, 

                            MalformedHunkHeader, 

                            MalformedPatchHeader, 

                            ContextLine, 

                            InsertLine,

                            RemoveLine, 

                            difference_index, 

                            get_patch_names,

                            hunk_from_header, 

                            iter_patched, 

                            parse_line,

                            parse_patch)

class PatchesTester(unittest.TestCase):

    def datafile(self, filename):

        data_path = os.path.join(os.path.dirname(__file__), 

                                 "test_patches_data", filename)

        return file(data_path, "rb")

    def testValidPatchHeader(self):

        """Parse a valid patch header"""

        lines = "--- orig/commands.py\n+++ mod/dommands.py\n".split('\n')

        (orig, mod) = get_patch_names(lines.__iter__())

        assert(orig == "orig/commands.py")

        assert(mod == "mod/dommands.py")

    def testInvalidPatchHeader(self):

        """Parse an invalid patch header"""

        lines = "-- orig/commands.py\n+++ mod/dommands.py".split('\n')

        self.assertRaises(MalformedPatchHeader, get_patch_names,

                          lines.__iter__())

    def testValidHunkHeader(self):

        """Parse a valid hunk header"""

        header = "@@ -34,11 +50,6 @@\n"

        hunk = hunk_from_header(header);

        assert (hunk.orig_pos == 34)

        assert (hunk.orig_range == 11)

        assert (hunk.mod_pos == 50)

        assert (hunk.mod_range == 6)

        assert (str(hunk) == header)

    def testValidHunkHeader2(self):

        """Parse a tricky, valid hunk header"""

        header = "@@ -1 +0,0 @@\n"

        hunk = hunk_from_header(header);

        assert (hunk.orig_pos == 1)

        assert (hunk.orig_range == 1)

        assert (hunk.mod_pos == 0)

        assert (hunk.mod_range == 0)

        assert (str(hunk) == header)

    def makeMalformed(self, header):

        self.assertRaises(MalformedHunkHeader, hunk_from_header, header)

    def testInvalidHeader(self):

        """Parse an invalid hunk header"""

        self.makeMalformed(" -34,11 +50,6 \n")

        self.makeMalformed("@@ +50,6 -34,11 @@\n")

        self.makeMalformed("@@ -34,11 +50,6 @@")

        self.makeMalformed("@@ -34.5,11 +50,6 @@\n")

        self.makeMalformed("@@-34,11 +50,6@@\n")

        self.makeMalformed("@@ 34,11 50,6 @@\n")

        self.makeMalformed("@@ -34,11 @@\n")

        self.makeMalformed("@@ -34,11 +50,6.5 @@\n")

        self.makeMalformed("@@ -34,11 +50,-6 @@\n")

    def lineThing(self,text, type):

        line = parse_line(text)

        assert(isinstance(line, type))

        assert(str(line)==text)

    def makeMalformedLine(self, text):

        self.assertRaises(MalformedLine, parse_line, text)

    def testValidLine(self):

        """Parse a valid hunk line"""

        self.lineThing(" hello\n", ContextLine)

        self.lineThing("+hello\n", InsertLine)

        self.lineThing("-hello\n", RemoveLine)

    def testMalformedLine(self):

        """Parse invalid valid hunk lines"""

        self.makeMalformedLine("hello\n")

    def compare_parsed(self, patchtext):

        lines = patchtext.splitlines(True)

        patch = parse_patch(lines.__iter__())

        pstr = str(patch)

        i = difference_index(patchtext, pstr)

        if i is not None:

            print "%i: \"%s\" != \"%s\"" % (i, patchtext[i], pstr[i])

        self.assertEqual (patchtext, str(patch))

    def testAll(self):

        """Test parsing a whole patch"""

        patchtext = self.datafile("patchtext.patch").read()

        self.compare_parsed(patchtext)

    def testInit(self):

        """Handle patches missing half the position, range tuple"""

        patchtext = \

"""--- orig/__vavg__.cl

+++ mod/__vavg__.cl

@@ -1 +1,2 @@

 __qbpsbezng__ = "erfgehpgherqgrkg ra"

+__qbp__ = Na nygreangr Nepu pbzznaqyvar vagresnpr

"""

        self.compare_parsed(patchtext)

    def testLineLookup(self):

        import sys

        """Make sure we can accurately look up mod line from orig"""

        patch = parse_patch(self.datafile("diff"))

        orig = list(self.datafile("orig"))

        mod = list(self.datafile("mod"))

        removals = []

        for i in range(len(orig)):

            mod_pos = patch.pos_in_mod(i)

            if mod_pos is None:

                removals.append(orig[i])

                continue

            assert(mod[mod_pos]==orig[i])

        rem_iter = removals.__iter__()

        for hunk in patch.hunks:

            for line in hunk.lines:

                if isinstance(line, RemoveLine):

                    next = rem_iter.next()

                    if line.contents != next:

                        sys.stdout.write(" orig:%spatch:%s" % (next,

                                         line.contents))

                    assert(line.contents == next)

        self.assertRaises(StopIteration, rem_iter.next)

    def testPatching(self):

        """Test a few patch files, and make sure they work."""

        files = [

            ('diff-2', 'orig-2', 'mod-2'),

            ('diff-3', 'orig-3', 'mod-3'),

            ('diff-4', 'orig-4', 'mod-4'),

            ('diff-5', 'orig-5', 'mod-5'),

            ('diff-6', 'orig-6', 'mod-6'),

        ]

        for diff, orig, mod in files:

            patch = self.datafile(diff)

            orig_lines = list(self.datafile(orig))

            mod_lines = list(self.datafile(mod))

            patched_file = IterableFile(iter_patched(orig_lines, patch))

            lines = []

            count = 0

            for patch_line in patched_file:

                self.assertEqual(patch_line, mod_lines[count])

                count += 1

            self.assertEqual(count, len(mod_lines))

    def testFirstLineRenumber(self):

        """Make sure we handle lines at the beginning of the hunk"""

        patch = parse_patch(self.datafile("insert_top.patch"))

        assert (patch.pos_in_mod(0)==1)

def test():

    patchesTestSuite = unittest.makeSuite(PatchesTester,'test')

    runner = unittest.TextTestRunner(verbosity=0)

    return runner.run(patchesTestSuite)

if __name__ == "__main__":

    test()