1
# Copyright (C) 2005, 2006 Canonical Ltd
1
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
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
34
from bzrlib.branch import (BzrBranch5,
33
from bzrlib.branch import (
37
BranchReferenceFormat,
36
42
from bzrlib.bzrdir import (BzrDirMetaFormat1, BzrDirMeta1,
37
43
BzrDir, BzrDirFormat)
38
44
from bzrlib.errors import (NotBranchError,
47
53
class TestDefaultFormat(TestCase):
49
55
def test_get_set_default_format(self):
50
old_format = bzrlib.branch.BranchFormat.get_default_format()
56
old_format = BranchFormat.get_default_format()
52
self.assertTrue(isinstance(old_format, bzrlib.branch.BzrBranchFormat5))
53
bzrlib.branch.BranchFormat.set_default_format(SampleBranchFormat())
58
self.assertTrue(isinstance(old_format, BzrBranchFormat5))
59
BranchFormat.set_default_format(SampleBranchFormat())
55
61
# the default branch format is used by the meta dir format
56
62
# which is not the default bzrdir format at this point
58
64
result = dir.create_branch()
59
65
self.assertEqual(result, 'A branch')
61
bzrlib.branch.BranchFormat.set_default_format(old_format)
62
self.assertEqual(old_format, bzrlib.branch.BranchFormat.get_default_format())
67
BranchFormat.set_default_format(old_format)
68
self.assertEqual(old_format, BranchFormat.get_default_format())
65
71
class TestBranchFormat5(TestCaseWithTransport):
99
105
# recursive section - that is, it appends the branch name.
102
class SampleBranchFormat(bzrlib.branch.BranchFormat):
108
class SampleBranchFormat(BranchFormat):
103
109
"""A sample format
105
111
this format is initializable, unsupported to aid in testing the
135
141
dir = format._matchingbzrdir.initialize(url)
136
142
dir.create_repository()
137
143
format.initialize(dir)
138
found_format = bzrlib.branch.BranchFormat.find_format(dir)
144
found_format = BranchFormat.find_format(dir)
139
145
self.failUnless(isinstance(found_format, format.__class__))
140
check_format(bzrlib.branch.BzrBranchFormat5(), "bar")
146
check_format(BzrBranchFormat5(), "bar")
142
148
def test_find_format_not_branch(self):
143
149
dir = bzrdir.BzrDirMetaFormat1().initialize(self.get_url())
144
150
self.assertRaises(NotBranchError,
145
bzrlib.branch.BranchFormat.find_format,
151
BranchFormat.find_format,
148
154
def test_find_format_unknown_format(self):
149
155
dir = bzrdir.BzrDirMetaFormat1().initialize(self.get_url())
150
156
SampleBranchFormat().initialize(dir)
151
157
self.assertRaises(UnknownFormatError,
152
bzrlib.branch.BranchFormat.find_format,
158
BranchFormat.find_format,
155
161
def test_register_unregister_format(self):
160
166
format.initialize(dir)
161
167
# register a format for it.
162
bzrlib.branch.BranchFormat.register_format(format)
168
BranchFormat.register_format(format)
163
169
# which branch.Open will refuse (not supported)
164
self.assertRaises(UnsupportedFormatError, bzrlib.branch.Branch.open, self.get_url())
170
self.assertRaises(UnsupportedFormatError, Branch.open, self.get_url())
165
171
self.make_branch_and_tree('foo')
166
172
# but open_downlevel will work
167
173
self.assertEqual(format.open(dir), bzrdir.BzrDir.open(self.get_url()).open_branch(unsupported=True))
168
174
# unregister the format
169
bzrlib.branch.BranchFormat.unregister_format(format)
175
BranchFormat.unregister_format(format)
170
176
self.make_branch_and_tree('bar')
172
178
def test_checkout_format(self):
184
190
format.set_branch_format(_mod_branch.BzrBranchFormat6())
185
191
branch = self.make_branch('a', format=format)
186
192
self.assertIsInstance(branch, _mod_branch.BzrBranch6)
187
branch = self.make_branch('b', format='experimental-branch6')
193
branch = self.make_branch('b', format='dirstate-with-subtree')
188
194
self.assertIsInstance(branch, _mod_branch.BzrBranch6)
189
195
branch = _mod_branch.Branch.open('a')
190
196
self.assertIsInstance(branch, _mod_branch.BzrBranch6)
192
198
def test_layout(self):
193
branch = self.make_branch('a', format='experimental-branch6')
199
branch = self.make_branch('a', format='dirstate-with-subtree')
194
200
self.failUnlessExists('a/.bzr/branch/last-revision')
195
201
self.failIfExists('a/.bzr/branch/revision-history')
197
203
def test_config(self):
198
204
"""Ensure that all configuration data is stored in the branch"""
199
branch = self.make_branch('a', format='experimental-branch6')
205
branch = self.make_branch('a', format='dirstate-with-subtree')
200
206
branch.set_parent('http://bazaar-vcs.org')
201
207
self.failIfExists('a/.bzr/branch/parent')
202
208
self.assertEqual('http://bazaar-vcs.org', branch.get_parent())
211
217
def test_set_revision_history(self):
212
218
tree = self.make_branch_and_memory_tree('.',
213
format='experimental-branch6')
219
format='dirstate-with-subtree')
214
220
tree.lock_write()
226
232
def test_append_revision(self):
227
233
tree = self.make_branch_and_tree('branch1',
228
format='experimental-branch6')
234
format='dirstate-with-subtree')
229
235
tree.lock_write()
232
237
tree.commit('foo', rev_id='foo')
233
238
tree.commit('bar', rev_id='bar')
234
239
tree.commit('baz', rev_id='baz')
258
def do_checkout_test(self, lightweight=False):
259
tree = self.make_branch_and_tree('source', format='dirstate-with-subtree')
260
subtree = self.make_branch_and_tree('source/subtree',
261
format='dirstate-with-subtree')
262
subsubtree = self.make_branch_and_tree('source/subtree/subsubtree',
263
format='dirstate-with-subtree')
264
self.build_tree(['source/subtree/file',
265
'source/subtree/subsubtree/file'])
266
subsubtree.add('file')
268
subtree.add_reference(subsubtree)
269
tree.add_reference(subtree)
270
tree.commit('a revision')
271
subtree.commit('a subtree file')
272
subsubtree.commit('a subsubtree file')
273
tree.branch.create_checkout('target', lightweight=lightweight)
274
self.failUnlessExists('target')
275
self.failUnlessExists('target/subtree')
276
self.failUnlessExists('target/subtree/file')
277
self.failUnlessExists('target/subtree/subsubtree/file')
278
subbranch = _mod_branch.Branch.open('target/subtree/subsubtree')
280
self.assertEndsWith(subbranch.base, 'source/subtree/subsubtree/')
282
self.assertEndsWith(subbranch.base, 'target/subtree/subsubtree/')
285
def test_checkout_with_references(self):
286
self.do_checkout_test()
288
def test_light_checkout_with_references(self):
289
self.do_checkout_test(lightweight=True)
254
291
class TestBranchReference(TestCaseWithTransport):
255
292
"""Tests for the branch reference facility."""
263
300
target_branch = dir.create_branch()
264
301
t.mkdir('branch')
265
302
branch_dir = bzrdirformat.initialize(self.get_url('branch'))
266
made_branch = bzrlib.branch.BranchReferenceFormat().initialize(branch_dir, target_branch)
303
made_branch = BranchReferenceFormat().initialize(branch_dir, target_branch)
267
304
self.assertEqual(made_branch.base, target_branch.base)
268
305
opened_branch = branch_dir.open_branch()
269
306
self.assertEqual(opened_branch.base, target_branch.base)
274
311
def test_constructor(self):
275
312
"""Check that creating a BranchHooks instance has the right defaults."""
276
hooks = bzrlib.branch.BranchHooks()
313
hooks = BranchHooks()
277
314
self.assertTrue("set_rh" in hooks, "set_rh not in %s" % hooks)
278
315
self.assertTrue("post_push" in hooks, "post_push not in %s" % hooks)
279
316
self.assertTrue("post_commit" in hooks, "post_commit not in %s" % hooks)
283
320
def test_installed_hooks_are_BranchHooks(self):
284
321
"""The installed hooks object should be a BranchHooks."""
285
322
# the installed hooks are saved in self._preserved_hooks.
286
self.assertIsInstance(self._preserved_hooks, bzrlib.branch.BranchHooks)
323
self.assertIsInstance(self._preserved_hooks, BranchHooks)
288
325
def test_install_hook_raises_unknown_hook(self):
289
326
"""install_hook should raise UnknownHook if a hook is unknown."""
290
hooks = bzrlib.branch.BranchHooks()
327
hooks = BranchHooks()
291
328
self.assertRaises(UnknownHook, hooks.install_hook, 'silly', None)
293
330
def test_install_hook_appends_known_hook(self):
294
331
"""install_hook should append the callable for known hooks."""
295
hooks = bzrlib.branch.BranchHooks()
332
hooks = BranchHooks()
296
333
hooks.install_hook('set_rh', None)
297
334
self.assertEqual(hooks['set_rh'], [None])
337
class TestPullResult(TestCase):
339
def test_pull_result_to_int(self):
340
# to support old code, the pull result can be used as an int
344
# this usage of results is not recommended for new code (because it
345
# doesn't describe very well what happened), but for api stability
346
# it's still supported
347
a = "%d revisions pulled" % r
348
self.assertEqual(a, "10 revisions pulled")
added
removed
bzrlib/tests/test_branch.py
