← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_branch.py

  • Committer: Ian Clatworthy
  • Date: 2007-11-27 21:17:06 UTC
  • mto: (3054.1.1 ianc-integration)
  • mto: This revision was merged to the branch mainline in revision 3055.
  • Revision ID: ian.clatworthy@internode.on.net-20071127211706-871zcqst0yi5tcvl
make fixes suggested by proof-readers

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_branch.py
1
 
# Copyright (C) 2005, 2006 Canonical Ltd
 
1
# Copyright (C) 2005, 2006, 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
24
24
 
25
25
from StringIO import StringIO
26
26
 
27
 
import bzrlib.branch
28
 
from bzrlib.branch import (BzrBranch5, 
29
 
                           BzrBranchFormat5)
30
 
import bzrlib.bzrdir as bzrdir
 
27
from bzrlib import (
 
28
    branch as _mod_branch,
 
29
    bzrdir,
 
30
    config,
 
31
    errors,
 
32
    trace,
 
33
    urlutils,
 
34
    )
 
35
from bzrlib.branch import (
 
36
    Branch,
 
37
    BranchHooks,
 
38
    BranchFormat,
 
39
    BranchReferenceFormat,
 
40
    BzrBranch5,
 
41
    BzrBranchFormat5,
 
42
    BzrBranchFormat6,
 
43
    PullResult,
 
44
    )
31
45
from bzrlib.bzrdir import (BzrDirMetaFormat1, BzrDirMeta1, 
32
46
                           BzrDir, BzrDirFormat)
33
47
from bzrlib.errors import (NotBranchError,
34
48
                           UnknownFormatError,
 
49
                           UnknownHook,
35
50
                           UnsupportedFormatError,
36
51
                           )
37
52
 
40
55
 
41
56
class TestDefaultFormat(TestCase):
42
57
 
 
58
    def test_default_format(self):
 
59
        # update this if you change the default branch format
 
60
        self.assertIsInstance(BranchFormat.get_default_format(),
 
61
                BzrBranchFormat6)
 
62
 
 
63
    def test_default_format_is_same_as_bzrdir_default(self):
 
64
        # XXX: it might be nice if there was only one place the default was
 
65
        # set, but at the moment that's not true -- mbp 20070814 -- 
 
66
        # https://bugs.launchpad.net/bzr/+bug/132376
 
67
        self.assertEqual(BranchFormat.get_default_format(),
 
68
                BzrDirFormat.get_default_format().get_branch_format())
 
69
 
43
70
    def test_get_set_default_format(self):
44
 
        old_format = bzrlib.branch.BranchFormat.get_default_format()
45
 
        # default is 5
46
 
        self.assertTrue(isinstance(old_format, bzrlib.branch.BzrBranchFormat5))
47
 
        bzrlib.branch.BranchFormat.set_default_format(SampleBranchFormat())
 
71
        # set the format and then set it back again
 
72
        old_format = BranchFormat.get_default_format()
 
73
        BranchFormat.set_default_format(SampleBranchFormat())
48
74
        try:
49
75
            # the default branch format is used by the meta dir format
50
76
            # which is not the default bzrdir format at this point
52
78
            result = dir.create_branch()
53
79
            self.assertEqual(result, 'A branch')
54
80
        finally:
55
 
            bzrlib.branch.BranchFormat.set_default_format(old_format)
56
 
        self.assertEqual(old_format, bzrlib.branch.BranchFormat.get_default_format())
 
81
            BranchFormat.set_default_format(old_format)
 
82
        self.assertEqual(old_format, BranchFormat.get_default_format())
57
83
 
58
84
 
59
85
class TestBranchFormat5(TestCaseWithTransport):
76
102
        finally:
77
103
            branch.unlock()
78
104
 
79
 
 
80
 
class SampleBranchFormat(bzrlib.branch.BranchFormat):
 
105
    def test_set_push_location(self):
 
106
        from bzrlib.config import (locations_config_filename,
 
107
                                   ensure_config_dir_exists)
 
108
        ensure_config_dir_exists()
 
109
        fn = locations_config_filename()
 
110
        # write correct newlines to locations.conf
 
111
        # by default ConfigObj uses native line-endings for new files
 
112
        # but uses already existing line-endings if file is not empty
 
113
        f = open(fn, 'wb')
 
114
        try:
 
115
            f.write('# comment\n')
 
116
        finally:
 
117
            f.close()
 
118
 
 
119
        branch = self.make_branch('.', format='knit')
 
120
        branch.set_push_location('foo')
 
121
        local_path = urlutils.local_path_from_url(branch.base[:-1])
 
122
        self.assertFileEqual("# comment\n"
 
123
                             "[%s]\n"
 
124
                             "push_location = foo\n"
 
125
                             "push_location:policy = norecurse" % local_path,
 
126
                             fn)
 
127
 
 
128
    # TODO RBC 20051029 test getting a push location from a branch in a
 
129
    # recursive section - that is, it appends the branch name.
 
130
 
 
131
 
 
132
class SampleBranchFormat(BranchFormat):
81
133
    """A sample format
82
134
 
83
135
    this format is initializable, unsupported to aid in testing the 
113
165
            dir = format._matchingbzrdir.initialize(url)
114
166
            dir.create_repository()
115
167
            format.initialize(dir)
116
 
            found_format = bzrlib.branch.BranchFormat.find_format(dir)
 
168
            found_format = BranchFormat.find_format(dir)
117
169
            self.failUnless(isinstance(found_format, format.__class__))
118
 
        check_format(bzrlib.branch.BzrBranchFormat5(), "bar")
 
170
        check_format(BzrBranchFormat5(), "bar")
119
171
        
120
172
    def test_find_format_not_branch(self):
121
173
        dir = bzrdir.BzrDirMetaFormat1().initialize(self.get_url())
122
174
        self.assertRaises(NotBranchError,
123
 
                          bzrlib.branch.BranchFormat.find_format,
 
175
                          BranchFormat.find_format,
124
176
                          dir)
125
177
 
126
178
    def test_find_format_unknown_format(self):
127
179
        dir = bzrdir.BzrDirMetaFormat1().initialize(self.get_url())
128
180
        SampleBranchFormat().initialize(dir)
129
181
        self.assertRaises(UnknownFormatError,
130
 
                          bzrlib.branch.BranchFormat.find_format,
 
182
                          BranchFormat.find_format,
131
183
                          dir)
132
184
 
133
185
    def test_register_unregister_format(self):
137
189
        # make a branch
138
190
        format.initialize(dir)
139
191
        # register a format for it.
140
 
        bzrlib.branch.BranchFormat.register_format(format)
 
192
        BranchFormat.register_format(format)
141
193
        # which branch.Open will refuse (not supported)
142
 
        self.assertRaises(UnsupportedFormatError, bzrlib.branch.Branch.open, self.get_url())
 
194
        self.assertRaises(UnsupportedFormatError, Branch.open, self.get_url())
 
195
        self.make_branch_and_tree('foo')
143
196
        # but open_downlevel will work
144
197
        self.assertEqual(format.open(dir), bzrdir.BzrDir.open(self.get_url()).open_branch(unsupported=True))
145
198
        # unregister the format
146
 
        bzrlib.branch.BranchFormat.unregister_format(format)
147
 
 
 
199
        BranchFormat.unregister_format(format)
 
200
        self.make_branch_and_tree('bar')
 
201
 
 
202
 
 
203
class TestBranch6(TestCaseWithTransport):
 
204
 
 
205
    def test_creation(self):
 
206
        format = BzrDirMetaFormat1()
 
207
        format.set_branch_format(_mod_branch.BzrBranchFormat6())
 
208
        branch = self.make_branch('a', format=format)
 
209
        self.assertIsInstance(branch, _mod_branch.BzrBranch6)
 
210
        branch = self.make_branch('b', format='dirstate-tags')
 
211
        self.assertIsInstance(branch, _mod_branch.BzrBranch6)
 
212
        branch = _mod_branch.Branch.open('a')
 
213
        self.assertIsInstance(branch, _mod_branch.BzrBranch6)
 
214
 
 
215
    def test_layout(self):
 
216
        branch = self.make_branch('a', format='dirstate-tags')
 
217
        self.failUnlessExists('a/.bzr/branch/last-revision')
 
218
        self.failIfExists('a/.bzr/branch/revision-history')
 
219
 
 
220
    def test_config(self):
 
221
        """Ensure that all configuration data is stored in the branch"""
 
222
        branch = self.make_branch('a', format='dirstate-tags')
 
223
        branch.set_parent('http://bazaar-vcs.org')
 
224
        self.failIfExists('a/.bzr/branch/parent')
 
225
        self.assertEqual('http://bazaar-vcs.org', branch.get_parent())
 
226
        branch.set_push_location('sftp://bazaar-vcs.org')
 
227
        config = branch.get_config()._get_branch_data_config()
 
228
        self.assertEqual('sftp://bazaar-vcs.org',
 
229
                         config.get_user_option('push_location'))
 
230
        branch.set_bound_location('ftp://bazaar-vcs.org')
 
231
        self.failIfExists('a/.bzr/branch/bound')
 
232
        self.assertEqual('ftp://bazaar-vcs.org', branch.get_bound_location())
 
233
 
 
234
    def test_set_revision_history(self):
 
235
        tree = self.make_branch_and_memory_tree('.',
 
236
            format='dirstate-tags')
 
237
        tree.lock_write()
 
238
        try:
 
239
            tree.add('.')
 
240
            tree.commit('foo', rev_id='foo')
 
241
            tree.commit('bar', rev_id='bar')
 
242
            tree.branch.set_revision_history(['foo', 'bar'])
 
243
            tree.branch.set_revision_history(['foo'])
 
244
            self.assertRaises(errors.NotLefthandHistory,
 
245
                              tree.branch.set_revision_history, ['bar'])
 
246
        finally:
 
247
            tree.unlock()
 
248
 
 
249
    def do_checkout_test(self, lightweight=False):
 
250
        tree = self.make_branch_and_tree('source', format='dirstate-with-subtree')
 
251
        subtree = self.make_branch_and_tree('source/subtree',
 
252
            format='dirstate-with-subtree')
 
253
        subsubtree = self.make_branch_and_tree('source/subtree/subsubtree',
 
254
            format='dirstate-with-subtree')
 
255
        self.build_tree(['source/subtree/file',
 
256
                         'source/subtree/subsubtree/file'])
 
257
        subsubtree.add('file')
 
258
        subtree.add('file')
 
259
        subtree.add_reference(subsubtree)
 
260
        tree.add_reference(subtree)
 
261
        tree.commit('a revision')
 
262
        subtree.commit('a subtree file')
 
263
        subsubtree.commit('a subsubtree file')
 
264
        tree.branch.create_checkout('target', lightweight=lightweight)
 
265
        self.failUnlessExists('target')
 
266
        self.failUnlessExists('target/subtree')
 
267
        self.failUnlessExists('target/subtree/file')
 
268
        self.failUnlessExists('target/subtree/subsubtree/file')
 
269
        subbranch = _mod_branch.Branch.open('target/subtree/subsubtree')
 
270
        if lightweight:
 
271
            self.assertEndsWith(subbranch.base, 'source/subtree/subsubtree/')
 
272
        else:
 
273
            self.assertEndsWith(subbranch.base, 'target/subtree/subsubtree/')
 
274
 
 
275
    def test_checkout_with_references(self):
 
276
        self.do_checkout_test()
 
277
 
 
278
    def test_light_checkout_with_references(self):
 
279
        self.do_checkout_test(lightweight=True)
 
280
 
 
281
    def test_set_push(self):
 
282
        branch = self.make_branch('source', format='dirstate-tags')
 
283
        branch.get_config().set_user_option('push_location', 'old',
 
284
            store=config.STORE_LOCATION)
 
285
        warnings = []
 
286
        def warning(*args):
 
287
            warnings.append(args[0] % args[1:])
 
288
        _warning = trace.warning
 
289
        trace.warning = warning
 
290
        try:
 
291
            branch.set_push_location('new')
 
292
        finally:
 
293
            trace.warning = _warning
 
294
        self.assertEqual(warnings[0], 'Value "new" is masked by "old" from '
 
295
                         'locations.conf')
148
296
 
149
297
class TestBranchReference(TestCaseWithTransport):
150
298
    """Tests for the branch reference facility."""
158
306
        target_branch = dir.create_branch()
159
307
        t.mkdir('branch')
160
308
        branch_dir = bzrdirformat.initialize(self.get_url('branch'))
161
 
        made_branch = bzrlib.branch.BranchReferenceFormat().initialize(branch_dir, target_branch)
 
309
        made_branch = BranchReferenceFormat().initialize(branch_dir, target_branch)
162
310
        self.assertEqual(made_branch.base, target_branch.base)
163
311
        opened_branch = branch_dir.open_branch()
164
312
        self.assertEqual(opened_branch.base, target_branch.base)
 
313
 
 
314
    def test_get_reference(self):
 
315
        """For a BranchReference, get_reference should reutrn the location."""
 
316
        branch = self.make_branch('target')
 
317
        checkout = branch.create_checkout('checkout', lightweight=True)
 
318
        reference_url = branch.bzrdir.root_transport.abspath('') + '/'
 
319
        # if the api for create_checkout changes to return different checkout types
 
320
        # then this file read will fail.
 
321
        self.assertFileEqual(reference_url, 'checkout/.bzr/branch/location')
 
322
        self.assertEqual(reference_url,
 
323
            _mod_branch.BranchReferenceFormat().get_reference(checkout.bzrdir))
 
324
 
 
325
 
 
326
class TestHooks(TestCase):
 
327
 
 
328
    def test_constructor(self):
 
329
        """Check that creating a BranchHooks instance has the right defaults."""
 
330
        hooks = BranchHooks()
 
331
        self.assertTrue("set_rh" in hooks, "set_rh not in %s" % hooks)
 
332
        self.assertTrue("post_push" in hooks, "post_push not in %s" % hooks)
 
333
        self.assertTrue("post_commit" in hooks, "post_commit not in %s" % hooks)
 
334
        self.assertTrue("pre_commit" in hooks, "pre_commit not in %s" % hooks)
 
335
        self.assertTrue("post_pull" in hooks, "post_pull not in %s" % hooks)
 
336
        self.assertTrue("post_uncommit" in hooks, "post_uncommit not in %s" % hooks)
 
337
 
 
338
    def test_installed_hooks_are_BranchHooks(self):
 
339
        """The installed hooks object should be a BranchHooks."""
 
340
        # the installed hooks are saved in self._preserved_hooks.
 
341
        self.assertIsInstance(self._preserved_hooks[_mod_branch.Branch], BranchHooks)
 
342
 
 
343
 
 
344
class TestPullResult(TestCase):
 
345
 
 
346
    def test_pull_result_to_int(self):
 
347
        # to support old code, the pull result can be used as an int
 
348
        r = PullResult()
 
349
        r.old_revno = 10
 
350
        r.new_revno = 20
 
351
        # this usage of results is not recommended for new code (because it
 
352
        # doesn't describe very well what happened), but for api stability
 
353
        # it's still supported
 
354
        a = "%d revisions pulled" % r
 
355
        self.assertEqual(a, "10 revisions pulled")