~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/blackbox/test_selftest.py

  • Committer: Vincent Ladeuil
  • Date: 2010-02-03 07:18:36 UTC
  • mto: (5008.1.1 integration)
  • mto: This revision was merged to the branch mainline in revision 5009.
  • Revision ID: v.ladeuil+lp@free.fr-20100203071836-u9b86q68fr9ri5s6
Fix NEWS.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005 by Canonical Ltd
 
1
# Copyright (C) 2005, 2007 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
 
# it under the terms of the GNU General Public License version 2 as published by
5
 
# the Free Software Foundation.
 
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.
6
7
#
7
8
# This program is distributed in the hope that it will be useful,
8
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
12
#
12
13
# You should have received a copy of the GNU General Public License
13
14
# along with this program; if not, write to the Free Software
14
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
15
16
 
16
17
"""UI tests for the test framework."""
17
18
 
18
 
import sys
19
 
 
20
 
import bzrlib
 
19
import bzrlib.transport
 
20
from bzrlib import (
 
21
    benchmarks,
 
22
    tests,
 
23
    )
21
24
from bzrlib.errors import ParamikoNotPresent
22
25
from bzrlib.tests import (
 
26
                          features,
23
27
                          TestCase,
24
28
                          TestCaseInTempDir,
25
29
                          TestSkipped,
26
30
                          )
27
 
from bzrlib.tests.blackbox import ExternalBase
28
 
 
29
 
 
30
 
class TestOptions(TestCase):
31
 
 
32
 
    current_test = None
 
31
 
 
32
 
 
33
class SelfTestPatch:
 
34
 
 
35
    def get_params_passed_to_core(self, cmdline):
 
36
        params = []
 
37
        def selftest(*args, **kwargs):
 
38
            """Capture the arguments selftest was run with."""
 
39
            params.append((args, kwargs))
 
40
            return True
 
41
        # Yes this prevents using threads to run the test suite in parallel,
 
42
        # however we don't have a clean dependency injector for commands, 
 
43
        # and even if we did - we'd still be testing that the glue is wired
 
44
        # up correctly. XXX: TODO: Solve this testing problem.
 
45
        original_selftest = tests.selftest
 
46
        tests.selftest = selftest
 
47
        try:
 
48
            self.run_bzr(cmdline)
 
49
            return params[0]
 
50
        finally:
 
51
            tests.selftest = original_selftest
 
52
 
 
53
 
 
54
class TestOptionsWritingToDisk(TestCaseInTempDir, SelfTestPatch):
 
55
 
 
56
    def test_benchmark_runs_benchmark_tests(self):
 
57
        """selftest --benchmark should change the suite factory."""
 
58
        params = self.get_params_passed_to_core('selftest --benchmark')
 
59
        self.assertEqual(benchmarks.test_suite,
 
60
            params[1]['test_suite_factory'])
 
61
        self.assertNotEqual(None, params[1]['bench_history'])
 
62
        benchfile = open(".perf_history", "rt")
 
63
        try:
 
64
            lines = benchfile.readlines()
 
65
        finally:
 
66
            benchfile.close()
 
67
        # Because we don't run the actual test code no output is made to the
 
68
        # file.
 
69
        self.assertEqual(0, len(lines))
 
70
 
 
71
 
 
72
class TestOptions(TestCase, SelfTestPatch):
 
73
 
 
74
    def test_load_list(self):
 
75
        params = self.get_params_passed_to_core('selftest --load-list foo')
 
76
        self.assertEqual('foo', params[1]['load_list'])
33
77
 
34
78
    def test_transport_set_to_sftp(self):
35
 
        # test the --transport option has taken effect from within the
36
 
        # test_transport test
37
 
        try:
38
 
            import bzrlib.transport.sftp
39
 
        except ParamikoNotPresent:
40
 
            raise TestSkipped("Paramiko not present")
41
 
        if TestOptions.current_test != "test_transport_set_to_sftp":
42
 
            return
 
79
        # Test that we can pass a transport to the selftest core - sftp
 
80
        # version.
 
81
        self.requireFeature(features.paramiko)
 
82
        params = self.get_params_passed_to_core('selftest --transport=sftp')
43
83
        self.assertEqual(bzrlib.transport.sftp.SFTPAbsoluteServer,
44
 
                         bzrlib.tests.default_transport)
 
84
            params[1]["transport"])
45
85
 
46
86
    def test_transport_set_to_memory(self):
47
 
        # test the --transport option has taken effect from within the
48
 
        # test_transport test
 
87
        # Test that we can pass a transport to the selftest core - memory
 
88
        # version.
49
89
        import bzrlib.transport.memory
50
 
        if TestOptions.current_test != "test_transport_set_to_memory":
51
 
            return
 
90
        params = self.get_params_passed_to_core('selftest --transport=memory')
52
91
        self.assertEqual(bzrlib.transport.memory.MemoryServer,
53
 
                         bzrlib.tests.default_transport)
54
 
 
55
 
    def test_transport(self):
56
 
        # test that --transport=sftp works
57
 
        try:
58
 
            import bzrlib.transport.sftp
59
 
        except ParamikoNotPresent:
60
 
            raise TestSkipped("Paramiko not present")
61
 
        old_transport = bzrlib.tests.default_transport
62
 
        old_root = TestCaseInTempDir.TEST_ROOT
63
 
        TestCaseInTempDir.TEST_ROOT = None
64
 
        try:
65
 
            TestOptions.current_test = "test_transport_set_to_sftp"
66
 
            stdout = self.capture('selftest --transport=sftp test_transport_set_to_sftp')
67
 
            
68
 
            self.assertContainsRe(stdout, 'Ran 1 test')
69
 
            self.assertEqual(old_transport, bzrlib.tests.default_transport)
70
 
 
71
 
            TestOptions.current_test = "test_transport_set_to_memory"
72
 
            stdout = self.capture('selftest --transport=memory test_transport_set_to_memory')
73
 
            self.assertContainsRe(stdout, 'Ran 1 test')
74
 
            self.assertEqual(old_transport, bzrlib.tests.default_transport)
 
92
            params[1]["transport"])
 
93
 
 
94
    def test_parameters_passed_to_core(self):
 
95
        params = self.get_params_passed_to_core('selftest --list-only')
 
96
        self.assertTrue("list_only" in params[1])
 
97
        params = self.get_params_passed_to_core('selftest --list-only selftest')
 
98
        self.assertTrue("list_only" in params[1])
 
99
        params = self.get_params_passed_to_core(['selftest', '--list-only',
 
100
            '--exclude', 'selftest'])
 
101
        self.assertTrue("list_only" in params[1])
 
102
        params = self.get_params_passed_to_core(['selftest', '--list-only',
 
103
            'selftest', '--randomize', 'now'])
 
104
        self.assertSubset(["list_only", "random_seed"], params[1])
 
105
 
 
106
    def test_starting_with(self):
 
107
        params = self.get_params_passed_to_core('selftest --starting-with foo')
 
108
        self.assertEqual(['foo'], params[1]['starting_with'])
 
109
 
 
110
    def test_starting_with_multiple_argument(self):
 
111
        params = self.get_params_passed_to_core(
 
112
            'selftest --starting-with foo --starting-with bar')
 
113
        self.assertEqual(['foo', 'bar'], params[1]['starting_with'])
 
114
 
 
115
    def test_subunit(self):
 
116
        self.requireFeature(features.subunit)
 
117
        params = self.get_params_passed_to_core('selftest --subunit')
 
118
        self.assertEqual(tests.SubUnitBzrRunner, params[1]['runner_class'])
 
119
 
 
120
    def _parse_test_list(self, lines, newlines_in_header=0):
 
121
        "Parse a list of lines into a tuple of 3 lists (header,body,footer)."
 
122
        in_header = newlines_in_header != 0
 
123
        in_footer = False
 
124
        header = []
 
125
        body = []
 
126
        footer = []
 
127
        header_newlines_found = 0
 
128
        for line in lines:
 
129
            if in_header:
 
130
                if line == '':
 
131
                    header_newlines_found += 1
 
132
                    if header_newlines_found >= newlines_in_header:
 
133
                        in_header = False
 
134
                        continue
 
135
                header.append(line)
 
136
            elif not in_footer:
 
137
                if line.startswith('-------'):
 
138
                    in_footer = True
 
139
                else:
 
140
                    body.append(line)
 
141
            else:
 
142
                footer.append(line)
 
143
        # If the last body line is blank, drop it off the list
 
144
        if len(body) > 0 and body[-1] == '':
 
145
            body.pop()
 
146
        return (header,body,footer)
 
147
 
 
148
    def test_list_only(self):
 
149
        # check that bzr selftest --list-only outputs no ui noise
 
150
        def selftest(*args, **kwargs):
 
151
            """Capture the arguments selftest was run with."""
 
152
            return True
 
153
        def outputs_nothing(cmdline):
 
154
            out,err = self.run_bzr(cmdline)
 
155
            (header,body,footer) = self._parse_test_list(out.splitlines())
 
156
            num_tests = len(body)
 
157
            self.assertLength(0, header)
 
158
            self.assertLength(0, footer)
 
159
            self.assertEqual('', err)
 
160
        # Yes this prevents using threads to run the test suite in parallel,
 
161
        # however we don't have a clean dependency injector for commands, 
 
162
        # and even if we did - we'd still be testing that the glue is wired
 
163
        # up correctly. XXX: TODO: Solve this testing problem.
 
164
        original_selftest = tests.selftest
 
165
        tests.selftest = selftest
 
166
        try:
 
167
            outputs_nothing('selftest --list-only')
 
168
            outputs_nothing('selftest --list-only selftest')
 
169
            outputs_nothing(['selftest', '--list-only', '--exclude', 'selftest'])
75
170
        finally:
76
 
            bzrlib.tests.default_transport = old_transport
77
 
            TestOptions.current_test = None
78
 
            TestCaseInTempDir.TEST_ROOT = old_root
79
 
 
80
 
    def test_benchmark_runs_benchmark_tests(self):
81
 
        """bzr selftest --benchmark should not run the default test suite."""
82
 
        # We test this by passing a regression test name to --benchmark, which
83
 
        # should result in 0 rests run.
84
 
        out, err = self.run_bzr('selftest', '--benchmark', 'workingtree_implementations')
85
 
        self.assertContainsRe(out, 'Ran 0 tests.*\n\nOK')
86
 
        self.assertEqual(
87
 
            'running tests...\ntests passed\n',
88
 
            err)
89
 
        
90
 
 
91
 
class TestRunBzr(ExternalBase):
92
 
 
93
 
    def run_bzr_captured(self, argv, retcode=0, encoding=None, stdin=None):
94
 
        self.stdin = stdin
95
 
 
96
 
    def test_stdin(self):
97
 
        # test that the stdin keyword to run_bzr is passed through to
98
 
        # run_bzr_captured as-is. We do this by overriding
99
 
        # run_bzr_captured in this class, and then calling run_bzr,
100
 
        # which is a convenience function for run_bzr_captured, so 
101
 
        # should invoke it.
102
 
        self.run_bzr('foo', 'bar', stdin='gam')
103
 
        self.assertEqual('gam', self.stdin)
104
 
        self.run_bzr('foo', 'bar', stdin='zippy')
105
 
        self.assertEqual('zippy', self.stdin)
106
 
 
107
 
 
108
 
class TestRunBzrCaptured(ExternalBase):
109
 
 
110
 
    def apply_redirected(self, stdin=None, stdout=None, stderr=None,
111
 
                         a_callable=None, *args, **kwargs):
112
 
        self.stdin = stdin
113
 
        self.factory_stdin = getattr(bzrlib.ui.ui_factory, "stdin", None)
114
 
        self.factory = bzrlib.ui.ui_factory
115
 
        stdout.write('foo\n')
116
 
        stderr.write('bar\n')
117
 
        return 0
118
 
 
119
 
    def test_stdin(self):
120
 
        # test that the stdin keyword to run_bzr_captured is passed through to
121
 
        # apply_redirected as a StringIO. We do this by overriding
122
 
        # apply_redirected in this class, and then calling run_bzr_captured,
123
 
        # which calls apply_redirected. 
124
 
        self.run_bzr_captured(['foo', 'bar'], stdin='gam')
125
 
        self.assertEqual('gam', self.stdin.read())
126
 
        self.assertTrue(self.stdin is self.factory_stdin)
127
 
        self.run_bzr_captured(['foo', 'bar'], stdin='zippy')
128
 
        self.assertEqual('zippy', self.stdin.read())
129
 
        self.assertTrue(self.stdin is self.factory_stdin)
130
 
 
131
 
    def test_ui_factory(self):
132
 
        # each invocation of self.run_bzr_captured should get its own UI
133
 
        # factory, which is an instance of TestUIFactory, with stdout and
134
 
        # stderr attached to the stdout and stderr of the invoked
135
 
        # run_bzr_captured
136
 
        current_factory = bzrlib.ui.ui_factory
137
 
        self.run_bzr_captured(['foo'])
138
 
        self.failIf(current_factory is self.factory)
139
 
        self.assertNotEqual(sys.stdout, self.factory.stdout)
140
 
        self.assertNotEqual(sys.stderr, self.factory.stderr)
141
 
        self.assertEqual('foo\n', self.factory.stdout.getvalue())
142
 
        self.assertEqual('bar\n', self.factory.stderr.getvalue())
143
 
        self.assertIsInstance(self.factory, bzrlib.tests.blackbox.TestUIFactory)
144
 
 
145
 
    def test_run_bzr_subprocess(self):
146
 
        """The run_bzr_helper_external comand behaves nicely."""
147
 
        result = self.run_bzr_subprocess('--version')
148
 
        result = self.run_bzr_subprocess('--version', retcode=None)
149
 
        self.assertContainsRe(result[0], 'is free software')
150
 
        self.assertRaises(AssertionError, self.run_bzr_subprocess, 
151
 
                          '--versionn')
152
 
        result = self.run_bzr_subprocess('--versionn', retcode=3)
153
 
        result = self.run_bzr_subprocess('--versionn', retcode=None)
154
 
        self.assertContainsRe(result[1], 'unknown command')
155
 
        err = self.run_bzr_subprocess('merge', '--merge-type', 'magic merge', 
156
 
                                      retcode=3)[1]
157
 
        self.assertContainsRe(err, 'No known merge type magic merge')
158
 
 
159
 
 
160
 
class TestRunBzrError(ExternalBase):
161
 
 
162
 
    def test_run_bzr_error(self):
163
 
        out, err = self.run_bzr_error(['^$'], 'rocks', retcode=0)
164
 
        self.assertEqual(out, 'it sure does!\n')
165
 
 
166
 
        out, err = self.run_bzr_error(["'foobarbaz' is not a versioned file"],
167
 
                                      'file-id', 'foobarbaz')
 
171
            tests.selftest = original_selftest
 
172
 
 
173
    def test_lsprof_tests(self):
 
174
        params = self.get_params_passed_to_core('selftest --lsprof-tests')
 
175
        self.assertEqual(True, params[1]["lsprof_tests"])