← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_diff.py

(mbp) merge bzr.dev to 0.8, prepare for release

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_diff.py
1
 
from bzrlib.selftest import TestCase
 
1
from cStringIO import StringIO
 
2
 
2
3
from bzrlib.diff import internal_diff
3
 
from cStringIO import StringIO
4
 
def udiff_lines(old, new):
 
4
from bzrlib.errors import BinaryFile
 
5
from bzrlib.tests import TestCase
 
6
 
 
7
 
 
8
def udiff_lines(old, new, allow_binary=False):
5
9
    output = StringIO()
6
 
    internal_diff('old', old, 'new', new, output)
 
10
    internal_diff('old', old, 'new', new, output, allow_binary)
7
11
    output.seek(0, 0)
8
12
    return output.readlines()
9
13
 
10
 
def check_patch(lines):
11
 
    assert len(lines) > 1, \
12
 
        "Not enough lines for a file header for patch:\n%s" % "".join(lines)
13
 
    assert lines[0].startswith ('---'), \
14
 
        'No orig line for patch:\n%s' % "".join(lines)
15
 
    assert lines[1].startswith ('+++'), \
16
 
        'No mod line for patch:\n%s' % "".join(lines)
17
 
    assert len(lines) > 2, \
18
 
        "No hunks for patch:\n%s" % "".join(lines)
19
 
    assert lines[2].startswith('@@'),\
20
 
        "No hunk header for patch:\n%s" % "".join(lines)
21
 
    assert '@@' in lines[2][2:], \
22
 
        "Unterminated hunk header for patch:\n%s" % "".join(lines)
23
 
 
24
14
class TestDiff(TestCase):
25
15
    def test_add_nl(self):
26
16
        """diff generates a valid diff for patches that add a newline"""
27
17
        lines = udiff_lines(['boo'], ['boo\n'])
28
 
        check_patch(lines)
29
 
        assert lines[4] == '\\ No newline at end of file\n', \
30
 
            "expected no-nl, got %r" % lines[4]
 
18
        self.check_patch(lines)
 
19
        self.assertEquals(lines[4], '\\ No newline at end of file\n')
 
20
            ## "expected no-nl, got %r" % lines[4]
31
21
 
32
22
    def test_add_nl_2(self):
33
23
        """diff generates a valid diff for patches that change last line and
34
24
        add a newline.
35
25
        """
36
26
        lines = udiff_lines(['boo'], ['goo\n'])
37
 
        check_patch(lines)
38
 
        assert lines[4] == '\\ No newline at end of file\n', \
39
 
            "expected no-nl, got %r" % lines[4]
 
27
        self.check_patch(lines)
 
28
        self.assertEquals(lines[4], '\\ No newline at end of file\n')
 
29
            ## "expected no-nl, got %r" % lines[4]
40
30
 
41
31
    def test_remove_nl(self):
42
32
        """diff generates a valid diff for patches that change last line and
43
33
        add a newline.
44
34
        """
45
35
        lines = udiff_lines(['boo\n'], ['boo'])
46
 
        check_patch(lines)
47
 
        assert lines[5] == '\\ No newline at end of file\n', \
48
 
            "expected no-nl, got %r" % lines[5]
 
36
        self.check_patch(lines)
 
37
        self.assertEquals(lines[5], '\\ No newline at end of file\n')
 
38
            ## "expected no-nl, got %r" % lines[5]
 
39
 
 
40
    def check_patch(self, lines):
 
41
        self.assert_(len(lines) > 1)
 
42
            ## "Not enough lines for a file header for patch:\n%s" % "".join(lines)
 
43
        self.assert_(lines[0].startswith ('---'))
 
44
            ## 'No orig line for patch:\n%s' % "".join(lines)
 
45
        self.assert_(lines[1].startswith ('+++'))
 
46
            ## 'No mod line for patch:\n%s' % "".join(lines)
 
47
        self.assert_(len(lines) > 2)
 
48
            ## "No hunks for patch:\n%s" % "".join(lines)
 
49
        self.assert_(lines[2].startswith('@@'))
 
50
            ## "No hunk header for patch:\n%s" % "".join(lines)
 
51
        self.assert_('@@' in lines[2][2:])
 
52
            ## "Unterminated hunk header for patch:\n%s" % "".join(lines)
 
53
 
 
54
    def test_binary_lines(self):
 
55
        self.assertRaises(BinaryFile, udiff_lines, [1023 * 'a' + '\x00'], [])
 
56
        self.assertRaises(BinaryFile, udiff_lines, [], [1023 * 'a' + '\x00'])
 
57
        udiff_lines([1023 * 'a' + '\x00'], [], allow_binary=True)
 
58
        udiff_lines([], [1023 * 'a' + '\x00'], allow_binary=True)