~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/selftest/testdiff.py

  • Committer: Martin Pool
  • Date: 2005-08-12 15:41:44 UTC
  • Revision ID: mbp@sourcefrog.net-20050812154144-bc98570a78b8f633
- merge in deferred revfile work

Show diffs side-by-side

added added

removed removed

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