/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/tests/script.py

  • Committer: Jelmer Vernooij
  • Date: 2017-06-08 23:30:31 UTC
  • mto: This revision was merged to the branch mainline in revision 6690.
  • Revision ID: jelmer@jelmer.uk-20170608233031-3qavls2o7a1pqllj
Update imports.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/tests/script.py
22
22
import doctest
23
23
import errno
24
24
import glob
25
 
import logging
26
25
import os
27
26
import shlex
28
 
import sys
29
27
import textwrap
30
28
 
31
29
from .. import (
32
30
    osutils,
33
31
    tests,
34
 
    trace,
35
32
    )
36
 
from ..tests import ui_testing
37
33
 
38
34
 
39
35
def split(s):
93
89
        lineno += 1
94
90
        # Keep a copy for error reporting
95
91
        orig = line
96
 
        comment = line.find('#')
 
92
        comment =  line.find('#')
97
93
        if comment >= 0:
98
94
            # Delete comments
99
95
            # NB: this syntax means comments are allowed inside output, which
142
138
 
143
139
    :param args: The command line arguments
144
140
 
145
 
    :return: A tuple containing:
 
141
    :return: A tuple containing: 
146
142
        - The file name redirected from or None
147
143
        - The file name redirected to or None
148
144
        - The mode to open the output file or None
167
163
            in_name = redirected_file_name('<', arg[1:], args)
168
164
        elif arg.startswith('>>'):
169
165
            out_name = redirected_file_name('>>', arg[2:], args)
170
 
            out_mode = 'a+'
 
166
            out_mode = 'ab+'
171
167
        elif arg.startswith('>',):
172
168
            out_name = redirected_file_name('>', arg[1:], args)
173
 
            out_mode = 'w+'
 
169
            out_mode = 'wb+'
174
170
        else:
175
171
            remaining.append(arg)
176
172
    return in_name, out_name, out_mode, remaining
178
174
 
179
175
class ScriptRunner(object):
180
176
    """Run a shell-like script from a test.
181
 
 
 
177
    
182
178
    Can be used as:
183
179
 
184
180
    from breezy.tests import script
236
232
        try:
237
233
            self._check_output(error, actual_error, test_case)
238
234
        except AssertionError as e:
239
 
            raise AssertionError(str(e)
240
 
                                 + " in stderr of running command %s" % cmd)
 
235
            raise AssertionError(str(e) +
 
236
                " in stderr of running command %s" % cmd)
241
237
        if retcode and not error and actual_error:
242
238
            test_case.fail('In \n\t%s\nUnexpected error: %s'
243
239
                           % (' '.join(cmd), actual_error))
251
247
                return
252
248
            else:
253
249
                test_case.fail('expected output: %r, but found nothing'
254
 
                               % (expected,))
 
250
                            % (expected,))
255
251
 
256
252
        null_output_matches_anything = getattr(
257
253
            self, 'null_output_matches_anything', False)
284
280
            # Strip the simple and double quotes since we don't care about
285
281
            # them.  We leave the backquotes in place though since they have a
286
282
            # different semantic.
287
 
            if arg[0] in ('"', "'") and arg[0] == arg[-1]:
 
283
            if arg[0] in  ('"', "'") and arg[0] == arg[-1]:
288
284
                yield arg[1:-1]
289
285
            else:
290
286
                if glob.has_magic(arg):
300
296
 
301
297
    def _read_input(self, input, in_name):
302
298
        if in_name is not None:
303
 
            infile = open(in_name, 'r')
 
299
            infile = open(in_name, 'rb')
304
300
            try:
305
301
                # Command redirection takes precedence over provided input
306
302
                input = infile.read()
319
315
        return output
320
316
 
321
317
    def do_brz(self, test_case, input, args):
322
 
        encoding = osutils.get_user_encoding()
323
 
        stdout = ui_testing.StringIOWithEncoding()
324
 
        stderr = ui_testing.StringIOWithEncoding()
325
 
        stdout.encoding = stderr.encoding = encoding
326
 
        handler = logging.StreamHandler(stderr)
327
 
        handler.setLevel(logging.INFO)
328
 
 
329
 
        logger = logging.getLogger('')
330
 
        logger.addHandler(handler)
331
 
        try:
332
 
            retcode = test_case._run_bzr_core(
333
 
                args, encoding=encoding, stdin=input, stdout=stdout,
334
 
                stderr=stderr, working_dir=None)
335
 
        finally:
336
 
            logger.removeHandler(handler)
337
 
 
338
 
        return retcode, stdout.getvalue(), stderr.getvalue()
 
318
        retcode, out, err = test_case._run_bzr_core(
 
319
            args, retcode=None, encoding=None, stdin=input, working_dir=None)
 
320
        return retcode, out, err
339
321
 
340
322
    def do_cat(self, test_case, input, args):
341
323
        (in_name, out_name, out_mode, args) = _scan_redirection_options(args)
422
404
        err = None
423
405
 
424
406
        def error(msg, path):
425
 
            return "rm: cannot remove '%s': %s\n" % (path, msg)
 
407
            return  "rm: cannot remove '%s': %s\n" % (path, msg)
426
408
 
427
409
        force, recursive = False, False
428
410
        opts = None
452
434
                        break
453
435
                elif e.errno == errno.ENOENT:
454
436
                    if not force:
455
 
                        err = error('No such file or directory', p)
 
437
                        err =  error('No such file or directory', p)
456
438
                        break
457
439
                else:
458
440
                    raise
464
446
 
465
447
    def do_mv(self, test_case, input, args):
466
448
        err = None
467
 
 
468
449
        def error(msg, src, dst):
469
450
            return "mv: cannot move %s to %s: %s\n" % (src, dst, msg)
470
451
 
488
469
        return retcode, None, err
489
470
 
490
471
 
 
472
 
491
473
class TestCaseWithMemoryTransportAndScript(tests.TestCaseWithMemoryTransport):
492
474
    """Helper class to experiment shell-like test and memory fs.
493
475
 
505
487
        self.overrideEnv('INSIDE_EMACS', '1')
506
488
 
507
489
    def run_script(self, script, null_output_matches_anything=False):
508
 
        return self.script_runner.run_script(self, script,
509
 
                                             null_output_matches_anything=null_output_matches_anything)
 
490
        return self.script_runner.run_script(self, script, 
 
491
                   null_output_matches_anything=null_output_matches_anything)
510
492
 
511
493
    def run_command(self, cmd, input, output, error):
512
494
        return self.script_runner.run_command(self, cmd, input, output, error)
540
522
 
541
523
    def run_script(self, script, null_output_matches_anything=False):
542
524
        return self.script_runner.run_script(self, script,
543
 
                                             null_output_matches_anything=null_output_matches_anything)
 
525
                   null_output_matches_anything=null_output_matches_anything)
544
526
 
545
527
    def run_command(self, cmd, input, output, error):
546
528
        return self.script_runner.run_command(self, cmd, input, output, error)
549
531
def run_script(test_case, script_string, null_output_matches_anything=False):
550
532
    """Run the given script within a testcase"""
551
533
    return ScriptRunner().run_script(test_case, script_string,
552
 
                                     null_output_matches_anything=null_output_matches_anything)
 
534
               null_output_matches_anything=null_output_matches_anything)
 
535