« back to all changes in this revision
Viewing changes to bzrlib/selftest/__init__.py
- browse files at revision 819
- compare with another revision
- download diff
- download tarball
- view history from revision 819
- Committer: Martin Pool
- Date: 2005-07-04 12:26:02 UTC
- Revision ID: mbp@sourcefrog.net-20050704122602-69901910521e62c3
- check command checks that all inventory-ids are the same as in the revision.
- files added:
- bzrlib/statcache.py
- plugins/rsync
- files removed:
- bzrlib/hashcache.py
- 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 = []
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
22
209
23
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
212
import bzrlib, bzrlib.store, bzrlib.inventory, bzrlib.branch, bzrlib.osutils, bzrlib.commands
28
213
29
214
import bzrlib.selftest.whitebox
30
215
import bzrlib.selftest.blackbox
31
216
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
217
import bzrlib.merge_core
39
218
from doctest import DocTestSuite
40
219
import os
41
220
import shutil
42
221
import time
43
222
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
for m in (bzrlib.selftest.whitebox,
52
bzrlib.selftest.versioning,
53
bzrlib.selftest.testinv,
54
bzrlib.selftest.testmerge3,
55
bzrlib.selftest.testhashcache,
56
bzrlib.selftest.teststatus,
57
bzrlib.selftest.blackbox,
58
bzrlib.selftest.testhashcache,
59
bzrlib.selftest.testrevisionnamespaces,
60
bzrlib.selftest.testbranch,
61
):
62
if m not in MODULES_TO_TEST:
63
MODULES_TO_TEST.append(m)
64
65
223
66
224
TestBase.BZRPATH = os.path.join(os.path.realpath(os.path.dirname(bzrlib.__path__[0])), 'bzr')
67
225
print '%-30s %s' % ('bzr binary', TestBase.BZRPATH)
68
226
227
_setup_test_log()
228
_setup_test_dir()
69
229
print
70
230
71
231
suite = TestSuite()
232
tl = TestLoader()
72
233
73
234
# should also test bzrlib.merge_core, but they seem to be out of date with
74
235
# the code.
75
236
76
77
# XXX: python2.3's TestLoader() doesn't seem to find all the
78
# tests; don't know why
79
for m in MODULES_TO_TEST:
80
suite.addTest(TestLoader().loadTestsFromModule(m))
81
82
for m in (MODULES_TO_DOCTEST):
237
for m in bzrlib.selftest.whitebox, \
238
bzrlib.selftest.versioning:
239
suite.addTest(tl.loadTestsFromModule(m))
240
241
for m in bzrlib.store, bzrlib.inventory, bzrlib.branch, bzrlib.osutils, \
242
bzrlib.commands:
83
243
suite.addTest(DocTestSuite(m))
84
244
85
for p in bzrlib.plugin.all_plugins:
86
if hasattr(p, 'test_suite'):
87
suite.addTest(p.test_suite())
88
89
suite.addTest(unittest.makeSuite(bzrlib.merge_core.MergeTest, 'test_'))
90
91
return run_suite(suite, 'testbzr')
92
93
245
suite.addTest(bzrlib.selftest.blackbox.suite())
246
247
# save stdout & stderr so there's no leakage from code-under-test
248
real_stdout = sys.stdout
249
real_stderr = sys.stderr
250
sys.stdout = sys.stderr = TestBase.TEST_LOG
251
try:
252
result = _MyResult(real_stdout)
253
suite.run(result)
254
finally:
255
sys.stdout = real_stdout
256
sys.stderr = real_stderr
257
258
_show_results(result)
259
260
return result.wasSuccessful()
261
262
263
264
265
def _setup_test_log():
266
import time
267
import os
268
269
log_filename = os.path.abspath('testbzr.log')
270
TestBase.TEST_LOG = open(log_filename, 'wt', buffering=1) # line buffered
271
272
print >>TestBase.TEST_LOG, "bzr tests run at " + time.ctime()
273
print '%-30s %s' % ('test log', log_filename)
274
275
276
def _setup_test_dir():
277
import os
278
import shutil
279
280
TestBase.ORIG_DIR = os.getcwdu()
281
TestBase.TEST_ROOT = os.path.abspath("testbzr.tmp")
282
283
print '%-30s %s' % ('running tests in', TestBase.TEST_ROOT)
284
285
if os.path.exists(TestBase.TEST_ROOT):
286
shutil.rmtree(TestBase.TEST_ROOT)
287
os.mkdir(TestBase.TEST_ROOT)
288
os.chdir(TestBase.TEST_ROOT)
289
290
# make a fake bzr directory there to prevent any tests propagating
291
# up onto the source directory's real branch
292
os.mkdir(os.path.join(TestBase.TEST_ROOT, '.bzr'))
293
294
295
296
def _show_results(result):
297
print
298
print '%4d tests run' % result.testsRun
299
print '%4d errors' % len(result.errors)
300
print '%4d failures' % len(result.failures)
301
302
303
304
def _show_test_failure(kind, case, exc_info, out):
305
from traceback import print_exception
306
307
print >>out, '-' * 60
308
print >>out, case
309
310
desc = case.shortDescription()
311
if desc:
312
print >>out, ' (%s)' % desc
313
314
print_exception(exc_info[0], exc_info[1], exc_info[2], None, out)
315
316
if isinstance(case, TestBase):
317
print >>out
318
print >>out, 'log from this test:'
319
print >>out, case._log_buf
320
321
print >>out, '-' * 60
322
94
323
Loggerhead is a web-based interface for Breezy
Version: 2.0.1