« back to all changes in this revision
Viewing changes to bzrlib/selftest/__init__.py
- browse files at revision 788
- compare with another revision
- download diff
- download tarball
- view history from revision 788
- Committer: Martin Pool
- Date: 2005-06-27 01:26:11 UTC
- Revision ID: mbp@sourcefrog.net-20050627012611-4effb7007553fde1
- tweak rsync upload script
- files added:
- bzrlib/statcache.py
- files removed:
- bzrlib/hashcache.py
- effbot
- effbot/org
- tools
- 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 TestBase, run_suite, InTempDir
19
20
MODULES_TO_TEST = []
21
MODULES_TO_DOCTEST = []
22
23
def selftest(verbose=False):
18
from unittest import TestResult, TestCase
19
20
try:
21
import shutil
22
from subprocess import call, Popen, PIPE
23
except ImportError, e:
24
sys.stderr.write("testbzr: sorry, this test suite requires the subprocess module\n"
25
"this is shipped with python2.4 and available separately for 2.3\n")
26
raise
27
28
29
class CommandFailed(Exception):
30
pass
31
32
33
class TestBase(TestCase):
34
"""Base class for bzr test cases.
35
36
Just defines some useful helper functions; doesn't actually test
37
anything.
38
"""
39
40
# TODO: Special methods to invoke bzr, so that we can run it
41
# through a specified Python intepreter
42
43
OVERRIDE_PYTHON = None # to run with alternative python 'python'
44
BZRPATH = 'bzr'
45
46
_log_buf = ""
47
48
49
def formcmd(self, cmd):
50
if isinstance(cmd, basestring):
51
cmd = cmd.split()
52
53
if cmd[0] == 'bzr':
54
cmd[0] = self.BZRPATH
55
if self.OVERRIDE_PYTHON:
56
cmd.insert(0, self.OVERRIDE_PYTHON)
57
58
self.log('$ %r' % cmd)
59
60
return cmd
61
62
63
def runcmd(self, cmd, retcode=0):
64
"""Run one command and check the return code.
65
66
Returns a tuple of (stdout,stderr) strings.
67
68
If a single string is based, it is split into words.
69
For commands that are not simple space-separated words, please
70
pass a list instead."""
71
cmd = self.formcmd(cmd)
72
73
self.log('$ ' + ' '.join(cmd))
74
actual_retcode = call(cmd, stdout=self.TEST_LOG, stderr=self.TEST_LOG)
75
76
if retcode != actual_retcode:
77
raise CommandFailed("test failed: %r returned %d, expected %d"
78
% (cmd, actual_retcode, retcode))
79
80
81
def backtick(self, cmd, retcode=0):
82
"""Run a command and return its output"""
83
cmd = self.formcmd(cmd)
84
child = Popen(cmd, stdout=PIPE, stderr=self.TEST_LOG)
85
outd, errd = child.communicate()
86
self.log(outd)
87
actual_retcode = child.wait()
88
89
outd = outd.replace('\r', '')
90
91
if retcode != actual_retcode:
92
raise CommandFailed("test failed: %r returned %d, expected %d"
93
% (cmd, actual_retcode, retcode))
94
95
return outd
96
97
98
99
def build_tree(self, shape):
100
"""Build a test tree according to a pattern.
101
102
shape is a sequence of file specifications. If the final
103
character is '/', a directory is created.
104
105
This doesn't add anything to a branch.
106
"""
107
# XXX: It's OK to just create them using forward slashes on windows?
108
import os
109
for name in shape:
110
assert isinstance(name, basestring)
111
if name[-1] == '/':
112
os.mkdir(name[:-1])
113
else:
114
f = file(name, 'wt')
115
print >>f, "contents of", name
116
f.close()
117
118
119
def log(self, msg):
120
"""Log a message to a progress file"""
121
self._log_buf = self._log_buf + str(msg) + '\n'
122
print >>self.TEST_LOG, msg
123
124
125
def check_inventory_shape(self, inv, shape):
126
"""
127
Compare an inventory to a list of expected names.
128
129
Fail if they are not precisely equal.
130
"""
131
extras = []
132
shape = list(shape) # copy
133
for path, ie in inv.entries():
134
name = path.replace('\\', '/')
135
if ie.kind == 'dir':
136
name = name + '/'
137
if name in shape:
138
shape.remove(name)
139
else:
140
extras.append(name)
141
if shape:
142
self.fail("expected paths not found in inventory: %r" % shape)
143
if extras:
144
self.fail("unexpected paths found in inventory: %r" % extras)
145
146
147
def check_file_contents(self, filename, expect):
148
self.log("check contents of file %s" % filename)
149
contents = file(filename, 'r').read()
150
if contents != expect:
151
self.log("expected: %r" % expected)
152
self.log("actually: %r" % contents)
153
self.fail("contents of %s not as expected")
154
155
156
157
class InTempDir(TestBase):
158
"""Base class for tests run in a temporary branch."""
159
def setUp(self):
160
import os
161
self.test_dir = os.path.join(self.TEST_ROOT, self.__class__.__name__)
162
os.mkdir(self.test_dir)
163
os.chdir(self.test_dir)
164
165
def tearDown(self):
166
import os
167
os.chdir(self.TEST_ROOT)
168
169
170
171
172
173
class _MyResult(TestResult):
174
"""
175
Custom TestResult.
176
177
No special behaviour for now.
178
"""
179
def __init__(self, out):
180
self.out = out
181
TestResult.__init__(self)
182
183
def startTest(self, test):
184
# TODO: Maybe show test.shortDescription somewhere?
185
print >>self.out, '%-60.60s' % test.id(),
186
self.out.flush()
187
TestResult.startTest(self, test)
188
189
def stopTest(self, test):
190
# print
191
TestResult.stopTest(self, test)
192
193
194
def addError(self, test, err):
195
print >>self.out, 'ERROR'
196
TestResult.addError(self, test, err)
197
_show_test_failure('error', test, err, self.out)
198
199
def addFailure(self, test, err):
200
print >>self.out, 'FAILURE'
201
TestResult.addFailure(self, test, err)
202
_show_test_failure('failure', test, err, self.out)
203
204
def addSuccess(self, test):
205
print >>self.out, 'OK'
206
TestResult.addSuccess(self, test)
207
208
209
210
def selftest():
24
211
from unittest import TestLoader, TestSuite
25
import bzrlib, bzrlib.store, bzrlib.inventory, bzrlib.branch
26
import bzrlib.osutils, bzrlib.commands, bzrlib.merge3, bzrlib.plugin
27
global MODULES_TO_TEST, MODULES_TO_DOCTEST
28
212
import bzrlib
29
213
import bzrlib.selftest.whitebox
30
214
import bzrlib.selftest.blackbox
31
215
import bzrlib.selftest.versioning
32
import bzrlib.selftest.testmerge3
33
import bzrlib.selftest.testhashcache
34
import bzrlib.selftest.testrevisionnamespaces
35
import bzrlib.selftest.testbranch
36
import bzrlib.selftest.teststatus
37
import bzrlib.selftest.testinv
38
import bzrlib.merge_core
39
216
from doctest import DocTestSuite
40
217
import os
41
218
import shutil
42
219
import time
43
220
import sys
44
import unittest
45
46
for m in (bzrlib.store, bzrlib.inventory, bzrlib.branch,
47
bzrlib.osutils, bzrlib.commands, bzrlib.merge3):
48
if m not in MODULES_TO_DOCTEST:
49
MODULES_TO_DOCTEST.append(m)
50
51
# ugly ugly!
52
for m in (bzrlib.selftest.whitebox,
53
bzrlib.selftest.versioning,
54
bzrlib.selftest.testinv,
55
bzrlib.selftest.testmerge3,
56
bzrlib.selftest.testhashcache,
57
bzrlib.selftest.teststatus,
58
bzrlib.selftest.blackbox,
59
bzrlib.selftest.testhashcache,
60
bzrlib.selftest.testrevisionnamespaces,
61
bzrlib.selftest.testbranch,
62
):
63
if m not in MODULES_TO_TEST:
64
MODULES_TO_TEST.append(m)
65
66
67
TestBase.BZRPATH = os.path.join(os.path.realpath(os.path.dirname(bzrlib.__path__[0])), 'bzr')
68
print '%-30s %s' % ('bzr binary', TestBase.BZRPATH)
69
221
222
_setup_test_log()
223
_setup_test_dir()
70
224
print
71
225
72
226
suite = TestSuite()
73
74
# should also test bzrlib.merge_core, but they seem to be out of date with
75
# the code.
76
77
78
# XXX: python2.3's TestLoader() doesn't seem to find all the
79
# tests; don't know why
80
for m in MODULES_TO_TEST:
81
suite.addTest(TestLoader().loadTestsFromModule(m))
82
83
for m in (MODULES_TO_DOCTEST):
227
tl = TestLoader()
228
229
for m in bzrlib.selftest.whitebox, \
230
bzrlib.selftest.versioning:
231
suite.addTest(tl.loadTestsFromModule(m))
232
233
for m in bzrlib.store, bzrlib.inventory, bzrlib.branch, bzrlib.osutils, \
234
bzrlib.commands:
84
235
suite.addTest(DocTestSuite(m))
85
236
86
for p in bzrlib.plugin.all_plugins:
87
if hasattr(p, 'test_suite'):
88
suite.addTest(p.test_suite())
89
90
suite.addTest(unittest.makeSuite(bzrlib.merge_core.MergeTest, 'test_'))
91
92
return run_suite(suite, 'testbzr', verbose=verbose)
93
94
237
suite.addTest(bzrlib.selftest.blackbox.suite())
238
239
# save stdout & stderr so there's no leakage from code-under-test
240
real_stdout = sys.stdout
241
real_stderr = sys.stderr
242
sys.stdout = sys.stderr = TestBase.TEST_LOG
243
try:
244
result = _MyResult(real_stdout)
245
suite.run(result)
246
finally:
247
sys.stdout = real_stdout
248
sys.stderr = real_stderr
249
250
_show_results(result)
251
252
return result.wasSuccessful()
253
254
255
256
257
def _setup_test_log():
258
import time
259
import os
260
261
log_filename = os.path.abspath('testbzr.log')
262
TestBase.TEST_LOG = open(log_filename, 'wt', buffering=1) # line buffered
263
264
print >>TestBase.TEST_LOG, "bzr tests run at " + time.ctime()
265
print '%-30s %s' % ('test log', log_filename)
266
267
268
def _setup_test_dir():
269
import os
270
import shutil
271
272
TestBase.ORIG_DIR = os.getcwdu()
273
TestBase.TEST_ROOT = os.path.abspath("testbzr.tmp")
274
275
print '%-30s %s' % ('running tests in', TestBase.TEST_ROOT)
276
277
if os.path.exists(TestBase.TEST_ROOT):
278
shutil.rmtree(TestBase.TEST_ROOT)
279
os.mkdir(TestBase.TEST_ROOT)
280
os.chdir(TestBase.TEST_ROOT)
281
282
# make a fake bzr directory there to prevent any tests propagating
283
# up onto the source directory's real branch
284
os.mkdir(os.path.join(TestBase.TEST_ROOT, '.bzr'))
285
286
287
288
def _show_results(result):
289
print
290
print '%4d tests run' % result.testsRun
291
print '%4d errors' % len(result.errors)
292
print '%4d failures' % len(result.failures)
293
294
295
296
def _show_test_failure(kind, case, exc_info, out):
297
from traceback import print_exception
298
299
print >>out, '-' * 60
300
print >>out, case
301
302
desc = case.shortDescription()
303
if desc:
304
print >>out, ' (%s)' % desc
305
306
print_exception(exc_info[0], exc_info[1], exc_info[2], None, out)
307
308
if isinstance(case, TestBase):
309
print >>out
310
print >>out, 'log from this test:'
311
print >>out, case._log_buf
312
313
print >>out, '-' * 60
314
95
315
Loggerhead is a web-based interface for Breezy
Version: 2.0.1