~bzr-pqm/bzr/bzr.dev

5452.4.3 by John Arbash Meinel
Merge bzr.dev to resolve bzr-2.3.txt (aka NEWS)
1
# Copyright (C) 2010 Canonical Ltd
5510.1.2 by Martin von Gagern
Avoid test-script command depending on testtools.
2
#
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
7
#
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
# GNU General Public License for more details.
12
#
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
17
"""Front-end command for shell-like test scripts.
18
5531.1.2 by Vincent Ladeuil
s/blank_output/null_output/
19
See doc/developers/testing.txt for more explanations.
5510.1.2 by Martin von Gagern
Avoid test-script command depending on testtools.
20
This module should be importable even if testtools aren't available.
21
"""
22
6379.6.7 by Jelmer Vernooij
Move importing from future until after doc string, otherwise the doc string will disappear.
23
from __future__ import absolute_import
24
5510.1.2 by Martin von Gagern
Avoid test-script command depending on testtools.
25
import os
26
5531.1.3 by Vincent Ladeuil
Implements --null-ouput for the test-script command.
27
from bzrlib import (
28
    commands,
29
    option,
30
    )
5510.1.2 by Martin von Gagern
Avoid test-script command depending on testtools.
31
5510.1.4 by Martin von Gagern
Minor polishing suggested by John A Meinel.
32
5510.1.2 by Martin von Gagern
Avoid test-script command depending on testtools.
33
class cmd_test_script(commands.Command):
34
    """Run a shell-like test from a file."""
35
36
    hidden = True
37
    takes_args = ['infile']
5531.1.3 by Vincent Ladeuil
Implements --null-ouput for the test-script command.
38
    takes_options = [
39
        option.Option('null-output',
40
                       help='Null command outputs match any output.'),
41
        ]
5510.1.2 by Martin von Gagern
Avoid test-script command depending on testtools.
42
43
    @commands.display_command
5531.1.3 by Vincent Ladeuil
Implements --null-ouput for the test-script command.
44
    def run(self, infile, null_output=False):
5510.1.4 by Martin von Gagern
Minor polishing suggested by John A Meinel.
45
        # local imports to defer testtools dependency
5510.1.3 by Martin von Gagern
Change lazy import to local import.
46
        from bzrlib import tests
47
        from bzrlib.tests.script import TestCaseWithTransportAndScript
5510.1.2 by Martin von Gagern
Avoid test-script command depending on testtools.
48
49
        f = open(infile)
50
        try:
51
            script = f.read()
52
        finally:
53
            f.close()
54
55
        class Test(TestCaseWithTransportAndScript):
56
57
            script = None # Set before running
58
59
            def test_it(self):
5531.1.3 by Vincent Ladeuil
Implements --null-ouput for the test-script command.
60
                self.run_script(script,
61
                                null_output_matches_anything=null_output)
5510.1.2 by Martin von Gagern
Avoid test-script command depending on testtools.
62
63
        runner = tests.TextTestRunner(stream=self.outf)
64
        test = Test('test_it')
65
        test.path = os.path.realpath(infile)
66
        res = runner.run(test)
67
        return len(res.errors) + len(res.failures)