15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18
# TODO: Perhaps there should be an API to find out if bzr running under the
19
# test suite -- some plugins might want to avoid making intrusive changes if
20
# this is the case. However, we want behaviour under to test to diverge as
21
# little as possible, so this should be used rarely if it's added at all.
22
# (Suggestion from j-a-meinel, 2005-11-24)
24
# NOTE: Some classes in here use camelCaseNaming() rather than
25
# underscore_naming(). That's for consistency with unittest; it's not the
26
# general style of bzrlib. Please continue that consistency when adding e.g.
27
# new assertFoo() methods.
30
from cStringIO import StringIO
45
import bzrlib.bzrdir as bzrdir
18
from testsweet import TestCase, run_suite, InTempDir
46
19
import bzrlib.commands
47
import bzrlib.errors as errors
48
import bzrlib.inventory
49
import bzrlib.iterablefile
53
import bzrlib.osutils as osutils
57
from bzrlib.transport import urlescape, get_transport
58
import bzrlib.transport
59
from bzrlib.transport.local import LocalRelpathServer
60
from bzrlib.transport.readonly import ReadonlyServer
61
from bzrlib.trace import mutter
62
from bzrlib.tests.TestUtil import TestLoader, TestSuite
63
from bzrlib.tests.treeshape import build_tree_contents
64
from bzrlib.workingtree import WorkingTree, WorkingTreeFormat2
66
default_transport = LocalRelpathServer
68
21
MODULES_TO_TEST = []
69
MODULES_TO_DOCTEST = [
81
def packages_to_test():
82
"""Return a list of packages to test.
84
The packages are not globally imported so that import failures are
85
triggered when running selftest, not when importing the command.
88
import bzrlib.tests.blackbox
89
import bzrlib.tests.branch_implementations
90
import bzrlib.tests.bzrdir_implementations
91
import bzrlib.tests.interrepository_implementations
92
import bzrlib.tests.interversionedfile_implementations
93
import bzrlib.tests.repository_implementations
94
import bzrlib.tests.revisionstore_implementations
95
import bzrlib.tests.workingtree_implementations
98
bzrlib.tests.blackbox,
99
bzrlib.tests.branch_implementations,
100
bzrlib.tests.bzrdir_implementations,
101
bzrlib.tests.interrepository_implementations,
102
bzrlib.tests.interversionedfile_implementations,
103
bzrlib.tests.repository_implementations,
104
bzrlib.tests.revisionstore_implementations,
105
bzrlib.tests.workingtree_implementations,
109
class _MyResult(unittest._TextTestResult):
110
"""Custom TestResult.
112
Shows output in a different format, including displaying runtime for tests.
116
def _elapsedTime(self):
117
return "%5dms" % (1000 * (time.time() - self._start_time))
119
def startTest(self, test):
120
unittest.TestResult.startTest(self, test)
121
# In a short description, the important words are in
122
# the beginning, but in an id, the important words are
124
SHOW_DESCRIPTIONS = False
126
width = osutils.terminal_width()
127
name_width = width - 15
129
if SHOW_DESCRIPTIONS:
130
what = test.shortDescription()
132
if len(what) > name_width:
133
what = what[:name_width-3] + '...'
136
if what.startswith('bzrlib.tests.'):
138
if len(what) > name_width:
139
what = '...' + what[3-name_width:]
140
what = what.ljust(name_width)
141
self.stream.write(what)
143
self._start_time = time.time()
145
def addError(self, test, err):
146
if isinstance(err[1], TestSkipped):
147
return self.addSkipped(test, err)
148
unittest.TestResult.addError(self, test, err)
150
self.stream.writeln("ERROR %s" % self._elapsedTime())
152
self.stream.write('E')
157
def addFailure(self, test, err):
158
unittest.TestResult.addFailure(self, test, err)
160
self.stream.writeln(" FAIL %s" % self._elapsedTime())
162
self.stream.write('F')
167
def addSuccess(self, test):
169
self.stream.writeln(' OK %s' % self._elapsedTime())
171
self.stream.write('~')
173
unittest.TestResult.addSuccess(self, test)
175
def addSkipped(self, test, skip_excinfo):
177
print >>self.stream, ' SKIP %s' % self._elapsedTime()
178
print >>self.stream, ' %s' % skip_excinfo[1]
180
self.stream.write('S')
182
# seems best to treat this as success from point-of-view of unittest
183
# -- it actually does nothing so it barely matters :)
184
unittest.TestResult.addSuccess(self, test)
186
def printErrorList(self, flavour, errors):
187
for test, err in errors:
188
self.stream.writeln(self.separator1)
189
self.stream.writeln("%s: %s" % (flavour, self.getDescription(test)))
190
if getattr(test, '_get_log', None) is not None:
192
print >>self.stream, \
193
('vvvv[log from %s]' % test.id()).ljust(78,'-')
194
print >>self.stream, test._get_log()
195
print >>self.stream, \
196
('^^^^[log from %s]' % test.id()).ljust(78,'-')
197
self.stream.writeln(self.separator2)
198
self.stream.writeln("%s" % err)
201
class TextTestRunner(unittest.TextTestRunner):
202
stop_on_failure = False
204
def _makeResult(self):
205
result = _MyResult(self.stream, self.descriptions, self.verbosity)
206
result.stop_early = self.stop_on_failure
210
def iter_suite_tests(suite):
211
"""Return all tests in a suite, recursing through nested suites"""
212
for item in suite._tests:
213
if isinstance(item, unittest.TestCase):
215
elif isinstance(item, unittest.TestSuite):
216
for r in iter_suite_tests(item):
219
raise Exception('unknown object %r inside test suite %r'
223
class TestSkipped(Exception):
224
"""Indicates that a test was intentionally skipped, rather than failing."""
228
class CommandFailed(Exception):
231
class TestCase(unittest.TestCase):
232
"""Base class for bzr unit tests.
234
Tests that need access to disk resources should subclass
235
TestCaseInTempDir not TestCase.
237
Error and debug log messages are redirected from their usual
238
location into a temporary file, the contents of which can be
239
retrieved by _get_log(). We use a real OS file, not an in-memory object,
240
so that it can also capture file IO. When the test completes this file
241
is read into memory and removed from disk.
243
There are also convenience functions to invoke bzr's command-line
244
routine, and to build and check bzr trees.
246
In addition to the usual method of overriding tearDown(), this class also
247
allows subclasses to register functions into the _cleanups list, which is
248
run in order as the object is torn down. It's less likely this will be
249
accidentally overlooked.
253
_log_file_name = None
256
def __init__(self, methodName='testMethod'):
257
super(TestCase, self).__init__(methodName)
261
unittest.TestCase.setUp(self)
262
self._cleanEnvironment()
263
bzrlib.trace.disable_default_logging()
266
def _ndiff_strings(self, a, b):
267
"""Return ndiff between two strings containing lines.
269
A trailing newline is added if missing to make the strings
271
if b and b[-1] != '\n':
273
if a and a[-1] != '\n':
275
difflines = difflib.ndiff(a.splitlines(True),
277
linejunk=lambda x: False,
278
charjunk=lambda x: False)
279
return ''.join(difflines)
281
def assertEqualDiff(self, a, b, message=None):
282
"""Assert two texts are equal, if not raise an exception.
284
This is intended for use with multi-line strings where it can
285
be hard to find the differences by eye.
287
# TODO: perhaps override assertEquals to call this for strings?
291
message = "texts not equal:\n"
292
raise AssertionError(message +
293
self._ndiff_strings(a, b))
295
def assertEqualMode(self, mode, mode_test):
296
self.assertEqual(mode, mode_test,
297
'mode mismatch %o != %o' % (mode, mode_test))
299
def assertStartsWith(self, s, prefix):
300
if not s.startswith(prefix):
301
raise AssertionError('string %r does not start with %r' % (s, prefix))
303
def assertEndsWith(self, s, suffix):
304
if not s.endswith(prefix):
305
raise AssertionError('string %r does not end with %r' % (s, suffix))
307
def assertContainsRe(self, haystack, needle_re):
308
"""Assert that a contains something matching a regular expression."""
309
if not re.search(needle_re, haystack):
310
raise AssertionError('pattern "%s" not found in "%s"'
311
% (needle_re, haystack))
313
def assertSubset(self, sublist, superlist):
314
"""Assert that every entry in sublist is present in superlist."""
316
for entry in sublist:
317
if entry not in superlist:
318
missing.append(entry)
320
raise AssertionError("value(s) %r not present in container %r" %
321
(missing, superlist))
323
def assertIs(self, left, right):
324
if not (left is right):
325
raise AssertionError("%r is not %r." % (left, right))
327
def assertTransportMode(self, transport, path, mode):
328
"""Fail if a path does not have mode mode.
330
If modes are not supported on this platform, the test is skipped.
332
if sys.platform == 'win32':
334
path_stat = transport.stat(path)
335
actual_mode = stat.S_IMODE(path_stat.st_mode)
336
self.assertEqual(mode, actual_mode,
337
'mode of %r incorrect (%o != %o)' % (path, mode, actual_mode))
339
def assertIsInstance(self, obj, kls):
340
"""Fail if obj is not an instance of kls"""
341
if not isinstance(obj, kls):
342
self.fail("%r is not an instance of %s" % (obj, kls))
344
def _startLogFile(self):
345
"""Send bzr and test log messages to a temporary file.
347
The file is removed as the test is torn down.
349
fileno, name = tempfile.mkstemp(suffix='.log', prefix='testbzr')
350
encoder, decoder, stream_reader, stream_writer = codecs.lookup('UTF-8')
351
self._log_file = stream_writer(os.fdopen(fileno, 'w+'))
352
self._log_nonce = bzrlib.trace.enable_test_log(self._log_file)
353
self._log_file_name = name
354
self.addCleanup(self._finishLogFile)
356
def _finishLogFile(self):
357
"""Finished with the log file.
359
Read contents into memory, close, and delete.
361
bzrlib.trace.disable_test_log(self._log_nonce)
362
self._log_file.seek(0)
363
self._log_contents = self._log_file.read()
364
self._log_file.close()
365
os.remove(self._log_file_name)
366
self._log_file = self._log_file_name = None
368
def addCleanup(self, callable):
369
"""Arrange to run a callable when this case is torn down.
371
Callables are run in the reverse of the order they are registered,
372
ie last-in first-out.
374
if callable in self._cleanups:
375
raise ValueError("cleanup function %r already registered on %s"
377
self._cleanups.append(callable)
379
def _cleanEnvironment(self):
382
'APPDATA': os.getcwd(),
387
self.addCleanup(self._restoreEnvironment)
388
for name, value in new_env.iteritems():
389
self._captureVar(name, value)
392
def _captureVar(self, name, newvalue):
393
"""Set an environment variable, preparing it to be reset when finished."""
394
self.__old_env[name] = os.environ.get(name, None)
396
if name in os.environ:
399
os.environ[name] = newvalue
402
def _restoreVar(name, value):
404
if name in os.environ:
407
os.environ[name] = value
409
def _restoreEnvironment(self):
410
for name, value in self.__old_env.iteritems():
411
self._restoreVar(name, value)
415
unittest.TestCase.tearDown(self)
417
def _runCleanups(self):
418
"""Run registered cleanup functions.
420
This should only be called from TestCase.tearDown.
422
# TODO: Perhaps this should keep running cleanups even if
424
for cleanup_fn in reversed(self._cleanups):
427
def log(self, *args):
431
"""Return as a string the log for this test"""
432
if self._log_file_name:
433
return open(self._log_file_name).read()
435
return self._log_contents
436
# TODO: Delete the log after it's been read in
438
def capture(self, cmd, retcode=0):
439
"""Shortcut that splits cmd into words, runs, and returns stdout"""
440
return self.run_bzr_captured(cmd.split(), retcode=retcode)[0]
442
def run_bzr_captured(self, argv, retcode=0):
443
"""Invoke bzr and return (stdout, stderr).
445
Useful for code that wants to check the contents of the
446
output, the way error messages are presented, etc.
448
This should be the main method for tests that want to exercise the
449
overall behavior of the bzr application (rather than a unit test
450
or a functional test of the library.)
452
Much of the old code runs bzr by forking a new copy of Python, but
453
that is slower, harder to debug, and generally not necessary.
455
This runs bzr through the interface that catches and reports
456
errors, and with logging set to something approximating the
457
default, so that error reporting can be checked.
459
argv -- arguments to invoke bzr
460
retcode -- expected return code, or None for don't-care.
464
self.log('run bzr: %s', ' '.join(argv))
465
# FIXME: don't call into logging here
466
handler = logging.StreamHandler(stderr)
467
handler.setFormatter(bzrlib.trace.QuietFormatter())
468
handler.setLevel(logging.INFO)
469
logger = logging.getLogger('')
470
logger.addHandler(handler)
472
result = self.apply_redirected(None, stdout, stderr,
473
bzrlib.commands.run_bzr_catch_errors,
476
logger.removeHandler(handler)
477
out = stdout.getvalue()
478
err = stderr.getvalue()
480
self.log('output:\n%s', out)
482
self.log('errors:\n%s', err)
483
if retcode is not None:
484
self.assertEquals(result, retcode)
22
MODULES_TO_DOCTEST = []
25
class BzrTestBase(InTempDir):
26
"""bzr-specific test base class"""
487
27
def run_bzr(self, *args, **kwargs):
488
"""Invoke bzr, as if it were run from the command line.
490
This should be the main method for tests that want to exercise the
491
overall behavior of the bzr application (rather than a unit test
492
or a functional test of the library.)
494
This sends the stdout/stderr results into the test's log,
495
where it may be useful for debugging. See also run_captured.
497
retcode = kwargs.pop('retcode', 0)
498
return self.run_bzr_captured(args, retcode)
500
def check_inventory_shape(self, inv, shape):
501
"""Compare an inventory to a list of expected names.
503
Fail if they are not precisely equal.
506
shape = list(shape) # copy
507
for path, ie in inv.entries():
508
name = path.replace('\\', '/')
516
self.fail("expected paths not found in inventory: %r" % shape)
518
self.fail("unexpected paths found in inventory: %r" % extras)
520
def apply_redirected(self, stdin=None, stdout=None, stderr=None,
521
a_callable=None, *args, **kwargs):
522
"""Call callable with redirected std io pipes.
524
Returns the return code."""
525
if not callable(a_callable):
526
raise ValueError("a_callable must be callable.")
530
if getattr(self, "_log_file", None) is not None:
531
stdout = self._log_file
535
if getattr(self, "_log_file", None is not None):
536
stderr = self._log_file
539
real_stdin = sys.stdin
540
real_stdout = sys.stdout
541
real_stderr = sys.stderr
546
return a_callable(*args, **kwargs)
548
sys.stdout = real_stdout
549
sys.stderr = real_stderr
550
sys.stdin = real_stdin
553
BzrTestBase = TestCase
556
class TestCaseInTempDir(TestCase):
557
"""Derived class that runs a test within a temporary directory.
559
This is useful for tests that need to create a branch, etc.
561
The directory is created in a slightly complex way: for each
562
Python invocation, a new temporary top-level directory is created.
563
All test cases create their own directory within that. If the
564
tests complete successfully, the directory is removed.
566
InTempDir is an old alias for FunctionalTestCase.
571
OVERRIDE_PYTHON = 'python'
573
def check_file_contents(self, filename, expect):
574
self.log("check contents of file %s" % filename)
575
contents = file(filename, 'r').read()
576
if contents != expect:
577
self.log("expected: %r" % expect)
578
self.log("actually: %r" % contents)
579
self.fail("contents of %s not as expected" % filename)
581
def _make_test_root(self):
582
if TestCaseInTempDir.TEST_ROOT is not None:
586
root = u'test%04d.tmp' % i
590
if e.errno == errno.EEXIST:
595
# successfully created
596
TestCaseInTempDir.TEST_ROOT = osutils.abspath(root)
598
# make a fake bzr directory there to prevent any tests propagating
599
# up onto the source directory's real branch
600
bzrdir.BzrDir.create_standalone_workingtree(TestCaseInTempDir.TEST_ROOT)
603
super(TestCaseInTempDir, self).setUp()
604
self._make_test_root()
605
_currentdir = os.getcwdu()
606
# shorten the name, to avoid test failures due to path length
607
short_id = self.id().replace('bzrlib.tests.', '') \
608
.replace('__main__.', '')[-100:]
609
# it's possible the same test class is run several times for
610
# parameterized tests, so make sure the names don't collide.
614
candidate_dir = '%s/%s.%d' % (self.TEST_ROOT, short_id, i)
616
candidate_dir = '%s/%s' % (self.TEST_ROOT, short_id)
617
if os.path.exists(candidate_dir):
621
self.test_dir = candidate_dir
622
os.mkdir(self.test_dir)
623
os.chdir(self.test_dir)
625
os.environ['HOME'] = self.test_dir
626
os.environ['APPDATA'] = self.test_dir
627
def _leaveDirectory():
628
os.chdir(_currentdir)
629
self.addCleanup(_leaveDirectory)
631
def build_tree(self, shape, line_endings='native', transport=None):
632
"""Build a test tree according to a pattern.
634
shape is a sequence of file specifications. If the final
635
character is '/', a directory is created.
637
This doesn't add anything to a branch.
638
:param line_endings: Either 'binary' or 'native'
639
in binary mode, exact contents are written
640
in native mode, the line endings match the
641
default platform endings.
643
:param transport: A transport to write to, for building trees on
644
VFS's. If the transport is readonly or None,
645
"." is opened automatically.
647
# XXX: It's OK to just create them using forward slashes on windows?
648
if transport is None or transport.is_readonly():
649
transport = get_transport(".")
651
self.assert_(isinstance(name, basestring))
653
transport.mkdir(urlescape(name[:-1]))
655
if line_endings == 'binary':
657
elif line_endings == 'native':
660
raise errors.BzrError('Invalid line ending request %r' % (line_endings,))
661
content = "contents of %s%s" % (name, end)
662
transport.put(urlescape(name), StringIO(content))
664
def build_tree_contents(self, shape):
665
build_tree_contents(shape)
667
def failUnlessExists(self, path):
668
"""Fail unless path, which may be abs or relative, exists."""
669
self.failUnless(osutils.lexists(path))
671
def failIfExists(self, path):
672
"""Fail if path, which may be abs or relative, exists."""
673
self.failIf(osutils.lexists(path))
675
def assertFileEqual(self, content, path):
676
"""Fail if path does not contain 'content'."""
677
self.failUnless(osutils.lexists(path))
678
self.assertEqualDiff(content, open(path, 'r').read())
681
class TestCaseWithTransport(TestCaseInTempDir):
682
"""A test case that provides get_url and get_readonly_url facilities.
684
These back onto two transport servers, one for readonly access and one for
687
If no explicit class is provided for readonly access, a
688
ReadonlyTransportDecorator is used instead which allows the use of non disk
689
based read write transports.
691
If an explicit class is provided for readonly access, that server and the
692
readwrite one must both define get_url() as resolving to os.getcwd().
695
def __init__(self, methodName='testMethod'):
696
super(TestCaseWithTransport, self).__init__(methodName)
697
self.__readonly_server = None
699
self.transport_server = default_transport
700
self.transport_readonly_server = None
702
def get_readonly_url(self, relpath=None):
703
"""Get a URL for the readonly transport.
705
This will either be backed by '.' or a decorator to the transport
706
used by self.get_url()
707
relpath provides for clients to get a path relative to the base url.
708
These should only be downwards relative, not upwards.
710
base = self.get_readonly_server().get_url()
711
if relpath is not None:
712
if not base.endswith('/'):
714
base = base + relpath
717
def get_readonly_server(self):
718
"""Get the server instance for the readonly transport
720
This is useful for some tests with specific servers to do diagnostics.
722
if self.__readonly_server is None:
723
if self.transport_readonly_server is None:
724
# readonly decorator requested
725
# bring up the server
727
self.__readonly_server = ReadonlyServer()
728
self.__readonly_server.setUp(self.__server)
730
self.__readonly_server = self.transport_readonly_server()
731
self.__readonly_server.setUp()
732
self.addCleanup(self.__readonly_server.tearDown)
733
return self.__readonly_server
735
def get_server(self):
736
"""Get the read/write server instance.
738
This is useful for some tests with specific servers that need
741
if self.__server is None:
742
self.__server = self.transport_server()
743
self.__server.setUp()
744
self.addCleanup(self.__server.tearDown)
747
def get_url(self, relpath=None):
748
"""Get a URL for the readwrite transport.
750
This will either be backed by '.' or to an equivalent non-file based
752
relpath provides for clients to get a path relative to the base url.
753
These should only be downwards relative, not upwards.
755
base = self.get_server().get_url()
756
if relpath is not None and relpath != '.':
757
if not base.endswith('/'):
759
base = base + relpath
762
def get_transport(self):
763
"""Return a writeable transport for the test scratch space"""
764
t = get_transport(self.get_url())
765
self.assertFalse(t.is_readonly())
768
def get_readonly_transport(self):
769
"""Return a readonly transport for the test scratch space
771
This can be used to test that operations which should only need
772
readonly access in fact do not try to write.
774
t = get_transport(self.get_readonly_url())
775
self.assertTrue(t.is_readonly())
778
def make_branch(self, relpath):
779
"""Create a branch on the transport at relpath."""
780
repo = self.make_repository(relpath)
781
return repo.bzrdir.create_branch()
783
def make_bzrdir(self, relpath):
785
url = self.get_url(relpath)
786
segments = relpath.split('/')
787
if segments and segments[-1] not in ('', '.'):
788
parent = self.get_url('/'.join(segments[:-1]))
789
t = get_transport(parent)
791
t.mkdir(segments[-1])
792
except errors.FileExists:
794
return bzrlib.bzrdir.BzrDir.create(url)
795
except errors.UninitializableFormat:
796
raise TestSkipped("Format %s is not initializable.")
798
def make_repository(self, relpath, shared=False):
799
"""Create a repository on our default transport at relpath."""
800
made_control = self.make_bzrdir(relpath)
801
return made_control.create_repository(shared=shared)
803
def make_branch_and_tree(self, relpath):
804
"""Create a branch on the transport and a tree locally.
808
# TODO: always use the local disk path for the working tree,
809
# this obviously requires a format that supports branch references
810
# so check for that by checking bzrdir.BzrDirFormat.get_default_format()
812
b = self.make_branch(relpath)
814
return b.bzrdir.create_workingtree()
815
except errors.NotLocalUrl:
816
# new formats - catch No tree error and create
817
# a branch reference and a checkout.
818
# old formats at that point - raise TestSkipped.
820
return WorkingTreeFormat2().initialize(bzrdir.BzrDir.open(relpath))
822
def assertIsDirectory(self, relpath, transport):
823
"""Assert that relpath within transport is a directory.
825
This may not be possible on all transports; in that case it propagates
826
a TransportNotPossible.
829
mode = transport.stat(relpath).st_mode
830
except errors.NoSuchFile:
831
self.fail("path %s is not a directory; no such file"
833
if not stat.S_ISDIR(mode):
834
self.fail("path %s is not a directory; has mode %#o"
838
class ChrootedTestCase(TestCaseWithTransport):
839
"""A support class that provides readonly urls outside the local namespace.
841
This is done by checking if self.transport_server is a MemoryServer. if it
842
is then we are chrooted already, if it is not then an HttpServer is used
845
TODO RBC 20060127: make this an option to TestCaseWithTransport so it can
846
be used without needed to redo it when a different
851
super(ChrootedTestCase, self).setUp()
852
if not self.transport_server == bzrlib.transport.memory.MemoryServer:
853
self.transport_readonly_server = bzrlib.transport.http.HttpServer
856
def filter_suite_by_re(suite, pattern):
858
filter_re = re.compile(pattern)
859
for test in iter_suite_tests(suite):
860
if filter_re.search(test.id()):
865
def run_suite(suite, name='test', verbose=False, pattern=".*",
866
stop_on_failure=False, keep_output=False,
868
TestCaseInTempDir._TEST_NAME = name
873
runner = TextTestRunner(stream=sys.stdout,
876
runner.stop_on_failure=stop_on_failure
878
suite = filter_suite_by_re(suite, pattern)
879
result = runner.run(suite)
880
# This is still a little bogus,
881
# but only a little. Folk not using our testrunner will
882
# have to delete their temp directories themselves.
883
test_root = TestCaseInTempDir.TEST_ROOT
884
if result.wasSuccessful() or not keep_output:
885
if test_root is not None:
886
print 'Deleting test root %s...' % test_root
888
shutil.rmtree(test_root)
892
print "Failed tests working directories are in '%s'\n" % TestCaseInTempDir.TEST_ROOT
893
return result.wasSuccessful()
896
def selftest(verbose=False, pattern=".*", stop_on_failure=True,
899
"""Run the whole test suite under the enhanced runner"""
900
global default_transport
901
if transport is None:
902
transport = default_transport
903
old_transport = default_transport
904
default_transport = transport
907
return run_suite(suite, 'testbzr', verbose=verbose, pattern=pattern,
908
stop_on_failure=stop_on_failure, keep_output=keep_output,
911
default_transport = old_transport
916
"""Build and return TestSuite for the whole program."""
28
retcode = kwargs.get('retcode', 0)
29
self.assertEquals(bzrlib.commands.run_bzr(args), retcode)
32
def selftest(verbose=False):
33
from unittest import TestLoader, TestSuite
34
import bzrlib, bzrlib.store, bzrlib.inventory, bzrlib.branch
35
import bzrlib.osutils, bzrlib.commands, bzrlib.merge3, bzrlib.plugin
917
36
from doctest import DocTestSuite
919
global MODULES_TO_DOCTEST
922
'bzrlib.tests.test_ancestry',
923
'bzrlib.tests.test_annotate',
924
'bzrlib.tests.test_api',
925
'bzrlib.tests.test_bad_files',
926
'bzrlib.tests.test_basis_inventory',
927
'bzrlib.tests.test_branch',
928
'bzrlib.tests.test_bzrdir',
929
'bzrlib.tests.test_command',
930
'bzrlib.tests.test_commit',
931
'bzrlib.tests.test_commit_merge',
932
'bzrlib.tests.test_config',
933
'bzrlib.tests.test_conflicts',
934
'bzrlib.tests.test_decorators',
935
'bzrlib.tests.test_diff',
936
'bzrlib.tests.test_doc_generate',
937
'bzrlib.tests.test_errors',
938
'bzrlib.tests.test_fetch',
939
'bzrlib.tests.test_gpg',
940
'bzrlib.tests.test_graph',
941
'bzrlib.tests.test_hashcache',
942
'bzrlib.tests.test_http',
943
'bzrlib.tests.test_identitymap',
944
'bzrlib.tests.test_inv',
945
'bzrlib.tests.test_knit',
946
'bzrlib.tests.test_lockdir',
947
'bzrlib.tests.test_lockable_files',
948
'bzrlib.tests.test_log',
949
'bzrlib.tests.test_merge',
950
'bzrlib.tests.test_merge3',
951
'bzrlib.tests.test_merge_core',
952
'bzrlib.tests.test_missing',
953
'bzrlib.tests.test_msgeditor',
954
'bzrlib.tests.test_nonascii',
955
'bzrlib.tests.test_options',
956
'bzrlib.tests.test_osutils',
957
'bzrlib.tests.test_permissions',
958
'bzrlib.tests.test_plugins',
959
'bzrlib.tests.test_progress',
960
'bzrlib.tests.test_reconcile',
961
'bzrlib.tests.test_repository',
962
'bzrlib.tests.test_revision',
963
'bzrlib.tests.test_revisionnamespaces',
964
'bzrlib.tests.test_revprops',
965
'bzrlib.tests.test_rio',
966
'bzrlib.tests.test_sampler',
967
'bzrlib.tests.test_selftest',
968
'bzrlib.tests.test_setup',
969
'bzrlib.tests.test_sftp_transport',
970
'bzrlib.tests.test_smart_add',
971
'bzrlib.tests.test_source',
972
'bzrlib.tests.test_store',
973
'bzrlib.tests.test_symbol_versioning',
974
'bzrlib.tests.test_testament',
975
'bzrlib.tests.test_trace',
976
'bzrlib.tests.test_transactions',
977
'bzrlib.tests.test_transform',
978
'bzrlib.tests.test_transport',
979
'bzrlib.tests.test_tsort',
980
'bzrlib.tests.test_ui',
981
'bzrlib.tests.test_uncommit',
982
'bzrlib.tests.test_upgrade',
983
'bzrlib.tests.test_versionedfile',
984
'bzrlib.tests.test_weave',
985
'bzrlib.tests.test_whitebox',
986
'bzrlib.tests.test_workingtree',
987
'bzrlib.tests.test_xml',
43
global MODULES_TO_TEST, MODULES_TO_DOCTEST
46
['bzrlib.selftest.whitebox',
47
'bzrlib.selftest.versioning',
48
'bzrlib.selftest.testinv',
49
'bzrlib.selftest.testmerge3',
50
'bzrlib.selftest.testhashcache',
51
'bzrlib.selftest.teststatus',
52
'bzrlib.selftest.testlog',
53
'bzrlib.selftest.blackbox',
54
'bzrlib.selftest.testrevisionnamespaces',
55
'bzrlib.selftest.testbranch',
56
'bzrlib.selftest.testrevision',
58
'bzrlib.selftest.testdiff',
989
test_transport_implementations = [
990
'bzrlib.tests.test_transport_implementations']
992
TestCase.BZRPATH = osutils.pathjoin(
993
osutils.realpath(osutils.dirname(bzrlib.__path__[0])), 'bzr')
994
print '%10s: %s' % ('bzr', osutils.realpath(sys.argv[0]))
995
print '%10s: %s' % ('bzrlib', bzrlib.__path__[0])
61
# XXX: should also test bzrlib.merge_core, but they seem to be out
62
# of date with the code.
64
for m in (bzrlib.store, bzrlib.inventory, bzrlib.branch,
65
bzrlib.osutils, bzrlib.commands, bzrlib.merge3):
66
if m not in MODULES_TO_DOCTEST:
67
MODULES_TO_DOCTEST.append(m)
70
TestCase.BZRPATH = os.path.join(os.path.realpath(os.path.dirname(bzrlib.__path__[0])), 'bzr')
71
print '%-30s %s' % ('bzr binary', TestCase.BZRPATH)
997
75
suite = TestSuite()
998
# python2.4's TestLoader.loadTestsFromNames gives very poor
999
# errors if it fails to load a named module - no indication of what's
1000
# actually wrong, just "no such module". We should probably override that
1001
# class, but for the moment just load them ourselves. (mbp 20051202)
1002
loader = TestLoader()
1003
from bzrlib.transport import TransportTestProviderAdapter
1004
adapter = TransportTestProviderAdapter()
1005
adapt_modules(test_transport_implementations, adapter, loader, suite)
1006
for mod_name in testmod_names:
1007
mod = _load_module_by_name(mod_name)
1008
suite.addTest(loader.loadTestsFromModule(mod))
1009
for package in packages_to_test():
1010
suite.addTest(package.test_suite())
77
suite.addTest(TestLoader().loadTestsFromNames(testmod_names))
1011
79
for m in MODULES_TO_TEST:
1012
suite.addTest(loader.loadTestsFromModule(m))
80
suite.addTest(TestLoader().loadTestsFromModule(m))
1013
82
for m in (MODULES_TO_DOCTEST):
1014
83
suite.addTest(DocTestSuite(m))
1015
for name, plugin in bzrlib.plugin.all_plugins().items():
1016
if getattr(plugin, 'test_suite', None) is not None:
1017
suite.addTest(plugin.test_suite())
1021
def adapt_modules(mods_list, adapter, loader, suite):
1022
"""Adapt the modules in mods_list using adapter and add to suite."""
1023
for mod_name in mods_list:
1024
mod = _load_module_by_name(mod_name)
1025
for test in iter_suite_tests(loader.loadTestsFromModule(mod)):
1026
suite.addTests(adapter.adapt(test))
1029
def _load_module_by_name(mod_name):
1030
parts = mod_name.split('.')
1031
module = __import__(mod_name)
1033
# for historical reasons python returns the top-level module even though
1034
# it loads the submodule; we need to walk down to get the one we want.
1036
module = getattr(module, parts.pop(0))
85
for p in bzrlib.plugin.all_plugins:
86
if hasattr(p, 'test_suite'):
87
suite.addTest(p.test_suite())
89
import bzrlib.merge_core
90
suite.addTest(unittest.makeSuite(bzrlib.merge_core.MergeTest, 'test_'))
92
return run_suite(suite, 'testbzr', verbose=verbose)
added
removed
bzrlib/selftest/__init__.py
