~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_branch.py

  • Committer: John Arbash Meinel
  • Author(s): Mark Hammond
  • Date: 2008-09-09 17:02:21 UTC
  • mto: This revision was merged to the branch mainline in revision 3697.
  • Revision ID: john@arbash-meinel.com-20080909170221-svim3jw2mrz0amp3
An updated transparent icon for bzr.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006-2012, 2016 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
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
"""Tests for the Branch facility that are not interface  tests.
18
18
 
19
 
For interface tests see `tests/per_branch/*.py`.
 
19
For interface tests see tests/branch_implementations/*.py.
20
20
 
21
21
For concrete class tests see this file, and for meta-branch tests
22
22
also see this file.
23
23
"""
24
24
 
25
 
from cStringIO import StringIO
 
25
from StringIO import StringIO
26
26
 
27
27
from bzrlib import (
28
28
    branch as _mod_branch,
29
29
    bzrdir,
30
30
    config,
31
 
    controldir,
32
31
    errors,
33
 
    tests,
34
32
    trace,
35
33
    urlutils,
36
34
    )
37
 
from bzrlib.branchfmt.fullhistory import (
 
35
from bzrlib.branch import (
 
36
    Branch,
 
37
    BranchHooks,
 
38
    BranchFormat,
 
39
    BranchReferenceFormat,
38
40
    BzrBranch5,
39
41
    BzrBranchFormat5,
 
42
    BzrBranchFormat6,
 
43
    PullResult,
40
44
    )
41
 
 
42
 
 
43
 
class TestDefaultFormat(tests.TestCase):
 
45
from bzrlib.bzrdir import (BzrDirMetaFormat1, BzrDirMeta1, 
 
46
                           BzrDir, BzrDirFormat)
 
47
from bzrlib.errors import (NotBranchError,
 
48
                           UnknownFormatError,
 
49
                           UnknownHook,
 
50
                           UnsupportedFormatError,
 
51
                           )
 
52
 
 
53
from bzrlib.tests import TestCase, TestCaseWithTransport
 
54
from bzrlib.transport import get_transport
 
55
 
 
56
 
 
57
class TestDefaultFormat(TestCase):
44
58
 
45
59
    def test_default_format(self):
46
60
        # update this if you change the default branch format
47
 
        self.assertIsInstance(_mod_branch.format_registry.get_default(),
48
 
                _mod_branch.BzrBranchFormat7)
 
61
        self.assertIsInstance(BranchFormat.get_default_format(),
 
62
                BzrBranchFormat6)
49
63
 
50
64
    def test_default_format_is_same_as_bzrdir_default(self):
51
65
        # XXX: it might be nice if there was only one place the default was
52
 
        # set, but at the moment that's not true -- mbp 20070814 --
 
66
        # set, but at the moment that's not true -- mbp 20070814 -- 
53
67
        # https://bugs.launchpad.net/bzr/+bug/132376
54
 
        self.assertEqual(
55
 
            _mod_branch.format_registry.get_default(),
56
 
            bzrdir.BzrDirFormat.get_default_format().get_branch_format())
 
68
        self.assertEqual(BranchFormat.get_default_format(),
 
69
                BzrDirFormat.get_default_format().get_branch_format())
57
70
 
58
71
    def test_get_set_default_format(self):
59
72
        # set the format and then set it back again
60
 
        old_format = _mod_branch.format_registry.get_default()
61
 
        _mod_branch.format_registry.set_default(SampleBranchFormat())
 
73
        old_format = BranchFormat.get_default_format()
 
74
        BranchFormat.set_default_format(SampleBranchFormat())
62
75
        try:
63
76
            # the default branch format is used by the meta dir format
64
77
            # which is not the default bzrdir format at this point
65
 
            dir = bzrdir.BzrDirMetaFormat1().initialize('memory:///')
 
78
            dir = BzrDirMetaFormat1().initialize('memory:///')
66
79
            result = dir.create_branch()
67
80
            self.assertEqual(result, 'A branch')
68
81
        finally:
69
 
            _mod_branch.format_registry.set_default(old_format)
70
 
        self.assertEqual(old_format,
71
 
                         _mod_branch.format_registry.get_default())
72
 
 
73
 
 
74
 
class TestBranchFormat5(tests.TestCaseWithTransport):
 
82
            BranchFormat.set_default_format(old_format)
 
83
        self.assertEqual(old_format, BranchFormat.get_default_format())
 
84
 
 
85
 
 
86
class TestBranchFormat5(TestCaseWithTransport):
75
87
    """Tests specific to branch format 5"""
76
88
 
77
89
    def test_branch_format_5_uses_lockdir(self):
78
90
        url = self.get_url()
79
 
        bdir = bzrdir.BzrDirMetaFormat1().initialize(url)
80
 
        bdir.create_repository()
81
 
        branch = BzrBranchFormat5().initialize(bdir)
 
91
        bzrdir = BzrDirMetaFormat1().initialize(url)
 
92
        bzrdir.create_repository()
 
93
        branch = bzrdir.create_branch()
82
94
        t = self.get_transport()
83
95
        self.log("branch instance is %r" % branch)
84
 
        self.assertTrue(isinstance(branch, BzrBranch5))
 
96
        self.assert_(isinstance(branch, BzrBranch5))
85
97
        self.assertIsDirectory('.', t)
86
98
        self.assertIsDirectory('.bzr/branch', t)
87
99
        self.assertIsDirectory('.bzr/branch/lock', t)
88
100
        branch.lock_write()
89
 
        self.addCleanup(branch.unlock)
90
 
        self.assertIsDirectory('.bzr/branch/lock/held', t)
 
101
        try:
 
102
            self.assertIsDirectory('.bzr/branch/lock/held', t)
 
103
        finally:
 
104
            branch.unlock()
91
105
 
92
106
    def test_set_push_location(self):
93
 
        conf = config.LocationConfig.from_string('# comment\n', '.', save=True)
 
107
        from bzrlib.config import (locations_config_filename,
 
108
                                   ensure_config_dir_exists)
 
109
        ensure_config_dir_exists()
 
110
        fn = locations_config_filename()
 
111
        # write correct newlines to locations.conf
 
112
        # by default ConfigObj uses native line-endings for new files
 
113
        # but uses already existing line-endings if file is not empty
 
114
        f = open(fn, 'wb')
 
115
        try:
 
116
            f.write('# comment\n')
 
117
        finally:
 
118
            f.close()
94
119
 
95
120
        branch = self.make_branch('.', format='knit')
96
121
        branch.set_push_location('foo')
99
124
                             "[%s]\n"
100
125
                             "push_location = foo\n"
101
126
                             "push_location:policy = norecurse\n" % local_path,
102
 
                             config.locations_config_filename())
 
127
                             fn)
103
128
 
104
129
    # TODO RBC 20051029 test getting a push location from a branch in a
105
130
    # recursive section - that is, it appends the branch name.
106
131
 
107
132
 
108
 
class SampleBranchFormat(_mod_branch.BranchFormatMetadir):
 
133
class SampleBranchFormat(BranchFormat):
109
134
    """A sample format
110
135
 
111
 
    this format is initializable, unsupported to aid in testing the
 
136
    this format is initializable, unsupported to aid in testing the 
112
137
    open and open_downlevel routines.
113
138
    """
114
139
 
115
 
    @classmethod
116
 
    def get_format_string(cls):
 
140
    def get_format_string(self):
117
141
        """See BzrBranchFormat.get_format_string()."""
118
142
        return "Sample branch format."
119
143
 
120
 
    def initialize(self, a_bzrdir, name=None, repository=None,
121
 
                   append_revisions_only=None):
 
144
    def initialize(self, a_bzrdir):
122
145
        """Format 4 branches cannot be created."""
123
 
        t = a_bzrdir.get_branch_transport(self, name=name)
 
146
        t = a_bzrdir.get_branch_transport(self)
124
147
        t.put_bytes('format', self.get_format_string())
125
148
        return 'A branch'
126
149
 
127
150
    def is_supported(self):
128
151
        return False
129
152
 
130
 
    def open(self, transport, name=None, _found=False, ignore_fallbacks=False,
131
 
             possible_transports=None):
 
153
    def open(self, transport, _found=False):
132
154
        return "opened branch."
133
155
 
134
156
 
135
 
# Demonstrating how lazy loading is often implemented:
136
 
# A constant string is created.
137
 
SampleSupportedBranchFormatString = "Sample supported branch format."
138
 
 
139
 
# And the format class can then reference the constant to avoid skew.
140
 
class SampleSupportedBranchFormat(_mod_branch.BranchFormatMetadir):
141
 
    """A sample supported format."""
142
 
 
143
 
    @classmethod
144
 
    def get_format_string(cls):
145
 
        """See BzrBranchFormat.get_format_string()."""
146
 
        return SampleSupportedBranchFormatString
147
 
 
148
 
    def initialize(self, a_bzrdir, name=None, append_revisions_only=None):
149
 
        t = a_bzrdir.get_branch_transport(self, name=name)
150
 
        t.put_bytes('format', self.get_format_string())
151
 
        return 'A branch'
152
 
 
153
 
    def open(self, transport, name=None, _found=False, ignore_fallbacks=False,
154
 
             possible_transports=None):
155
 
        return "opened supported branch."
156
 
 
157
 
 
158
 
class SampleExtraBranchFormat(_mod_branch.BranchFormat):
159
 
    """A sample format that is not usable in a metadir."""
160
 
 
161
 
    def get_format_string(self):
162
 
        # This format is not usable in a metadir.
163
 
        return None
164
 
 
165
 
    def network_name(self):
166
 
        # Network name always has to be provided.
167
 
        return "extra"
168
 
 
169
 
    def initialize(self, a_bzrdir, name=None):
170
 
        raise NotImplementedError(self.initialize)
171
 
 
172
 
    def open(self, transport, name=None, _found=False, ignore_fallbacks=False,
173
 
             possible_transports=None):
174
 
        raise NotImplementedError(self.open)
175
 
 
176
 
 
177
 
class TestBzrBranchFormat(tests.TestCaseWithTransport):
 
157
class TestBzrBranchFormat(TestCaseWithTransport):
178
158
    """Tests for the BzrBranchFormat facility."""
179
159
 
180
160
    def test_find_format(self):
181
161
        # is the right format object found for a branch?
182
162
        # create a branch with a few known format objects.
183
 
        # this is not quite the same as
 
163
        # this is not quite the same as 
184
164
        self.build_tree(["foo/", "bar/"])
185
165
        def check_format(format, url):
186
166
            dir = format._matchingbzrdir.initialize(url)
187
167
            dir.create_repository()
188
168
            format.initialize(dir)
189
 
            found_format = _mod_branch.BranchFormatMetadir.find_format(dir)
190
 
            self.assertIsInstance(found_format, format.__class__)
 
169
            found_format = BranchFormat.find_format(dir)
 
170
            self.failUnless(isinstance(found_format, format.__class__))
191
171
        check_format(BzrBranchFormat5(), "bar")
192
 
 
193
 
    def test_find_format_factory(self):
194
 
        dir = bzrdir.BzrDirMetaFormat1().initialize(self.get_url())
195
 
        SampleSupportedBranchFormat().initialize(dir)
196
 
        factory = _mod_branch.MetaDirBranchFormatFactory(
197
 
            SampleSupportedBranchFormatString,
198
 
            "bzrlib.tests.test_branch", "SampleSupportedBranchFormat")
199
 
        _mod_branch.format_registry.register(factory)
200
 
        self.addCleanup(_mod_branch.format_registry.remove, factory)
201
 
        b = _mod_branch.Branch.open(self.get_url())
202
 
        self.assertEqual(b, "opened supported branch.")
203
 
 
204
 
    def test_from_string(self):
205
 
        self.assertIsInstance(
206
 
            SampleBranchFormat.from_string("Sample branch format."),
207
 
            SampleBranchFormat)
208
 
        self.assertRaises(AssertionError,
209
 
            SampleBranchFormat.from_string, "Different branch format.")
210
 
 
 
172
        
211
173
    def test_find_format_not_branch(self):
212
174
        dir = bzrdir.BzrDirMetaFormat1().initialize(self.get_url())
213
 
        self.assertRaises(errors.NotBranchError,
214
 
                          _mod_branch.BranchFormatMetadir.find_format,
 
175
        self.assertRaises(NotBranchError,
 
176
                          BranchFormat.find_format,
215
177
                          dir)
216
178
 
217
179
    def test_find_format_unknown_format(self):
218
180
        dir = bzrdir.BzrDirMetaFormat1().initialize(self.get_url())
219
181
        SampleBranchFormat().initialize(dir)
220
 
        self.assertRaises(errors.UnknownFormatError,
221
 
                          _mod_branch.BranchFormatMetadir.find_format,
 
182
        self.assertRaises(UnknownFormatError,
 
183
                          BranchFormat.find_format,
222
184
                          dir)
223
185
 
224
 
    def test_find_format_with_features(self):
225
 
        tree = self.make_branch_and_tree('.', format='2a')
226
 
        tree.branch.update_feature_flags({"name": "optional"})
227
 
        found_format = _mod_branch.BranchFormatMetadir.find_format(tree.bzrdir)
228
 
        self.assertIsInstance(found_format, _mod_branch.BranchFormatMetadir)
229
 
        self.assertEqual(found_format.features.get("name"), "optional")
230
 
        tree.branch.update_feature_flags({"name": None})
231
 
        branch = _mod_branch.Branch.open('.')
232
 
        self.assertEqual(branch._format.features, {})
233
 
 
234
 
 
235
 
class TestBranchFormatRegistry(tests.TestCase):
236
 
 
237
 
    def setUp(self):
238
 
        super(TestBranchFormatRegistry, self).setUp()
239
 
        self.registry = _mod_branch.BranchFormatRegistry()
240
 
 
241
 
    def test_default(self):
242
 
        self.assertIs(None, self.registry.get_default())
243
 
        format = SampleBranchFormat()
244
 
        self.registry.set_default(format)
245
 
        self.assertEqual(format, self.registry.get_default())
246
 
 
247
186
    def test_register_unregister_format(self):
248
187
        format = SampleBranchFormat()
249
 
        self.registry.register(format)
250
 
        self.assertEqual(format,
251
 
            self.registry.get("Sample branch format."))
252
 
        self.registry.remove(format)
253
 
        self.assertRaises(KeyError, self.registry.get,
254
 
            "Sample branch format.")
255
 
 
256
 
    def test_get_all(self):
257
 
        format = SampleBranchFormat()
258
 
        self.assertEqual([], self.registry._get_all())
259
 
        self.registry.register(format)
260
 
        self.assertEqual([format], self.registry._get_all())
261
 
 
262
 
    def test_register_extra(self):
263
 
        format = SampleExtraBranchFormat()
264
 
        self.assertEqual([], self.registry._get_all())
265
 
        self.registry.register_extra(format)
266
 
        self.assertEqual([format], self.registry._get_all())
267
 
 
268
 
    def test_register_extra_lazy(self):
269
 
        self.assertEqual([], self.registry._get_all())
270
 
        self.registry.register_extra_lazy("bzrlib.tests.test_branch",
271
 
            "SampleExtraBranchFormat")
272
 
        formats = self.registry._get_all()
273
 
        self.assertEqual(1, len(formats))
274
 
        self.assertIsInstance(formats[0], SampleExtraBranchFormat)
275
 
 
276
 
 
277
 
#Used by TestMetaDirBranchFormatFactory 
278
 
FakeLazyFormat = None
279
 
 
280
 
 
281
 
class TestMetaDirBranchFormatFactory(tests.TestCase):
282
 
 
283
 
    def test_get_format_string_does_not_load(self):
284
 
        """Formats have a static format string."""
285
 
        factory = _mod_branch.MetaDirBranchFormatFactory("yo", None, None)
286
 
        self.assertEqual("yo", factory.get_format_string())
287
 
 
288
 
    def test_call_loads(self):
289
 
        # __call__ is used by the network_format_registry interface to get a
290
 
        # Format.
291
 
        global FakeLazyFormat
292
 
        del FakeLazyFormat
293
 
        factory = _mod_branch.MetaDirBranchFormatFactory(None,
294
 
            "bzrlib.tests.test_branch", "FakeLazyFormat")
295
 
        self.assertRaises(AttributeError, factory)
296
 
 
297
 
    def test_call_returns_call_of_referenced_object(self):
298
 
        global FakeLazyFormat
299
 
        FakeLazyFormat = lambda:'called'
300
 
        factory = _mod_branch.MetaDirBranchFormatFactory(None,
301
 
            "bzrlib.tests.test_branch", "FakeLazyFormat")
302
 
        self.assertEqual('called', factory())
 
188
        # make a control dir
 
189
        dir = bzrdir.BzrDirMetaFormat1().initialize(self.get_url())
 
190
        # make a branch
 
191
        format.initialize(dir)
 
192
        # register a format for it.
 
193
        BranchFormat.register_format(format)
 
194
        # which branch.Open will refuse (not supported)
 
195
        self.assertRaises(UnsupportedFormatError, Branch.open, self.get_url())
 
196
        self.make_branch_and_tree('foo')
 
197
        # but open_downlevel will work
 
198
        self.assertEqual(format.open(dir), bzrdir.BzrDir.open(self.get_url()).open_branch(unsupported=True))
 
199
        # unregister the format
 
200
        BranchFormat.unregister_format(format)
 
201
        self.make_branch_and_tree('bar')
303
202
 
304
203
 
305
204
class TestBranch67(object):
315
214
        raise NotImplementedError(self.get_class)
316
215
 
317
216
    def test_creation(self):
318
 
        format = bzrdir.BzrDirMetaFormat1()
 
217
        format = BzrDirMetaFormat1()
319
218
        format.set_branch_format(_mod_branch.BzrBranchFormat6())
320
219
        branch = self.make_branch('a', format=format)
321
220
        self.assertIsInstance(branch, self.get_class())
326
225
 
327
226
    def test_layout(self):
328
227
        branch = self.make_branch('a', format=self.get_format_name())
329
 
        self.assertPathExists('a/.bzr/branch/last-revision')
330
 
        self.assertPathDoesNotExist('a/.bzr/branch/revision-history')
331
 
        self.assertPathDoesNotExist('a/.bzr/branch/references')
 
228
        self.failUnlessExists('a/.bzr/branch/last-revision')
 
229
        self.failIfExists('a/.bzr/branch/revision-history')
332
230
 
333
231
    def test_config(self):
334
232
        """Ensure that all configuration data is stored in the branch"""
335
233
        branch = self.make_branch('a', format=self.get_format_name())
336
 
        branch.set_parent('http://example.com')
337
 
        self.assertPathDoesNotExist('a/.bzr/branch/parent')
338
 
        self.assertEqual('http://example.com', branch.get_parent())
339
 
        branch.set_push_location('sftp://example.com')
340
 
        conf = branch.get_config_stack()
341
 
        self.assertEqual('sftp://example.com', conf.get('push_location'))
342
 
        branch.set_bound_location('ftp://example.com')
343
 
        self.assertPathDoesNotExist('a/.bzr/branch/bound')
344
 
        self.assertEqual('ftp://example.com', branch.get_bound_location())
 
234
        branch.set_parent('http://bazaar-vcs.org')
 
235
        self.failIfExists('a/.bzr/branch/parent')
 
236
        self.assertEqual('http://bazaar-vcs.org', branch.get_parent())
 
237
        branch.set_push_location('sftp://bazaar-vcs.org')
 
238
        config = branch.get_config()._get_branch_data_config()
 
239
        self.assertEqual('sftp://bazaar-vcs.org',
 
240
                         config.get_user_option('push_location'))
 
241
        branch.set_bound_location('ftp://bazaar-vcs.org')
 
242
        self.failIfExists('a/.bzr/branch/bound')
 
243
        self.assertEqual('ftp://bazaar-vcs.org', branch.get_bound_location())
 
244
 
 
245
    def test_set_revision_history(self):
 
246
        builder = self.make_branch_builder('.', format=self.get_format_name())
 
247
        builder.build_snapshot('foo', None,
 
248
            [('add', ('', None, 'directory', None))],
 
249
            message='foo')
 
250
        builder.build_snapshot('bar', None, [], message='bar')
 
251
        branch = builder.get_branch()
 
252
        branch.lock_write()
 
253
        self.addCleanup(branch.unlock)
 
254
        branch.set_revision_history(['foo', 'bar'])
 
255
        branch.set_revision_history(['foo'])
 
256
        self.assertRaises(errors.NotLefthandHistory,
 
257
                          branch.set_revision_history, ['bar'])
345
258
 
346
259
    def do_checkout_test(self, lightweight=False):
347
260
        tree = self.make_branch_and_tree('source',
360
273
        subtree.commit('a subtree file')
361
274
        subsubtree.commit('a subsubtree file')
362
275
        tree.branch.create_checkout('target', lightweight=lightweight)
363
 
        self.assertPathExists('target')
364
 
        self.assertPathExists('target/subtree')
365
 
        self.assertPathExists('target/subtree/file')
366
 
        self.assertPathExists('target/subtree/subsubtree/file')
 
276
        self.failUnlessExists('target')
 
277
        self.failUnlessExists('target/subtree')
 
278
        self.failUnlessExists('target/subtree/file')
 
279
        self.failUnlessExists('target/subtree/subsubtree/file')
367
280
        subbranch = _mod_branch.Branch.open('target/subtree/subsubtree')
368
281
        if lightweight:
369
282
            self.assertEndsWith(subbranch.base, 'source/subtree/subsubtree/')
376
289
    def test_light_checkout_with_references(self):
377
290
        self.do_checkout_test(lightweight=True)
378
291
 
379
 
 
380
 
class TestBranch6(TestBranch67, tests.TestCaseWithTransport):
 
292
    def test_set_push(self):
 
293
        branch = self.make_branch('source', format=self.get_format_name())
 
294
        branch.get_config().set_user_option('push_location', 'old',
 
295
            store=config.STORE_LOCATION)
 
296
        warnings = []
 
297
        def warning(*args):
 
298
            warnings.append(args[0] % args[1:])
 
299
        _warning = trace.warning
 
300
        trace.warning = warning
 
301
        try:
 
302
            branch.set_push_location('new')
 
303
        finally:
 
304
            trace.warning = _warning
 
305
        self.assertEqual(warnings[0], 'Value "new" is masked by "old" from '
 
306
                         'locations.conf')
 
307
 
 
308
 
 
309
class TestBranch6(TestBranch67, TestCaseWithTransport):
381
310
 
382
311
    def get_class(self):
383
312
        return _mod_branch.BzrBranch6
398
327
        self.assertRaises(errors.UnstackableBranchFormat, branch.get_stacked_on_url)
399
328
 
400
329
 
401
 
class TestBranch7(TestBranch67, tests.TestCaseWithTransport):
 
330
class TestBranch7(TestBranch67, TestCaseWithTransport):
402
331
 
403
332
    def get_class(self):
404
333
        return _mod_branch.BzrBranch7
405
334
 
406
335
    def get_format_name(self):
407
 
        return "1.9"
 
336
        return "development"
408
337
 
409
338
    def get_format_name_subtree(self):
410
339
        return "development-subtree"
448
377
        self.assertTrue(branch.repository.has_revision(revid))
449
378
 
450
379
 
451
 
class BzrBranch8(tests.TestCaseWithTransport):
452
 
 
453
 
    def make_branch(self, location, format=None):
454
 
        if format is None:
455
 
            format = controldir.format_registry.make_bzrdir('1.9')
456
 
            format.set_branch_format(_mod_branch.BzrBranchFormat8())
457
 
        return tests.TestCaseWithTransport.make_branch(
458
 
            self, location, format=format)
459
 
 
460
 
    def create_branch_with_reference(self):
461
 
        branch = self.make_branch('branch')
462
 
        branch._set_all_reference_info({'file-id': ('path', 'location')})
463
 
        return branch
464
 
 
465
 
    @staticmethod
466
 
    def instrument_branch(branch, gets):
467
 
        old_get = branch._transport.get
468
 
        def get(*args, **kwargs):
469
 
            gets.append((args, kwargs))
470
 
            return old_get(*args, **kwargs)
471
 
        branch._transport.get = get
472
 
 
473
 
    def test_reference_info_caching_read_locked(self):
474
 
        gets = []
475
 
        branch = self.create_branch_with_reference()
476
 
        branch.lock_read()
477
 
        self.addCleanup(branch.unlock)
478
 
        self.instrument_branch(branch, gets)
479
 
        branch.get_reference_info('file-id')
480
 
        branch.get_reference_info('file-id')
481
 
        self.assertEqual(1, len(gets))
482
 
 
483
 
    def test_reference_info_caching_read_unlocked(self):
484
 
        gets = []
485
 
        branch = self.create_branch_with_reference()
486
 
        self.instrument_branch(branch, gets)
487
 
        branch.get_reference_info('file-id')
488
 
        branch.get_reference_info('file-id')
489
 
        self.assertEqual(2, len(gets))
490
 
 
491
 
    def test_reference_info_caching_write_locked(self):
492
 
        gets = []
493
 
        branch = self.make_branch('branch')
494
 
        branch.lock_write()
495
 
        self.instrument_branch(branch, gets)
496
 
        self.addCleanup(branch.unlock)
497
 
        branch._set_all_reference_info({'file-id': ('path2', 'location2')})
498
 
        path, location = branch.get_reference_info('file-id')
499
 
        self.assertEqual(0, len(gets))
500
 
        self.assertEqual('path2', path)
501
 
        self.assertEqual('location2', location)
502
 
 
503
 
    def test_reference_info_caches_cleared(self):
504
 
        branch = self.make_branch('branch')
505
 
        branch.lock_write()
506
 
        branch.set_reference_info('file-id', 'path2', 'location2')
507
 
        branch.unlock()
508
 
        doppelganger = _mod_branch.Branch.open('branch')
509
 
        doppelganger.set_reference_info('file-id', 'path3', 'location3')
510
 
        self.assertEqual(('path3', 'location3'),
511
 
                         branch.get_reference_info('file-id'))
512
 
 
513
 
    def _recordParentMapCalls(self, repo):
514
 
        self._parent_map_calls = []
515
 
        orig_get_parent_map = repo.revisions.get_parent_map
516
 
        def get_parent_map(q):
517
 
            q = list(q)
518
 
            self._parent_map_calls.extend([e[0] for e in q])
519
 
            return orig_get_parent_map(q)
520
 
        repo.revisions.get_parent_map = get_parent_map
521
 
 
522
 
 
523
 
class TestBranchReference(tests.TestCaseWithTransport):
 
380
class TestBranchReference(TestCaseWithTransport):
524
381
    """Tests for the branch reference facility."""
525
382
 
526
383
    def test_create_open_reference(self):
527
384
        bzrdirformat = bzrdir.BzrDirMetaFormat1()
528
 
        t = self.get_transport()
 
385
        t = get_transport(self.get_url('.'))
529
386
        t.mkdir('repo')
530
387
        dir = bzrdirformat.initialize(self.get_url('repo'))
531
388
        dir.create_repository()
532
389
        target_branch = dir.create_branch()
533
390
        t.mkdir('branch')
534
391
        branch_dir = bzrdirformat.initialize(self.get_url('branch'))
535
 
        made_branch = _mod_branch.BranchReferenceFormat().initialize(
536
 
            branch_dir, target_branch=target_branch)
 
392
        made_branch = BranchReferenceFormat().initialize(branch_dir, target_branch)
537
393
        self.assertEqual(made_branch.base, target_branch.base)
538
394
        opened_branch = branch_dir.open_branch()
539
395
        self.assertEqual(opened_branch.base, target_branch.base)
540
396
 
541
397
    def test_get_reference(self):
542
 
        """For a BranchReference, get_reference should return the location."""
 
398
        """For a BranchReference, get_reference should reutrn the location."""
543
399
        branch = self.make_branch('target')
544
400
        checkout = branch.create_checkout('checkout', lightweight=True)
545
401
        reference_url = branch.bzrdir.root_transport.abspath('') + '/'
550
406
            _mod_branch.BranchReferenceFormat().get_reference(checkout.bzrdir))
551
407
 
552
408
 
553
 
class TestHooks(tests.TestCaseWithTransport):
 
409
class TestHooks(TestCase):
554
410
 
555
411
    def test_constructor(self):
556
412
        """Check that creating a BranchHooks instance has the right defaults."""
557
 
        hooks = _mod_branch.BranchHooks()
 
413
        hooks = BranchHooks()
 
414
        self.assertTrue("set_rh" in hooks, "set_rh not in %s" % hooks)
558
415
        self.assertTrue("post_push" in hooks, "post_push not in %s" % hooks)
559
416
        self.assertTrue("post_commit" in hooks, "post_commit not in %s" % hooks)
560
417
        self.assertTrue("pre_commit" in hooks, "pre_commit not in %s" % hooks)
561
418
        self.assertTrue("post_pull" in hooks, "post_pull not in %s" % hooks)
562
 
        self.assertTrue("post_uncommit" in hooks,
563
 
                        "post_uncommit not in %s" % hooks)
 
419
        self.assertTrue("post_uncommit" in hooks, "post_uncommit not in %s" % hooks)
564
420
        self.assertTrue("post_change_branch_tip" in hooks,
565
421
                        "post_change_branch_tip not in %s" % hooks)
566
 
        self.assertTrue("post_branch_init" in hooks,
567
 
                        "post_branch_init not in %s" % hooks)
568
 
        self.assertTrue("post_switch" in hooks,
569
 
                        "post_switch not in %s" % hooks)
570
422
 
571
423
    def test_installed_hooks_are_BranchHooks(self):
572
424
        """The installed hooks object should be a BranchHooks."""
573
425
        # the installed hooks are saved in self._preserved_hooks.
574
 
        self.assertIsInstance(self._preserved_hooks[_mod_branch.Branch][1],
575
 
                              _mod_branch.BranchHooks)
576
 
 
577
 
    def test_post_branch_init_hook(self):
578
 
        calls = []
579
 
        _mod_branch.Branch.hooks.install_named_hook('post_branch_init',
580
 
            calls.append, None)
581
 
        self.assertLength(0, calls)
582
 
        branch = self.make_branch('a')
583
 
        self.assertLength(1, calls)
584
 
        params = calls[0]
585
 
        self.assertIsInstance(params, _mod_branch.BranchInitHookParams)
586
 
        self.assertTrue(hasattr(params, 'bzrdir'))
587
 
        self.assertTrue(hasattr(params, 'branch'))
588
 
 
589
 
    def test_post_branch_init_hook_repr(self):
590
 
        param_reprs = []
591
 
        _mod_branch.Branch.hooks.install_named_hook('post_branch_init',
592
 
            lambda params: param_reprs.append(repr(params)), None)
593
 
        branch = self.make_branch('a')
594
 
        self.assertLength(1, param_reprs)
595
 
        param_repr = param_reprs[0]
596
 
        self.assertStartsWith(param_repr, '<BranchInitHookParams of ')
597
 
 
598
 
    def test_post_switch_hook(self):
599
 
        from bzrlib import switch
600
 
        calls = []
601
 
        _mod_branch.Branch.hooks.install_named_hook('post_switch',
602
 
            calls.append, None)
603
 
        tree = self.make_branch_and_tree('branch-1')
604
 
        self.build_tree(['branch-1/file-1'])
605
 
        tree.add('file-1')
606
 
        tree.commit('rev1')
607
 
        to_branch = tree.bzrdir.sprout('branch-2').open_branch()
608
 
        self.build_tree(['branch-1/file-2'])
609
 
        tree.add('file-2')
610
 
        tree.remove('file-1')
611
 
        tree.commit('rev2')
612
 
        checkout = tree.branch.create_checkout('checkout')
613
 
        self.assertLength(0, calls)
614
 
        switch.switch(checkout.bzrdir, to_branch)
615
 
        self.assertLength(1, calls)
616
 
        params = calls[0]
617
 
        self.assertIsInstance(params, _mod_branch.SwitchHookParams)
618
 
        self.assertTrue(hasattr(params, 'to_branch'))
619
 
        self.assertTrue(hasattr(params, 'revision_id'))
620
 
 
621
 
 
622
 
class TestBranchOptions(tests.TestCaseWithTransport):
623
 
 
624
 
    def setUp(self):
625
 
        super(TestBranchOptions, self).setUp()
626
 
        self.branch = self.make_branch('.')
627
 
        self.config_stack = self.branch.get_config_stack()
628
 
 
629
 
    def check_append_revisions_only(self, expected_value, value=None):
630
 
        """Set append_revisions_only in config and check its interpretation."""
631
 
        if value is not None:
632
 
            self.config_stack.set('append_revisions_only', value)
633
 
        self.assertEqual(expected_value,
634
 
                         self.branch.get_append_revisions_only())
635
 
 
636
 
    def test_valid_append_revisions_only(self):
637
 
        self.assertEqual(None,
638
 
                          self.config_stack.get('append_revisions_only'))
639
 
        self.check_append_revisions_only(None)
640
 
        self.check_append_revisions_only(False, 'False')
641
 
        self.check_append_revisions_only(True, 'True')
642
 
        # The following values will cause compatibility problems on projects
643
 
        # using older bzr versions (<2.2) but are accepted
644
 
        self.check_append_revisions_only(False, 'false')
645
 
        self.check_append_revisions_only(True, 'true')
646
 
 
647
 
    def test_invalid_append_revisions_only(self):
648
 
        """Ensure warning is noted on invalid settings"""
649
 
        self.warnings = []
650
 
        def warning(*args):
651
 
            self.warnings.append(args[0] % args[1:])
652
 
        self.overrideAttr(trace, 'warning', warning)
653
 
        self.check_append_revisions_only(None, 'not-a-bool')
654
 
        self.assertLength(1, self.warnings)
655
 
        self.assertEqual(
656
 
            'Value "not-a-bool" is not valid for "append_revisions_only"',
657
 
            self.warnings[0])
658
 
 
659
 
    def test_use_fresh_values(self):
660
 
        copy = _mod_branch.Branch.open(self.branch.base)
661
 
        copy.lock_write()
662
 
        try:
663
 
            copy.get_config_stack().set('foo', 'bar')
664
 
        finally:
665
 
            copy.unlock()
666
 
        self.assertFalse(self.branch.is_locked())
667
 
        # Since the branch is locked, the option value won't be saved on disk
668
 
        # so trying to access the config of locked branch via another older
669
 
        # non-locked branch object pointing to the same branch is not supported
670
 
        self.assertEqual(None, self.branch.get_config_stack().get('foo'))
671
 
        # Using a newly created branch object works as expected
672
 
        fresh = _mod_branch.Branch.open(self.branch.base)
673
 
        self.assertEqual('bar', fresh.get_config_stack().get('foo'))
674
 
 
675
 
    def test_set_from_config_get_from_config_stack(self):
676
 
        self.branch.lock_write()
677
 
        self.addCleanup(self.branch.unlock)
678
 
        self.branch.get_config().set_user_option('foo', 'bar')
679
 
        result = self.branch.get_config_stack().get('foo')
680
 
        # https://bugs.launchpad.net/bzr/+bug/948344
681
 
        self.expectFailure('BranchStack uses cache after set_user_option',
682
 
                           self.assertEqual, 'bar', result)
683
 
 
684
 
    def test_set_from_config_stack_get_from_config(self):
685
 
        self.branch.lock_write()
686
 
        self.addCleanup(self.branch.unlock)
687
 
        self.branch.get_config_stack().set('foo', 'bar')
688
 
        # Since the branch is locked, the option value won't be saved on disk
689
 
        # so mixing get() and get_user_option() is broken by design.
690
 
        self.assertEqual(None,
691
 
                         self.branch.get_config().get_user_option('foo'))
692
 
 
693
 
    def test_set_delays_write_when_branch_is_locked(self):
694
 
        self.branch.lock_write()
695
 
        self.addCleanup(self.branch.unlock)
696
 
        self.branch.get_config_stack().set('foo', 'bar')
697
 
        copy = _mod_branch.Branch.open(self.branch.base)
698
 
        result = copy.get_config_stack().get('foo')
699
 
        # Accessing from a different branch object is like accessing from a
700
 
        # different process: the option has not been saved yet and the new
701
 
        # value cannot be seen.
702
 
        self.assertIs(None, result)
703
 
 
704
 
 
705
 
class TestPullResult(tests.TestCase):
706
 
 
707
 
    def test_report_changed(self):
708
 
        r = _mod_branch.PullResult()
709
 
        r.old_revid = "old-revid"
 
426
        self.assertIsInstance(self._preserved_hooks[_mod_branch.Branch], BranchHooks)
 
427
 
 
428
 
 
429
class TestPullResult(TestCase):
 
430
 
 
431
    def test_pull_result_to_int(self):
 
432
        # to support old code, the pull result can be used as an int
 
433
        r = PullResult()
710
434
        r.old_revno = 10
711
 
        r.new_revid = "new-revid"
712
435
        r.new_revno = 20
713
 
        f = StringIO()
714
 
        r.report(f)
715
 
        self.assertEqual("Now on revision 20.\n", f.getvalue())
716
 
        self.assertEqual("Now on revision 20.\n", f.getvalue())
717
 
 
718
 
    def test_report_unchanged(self):
719
 
        r = _mod_branch.PullResult()
720
 
        r.old_revid = "same-revid"
721
 
        r.new_revid = "same-revid"
722
 
        f = StringIO()
723
 
        r.report(f)
724
 
        self.assertEqual("No revisions or tags to pull.\n", f.getvalue())
 
436
        # this usage of results is not recommended for new code (because it
 
437
        # doesn't describe very well what happened), but for api stability
 
438
        # it's still supported
 
439
        a = "%d revisions pulled" % r
 
440
        self.assertEqual(a, "10 revisions pulled")