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 (
27
from bzrlib.tests.blackbox import ExternalBase
30
class TestOptions(TestCase):
34
def test_transport_set_to_sftp(self):
35
# test the --transport option has taken effect from within the
38
import bzrlib.transport.sftp
39
except ParamikoNotPresent:
40
raise TestSkipped("Paramiko not present")
41
if TestOptions.current_test != "test_transport_set_to_sftp":
43
self.assertEqual(bzrlib.transport.sftp.SFTPAbsoluteServer,
44
bzrlib.tests.default_transport)
46
def test_transport_set_to_memory(self):
47
# test the --transport option has taken effect from within the
49
import bzrlib.transport.memory
50
if TestOptions.current_test != "test_transport_set_to_memory":
52
self.assertEqual(bzrlib.transport.memory.MemoryServer,
53
bzrlib.tests.default_transport)
55
def test_transport(self):
56
# test that --transport=sftp works
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
65
TestOptions.current_test = "test_transport_set_to_sftp"
66
stdout = self.capture('selftest --transport=sftp test_transport_set_to_sftp')
68
self.assertContainsRe(stdout, 'Ran 1 test')
69
self.assertEqual(old_transport, bzrlib.tests.default_transport)
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)
76
bzrlib.tests.default_transport = old_transport
77
TestOptions.current_test = None
78
TestCaseInTempDir.TEST_ROOT = old_root
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')
87
'running tests...\nRunning tests: .\nCleaning up: .\ntests passed\n',
91
class TestRunBzr(ExternalBase):
93
def run_bzr_captured(self, argv, retcode=0, stdin=None):
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
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)
108
class TestRunBzrCaptured(ExternalBase):
110
def apply_redirected(self, stdin=None, stdout=None, stderr=None,
111
a_callable=None, *args, **kwargs):
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')
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)
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
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)
added
removed
bzrlib/tests/blackbox/test_selftest.py
