1
 
from bzrlib.selftest import TestBase
 
2
 
from bzrlib.diff import internal_diff
 
3
 
from cStringIO import StringIO
 
4
 
def udiff_lines(old, new):
 
6
 
    internal_diff('old', old, 'new', new, output)
 
8
 
    return output.readlines()
 
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)
 
24
 
class AddNL(TestBase):
 
26
 
    diff generates a valid diff for patches that add a newline
 
29
 
        lines = udiff_lines(['boo'], ['boo\n'])
 
31
 
        assert lines[4] == '\\ No newline at end of file\n', \
 
32
 
            "expected no-nl, got %r" % lines[4]
 
35
 
class AddNL2(TestBase):
 
37
 
    diff generates a valid diff for patches that change last line and add a
 
41
 
        lines = udiff_lines(['boo'], ['goo\n'])
 
43
 
        assert lines[4] == '\\ No newline at end of file\n', \
 
44
 
            "expected no-nl, got %r" % lines[4]
 
46
 
class RemoveNL(TestBase):
 
48
 
    diff generates a valid diff for patches that change last line and add a
 
52
 
        lines = udiff_lines(['boo\n'], ['boo'])
 
54
 
        assert lines[5] == '\\ No newline at end of file\n', \
 
55
 
            "expected no-nl, got %r" % lines[5]