← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_repository.py

  • Committer: Martin
  • Date: 2011-03-22 00:33:21 UTC
  • mto: This revision was merged to the branch mainline in revision 5731.
  • Revision ID: gzlist@googlemail.com-20110322003321-h3n4q806g759tk7v
Tweak function descriptions in delta header for clarity

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_repository.py
26
26
import sys
27
27
 
28
28
import bzrlib
29
 
from bzrlib.errors import (NoSuchFile,
30
 
                           UnknownFormatError,
31
 
                           UnsupportedFormatError,
32
 
                           )
 
29
from bzrlib.errors import (
 
30
    NoSuchFile,
 
31
    UnknownFormatError,
 
32
    UnsupportedFormatError,
 
33
    )
33
34
from bzrlib import (
 
35
    btree_index,
34
36
    graph,
 
37
    symbol_versioning,
35
38
    tests,
 
39
    transport,
36
40
    )
37
41
from bzrlib.btree_index import BTreeBuilder, BTreeGraphIndex
38
42
from bzrlib.index import GraphIndex
41
45
    TestCase,
42
46
    TestCaseWithTransport,
43
47
    )
44
 
from bzrlib.transport import (
45
 
    get_transport,
46
 
    )
47
48
from bzrlib import (
48
49
    bzrdir,
49
50
    errors,
68
69
    def test_get_set_default_format(self):
69
70
        old_default = bzrdir.format_registry.get('default')
70
71
        private_default = old_default().repository_format.__class__
71
 
        old_format = repository.RepositoryFormat.get_default_format()
 
72
        old_format = repository.format_registry.get_default()
72
73
        self.assertTrue(isinstance(old_format, private_default))
73
74
        def make_sample_bzrdir():
74
75
            my_bzrdir = bzrdir.BzrDirMetaFormat1()
88
89
            bzrdir.format_registry.remove('default')
89
90
            bzrdir.format_registry.remove('sample')
90
91
            bzrdir.format_registry.register('default', old_default, '')
91
 
        self.assertIsInstance(repository.RepositoryFormat.get_default_format(),
 
92
        self.assertIsInstance(repository.format_registry.get_default(),
92
93
                              old_format.__class__)
93
94
 
94
95
 
116
117
        return "opened repository."
117
118
 
118
119
 
 
120
class SampleExtraRepositoryFormat(repository.RepositoryFormat):
 
121
    """A sample format that can not be used in a metadir
 
122
 
 
123
    """
 
124
 
 
125
    def get_format_string(self):
 
126
        raise NotImplementedError
 
127
 
 
128
 
119
129
class TestRepositoryFormat(TestCaseWithTransport):
120
130
    """Tests for the Repository format detection used by the bzr meta dir facility.BzrBranchFormat facility."""
121
131
 
127
137
        def check_format(format, url):
128
138
            dir = format._matchingbzrdir.initialize(url)
129
139
            format.initialize(dir)
130
 
            t = get_transport(url)
 
140
            t = transport.get_transport(url)
131
141
            found_format = repository.RepositoryFormat.find_format(dir)
132
142
            self.failUnless(isinstance(found_format, format.__class__))
133
 
        check_format(weaverepo.RepositoryFormat7(), "bar")
 
143
        check_format(repository.format_registry.get_default(), "bar")
134
144
 
135
145
    def test_find_format_no_repository(self):
136
146
        dir = bzrdir.BzrDirMetaFormat1().initialize(self.get_url())
146
156
                          dir)
147
157
 
148
158
    def test_register_unregister_format(self):
 
159
        # Test deprecated format registration functions
149
160
        format = SampleRepositoryFormat()
150
161
        # make a control dir
151
162
        dir = bzrdir.BzrDirMetaFormat1().initialize(self.get_url())
152
163
        # make a repo
153
164
        format.initialize(dir)
154
165
        # register a format for it.
155
 
        repository.RepositoryFormat.register_format(format)
 
166
        self.applyDeprecated(symbol_versioning.deprecated_in((2, 4, 0)),
 
167
            repository.RepositoryFormat.register_format, format)
156
168
        # which repository.Open will refuse (not supported)
157
 
        self.assertRaises(UnsupportedFormatError, repository.Repository.open, self.get_url())
 
169
        self.assertRaises(UnsupportedFormatError, repository.Repository.open,
 
170
            self.get_url())
158
171
        # but open(unsupported) will work
159
172
        self.assertEqual(format.open(dir), "opened repository.")
160
173
        # unregister the format
161
 
        repository.RepositoryFormat.unregister_format(format)
 
174
        self.applyDeprecated(symbol_versioning.deprecated_in((2, 4, 0)),
 
175
            repository.RepositoryFormat.unregister_format, format)
 
176
 
 
177
 
 
178
class TestRepositoryFormatRegistry(TestCase):
 
179
 
 
180
    def setUp(self):
 
181
        super(TestRepositoryFormatRegistry, self).setUp()
 
182
        self.registry = repository.RepositoryFormatRegistry()
 
183
 
 
184
    def test_register_unregister_format(self):
 
185
        format = SampleRepositoryFormat()
 
186
        self.registry.register(format)
 
187
        self.assertEquals(format, self.registry.get("Sample .bzr repository format."))
 
188
        self.registry.remove(format)
 
189
        self.assertRaises(KeyError, self.registry.get, "Sample .bzr repository format.")
 
190
 
 
191
    def test_get_all(self):
 
192
        format = SampleRepositoryFormat()
 
193
        self.assertEquals([], self.registry._get_all())
 
194
        self.registry.register(format)
 
195
        self.assertEquals([format], self.registry._get_all())
 
196
 
 
197
    def test_register_extra(self):
 
198
        format = SampleExtraRepositoryFormat()
 
199
        self.assertEquals([], self.registry._get_all())
 
200
        self.registry.register_extra(format)
 
201
        self.assertEquals([format], self.registry._get_all())
 
202
 
 
203
    def test_register_extra_lazy(self):
 
204
        self.assertEquals([], self.registry._get_all())
 
205
        self.registry.register_extra_lazy("bzrlib.tests.test_repository",
 
206
            "SampleExtraRepositoryFormat")
 
207
        formats = self.registry._get_all()
 
208
        self.assertEquals(1, len(formats))
 
209
        self.assertIsInstance(formats[0], SampleExtraRepositoryFormat)
162
210
 
163
211
 
164
212
class TestFormat6(TestCaseWithTransport):
585
633
                                ]
586
634
        repo_a = self.make_repository('a')
587
635
        repo_b = self.make_repository('b')
588
 
        is_compatible = repository.InterWeaveRepo.is_compatible
 
636
        is_compatible = weaverepo.InterWeaveRepo.is_compatible
589
637
        for source in incompatible_formats:
590
638
            # force incompatible left then right
591
639
            repo_a._format = source
597
645
            for target in formats:
598
646
                repo_b._format = target
599
647
                self.assertTrue(is_compatible(repo_a, repo_b))
600
 
        self.assertEqual(repository.InterWeaveRepo,
 
648
        self.assertEqual(weaverepo.InterWeaveRepo,
601
649
                         repository.InterRepository.get(repo_a,
602
650
                                                        repo_b).__class__)
603
651
 
604
652
 
 
653
class TestRepositoryFormat1(knitrepo.RepositoryFormatKnit1):
 
654
 
 
655
    def get_format_string(self):
 
656
        return "Test Format 1"
 
657
 
 
658
 
 
659
class TestRepositoryFormat2(knitrepo.RepositoryFormatKnit1):
 
660
 
 
661
    def get_format_string(self):
 
662
        return "Test Format 2"
 
663
 
 
664
 
605
665
class TestRepositoryConverter(TestCaseWithTransport):
606
666
 
607
667
    def test_convert_empty(self):
608
 
        t = get_transport(self.get_url('.'))
 
668
        source_format = TestRepositoryFormat1()
 
669
        target_format = TestRepositoryFormat2()
 
670
        repository.format_registry.register(source_format)
 
671
        self.addCleanup(repository.format_registry.remove,
 
672
            source_format)
 
673
        repository.format_registry.register(target_format)
 
674
        self.addCleanup(repository.format_registry.remove,
 
675
            target_format)
 
676
        t = self.get_transport()
609
677
        t.mkdir('repository')
610
678
        repo_dir = bzrdir.BzrDirMetaFormat1().initialize('repository')
611
 
        repo = weaverepo.RepositoryFormat7().initialize(repo_dir)
612
 
        target_format = knitrepo.RepositoryFormatKnit1()
 
679
        repo = TestRepositoryFormat1().initialize(repo_dir)
613
680
        converter = repository.CopyConverter(target_format)
614
681
        pb = bzrlib.ui.ui_factory.nested_progress_bar()
615
682
        try:
620
687
        self.assertTrue(isinstance(target_format, repo._format.__class__))
621
688
 
622
689
 
623
 
class TestMisc(TestCase):
624
 
 
625
 
    def test_unescape_xml(self):
626
 
        """We get some kind of error when malformed entities are passed"""
627
 
        self.assertRaises(KeyError, repository._unescape_xml, 'foo&bar;')
628
 
 
629
 
 
630
690
class TestRepositoryFormatKnit3(TestCaseWithTransport):
631
691
 
632
692
    def test_attribute__fetch_order(self):
681
741
 
682
742
class Test2a(tests.TestCaseWithMemoryTransport):
683
743
 
 
744
    def test_chk_bytes_uses_custom_btree_parser(self):
 
745
        mt = self.make_branch_and_memory_tree('test', format='2a')
 
746
        mt.lock_write()
 
747
        self.addCleanup(mt.unlock)
 
748
        mt.add([''], ['root-id'])
 
749
        mt.commit('first')
 
750
        index = mt.branch.repository.chk_bytes._index._graph_index._indices[0]
 
751
        self.assertEqual(btree_index._gcchk_factory, index._leaf_factory)
 
752
        # It should also work if we re-open the repo
 
753
        repo = mt.branch.repository.bzrdir.open_repository()
 
754
        repo.lock_read()
 
755
        self.addCleanup(repo.unlock)
 
756
        index = repo.chk_bytes._index._graph_index._indices[0]
 
757
        self.assertEqual(btree_index._gcchk_factory, index._leaf_factory)
 
758
 
684
759
    def test_fetch_combines_groups(self):
685
760
        builder = self.make_branch_builder('source', format='2a')
686
761
        builder.start_series()
956
1031
 
957
1032
    def setUp(self):
958
1033
        super(TestDevelopment6FindParentIdsOfRevisions, self).setUp()
959
 
        self.builder = self.make_branch_builder('source',
960
 
            format='development6-rich-root')
 
1034
        self.builder = self.make_branch_builder('source')
961
1035
        self.builder.start_series()
962
1036
        self.builder.build_snapshot('initial', None,
963
1037
            [('add', ('', 'tree-root', 'directory', None))])
1644
1718
        self.orig_pack = target.pack
1645
1719
        target.pack = self.log_pack
1646
1720
        search = target.search_missing_revision_ids(
1647
 
            source_tree.branch.repository, tip)
 
1721
            source_tree.branch.repository, revision_ids=[tip])
1648
1722
        stream = source.get_stream(search)
1649
1723
        from_format = source_tree.branch.repository._format
1650
1724
        sink = target._get_sink()