← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/__init__.py

Add TestCaseWithTransport class that provides tests with read and write transport pairs.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/__init__.py
47
47
import bzrlib.trace
48
48
from bzrlib.transport import urlescape
49
49
import bzrlib.transport
 
50
from bzrlib.transport.readonly import ReadonlyServer
50
51
from bzrlib.trace import mutter
51
52
from bzrlib.tests.TestUtil import TestLoader, TestSuite
52
53
from bzrlib.tests.treeshape import build_tree_contents
237
238
    _log_file_name = None
238
239
    _log_contents = ''
239
240
 
 
241
    def __init__(self, methodName='testMethod'):
 
242
        super(TestCase, self).__init__(methodName)
 
243
        self._cleanups = []
 
244
 
240
245
    def setUp(self):
241
246
        unittest.TestCase.setUp(self)
242
 
        self._cleanups = []
243
247
        self._cleanEnvironment()
244
248
        bzrlib.trace.disable_default_logging()
245
249
        self._startLogFile()
628
632
        self.assertEqualDiff(content, open(path, 'r').read())
629
633
 
630
634
 
 
635
class TestCaseWithTransport(TestCaseInTempDir):
 
636
    """A test case that provides get_url and get_readonly_url facilities.
 
637
 
 
638
    These back onto two transport servers, one for readonly access and one for
 
639
    read write access.
 
640
 
 
641
    If no explicit class is provided for readonly access, a
 
642
    ReadonlyTransportDecorator is used instead which allows the use of non disk
 
643
    based read write transports.
 
644
 
 
645
    If an explicit class is provided for readonly access, that server and the 
 
646
    readwrite one must both define get_url() as resolving to os.getcwd().
 
647
    """
 
648
 
 
649
    def __init__(self, methodName='testMethod'):
 
650
        super(TestCaseWithTransport, self).__init__(methodName)
 
651
        self.__readonly_server = None
 
652
        self.__server = None
 
653
 
 
654
    def get_readonly_url(self):
 
655
        """Get a URL for the readonly transport.
 
656
 
 
657
        This will either be backed by '.' or a decorator to the transport 
 
658
        used by self.get_url()
 
659
        """
 
660
        if self.__readonly_server is None:
 
661
            if self.transport_readonly_server is None:
 
662
                # readonly decorator requested
 
663
                # bring up the server
 
664
                self.get_url()
 
665
                self.__readonly_server = ReadonlyServer()
 
666
                self.__readonly_server.setUp(self.__server)
 
667
            else:
 
668
                self.__readonly_server = self.transport_readonly_server()
 
669
                self.__readonly_server.setUp()
 
670
            self.addCleanup(self.__readonly_server.tearDown)
 
671
        return self.__readonly_server.get_url()
 
672
 
 
673
    def get_url(self):
 
674
        """Get a URL for the readwrite transport.
 
675
 
 
676
        This will either be backed by '.' or to an equivalent non-file based
 
677
        facility.
 
678
        """
 
679
        if self.__server is None:
 
680
            self.__server = self.transport_server()
 
681
            self.__server.setUp()
 
682
            self.addCleanup(self.__server.tearDown)
 
683
        return self.__server.get_url()
 
684
    
 
685
 
 
686
 
631
687
def filter_suite_by_re(suite, pattern):
632
688
    result = TestSuite()
633
689
    filter_re = re.compile(pattern)
760
816
    from bzrlib.branch import BranchTestProviderAdapter
761
817
    adapter = BranchTestProviderAdapter(
762
818
        bzrlib.transport.local.LocalRelpathServer,
763
 
        bzrlib.transport.local.LocalRelpathServer,
 
819
        # None here will cause a readonly decorator to be created
 
820
        # by the TestCaseWithTransport.get_readonly_transport method.
 
821
        None,
764
822
        bzrlib.branch.BzrBranchFormat._formats.values())
765
823
    for mod_name in test_branch_implementations:
766
824
        mod = _load_module_by_name(mod_name)