← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_info.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2008-09-03 22:30:56 UTC
  • mfrom: (3644.2.13 index_builder_cleanup)
  • Revision ID: pqm@pqm.ubuntu.com-20080903223056-b108iytb38xkznci
(jam) Streamline BTreeBuilder.add_node et al to make btree creation
        faster.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_info.py
1
 
# Copyright (C) 2007-2012 Canonical Ltd
 
1
# Copyright (C) 2007 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
import sys
 
18
from urllib import quote
18
19
 
19
20
from bzrlib import (
20
21
    branch as _mod_branch,
21
 
    controldir,
 
22
    bzrdir,
22
23
    info,
23
24
    tests,
24
25
    workingtree,
31
32
    def test_describe_standalone_layout(self):
32
33
        tree = self.make_branch_and_tree('tree')
33
34
        self.assertEqual('Empty control directory', info.describe_layout())
34
 
        self.assertEqual(
35
 
            'Unshared repository with trees and colocated branches',
36
 
            info.describe_layout(tree.branch.repository, control=tree.bzrdir))
 
35
        self.assertEqual('Unshared repository with trees',
 
36
            info.describe_layout(tree.branch.repository))
37
37
        tree.branch.repository.set_make_working_trees(False)
38
 
        self.assertEqual('Unshared repository with colocated branches',
39
 
            info.describe_layout(tree.branch.repository, control=tree.bzrdir))
 
38
        self.assertEqual('Unshared repository',
 
39
            info.describe_layout(tree.branch.repository))
40
40
        self.assertEqual('Standalone branch',
41
 
            info.describe_layout(tree.branch.repository, tree.branch,
42
 
                control=tree.bzrdir))
 
41
            info.describe_layout(tree.branch.repository, tree.branch))
43
42
        self.assertEqual('Standalone branchless tree',
44
 
            info.describe_layout(tree.branch.repository, None, tree,
45
 
                control=tree.bzrdir))
 
43
            info.describe_layout(tree.branch.repository, None, tree))
46
44
        self.assertEqual('Standalone tree',
47
 
            info.describe_layout(tree.branch.repository, tree.branch, tree,
48
 
                control=tree.bzrdir))
 
45
            info.describe_layout(tree.branch.repository, tree.branch, tree))
49
46
        tree.branch.bind(tree.branch)
50
47
        self.assertEqual('Bound branch',
51
 
            info.describe_layout(tree.branch.repository, tree.branch,
52
 
                control=tree.bzrdir))
 
48
            info.describe_layout(tree.branch.repository, tree.branch))
53
49
        self.assertEqual('Checkout',
54
 
            info.describe_layout(tree.branch.repository, tree.branch, tree,
55
 
                control=tree.bzrdir))
 
50
            info.describe_layout(tree.branch.repository, tree.branch, tree))
56
51
        checkout = tree.branch.create_checkout('checkout', lightweight=True)
57
52
        self.assertEqual('Lightweight checkout',
58
53
            info.describe_layout(checkout.branch.repository, checkout.branch,
59
 
                                 checkout, control=tree.bzrdir))
 
54
                                 checkout))
60
55
 
61
56
    def test_describe_repository_layout(self):
62
57
        repository = self.make_repository('.', shared=True)
63
 
        tree = controldir.ControlDir.create_branch_convenience('tree',
 
58
        tree = bzrdir.BzrDir.create_branch_convenience('tree',
64
59
            force_new_tree=True).bzrdir.open_workingtree()
65
 
        self.assertEqual('Shared repository with trees and colocated branches',
66
 
            info.describe_layout(tree.branch.repository, control=tree.bzrdir))
 
60
        self.assertEqual('Shared repository with trees',
 
61
            info.describe_layout(tree.branch.repository))
67
62
        repository.set_make_working_trees(False)
68
 
        self.assertEqual('Shared repository with colocated branches',
69
 
            info.describe_layout(tree.branch.repository, control=tree.bzrdir))
 
63
        self.assertEqual('Shared repository',
 
64
            info.describe_layout(tree.branch.repository))
70
65
        self.assertEqual('Repository branch',
71
 
            info.describe_layout(tree.branch.repository, tree.branch,
72
 
                control=tree.bzrdir))
 
66
            info.describe_layout(tree.branch.repository, tree.branch))
73
67
        self.assertEqual('Repository branchless tree',
74
 
            info.describe_layout(tree.branch.repository, None, tree,
75
 
                control=tree.bzrdir))
 
68
            info.describe_layout(tree.branch.repository, None, tree))
76
69
        self.assertEqual('Repository tree',
77
 
            info.describe_layout(tree.branch.repository, tree.branch, tree,
78
 
                control=tree.bzrdir))
 
70
            info.describe_layout(tree.branch.repository, tree.branch, tree))
79
71
        tree.branch.bind(tree.branch)
80
72
        self.assertEqual('Repository checkout',
81
 
            info.describe_layout(tree.branch.repository, tree.branch, tree,
82
 
                control=tree.bzrdir))
 
73
            info.describe_layout(tree.branch.repository, tree.branch, tree))
83
74
        checkout = tree.branch.create_checkout('checkout', lightweight=True)
84
75
        self.assertEqual('Lightweight checkout',
85
76
            info.describe_layout(checkout.branch.repository, checkout.branch,
86
 
                                 checkout, control=tree.bzrdir))
 
77
                                 checkout))
87
78
 
88
79
    def assertTreeDescription(self, format):
89
80
        """Assert a tree's format description matches expectations"""
100
91
        # this ought to be easier...
101
92
        branch.create_checkout('%s_co' % format,
102
93
            lightweight=True).bzrdir.destroy_workingtree()
103
 
        control = controldir.ControlDir.open('%s_co' % format)
 
94
        control = bzrdir.BzrDir.open('%s_co' % format)
104
95
        old_format = control._format.workingtree_format
105
96
        try:
106
97
            control._format.workingtree_format = \
107
 
                controldir.format_registry.make_bzrdir(format).workingtree_format
 
98
                bzrdir.format_registry.make_bzrdir(format).workingtree_format
108
99
            control.create_workingtree()
109
100
            tree = workingtree.WorkingTree.open('%s_co' % format)
110
101
            format_description = info.describe_format(tree.bzrdir,
134
125
            repo, None, None))
135
126
 
136
127
    def test_describe_tree_format(self):
137
 
        for key in controldir.format_registry.keys():
138
 
            if key in controldir.format_registry.aliases():
 
128
        for key in bzrdir.format_registry.keys():
 
129
            if key in bzrdir.format_registry.aliases():
139
130
                continue
140
131
            self.assertTreeDescription(key)
141
132
 
142
133
    def test_describe_checkout_format(self):
143
 
        for key in controldir.format_registry.keys():
144
 
            if key in controldir.format_registry.aliases():
 
134
        for key in bzrdir.format_registry.keys():
 
135
            if key in bzrdir.format_registry.aliases():
145
136
                # Aliases will not describe correctly in the UI because the
146
137
                # real format is found.
147
138
                continue
148
139
            # legacy: weave does not support checkouts
149
140
            if key == 'weave':
150
141
                continue
151
 
            if controldir.format_registry.get_info(key).experimental:
 
142
            if bzrdir.format_registry.get_info(key).experimental:
152
143
                # We don't require that experimental formats support checkouts
153
144
                # or describe correctly in the UI.
154
145
                continue
155
 
            if controldir.format_registry.get_info(key).hidden:
156
 
                continue
157
146
            expected = None
158
 
            if key in ('pack-0.92',):
159
 
                expected = 'pack-0.92'
160
 
            elif key in ('knit', 'metaweave'):
161
 
                if 'metaweave' in controldir.format_registry:
162
 
                    expected = 'knit or metaweave'
163
 
                else:
164
 
                    expected = 'knit'
165
 
            elif key in ('1.14', '1.14-rich-root'):
166
 
                expected = '1.14 or 1.14-rich-root'
 
147
            if key in ('dirstate', 'dirstate-tags', 'dirstate-with-subtree',
 
148
                'pack-0.92', 'pack-0.92-subtree', 'rich-root',
 
149
                'rich-root-pack', '1.6', '1.6.1-rich-root'):
 
150
                expected = '1.6 or 1.6.1-rich-root or ' \
 
151
                    'dirstate or dirstate-tags or pack-0.92 or'\
 
152
                    ' rich-root or rich-root-pack'
 
153
            if key in ('knit', 'metaweave'):
 
154
                expected = 'knit or metaweave'
167
155
            self.assertCheckoutDescription(key, expected)
168
156
 
169
157
    def test_describe_branch_format(self):
170
 
        for key in controldir.format_registry.keys():
171
 
            if key in controldir.format_registry.aliases():
172
 
                continue
173
 
            if controldir.format_registry.get_info(key).hidden:
 
158
        for key in bzrdir.format_registry.keys():
 
159
            if key in bzrdir.format_registry.aliases():
174
160
                continue
175
161
            expected = None
176
162
            if key in ('dirstate', 'knit'):
177
163
                expected = 'dirstate or knit'
178
 
            elif key in ('1.14',):
179
 
                expected = '1.14'
180
 
            elif key in ('1.14-rich-root',):
181
 
                expected = '1.14-rich-root'
182
164
            self.assertBranchDescription(key, expected)
183
165
 
184
166
    def test_describe_repo_format(self):
185
 
        for key in controldir.format_registry.keys():
186
 
            if key in controldir.format_registry.aliases():
187
 
                continue
188
 
            if controldir.format_registry.get_info(key).hidden:
 
167
        for key in bzrdir.format_registry.keys():
 
168
            if key in bzrdir.format_registry.aliases():
189
169
                continue
190
170
            expected = None
191
171
            if key in ('dirstate', 'knit', 'dirstate-tags'):
192
172
                expected = 'dirstate or dirstate-tags or knit'
193
 
            elif key in ('1.14',):
194
 
                expected = '1.14'
195
 
            elif key in ('1.14-rich-root',):
196
 
                expected = '1.14-rich-root'
197
173
            self.assertRepoDescription(key, expected)
198
174
 
199
 
        format = controldir.format_registry.make_bzrdir('knit')
 
175
        format = bzrdir.format_registry.make_bzrdir('metaweave')
200
176
        format.set_branch_format(_mod_branch.BzrBranchFormat6())
201
177
        tree = self.make_branch_and_tree('unknown', format=format)
202
178
        self.assertEqual('unnamed', info.describe_format(tree.bzrdir,
203
179
            tree.branch.repository, tree.branch, tree))
204
180
 
205
 
    def test_gather_location_controldir_only(self):
206
 
        bzrdir = self.make_bzrdir('.')
207
 
        self.assertEqual([('control directory', bzrdir.user_url)],
208
 
            info.gather_location_info(control=bzrdir))
209
 
 
210
181
    def test_gather_location_standalone(self):
211
182
        tree = self.make_branch_and_tree('tree')
212
183
        self.assertEqual([('branch root', tree.bzrdir.root_transport.base)],
213
 
            info.gather_location_info(
214
 
                tree.branch.repository, tree.branch, tree, control=tree.bzrdir))
 
184
            info.gather_location_info(tree.branch.repository, tree.branch,
 
185
                                      tree))
215
186
        self.assertEqual([('branch root', tree.bzrdir.root_transport.base)],
216
 
            info.gather_location_info(
217
 
                tree.branch.repository, tree.branch, control=tree.bzrdir))
 
187
            info.gather_location_info(tree.branch.repository, tree.branch))
218
188
        return tree
219
189
 
220
190
    def test_gather_location_repo(self):
221
191
        srepo = self.make_repository('shared', shared=True)
222
 
        self.assertEqual(
223
 
            [('shared repository', srepo.bzrdir.root_transport.base)],
224
 
            info.gather_location_info(srepo, control=srepo.bzrdir))
 
192
        self.assertEqual([('shared repository',
 
193
                          srepo.bzrdir.root_transport.base)],
 
194
                          info.gather_location_info(srepo))
225
195
        urepo = self.make_repository('unshared')
226
 
        self.assertEqual(
227
 
            [('repository', urepo.bzrdir.root_transport.base)],
228
 
            info.gather_location_info(urepo, control=urepo.bzrdir))
 
196
        self.assertEqual([('repository',
 
197
                          urepo.bzrdir.root_transport.base)],
 
198
                          info.gather_location_info(urepo))
229
199
 
230
200
    def test_gather_location_repo_branch(self):
231
201
        srepo = self.make_repository('shared', shared=True)
232
 
        self.assertEqual(
233
 
            [('shared repository', srepo.bzrdir.root_transport.base)],
234
 
            info.gather_location_info(srepo, control=srepo.bzrdir))
 
202
        self.assertEqual([('shared repository',
 
203
                          srepo.bzrdir.root_transport.base)],
 
204
                          info.gather_location_info(srepo))
235
205
        tree = self.make_branch_and_tree('shared/tree')
236
 
        self.assertEqual(
237
 
            [('shared repository', srepo.bzrdir.root_transport.base),
238
 
             ('repository branch', tree.branch.base)],
239
 
            info.gather_location_info(srepo, tree.branch, tree, srepo.bzrdir))
 
206
        self.assertEqual([('shared repository',
 
207
                          srepo.bzrdir.root_transport.base),
 
208
                          ('repository branch', tree.branch.base)],
 
209
                          info.gather_location_info(srepo, tree.branch, tree))
240
210
 
241
211
    def test_gather_location_light_checkout(self):
242
212
        tree = self.make_branch_and_tree('tree')
275
245
             self.gather_tree_location_info(shared_checkout))
276
246
 
277
247
    def gather_tree_location_info(self, tree):
278
 
        return info.gather_location_info(
279
 
            tree.branch.repository, tree.branch, tree, tree.bzrdir)
 
248
        return info.gather_location_info(tree.branch.repository, tree.branch,
 
249
                                         tree)
280
250
 
281
251
    def test_gather_location_bound(self):
282
252
        branch = self.make_branch('branch')
285
255
        self.assertEqual(
286
256
            [('branch root', bound_branch.bzrdir.root_transport.base),
287
257
             ('bound to branch', branch.bzrdir.root_transport.base)],
288
 
            info.gather_location_info(
289
 
                bound_branch.repository, bound_branch, control=bound_branch.bzrdir)
290
 
        )
291
 
 
292
 
    def test_gather_location_bound_in_repository(self):
293
 
        repo = self.make_repository('repo', shared=True)
294
 
        repo.set_make_working_trees(False)
295
 
        branch = self.make_branch('branch')
296
 
        bound_branch = controldir.ControlDir.create_branch_convenience(
297
 
            'repo/bound_branch')
298
 
        bound_branch.bind(branch)
299
 
        self.assertEqual(
300
 
            [('shared repository', bound_branch.repository.bzrdir.user_url),
301
 
             ('repository branch', bound_branch.bzrdir.user_url),
302
 
             ('bound to branch', branch.bzrdir.user_url)],
303
258
            info.gather_location_info(bound_branch.repository, bound_branch)
304
259
        )
305
260
 
322
277
 
323
278
    def test_gather_related_braches(self):
324
279
        branch = self.make_branch('.')
325
 
        branch.lock_write()
326
 
        try:
327
 
            branch.set_public_branch('baz')
328
 
            branch.set_push_location('bar')
329
 
            branch.set_parent('foo')
330
 
            branch.set_submit_branch('qux')
331
 
        finally:
332
 
            branch.unlock()
 
280
        branch.set_public_branch('baz')
 
281
        branch.set_push_location('bar')
 
282
        branch.set_parent('foo')
 
283
        branch.set_submit_branch('qux')
333
284
        self.assertEqual(
334
285
            [('public branch', 'baz'), ('push branch', 'bar'),
335
286
             ('parent branch', 'foo'), ('submit branch', 'qux')],