← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/script.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-09-13 12:30:52 UTC
  • mfrom: (5417.1.6 scripts)
  • Revision ID: pqm@pqm.ubuntu.com-20100913123052-cxsnsnc6bca12cq8
(mbp) improve scriptrunner handling of blanks, newlines,
 and lines without newlines (Martin Pool)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/script.py
56
56
    Input lines start with '<'.
57
57
    Output lines start with nothing.
58
58
    Error lines start with '2>'.
 
59
 
 
60
    :return: A sequence of ([args], input, output, errors), where the args are
 
61
        split in to words, and the input, output, and errors are just strings,
 
62
        typically containing newlines.
59
63
    """
60
64
 
61
65
    commands = []
75
79
    lineno = 0
76
80
    input, output, error = None, None, None
77
81
    text = textwrap.dedent(text)
78
 
    for line in text.split('\n'):
 
82
    lines = text.split('\n')
 
83
    # to make use of triple-quoted strings easier, we ignore a blank line
 
84
    # right at the start and right at the end; the rest are meaningful
 
85
    if lines and lines[0] == '':
 
86
        del lines[0]
 
87
    if lines and lines[-1] == '':
 
88
        del lines[-1]
 
89
    for line in lines:
79
90
        lineno += 1
80
91
        # Keep a copy for error reporting
81
92
        orig = line
82
93
        comment =  line.find('#')
83
94
        if comment >= 0:
84
95
            # Delete comments
 
96
            # NB: this syntax means comments are allowed inside output, which
 
97
            # may be confusing...
85
98
            line = line[0:comment]
86
99
            line = line.rstrip()
87
 
        if line == '':
88
 
            # Ignore empty lines
89
 
            continue
 
100
            if line == '':
 
101
                continue
90
102
        if line.startswith('$'):
91
103
            # Time to output the current command
92
104
            add_command(cmd_cur, input, output, error)
232
244
            # 'expected' parameter. So we just fallback to our good old
233
245
            # assertEqualDiff since we know there *are* differences and the
234
246
            # output should be decently readable.
235
 
            test_case.assertEqualDiff(expected, actual)
 
247
            #
 
248
            # As a special case, we allow output that's missing a final
 
249
            # newline to match an expected string that does have one, so that
 
250
            # we can match a prompt printed on one line, then input given on
 
251
            # the next line.
 
252
            if expected == actual + '\n':
 
253
                pass
 
254
            else:
 
255
                test_case.assertEqualDiff(expected, actual)
236
256
 
237
257
    def _pre_process_args(self, args):
238
258
        new_args = []