[Pkg-bazaar-commits] ./bzr/unstable r978: - new BzrTestBase class allowing running bzr in-process

[Martin Pool]

------------------------------------------------------------
revno: 978
committer: Martin Pool <mbp at sourcefrog.net>
timestamp: Mon 2005-07-25 16:25:12 -0300
message:
  - new BzrTestBase class allowing running bzr in-process
modified:
  bzrlib/selftest/__init__.py
-------------- next part --------------
=== modified file 'bzrlib/selftest/__init__.py'
--- a/bzrlib/selftest/__init__.py	2005-07-23 00:18:14 +0000
+++ b/bzrlib/selftest/__init__.py	2005-07-25 19:25:12 +0000
@@ -16,10 +16,19 @@
 
 
 from testsweet import TestBase, run_suite, InTempDir
+import bzrlib.commands
 
 MODULES_TO_TEST = []
 MODULES_TO_DOCTEST = []
 
+
+class BzrTestBase(InTempDir):
+    """bzr-specific test base class"""
+    def run_bzr(self, *args, **kwargs):
+        retcode = kwargs.get('retcode', 0)
+        self.assertEquals(bzrlib.commands.run_bzr(args), retcode)
+        
+
 def selftest(verbose=False):
     from unittest import TestLoader, TestSuite
     import bzrlib, bzrlib.store, bzrlib.inventory, bzrlib.branch