~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_diff.py

  • Committer: Martin Pool
  • Date: 2005-05-03 07:48:54 UTC
  • Revision ID: mbp@sourcefrog.net-20050503074854-adb6f9d6382e27a9
- sketchy experiments in bash and zsh completion

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
from cStringIO import StringIO
2
 
 
3
 
from bzrlib.diff import internal_diff
4
 
from bzrlib.errors import BinaryFile
5
 
from bzrlib.tests import TestCase
6
 
 
7
 
 
8
 
def udiff_lines(old, new, allow_binary=False):
9
 
    output = StringIO()
10
 
    internal_diff('old', old, 'new', new, output, allow_binary)
11
 
    output.seek(0, 0)
12
 
    return output.readlines()
13
 
 
14
 
class TestDiff(TestCase):
15
 
    def test_add_nl(self):
16
 
        """diff generates a valid diff for patches that add a newline"""
17
 
        lines = udiff_lines(['boo'], ['boo\n'])
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]
21
 
 
22
 
    def test_add_nl_2(self):
23
 
        """diff generates a valid diff for patches that change last line and
24
 
        add a newline.
25
 
        """
26
 
        lines = udiff_lines(['boo'], ['goo\n'])
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]
30
 
 
31
 
    def test_remove_nl(self):
32
 
        """diff generates a valid diff for patches that change last line and
33
 
        add a newline.
34
 
        """
35
 
        lines = udiff_lines(['boo\n'], ['boo'])
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)