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
81
class TestRunBzr(ExternalBase):
83
def run_bzr_captured(self, argv, retcode=0, stdin=None):
87
# test that the stdin keyword to run_bzr is passed through to
88
# run_bzr_captured as-is. We do this by overriding
89
# run_bzr_captured in this class, and then calling run_bzr,
90
# which is a convenience function for run_bzr_captured, so
92
self.run_bzr('foo', 'bar', stdin='gam')
93
self.assertEqual('gam', self.stdin)
94
self.run_bzr('foo', 'bar', stdin='zippy')
95
self.assertEqual('zippy', self.stdin)
98
class TestRunBzrCaptured(ExternalBase):
100
def apply_redirected(self, stdin=None, stdout=None, stderr=None,
101
a_callable=None, *args, **kwargs):
103
self.factory_stdin = getattr(bzrlib.ui.ui_factory, "stdin", None)
104
self.factory = bzrlib.ui.ui_factory
105
stdout.write('foo\n')
106
stderr.write('bar\n')
109
def test_stdin(self):
110
# test that the stdin keyword to run_bzr_captured is passed through to
111
# apply_redirected as a StringIO. We do this by overriding
112
# apply_redirected in this class, and then calling run_bzr_captured,
113
# which calls apply_redirected.
114
self.run_bzr_captured(['foo', 'bar'], stdin='gam')
115
self.assertEqual('gam', self.stdin.read())
116
self.assertTrue(self.stdin is self.factory_stdin)
117
self.run_bzr_captured(['foo', 'bar'], stdin='zippy')
118
self.assertEqual('zippy', self.stdin.read())
119
self.assertTrue(self.stdin is self.factory_stdin)
121
def test_ui_factory(self):
122
# each invocation of self.run_bzr_captured should get its own UI
123
# factory, which is an instance of TestUIFactory, with stdout and
124
# stderr attached to the stdout and stderr of the invoked
126
current_factory = bzrlib.ui.ui_factory
127
self.run_bzr_captured(['foo'])
128
self.failIf(current_factory is self.factory)
129
self.assertNotEqual(sys.stdout, self.factory.stdout)
130
self.assertNotEqual(sys.stderr, self.factory.stderr)
131
self.assertEqual('foo\n', self.factory.stdout.getvalue())
132
self.assertEqual('bar\n', self.factory.stderr.getvalue())
133
self.assertIsInstance(self.factory, bzrlib.tests.blackbox.TestUIFactory)
added
removed
bzrlib/tests/blackbox/test_selftest.py
