1
# Copyright (C) 2005 by Canonical Ltd
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.
7
# This program is distributed in the hope that it will be useful,
8
# but WITHOUT ANY WARRANTY; without even the implied warranty of
9
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
10
# GNU General Public License for more details.
12
# You should have received a copy of the GNU General Public License
13
# along with this program; if not, write to the Free Software
14
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
16
"""UI tests for the test framework."""
21
from bzrlib.errors import ParamikoNotPresent
22
from bzrlib.tests import (
25
TestCaseWithTransport,
28
from bzrlib.tests.blackbox import ExternalBase
31
class TestOptions(TestCase):
35
def test_transport_set_to_sftp(self):
36
# test the --transport option has taken effect from within the
39
import bzrlib.transport.sftp
40
except ParamikoNotPresent:
41
raise TestSkipped("Paramiko not present")
42
if TestOptions.current_test != "test_transport_set_to_sftp":
44
self.assertEqual(bzrlib.transport.sftp.SFTPAbsoluteServer,
45
bzrlib.tests.default_transport)
47
def test_transport_set_to_memory(self):
48
# test the --transport option has taken effect from within the
50
import bzrlib.transport.memory
51
if TestOptions.current_test != "test_transport_set_to_memory":
53
self.assertEqual(bzrlib.transport.memory.MemoryServer,
54
bzrlib.tests.default_transport)
56
def test_transport(self):
57
# test that --transport=sftp works
59
import bzrlib.transport.sftp
60
except ParamikoNotPresent:
61
raise TestSkipped("Paramiko not present")
62
old_transport = bzrlib.tests.default_transport
63
old_root = TestCaseInTempDir.TEST_ROOT
64
TestCaseInTempDir.TEST_ROOT = None
66
TestOptions.current_test = "test_transport_set_to_sftp"
67
stdout = self.capture('selftest --transport=sftp test_transport_set_to_sftp')
69
self.assertContainsRe(stdout, 'Ran 1 test')
70
self.assertEqual(old_transport, bzrlib.tests.default_transport)
72
TestOptions.current_test = "test_transport_set_to_memory"
73
stdout = self.capture('selftest --transport=memory test_transport_set_to_memory')
74
self.assertContainsRe(stdout, 'Ran 1 test')
75
self.assertEqual(old_transport, bzrlib.tests.default_transport)
77
bzrlib.tests.default_transport = old_transport
78
TestOptions.current_test = None
79
TestCaseInTempDir.TEST_ROOT = old_root
82
class TestRunBzr(ExternalBase):
84
def run_bzr_captured(self, argv, retcode=0, encoding=None, stdin=None):
88
# test that the stdin keyword to run_bzr is passed through to
89
# run_bzr_captured as-is. We do this by overriding
90
# run_bzr_captured in this class, and then calling run_bzr,
91
# which is a convenience function for run_bzr_captured, so
93
self.run_bzr('foo', 'bar', stdin='gam')
94
self.assertEqual('gam', self.stdin)
95
self.run_bzr('foo', 'bar', stdin='zippy')
96
self.assertEqual('zippy', self.stdin)
99
class TestBenchmarkTests(TestCaseWithTransport):
101
def test_benchmark_runs_benchmark_tests(self):
102
"""bzr selftest --benchmark should not run the default test suite."""
103
# We test this by passing a regression test name to --benchmark, which
104
# should result in 0 rests run.
105
old_root = TestCaseInTempDir.TEST_ROOT
107
TestCaseInTempDir.TEST_ROOT = None
108
out, err = self.run_bzr('selftest', '--benchmark', 'workingtree_implementations')
110
TestCaseInTempDir.TEST_ROOT = old_root
111
self.assertContainsRe(out, 'Ran 0 tests.*\n\nOK')
113
'running tests...\ntests passed\n',
115
benchfile = open(".perf_history", "rt")
117
lines = benchfile.readlines()
120
self.assertEqual(1, len(lines))
121
self.assertContainsRe(lines[0], "--date [0-9.]+")
124
class TestRunBzrCaptured(ExternalBase):
126
def apply_redirected(self, stdin=None, stdout=None, stderr=None,
127
a_callable=None, *args, **kwargs):
129
self.factory_stdin = getattr(bzrlib.ui.ui_factory, "stdin", None)
130
self.factory = bzrlib.ui.ui_factory
131
stdout.write('foo\n')
132
stderr.write('bar\n')
135
def test_stdin(self):
136
# test that the stdin keyword to run_bzr_captured is passed through to
137
# apply_redirected as a StringIO. We do this by overriding
138
# apply_redirected in this class, and then calling run_bzr_captured,
139
# which calls apply_redirected.
140
self.run_bzr_captured(['foo', 'bar'], stdin='gam')
141
self.assertEqual('gam', self.stdin.read())
142
self.assertTrue(self.stdin is self.factory_stdin)
143
self.run_bzr_captured(['foo', 'bar'], stdin='zippy')
144
self.assertEqual('zippy', self.stdin.read())
145
self.assertTrue(self.stdin is self.factory_stdin)
147
def test_ui_factory(self):
148
# each invocation of self.run_bzr_captured should get its own UI
149
# factory, which is an instance of TestUIFactory, with stdout and
150
# stderr attached to the stdout and stderr of the invoked
152
current_factory = bzrlib.ui.ui_factory
153
self.run_bzr_captured(['foo'])
154
self.failIf(current_factory is self.factory)
155
self.assertNotEqual(sys.stdout, self.factory.stdout)
156
self.assertNotEqual(sys.stderr, self.factory.stderr)
157
self.assertEqual('foo\n', self.factory.stdout.getvalue())
158
self.assertEqual('bar\n', self.factory.stderr.getvalue())
159
self.assertIsInstance(self.factory, bzrlib.tests.blackbox.TestUIFactory)
161
def test_run_bzr_subprocess(self):
162
"""The run_bzr_helper_external comand behaves nicely."""
163
result = self.run_bzr_subprocess('--version')
164
result = self.run_bzr_subprocess('--version', retcode=None)
165
self.assertContainsRe(result[0], 'is free software')
166
self.assertRaises(AssertionError, self.run_bzr_subprocess,
168
result = self.run_bzr_subprocess('--versionn', retcode=3)
169
result = self.run_bzr_subprocess('--versionn', retcode=None)
170
self.assertContainsRe(result[1], 'unknown command')
171
err = self.run_bzr_subprocess('merge', '--merge-type', 'magic merge',
173
self.assertContainsRe(err, 'No known merge type magic merge')
176
class TestRunBzrError(ExternalBase):
178
def test_run_bzr_error(self):
179
out, err = self.run_bzr_error(['^$'], 'rocks', retcode=0)
180
self.assertEqual(out, 'it sure does!\n')
182
out, err = self.run_bzr_error(["'foobarbaz' is not a versioned file"],
183
'file-id', 'foobarbaz')
added
removed
bzrlib/tests/blackbox/test_selftest.py
