← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/__init__.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2006-02-18 02:33:47 UTC
  • mfrom: (1534.1.24 integration)
  • Revision ID: pqm@pqm.ubuntu.com-20060218023347-0952c65f668bfd68
Merge Robert Collins integration.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/__init__.py
21
21
# little as possible, so this should be used rarely if it's added at all.
22
22
# (Suggestion from j-a-meinel, 2005-11-24)
23
23
 
 
24
import codecs
24
25
from cStringIO import StringIO
25
26
import difflib
26
27
import errno
33
34
import tempfile
34
35
import unittest
35
36
import time
36
 
import codecs
 
37
 
37
38
 
38
39
import bzrlib.branch
 
40
import bzrlib.bzrdir as bzrdir
39
41
import bzrlib.commands
40
 
from bzrlib.errors import (BzrError,
41
 
                           FileExists,
42
 
                           UninitializableFormat,
43
 
                           )
 
42
import bzrlib.errors as errors
44
43
import bzrlib.inventory
45
44
import bzrlib.iterablefile
46
45
import bzrlib.merge3
56
55
from bzrlib.trace import mutter
57
56
from bzrlib.tests.TestUtil import TestLoader, TestSuite
58
57
from bzrlib.tests.treeshape import build_tree_contents
59
 
from bzrlib.workingtree import WorkingTree
 
58
from bzrlib.workingtree import WorkingTree, WorkingTreeFormat2
60
59
 
61
60
default_transport = LocalRelpathServer
62
61
 
81
80
    import bzrlib.doc
82
81
    import bzrlib.tests.blackbox
83
82
    import bzrlib.tests.branch_implementations
 
83
    import bzrlib.tests.bzrdir_implementations
 
84
    import bzrlib.tests.repository_implementations
 
85
    import bzrlib.tests.workingtree_implementations
84
86
    return [
85
87
            bzrlib.doc,
 
88
            bzrlib.tests.blackbox,
86
89
            bzrlib.tests.branch_implementations,
 
90
            bzrlib.tests.bzrdir_implementations,
 
91
            bzrlib.tests.repository_implementations,
 
92
            bzrlib.tests.workingtree_implementations,
87
93
            ]
88
94
 
89
95
 
259
265
                                  charjunk=lambda x: False)
260
266
        return ''.join(difflines)
261
267
 
262
 
    def assertEqualDiff(self, a, b):
 
268
    def assertEqualDiff(self, a, b, message=None):
263
269
        """Assert two texts are equal, if not raise an exception.
264
270
        
265
271
        This is intended for use with multi-line strings where it can 
268
274
        # TODO: perhaps override assertEquals to call this for strings?
269
275
        if a == b:
270
276
            return
271
 
        raise AssertionError("texts not equal:\n" + 
 
277
        if message is None:
 
278
            message = "texts not equal:\n"
 
279
        raise AssertionError(message + 
272
280
                             self._ndiff_strings(a, b))      
273
281
        
 
282
    def assertEqualMode(self, mode, mode_test):
 
283
        self.assertEqual(mode, mode_test,
 
284
                         'mode mismatch %o != %o' % (mode, mode_test))
 
285
 
274
286
    def assertStartsWith(self, s, prefix):
275
287
        if not s.startswith(prefix):
276
288
            raise AssertionError('string %r does not start with %r' % (s, prefix))
613
625
                elif line_endings == 'native':
614
626
                    end = os.linesep
615
627
                else:
616
 
                    raise BzrError('Invalid line ending request %r' % (line_endings,))
 
628
                    raise errors.BzrError('Invalid line ending request %r' % (line_endings,))
617
629
                content = "contents of %s%s" % (name, end)
618
630
                transport.put(urlescape(name), StringIO(content))
619
631
 
663
675
        relpath provides for clients to get a path relative to the base url.
664
676
        These should only be downwards relative, not upwards.
665
677
        """
 
678
        base = self.get_readonly_server().get_url()
 
679
        if relpath is not None:
 
680
            if not base.endswith('/'):
 
681
                base = base + '/'
 
682
            base = base + relpath
 
683
        return base
 
684
 
 
685
    def get_readonly_server(self):
 
686
        """Get the server instance for the readonly transport
 
687
 
 
688
        This is useful for some tests with specific servers to do diagnostics.
 
689
        """
666
690
        if self.__readonly_server is None:
667
691
            if self.transport_readonly_server is None:
668
692
                # readonly decorator requested
674
698
                self.__readonly_server = self.transport_readonly_server()
675
699
                self.__readonly_server.setUp()
676
700
            self.addCleanup(self.__readonly_server.tearDown)
677
 
        base = self.__readonly_server.get_url()
678
 
        if relpath is not None:
679
 
            if not base.endswith('/'):
680
 
                base = base + '/'
681
 
            base = base + relpath
682
 
        return base
 
701
        return self.__readonly_server
 
702
 
 
703
    def get_server(self):
 
704
        """Get the read/write server instance.
 
705
 
 
706
        This is useful for some tests with specific servers that need
 
707
        diagnostics.
 
708
        """
 
709
        if self.__server is None:
 
710
            self.__server = self.transport_server()
 
711
            self.__server.setUp()
 
712
            self.addCleanup(self.__server.tearDown)
 
713
        return self.__server
683
714
 
684
715
    def get_url(self, relpath=None):
685
716
        """Get a URL for the readwrite transport.
689
720
        relpath provides for clients to get a path relative to the base url.
690
721
        These should only be downwards relative, not upwards.
691
722
        """
692
 
        if self.__server is None:
693
 
            self.__server = self.transport_server()
694
 
            self.__server.setUp()
695
 
            self.addCleanup(self.__server.tearDown)
696
 
        base = self.__server.get_url()
 
723
        base = self.get_server().get_url()
697
724
        if relpath is not None and relpath != '.':
698
725
            if not base.endswith('/'):
699
726
                base = base + '/'
702
729
 
703
730
    def make_branch(self, relpath):
704
731
        """Create a branch on the transport at relpath."""
 
732
        repo = self.make_repository(relpath)
 
733
        return repo.bzrdir.create_branch()
 
734
 
 
735
    def make_bzrdir(self, relpath):
705
736
        try:
706
737
            url = self.get_url(relpath)
707
 
            segments = relpath.split('/')
 
738
            segments = url.split('/')
708
739
            if segments and segments[-1] not in ('', '.'):
709
 
                parent = self.get_url('/'.join(segments[:-1]))
 
740
                parent = '/'.join(segments[:-1])
710
741
                t = bzrlib.transport.get_transport(parent)
711
742
                try:
712
743
                    t.mkdir(segments[-1])
713
 
                except FileExists:
 
744
                except errors.FileExists:
714
745
                    pass
715
 
            return bzrlib.branch.Branch.create(url)
716
 
        except UninitializableFormat:
 
746
            return bzrlib.bzrdir.BzrDir.create(url)
 
747
        except errors.UninitializableFormat:
717
748
            raise TestSkipped("Format %s is not initializable.")
718
749
 
 
750
    def make_repository(self, relpath, shared=False):
 
751
        """Create a repository on our default transport at relpath."""
 
752
        made_control = self.make_bzrdir(relpath)
 
753
        return made_control.create_repository(shared=shared)
 
754
 
719
755
    def make_branch_and_tree(self, relpath):
720
756
        """Create a branch on the transport and a tree locally.
721
757
 
722
758
        Returns the tree.
723
759
        """
 
760
        # TODO: always use the local disk path for the working tree,
 
761
        # this obviously requires a format that supports branch references
 
762
        # so check for that by checking bzrdir.BzrDirFormat.get_default_format()
 
763
        # RBC 20060208
724
764
        b = self.make_branch(relpath)
725
 
        return WorkingTree.create(b, relpath)
 
765
        try:
 
766
            return b.bzrdir.create_workingtree()
 
767
        except errors.NotLocalUrl:
 
768
            # new formats - catch No tree error and create
 
769
            # a branch reference and a checkout.
 
770
            # old formats at that point - raise TestSkipped.
 
771
            # TODO: rbc 20060208
 
772
            return WorkingTreeFormat2().initialize(bzrdir.BzrDir.open(relpath))
726
773
 
727
774
 
728
775
class ChrootedTestCase(TestCaseWithTransport):
810
857
                   'bzrlib.tests.test_bad_files',
811
858
                   'bzrlib.tests.test_basis_inventory',
812
859
                   'bzrlib.tests.test_branch',
 
860
                   'bzrlib.tests.test_bzrdir',
813
861
                   'bzrlib.tests.test_command',
814
862
                   'bzrlib.tests.test_commit',
815
863
                   'bzrlib.tests.test_commit_merge',
816
864
                   'bzrlib.tests.test_config',
817
865
                   'bzrlib.tests.test_conflicts',
 
866
                   'bzrlib.tests.test_decorators',
818
867
                   'bzrlib.tests.test_diff',
819
 
                   'bzrlib.tests.test_decorators',
 
868
                   'bzrlib.tests.test_doc_generate',
 
869
                   'bzrlib.tests.test_errors',
820
870
                   'bzrlib.tests.test_fetch',
821
 
                   'bzrlib.tests.test_fileid_involved',
822
871
                   'bzrlib.tests.test_gpg',
823
872
                   'bzrlib.tests.test_graph',
824
873
                   'bzrlib.tests.test_hashcache',
835
884
                   'bzrlib.tests.test_nonascii',
836
885
                   'bzrlib.tests.test_options',
837
886
                   'bzrlib.tests.test_osutils',
838
 
                   'bzrlib.tests.test_parent',
839
887
                   'bzrlib.tests.test_permissions',
840
888
                   'bzrlib.tests.test_plugins',
 
889
                   'bzrlib.tests.test_repository',
841
890
                   'bzrlib.tests.test_revision',
842
891
                   'bzrlib.tests.test_revisionnamespaces',
843
892
                   'bzrlib.tests.test_revprops',