/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 bzrlib/tests/blackbox/test_selftest.py

  • Committer: Michael Ellerman
  • Date: 2006-05-31 08:44:29 UTC
  • mto: (1711.2.63 jam-integration)
  • mto: This revision was merged to the branch mainline in revision 1792.
  • Revision ID: michael@ellerman.id.au-20060531084429-35e5429abda9f560
Add optional location to ancestry and fix behaviour for checkouts.

This adds an optional location parameter to the ancestry command. It also
changes the behaviour of ancestry on checkouts such that if they have
been created with a subset of the branch history, only the subset is
shown by 'bzr ancestry'. Tests for all of that as well.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006-2010 Canonical Ltd
 
1
# Copyright (C) 2005 by 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 as published by
5
 
# the Free Software Foundation; either version 2 of the License, or
6
 
# (at your option) any later version.
 
4
# it under the terms of the GNU General Public License version 2 as published by
 
5
# the Free Software Foundation.
7
6
#
8
7
# This program is distributed in the hope that it will be useful,
9
8
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
11
#
13
12
# You should have received a copy of the GNU General Public License
14
13
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
14
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
15
 
17
16
"""UI tests for the test framework."""
18
17
 
19
 
from bzrlib import (
20
 
    benchmarks,
21
 
    tests,
22
 
    )
 
18
import sys
 
19
 
 
20
import bzrlib
 
21
from bzrlib.errors import ParamikoNotPresent
23
22
from bzrlib.tests import (
24
 
    features,
25
 
    )
26
 
from bzrlib.transport import memory
27
 
 
28
 
class SelfTestPatch:
29
 
 
30
 
    def get_params_passed_to_core(self, cmdline):
31
 
        params = []
32
 
        def selftest(*args, **kwargs):
33
 
            """Capture the arguments selftest was run with."""
34
 
            params.append((args, kwargs))
35
 
            return True
36
 
        # Yes this prevents using threads to run the test suite in parallel,
37
 
        # however we don't have a clean dependency injector for commands, 
38
 
        # and even if we did - we'd still be testing that the glue is wired
39
 
        # up correctly. XXX: TODO: Solve this testing problem.
40
 
        original_selftest = tests.selftest
41
 
        tests.selftest = selftest
42
 
        try:
43
 
            self.run_bzr(cmdline)
44
 
            return params[0]
45
 
        finally:
46
 
            tests.selftest = original_selftest
47
 
 
48
 
 
49
 
class TestOptionsWritingToDisk(tests.TestCaseInTempDir, SelfTestPatch):
50
 
 
51
 
    def test_benchmark_runs_benchmark_tests(self):
52
 
        """selftest --benchmark should change the suite factory."""
53
 
        params = self.get_params_passed_to_core('selftest --benchmark')
54
 
        self.assertEqual(benchmarks.test_suite,
55
 
            params[1]['test_suite_factory'])
56
 
        self.assertNotEqual(None, params[1]['bench_history'])
57
 
        benchfile = open(".perf_history", "rt")
58
 
        try:
59
 
            lines = benchfile.readlines()
60
 
        finally:
61
 
            benchfile.close()
62
 
        # Because we don't run the actual test code no output is made to the
63
 
        # file.
64
 
        self.assertEqual(0, len(lines))
65
 
 
66
 
 
67
 
class TestOptions(tests.TestCase, SelfTestPatch):
68
 
 
69
 
    def test_load_list(self):
70
 
        params = self.get_params_passed_to_core('selftest --load-list foo')
71
 
        self.assertEqual('foo', params[1]['load_list'])
 
23
                          TestCase,
 
24
                          TestCaseInTempDir,
 
25
                          TestSkipped,
 
26
                          )
 
27
from bzrlib.tests.blackbox import ExternalBase
 
28
 
 
29
 
 
30
class TestOptions(TestCase):
 
31
 
 
32
    current_test = None
72
33
 
73
34
    def test_transport_set_to_sftp(self):
74
 
        # Test that we can pass a transport to the selftest core - sftp
75
 
        # version.
76
 
        self.requireFeature(features.paramiko)
77
 
        from bzrlib.tests import stub_sftp
78
 
        params = self.get_params_passed_to_core('selftest --transport=sftp')
79
 
        self.assertEqual(stub_sftp.SFTPAbsoluteServer,
80
 
            params[1]["transport"])
 
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
 
43
        self.assertEqual(bzrlib.transport.sftp.SFTPAbsoluteServer,
 
44
                         bzrlib.tests.default_transport)
81
45
 
82
46
    def test_transport_set_to_memory(self):
83
 
        # Test that we can pass a transport to the selftest core - memory
84
 
        # version.
85
 
        params = self.get_params_passed_to_core('selftest --transport=memory')
86
 
        self.assertEqual(memory.MemoryServer, params[1]["transport"])
87
 
 
88
 
    def test_parameters_passed_to_core(self):
89
 
        params = self.get_params_passed_to_core('selftest --list-only')
90
 
        self.assertTrue("list_only" in params[1])
91
 
        params = self.get_params_passed_to_core('selftest --list-only selftest')
92
 
        self.assertTrue("list_only" in params[1])
93
 
        params = self.get_params_passed_to_core(['selftest', '--list-only',
94
 
            '--exclude', 'selftest'])
95
 
        self.assertTrue("list_only" in params[1])
96
 
        params = self.get_params_passed_to_core(['selftest', '--list-only',
97
 
            'selftest', '--randomize', 'now'])
98
 
        self.assertSubset(["list_only", "random_seed"], params[1])
99
 
 
100
 
    def test_starting_with(self):
101
 
        params = self.get_params_passed_to_core('selftest --starting-with foo')
102
 
        self.assertEqual(['foo'], params[1]['starting_with'])
103
 
 
104
 
    def test_starting_with_multiple_argument(self):
105
 
        params = self.get_params_passed_to_core(
106
 
            'selftest --starting-with foo --starting-with bar')
107
 
        self.assertEqual(['foo', 'bar'], params[1]['starting_with'])
108
 
 
109
 
    def test_subunit(self):
110
 
        self.requireFeature(features.subunit)
111
 
        params = self.get_params_passed_to_core('selftest --subunit')
112
 
        self.assertEqual(tests.SubUnitBzrRunner, params[1]['runner_class'])
113
 
 
114
 
    def _parse_test_list(self, lines, newlines_in_header=0):
115
 
        "Parse a list of lines into a tuple of 3 lists (header,body,footer)."
116
 
        in_header = newlines_in_header != 0
117
 
        in_footer = False
118
 
        header = []
119
 
        body = []
120
 
        footer = []
121
 
        header_newlines_found = 0
122
 
        for line in lines:
123
 
            if in_header:
124
 
                if line == '':
125
 
                    header_newlines_found += 1
126
 
                    if header_newlines_found >= newlines_in_header:
127
 
                        in_header = False
128
 
                        continue
129
 
                header.append(line)
130
 
            elif not in_footer:
131
 
                if line.startswith('-------'):
132
 
                    in_footer = True
133
 
                else:
134
 
                    body.append(line)
135
 
            else:
136
 
                footer.append(line)
137
 
        # If the last body line is blank, drop it off the list
138
 
        if len(body) > 0 and body[-1] == '':
139
 
            body.pop()
140
 
        return (header,body,footer)
141
 
 
142
 
    def test_list_only(self):
143
 
        # check that bzr selftest --list-only outputs no ui noise
144
 
        def selftest(*args, **kwargs):
145
 
            """Capture the arguments selftest was run with."""
146
 
            return True
147
 
        def outputs_nothing(cmdline):
148
 
            out,err = self.run_bzr(cmdline)
149
 
            (header,body,footer) = self._parse_test_list(out.splitlines())
150
 
            num_tests = len(body)
151
 
            self.assertLength(0, header)
152
 
            self.assertLength(0, footer)
153
 
            self.assertEqual('', err)
154
 
        # Yes this prevents using threads to run the test suite in parallel,
155
 
        # however we don't have a clean dependency injector for commands, 
156
 
        # and even if we did - we'd still be testing that the glue is wired
157
 
        # up correctly. XXX: TODO: Solve this testing problem.
158
 
        original_selftest = tests.selftest
159
 
        tests.selftest = selftest
160
 
        try:
161
 
            outputs_nothing('selftest --list-only')
162
 
            outputs_nothing('selftest --list-only selftest')
163
 
            outputs_nothing(['selftest', '--list-only', '--exclude', 'selftest'])
 
47
        # test the --transport option has taken effect from within the
 
48
        # test_transport test
 
49
        import bzrlib.transport.memory
 
50
        if TestOptions.current_test != "test_transport_set_to_memory":
 
51
            return
 
52
        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)
164
75
        finally:
165
 
            tests.selftest = original_selftest
166
 
 
167
 
    def test_lsprof_tests(self):
168
 
        params = self.get_params_passed_to_core('selftest --lsprof-tests')
169
 
        self.assertEqual(True, params[1]["lsprof_tests"])
 
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, 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)