« back to all changes in this revision
Viewing changes to bzrlib/selftest/__init__.py
- browse files at revision 974.1.52
- compare with another revision
- compare with revision 1185.46.2
- download diff
- download tarball
- view history from revision 974.1.52
- Committer: aaron.bentley at utoronto
- Date: 2005-09-04 02:59:56 UTC
- mfrom: (1172)
- mto: (1185.3.4)
- mto: This revision was merged to the branch mainline in revision 1178.
- Revision ID: aaron.bentley@utoronto.ca-20050904025956-776ba4f07de97700
Merged mpool's latest changes (~0.0.7)
- files added:
- bzrlib/builtins.py
- files removed:
- bzrlib/plugins/checkperms
- files modified:
- .bzrignore
added
removed
bzrlib/selftest/__init__.py
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
16
16
17
17
18
from testsweet import TestCase, run_suite, InTempDir, FunctionalTestCase
18
import logging
19
import unittest
20
import tempfile
21
import os
22
import sys
23
import subprocess
24
25
from testsweet import run_suite
19
26
import bzrlib.commands
27
28
import bzrlib.trace
20
29
import bzrlib.fetch
21
import bzrlib.selftest.teststore
30
22
31
23
32
MODULES_TO_TEST = []
24
33
MODULES_TO_DOCTEST = []
25
34
26
27
class BzrTestBase(InTempDir):
28
"""bzr-specific test base class"""
35
from logging import debug, warning, error
36
37
class CommandFailed(Exception):
38
pass
39
40
class TestCase(unittest.TestCase):
41
"""Base class for bzr unit tests.
42
43
Tests that need access to disk resources should subclass
44
TestCaseInTempDir not TestCase.
45
46
Error and debug log messages are redirected from their usual
47
location into a temporary file, the contents of which can be
48
retrieved by _get_log().
49
50
There are also convenience functions to invoke bzr's command-line
51
routine, and to build and check bzr trees."""
52
53
BZRPATH = 'bzr'
54
55
def setUp(self):
56
# this replaces the default testsweet.TestCase; we don't want logging changed
57
unittest.TestCase.setUp(self)
58
bzrlib.trace.disable_default_logging()
59
self._enable_file_logging()
60
61
62
def _enable_file_logging(self):
63
fileno, name = tempfile.mkstemp(suffix='.log', prefix='testbzr')
64
65
self._log_file = os.fdopen(fileno, 'w+')
66
67
hdlr = logging.StreamHandler(self._log_file)
68
hdlr.setLevel(logging.DEBUG)
69
hdlr.setFormatter(logging.Formatter('%(levelname)4.4s %(message)s'))
70
logging.getLogger('').addHandler(hdlr)
71
logging.getLogger('').setLevel(logging.DEBUG)
72
self._log_hdlr = hdlr
73
debug('opened log file %s', name)
74
75
self._log_file_name = name
76
77
78
def tearDown(self):
79
logging.getLogger('').removeHandler(self._log_hdlr)
80
bzrlib.trace.enable_default_logging()
81
logging.debug('%s teardown', self.id())
82
self._log_file.close()
83
unittest.TestCase.tearDown(self)
84
85
86
def log(self, *args):
87
logging.debug(*args)
88
89
def _get_log(self):
90
"""Return as a string the log for this test"""
91
return open(self._log_file_name).read()
92
29
93
def run_bzr(self, *args, **kwargs):
94
"""Invoke bzr, as if it were run from the command line.
95
96
This should be the main method for tests that want to exercise the
97
overall behavior of the bzr application (rather than a unit test
98
or a functional test of the library.)
99
100
Much of the old code runs bzr by forking a new copy of Python, but
101
that is slower, harder to debug, and generally not necessary.
102
"""
30
103
retcode = kwargs.get('retcode', 0)
31
104
result = self.apply_redirected(None, None, None,
32
105
bzrlib.commands.run_bzr, args)
33
106
self.assertEquals(result, retcode)
34
107
108
109
def check_inventory_shape(self, inv, shape):
110
"""
111
Compare an inventory to a list of expected names.
112
113
Fail if they are not precisely equal.
114
"""
115
extras = []
116
shape = list(shape) # copy
117
for path, ie in inv.entries():
118
name = path.replace('\\', '/')
119
if ie.kind == 'dir':
120
name = name + '/'
121
if name in shape:
122
shape.remove(name)
123
else:
124
extras.append(name)
125
if shape:
126
self.fail("expected paths not found in inventory: %r" % shape)
127
if extras:
128
self.fail("unexpected paths found in inventory: %r" % extras)
129
130
def apply_redirected(self, stdin=None, stdout=None, stderr=None,
131
a_callable=None, *args, **kwargs):
132
"""Call callable with redirected std io pipes.
133
134
Returns the return code."""
135
from StringIO import StringIO
136
if not callable(a_callable):
137
raise ValueError("a_callable must be callable.")
138
if stdin is None:
139
stdin = StringIO("")
140
if stdout is None:
141
stdout = self._log_file
142
if stderr is None:
143
stderr = self._log_file
144
real_stdin = sys.stdin
145
real_stdout = sys.stdout
146
real_stderr = sys.stderr
147
try:
148
sys.stdout = stdout
149
sys.stderr = stderr
150
sys.stdin = stdin
151
return a_callable(*args, **kwargs)
152
finally:
153
sys.stdout = real_stdout
154
sys.stderr = real_stderr
155
sys.stdin = real_stdin
156
157
158
BzrTestBase = TestCase
159
160
161
class TestCaseInTempDir(TestCase):
162
"""Derived class that runs a test within a temporary directory.
163
164
This is useful for tests that need to create a branch, etc.
165
166
The directory is created in a slightly complex way: for each
167
Python invocation, a new temporary top-level directory is created.
168
All test cases create their own directory within that. If the
169
tests complete successfully, the directory is removed.
170
171
InTempDir is an old alias for FunctionalTestCase.
172
"""
173
174
TEST_ROOT = None
175
_TEST_NAME = 'test'
176
OVERRIDE_PYTHON = 'python'
177
178
def check_file_contents(self, filename, expect):
179
self.log("check contents of file %s" % filename)
180
contents = file(filename, 'r').read()
181
if contents != expect:
182
self.log("expected: %r" % expect)
183
self.log("actually: %r" % contents)
184
self.fail("contents of %s not as expected")
185
186
def _make_test_root(self):
187
import os
188
import shutil
189
import tempfile
190
191
if TestCaseInTempDir.TEST_ROOT is not None:
192
return
193
TestCaseInTempDir.TEST_ROOT = os.path.abspath(
194
tempfile.mkdtemp(suffix='.tmp',
195
prefix=self._TEST_NAME + '-',
196
dir=os.curdir))
197
198
# make a fake bzr directory there to prevent any tests propagating
199
# up onto the source directory's real branch
200
os.mkdir(os.path.join(TestCaseInTempDir.TEST_ROOT, '.bzr'))
201
202
def setUp(self):
203
super(TestCaseInTempDir, self).setUp()
204
import os
205
self._make_test_root()
206
self._currentdir = os.getcwdu()
207
self.test_dir = os.path.join(self.TEST_ROOT, self.id())
208
os.mkdir(self.test_dir)
209
os.chdir(self.test_dir)
210
211
def tearDown(self):
212
import os
213
os.chdir(self._currentdir)
214
super(TestCaseInTempDir, self).tearDown()
215
216
def _formcmd(self, cmd):
217
if isinstance(cmd, basestring):
218
cmd = cmd.split()
219
if cmd[0] == 'bzr':
220
cmd[0] = self.BZRPATH
221
if self.OVERRIDE_PYTHON:
222
cmd.insert(0, self.OVERRIDE_PYTHON)
223
self.log('$ %r' % cmd)
224
return cmd
225
226
def runcmd(self, cmd, retcode=0):
227
"""Run one command and check the return code.
228
229
Returns a tuple of (stdout,stderr) strings.
230
231
If a single string is based, it is split into words.
232
For commands that are not simple space-separated words, please
233
pass a list instead."""
234
cmd = self._formcmd(cmd)
235
self.log('$ ' + ' '.join(cmd))
236
actual_retcode = subprocess.call(cmd, stdout=self._log_file,
237
stderr=self._log_file)
238
if retcode != actual_retcode:
239
raise CommandFailed("test failed: %r returned %d, expected %d"
240
% (cmd, actual_retcode, retcode))
241
242
def backtick(self, cmd, retcode=0):
243
"""Run a command and return its output"""
244
cmd = self._formcmd(cmd)
245
child = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=self._log_file)
246
outd, errd = child.communicate()
247
self.log(outd)
248
actual_retcode = child.wait()
249
250
outd = outd.replace('\r', '')
251
252
if retcode != actual_retcode:
253
raise CommandFailed("test failed: %r returned %d, expected %d"
254
% (cmd, actual_retcode, retcode))
255
256
return outd
257
258
259
260
def build_tree(self, shape):
261
"""Build a test tree according to a pattern.
262
263
shape is a sequence of file specifications. If the final
264
character is '/', a directory is created.
265
266
This doesn't add anything to a branch.
267
"""
268
# XXX: It's OK to just create them using forward slashes on windows?
269
import os
270
for name in shape:
271
assert isinstance(name, basestring)
272
if name[-1] == '/':
273
os.mkdir(name[:-1])
274
else:
275
f = file(name, 'wt')
276
print >>f, "contents of", name
277
f.close()
278
279
280
281
class MetaTestLog(TestCase):
282
def test_logging(self):
283
"""Test logs are captured when a test fails."""
284
logging.info('an info message')
285
warning('something looks dodgy...')
286
logging.debug('hello, test is running')
287
##assert 0
288
35
289
36
290
def selftest(verbose=False, pattern=".*"):
37
291
return run_suite(test_suite(), 'testbzr', verbose=verbose, pattern=pattern)
46
300
import shutil
47
301
import time
48
302
import sys
49
import unittest
50
303
51
304
global MODULES_TO_TEST, MODULES_TO_DOCTEST
52
305
53
306
testmod_names = \
54
['bzrlib.selftest.whitebox',
307
['bzrlib.selftest.MetaTestLog',
308
'bzrlib.selftest.test_parent',
309
'bzrlib.selftest.testinv',
310
'bzrlib.selftest.testfetch',
55
311
'bzrlib.selftest.versioning',
56
'bzrlib.selftest.testfetch',
57
'bzrlib.selftest.testinv',
312
'bzrlib.selftest.whitebox',
58
313
'bzrlib.selftest.testmerge3',
59
314
'bzrlib.selftest.testhashcache',
60
315
'bzrlib.selftest.teststatus',
Loggerhead is a web-based interface for Breezy
Version: 2.0.1