52
62
from bzrlib.tests.test_http import TestWithTransport_pycurl
53
from bzrlib.transport import get_transport
63
from bzrlib.transport import (
54
67
from bzrlib.transport.http._urllib import HttpTransport_urllib
55
from bzrlib.transport.memory import MemoryServer
56
68
from bzrlib.transport.nosmart import NoSmartTransportDecorator
57
69
from bzrlib.transport.readonly import ReadonlyTransportDecorator
58
from bzrlib.repofmt import knitrepo, weaverepo, pack_repo
70
from bzrlib.repofmt import knitrepo, knitpack_repo
61
73
class TestDefaultFormat(TestCase):
63
75
def test_get_set_default_format(self):
64
76
old_format = bzrdir.BzrDirFormat.get_default_format()
65
# default is BzrDirFormat6
66
self.failUnless(isinstance(old_format, bzrdir.BzrDirMetaFormat1))
67
bzrdir.BzrDirFormat._set_default_format(SampleBzrDirFormat())
77
# default is BzrDirMetaFormat1
78
self.assertIsInstance(old_format, bzrdir.BzrDirMetaFormat1)
79
controldir.ControlDirFormat._set_default_format(SampleBzrDirFormat())
68
80
# creating a bzr dir should now create an instrumented dir.
70
82
result = bzrdir.BzrDir.create('memory:///')
71
self.failUnless(isinstance(result, SampleBzrDir))
83
self.assertIsInstance(result, SampleBzrDir)
73
bzrdir.BzrDirFormat._set_default_format(old_format)
85
controldir.ControlDirFormat._set_default_format(old_format)
74
86
self.assertEqual(old_format, bzrdir.BzrDirFormat.get_default_format())
89
class DeprecatedBzrDirFormat(bzrdir.BzrDirFormat):
90
"""A deprecated bzr dir format."""
77
93
class TestFormatRegistry(TestCase):
79
95
def make_format_registry(self):
80
my_format_registry = bzrdir.BzrDirFormatRegistry()
81
my_format_registry.register('weave', bzrdir.BzrDirFormat6,
82
'Pre-0.8 format. Slower and does not support checkouts or shared'
83
' repositories', deprecated=True)
84
my_format_registry.register_lazy('lazy', 'bzrlib.bzrdir',
85
'BzrDirFormat6', 'Format registered lazily', deprecated=True)
86
my_format_registry.register_metadir('knit',
96
my_format_registry = controldir.ControlDirFormatRegistry()
97
my_format_registry.register('deprecated', DeprecatedBzrDirFormat,
98
'Some format. Slower and unawesome and deprecated.',
100
my_format_registry.register_lazy('lazy', 'bzrlib.tests.test_bzrdir',
101
'DeprecatedBzrDirFormat', 'Format registered lazily',
103
bzrdir.register_metadir(my_format_registry, 'knit',
87
104
'bzrlib.repofmt.knitrepo.RepositoryFormatKnit1',
88
105
'Format using knits',
90
107
my_format_registry.set_default('knit')
91
my_format_registry.register_metadir(
108
bzrdir.register_metadir(my_format_registry,
93
110
'bzrlib.repofmt.knitrepo.RepositoryFormatKnit3',
94
111
'Experimental successor to knit. Use at your own risk.',
95
112
branch_format='bzrlib.branch.BzrBranchFormat6',
96
113
experimental=True)
97
my_format_registry.register_metadir(
114
bzrdir.register_metadir(my_format_registry,
99
116
'bzrlib.repofmt.knitrepo.RepositoryFormatKnit3',
100
117
'Experimental successor to knit. Use at your own risk.',
101
118
branch_format='bzrlib.branch.BzrBranchFormat6', hidden=True)
102
my_format_registry.register('hiddenweave', bzrdir.BzrDirFormat6,
103
'Pre-0.8 format. Slower and does not support checkouts or shared'
104
' repositories', hidden=True)
105
my_format_registry.register_lazy('hiddenlazy', 'bzrlib.bzrdir',
106
'BzrDirFormat6', 'Format registered lazily', deprecated=True,
119
my_format_registry.register('hiddendeprecated', DeprecatedBzrDirFormat,
120
'Old format. Slower and does not support things. ', hidden=True)
121
my_format_registry.register_lazy('hiddenlazy', 'bzrlib.tests.test_bzrdir',
122
'DeprecatedBzrDirFormat', 'Format registered lazily',
123
deprecated=True, hidden=True)
108
124
return my_format_registry
110
126
def test_format_registry(self):
111
127
my_format_registry = self.make_format_registry()
112
128
my_bzrdir = my_format_registry.make_bzrdir('lazy')
113
self.assertIsInstance(my_bzrdir, bzrdir.BzrDirFormat6)
114
my_bzrdir = my_format_registry.make_bzrdir('weave')
115
self.assertIsInstance(my_bzrdir, bzrdir.BzrDirFormat6)
129
self.assertIsInstance(my_bzrdir, DeprecatedBzrDirFormat)
130
my_bzrdir = my_format_registry.make_bzrdir('deprecated')
131
self.assertIsInstance(my_bzrdir, DeprecatedBzrDirFormat)
116
132
my_bzrdir = my_format_registry.make_bzrdir('default')
117
133
self.assertIsInstance(my_bzrdir.repository_format,
118
134
knitrepo.RepositoryFormatKnit1)
155
170
self.assertNotContainsRe(new, 'hidden')
157
172
def test_set_default_repository(self):
158
default_factory = bzrdir.format_registry.get('default')
159
old_default = [k for k, v in bzrdir.format_registry.iteritems()
173
default_factory = controldir.format_registry.get('default')
174
old_default = [k for k, v in controldir.format_registry.iteritems()
160
175
if v == default_factory and k != 'default'][0]
161
bzrdir.format_registry.set_default_repository('dirstate-with-subtree')
176
controldir.format_registry.set_default_repository('dirstate-with-subtree')
163
self.assertIs(bzrdir.format_registry.get('dirstate-with-subtree'),
164
bzrdir.format_registry.get('default'))
178
self.assertIs(controldir.format_registry.get('dirstate-with-subtree'),
179
controldir.format_registry.get('default'))
166
repository.RepositoryFormat.get_default_format().__class__,
181
repository.format_registry.get_default().__class__,
167
182
knitrepo.RepositoryFormatKnit3)
169
bzrdir.format_registry.set_default_repository(old_default)
184
controldir.format_registry.set_default_repository(old_default)
171
186
def test_aliases(self):
172
a_registry = bzrdir.BzrDirFormatRegistry()
173
a_registry.register('weave', bzrdir.BzrDirFormat6,
174
'Pre-0.8 format. Slower and does not support checkouts or shared'
175
' repositories', deprecated=True)
176
a_registry.register('weavealias', bzrdir.BzrDirFormat6,
177
'Pre-0.8 format. Slower and does not support checkouts or shared'
178
' repositories', deprecated=True, alias=True)
179
self.assertEqual(frozenset(['weavealias']), a_registry.aliases())
187
a_registry = controldir.ControlDirFormatRegistry()
188
a_registry.register('deprecated', DeprecatedBzrDirFormat,
189
'Old format. Slower and does not support stuff',
191
a_registry.register('deprecatedalias', DeprecatedBzrDirFormat,
192
'Old format. Slower and does not support stuff',
193
deprecated=True, alias=True)
194
self.assertEqual(frozenset(['deprecatedalias']), a_registry.aliases())
182
197
class SampleBranch(bzrlib.branch.Branch):
243
278
def test_find_format(self):
244
279
# is the right format object found for a branch?
245
280
# create a branch with a few known format objects.
246
# this is not quite the same as
247
t = get_transport(self.get_url())
281
bzrdir.BzrProber.formats.register(BzrDirFormatTest1.get_format_string(),
283
self.addCleanup(bzrdir.BzrProber.formats.remove,
284
BzrDirFormatTest1.get_format_string())
285
bzrdir.BzrProber.formats.register(BzrDirFormatTest2.get_format_string(),
287
self.addCleanup(bzrdir.BzrProber.formats.remove,
288
BzrDirFormatTest2.get_format_string())
289
t = self.get_transport()
248
290
self.build_tree(["foo/", "bar/"], transport=t)
249
291
def check_format(format, url):
250
292
format.initialize(url)
251
t = get_transport(url)
293
t = _mod_transport.get_transport_from_path(url)
252
294
found_format = bzrdir.BzrDirFormat.find_format(t)
253
self.failUnless(isinstance(found_format, format.__class__))
254
check_format(bzrdir.BzrDirFormat5(), "foo")
255
check_format(bzrdir.BzrDirFormat6(), "bar")
295
self.assertIsInstance(found_format, format.__class__)
296
check_format(BzrDirFormatTest1(), "foo")
297
check_format(BzrDirFormatTest2(), "bar")
257
299
def test_find_format_nothing_there(self):
258
300
self.assertRaises(NotBranchError,
259
301
bzrdir.BzrDirFormat.find_format,
302
_mod_transport.get_transport_from_path('.'))
262
304
def test_find_format_unknown_format(self):
263
t = get_transport(self.get_url())
305
t = self.get_transport()
265
307
t.put_bytes('.bzr/branch-format', '')
266
308
self.assertRaises(UnknownFormatError,
267
309
bzrdir.BzrDirFormat.find_format,
310
_mod_transport.get_transport_from_path('.'))
270
312
def test_register_unregister_format(self):
271
313
format = SampleBzrDirFormat()
474
517
# Clone source into directory
475
518
target = source_bzrdir.clone(self.get_url('parent/target'))
520
def test_format_initialize_on_transport_ex_stacked_on(self):
521
# trunk is a stackable format. Note that its in the same server area
522
# which is what launchpad does, but not sufficient to exercise the
524
trunk = self.make_branch('trunk', format='1.9')
525
t = self.get_transport('stacked')
526
old_fmt = controldir.format_registry.make_bzrdir('pack-0.92')
527
repo_name = old_fmt.repository_format.network_name()
528
# Should end up with a 1.9 format (stackable)
529
repo, control, require_stacking, repo_policy = \
530
old_fmt.initialize_on_transport_ex(t,
531
repo_format_name=repo_name, stacked_on='../trunk',
534
# Repositories are open write-locked
535
self.assertTrue(repo.is_write_locked())
536
self.addCleanup(repo.unlock)
538
repo = control.open_repository()
539
self.assertIsInstance(control, bzrdir.BzrDir)
540
opened = bzrdir.BzrDir.open(t.base)
541
if not isinstance(old_fmt, remote.RemoteBzrDirFormat):
542
self.assertEqual(control._format.network_name(),
543
old_fmt.network_name())
544
self.assertEqual(control._format.network_name(),
545
opened._format.network_name())
546
self.assertEqual(control.__class__, opened.__class__)
547
self.assertLength(1, repo._fallback_repositories)
477
549
def test_sprout_obeys_stacking_policy(self):
478
550
child_branch, new_child_transport = self.prepare_default_stacking()
479
551
new_child = child_branch.bzrdir.sprout(new_child_transport.base)
670
742
self.assertEqual(relpath, 'baz')
672
744
def test_open_containing_from_transport(self):
673
self.assertRaises(NotBranchError, bzrdir.BzrDir.open_containing_from_transport,
674
get_transport(self.get_readonly_url('')))
675
self.assertRaises(NotBranchError, bzrdir.BzrDir.open_containing_from_transport,
676
get_transport(self.get_readonly_url('g/p/q')))
745
self.assertRaises(NotBranchError,
746
bzrdir.BzrDir.open_containing_from_transport,
747
_mod_transport.get_transport_from_url(self.get_readonly_url('')))
748
self.assertRaises(NotBranchError,
749
bzrdir.BzrDir.open_containing_from_transport,
750
_mod_transport.get_transport_from_url(
751
self.get_readonly_url('g/p/q')))
677
752
control = bzrdir.BzrDir.create(self.get_url())
678
753
branch, relpath = bzrdir.BzrDir.open_containing_from_transport(
679
get_transport(self.get_readonly_url('')))
754
_mod_transport.get_transport_from_url(
755
self.get_readonly_url('')))
680
756
self.assertEqual('', relpath)
681
757
branch, relpath = bzrdir.BzrDir.open_containing_from_transport(
682
get_transport(self.get_readonly_url('g/p/q')))
758
_mod_transport.get_transport_from_url(
759
self.get_readonly_url('g/p/q')))
683
760
self.assertEqual('g/p/q', relpath)
685
762
def test_open_containing_tree_or_branch(self):
729
806
# transport pointing at bzrdir should give a bzrdir with root transport
730
807
# set to the given transport
731
808
control = bzrdir.BzrDir.create(self.get_url())
732
transport = get_transport(self.get_url())
733
opened_bzrdir = bzrdir.BzrDir.open_from_transport(transport)
734
self.assertEqual(transport.base, opened_bzrdir.root_transport.base)
809
t = self.get_transport()
810
opened_bzrdir = bzrdir.BzrDir.open_from_transport(t)
811
self.assertEqual(t.base, opened_bzrdir.root_transport.base)
735
812
self.assertIsInstance(opened_bzrdir, bzrdir.BzrDir)
737
814
def test_open_from_transport_no_bzrdir(self):
738
transport = get_transport(self.get_url())
739
self.assertRaises(NotBranchError, bzrdir.BzrDir.open_from_transport,
815
t = self.get_transport()
816
self.assertRaises(NotBranchError, bzrdir.BzrDir.open_from_transport, t)
742
818
def test_open_from_transport_bzrdir_in_parent(self):
743
819
control = bzrdir.BzrDir.create(self.get_url())
744
transport = get_transport(self.get_url())
745
transport.mkdir('subdir')
746
transport = transport.clone('subdir')
747
self.assertRaises(NotBranchError, bzrdir.BzrDir.open_from_transport,
820
t = self.get_transport()
822
t = t.clone('subdir')
823
self.assertRaises(NotBranchError, bzrdir.BzrDir.open_from_transport, t)
750
825
def test_sprout_recursive(self):
751
826
tree = self.make_branch_and_tree('tree1',
752
format='dirstate-with-subtree')
827
format='development-subtree')
753
828
sub_tree = self.make_branch_and_tree('tree1/subtree',
754
format='dirstate-with-subtree')
829
format='development-subtree')
755
830
sub_tree.set_root_id('subtree-root')
756
831
tree.add_reference(sub_tree)
757
832
self.build_tree(['tree1/subtree/file'])
770
845
branch = self.make_branch('branch', format='knit')
771
846
format = branch.bzrdir.cloning_metadir()
772
847
self.assertIsInstance(format.workingtree_format,
773
workingtree.WorkingTreeFormat3)
848
workingtree_4.WorkingTreeFormat6)
775
850
def test_sprout_recursive_treeless(self):
776
851
tree = self.make_branch_and_tree('tree1',
777
format='dirstate-with-subtree')
852
format='development-subtree')
778
853
sub_tree = self.make_branch_and_tree('tree1/subtree',
779
format='dirstate-with-subtree')
854
format='development-subtree')
780
855
tree.add_reference(sub_tree)
781
856
self.build_tree(['tree1/subtree/file'])
782
857
sub_tree.add('file')
783
858
tree.commit('Initial commit')
859
# The following line force the orhaning to reveal bug #634470
860
tree.branch.get_config_stack().set(
861
'bzr.transform.orphan_policy', 'move')
784
862
tree.bzrdir.destroy_workingtree()
863
# FIXME: subtree/.bzr is left here which allows the test to pass (or
864
# fail :-( ) -- vila 20100909
785
865
repo = self.make_repository('repo', shared=True,
786
format='dirstate-with-subtree')
866
format='development-subtree')
787
867
repo.set_make_working_trees(False)
788
tree.bzrdir.sprout('repo/tree2')
789
self.failUnlessExists('repo/tree2/subtree')
790
self.failIfExists('repo/tree2/subtree/file')
868
# FIXME: we just deleted the workingtree and now we want to use it ????
869
# At a minimum, we should use tree.branch below (but this fails too
870
# currently) or stop calling this test 'treeless'. Specifically, I've
871
# turn the line below into an assertRaises when 'subtree/.bzr' is
872
# orphaned and sprout tries to access the branch there (which is left
873
# by bzrdir.BzrDirMeta1.destroy_workingtree when it ignores the
874
# [DeletingParent('Not deleting', u'subtree', None)] conflict). See bug
875
# #634470. -- vila 20100909
876
self.assertRaises(errors.NotBranchError,
877
tree.bzrdir.sprout, 'repo/tree2')
878
# self.assertPathExists('repo/tree2/subtree')
879
# self.assertPathDoesNotExist('repo/tree2/subtree/file')
792
881
def make_foo_bar_baz(self):
793
882
foo = bzrdir.BzrDir.create_branch_convenience('foo').bzrdir
798
887
def test_find_bzrdirs(self):
799
888
foo, bar, baz = self.make_foo_bar_baz()
800
transport = get_transport(self.get_url())
801
self.assertEqualBzrdirs([baz, foo, bar],
802
bzrdir.BzrDir.find_bzrdirs(transport))
889
t = self.get_transport()
890
self.assertEqualBzrdirs([baz, foo, bar], bzrdir.BzrDir.find_bzrdirs(t))
892
def make_fake_permission_denied_transport(self, transport, paths):
893
"""Create a transport that raises PermissionDenied for some paths."""
896
raise errors.PermissionDenied(path)
898
path_filter_server = pathfilter.PathFilteringServer(transport, filter)
899
path_filter_server.start_server()
900
self.addCleanup(path_filter_server.stop_server)
901
path_filter_transport = pathfilter.PathFilteringTransport(
902
path_filter_server, '.')
903
return (path_filter_server, path_filter_transport)
905
def assertBranchUrlsEndWith(self, expect_url_suffix, actual_bzrdirs):
906
"""Check that each branch url ends with the given suffix."""
907
for actual_bzrdir in actual_bzrdirs:
908
self.assertEndsWith(actual_bzrdir.user_url, expect_url_suffix)
910
def test_find_bzrdirs_permission_denied(self):
911
foo, bar, baz = self.make_foo_bar_baz()
912
t = self.get_transport()
913
path_filter_server, path_filter_transport = \
914
self.make_fake_permission_denied_transport(t, ['foo'])
916
self.assertBranchUrlsEndWith('/baz/',
917
bzrdir.BzrDir.find_bzrdirs(path_filter_transport))
919
smart_transport = self.make_smart_server('.',
920
backing_server=path_filter_server)
921
self.assertBranchUrlsEndWith('/baz/',
922
bzrdir.BzrDir.find_bzrdirs(smart_transport))
804
924
def test_find_bzrdirs_list_current(self):
805
925
def list_current(transport):
806
926
return [s for s in transport.list_dir('') if s != 'baz']
808
928
foo, bar, baz = self.make_foo_bar_baz()
809
transport = get_transport(self.get_url())
810
self.assertEqualBzrdirs([foo, bar],
811
bzrdir.BzrDir.find_bzrdirs(transport,
812
list_current=list_current))
929
t = self.get_transport()
930
self.assertEqualBzrdirs(
932
bzrdir.BzrDir.find_bzrdirs(t, list_current=list_current))
815
934
def test_find_bzrdirs_evaluate(self):
816
935
def evaluate(bzrdir):
818
937
repo = bzrdir.open_repository()
819
except NoRepositoryPresent:
938
except errors.NoRepositoryPresent:
820
939
return True, bzrdir.root_transport.base
822
941
return False, bzrdir.root_transport.base
824
943
foo, bar, baz = self.make_foo_bar_baz()
825
transport = get_transport(self.get_url())
944
t = self.get_transport()
826
945
self.assertEqual([baz.root_transport.base, foo.root_transport.base],
827
list(bzrdir.BzrDir.find_bzrdirs(transport,
946
list(bzrdir.BzrDir.find_bzrdirs(t, evaluate=evaluate)))
830
948
def assertEqualBzrdirs(self, first, second):
831
949
first = list(first)
838
956
root = self.make_repository('', shared=True)
839
957
foo, bar, baz = self.make_foo_bar_baz()
840
958
qux = self.make_bzrdir('foo/qux')
841
transport = get_transport(self.get_url())
842
branches = bzrdir.BzrDir.find_branches(transport)
959
t = self.get_transport()
960
branches = bzrdir.BzrDir.find_branches(t)
843
961
self.assertEqual(baz.root_transport.base, branches[0].base)
844
962
self.assertEqual(foo.root_transport.base, branches[1].base)
845
963
self.assertEqual(bar.root_transport.base, branches[2].base)
847
965
# ensure this works without a top-level repo
848
branches = bzrdir.BzrDir.find_branches(transport.clone('foo'))
966
branches = bzrdir.BzrDir.find_branches(t.clone('foo'))
849
967
self.assertEqual(foo.root_transport.base, branches[0].base)
850
968
self.assertEqual(bar.root_transport.base, branches[1].base)
971
class TestMissingRepoBranchesSkipped(TestCaseWithMemoryTransport):
973
def test_find_bzrdirs_missing_repo(self):
974
t = self.get_transport()
975
arepo = self.make_repository('arepo', shared=True)
976
abranch_url = arepo.user_url + '/abranch'
977
abranch = bzrdir.BzrDir.create(abranch_url).create_branch()
978
t.delete_tree('arepo/.bzr')
979
self.assertRaises(errors.NoRepositoryPresent,
980
branch.Branch.open, abranch_url)
981
self.make_branch('baz')
982
for actual_bzrdir in bzrdir.BzrDir.find_branches(t):
983
self.assertEndsWith(actual_bzrdir.user_url, '/baz/')
853
986
class TestMeta1DirFormat(TestCaseWithTransport):
854
987
"""Tests specific to the meta1 dir format."""
881
1015
Metadirs should compare equal iff they have the same repo, branch and
884
mydir = bzrdir.format_registry.make_bzrdir('knit')
1018
mydir = controldir.format_registry.make_bzrdir('knit')
885
1019
self.assertEqual(mydir, mydir)
886
1020
self.assertFalse(mydir != mydir)
887
otherdir = bzrdir.format_registry.make_bzrdir('knit')
1021
otherdir = controldir.format_registry.make_bzrdir('knit')
888
1022
self.assertEqual(otherdir, mydir)
889
1023
self.assertFalse(otherdir != mydir)
890
otherdir2 = bzrdir.format_registry.make_bzrdir('dirstate-with-subtree')
1024
otherdir2 = controldir.format_registry.make_bzrdir('development-subtree')
891
1025
self.assertNotEqual(otherdir2, mydir)
892
1026
self.assertFalse(otherdir2 == mydir)
1028
def test_with_features(self):
1029
tree = self.make_branch_and_tree('tree', format='2a')
1030
tree.bzrdir.update_feature_flags({"bar": "required"})
1031
self.assertRaises(errors.MissingFeature, bzrdir.BzrDir.open, 'tree')
1032
bzrdir.BzrDirMetaFormat1.register_feature('bar')
1033
self.addCleanup(bzrdir.BzrDirMetaFormat1.unregister_feature, 'bar')
1034
dir = bzrdir.BzrDir.open('tree')
1035
self.assertEquals("required", dir._format.features.get("bar"))
1036
tree.bzrdir.update_feature_flags({"bar": None, "nonexistant": None})
1037
dir = bzrdir.BzrDir.open('tree')
1038
self.assertEquals({}, dir._format.features)
894
1040
def test_needs_conversion_different_working_tree(self):
895
1041
# meta1dirs need an conversion if any element is not the default.
896
new_format = bzrdir.format_registry.make_bzrdir('dirstate')
1042
new_format = controldir.format_registry.make_bzrdir('dirstate')
897
1043
tree = self.make_branch_and_tree('tree', format='knit')
898
1044
self.assertTrue(tree.bzrdir.needs_format_conversion(
901
1047
def test_initialize_on_format_uses_smart_transport(self):
902
1048
self.setup_smart_server_with_call_log()
903
new_format = bzrdir.format_registry.make_bzrdir('dirstate')
1049
new_format = controldir.format_registry.make_bzrdir('dirstate')
904
1050
transport = self.get_transport('target')
905
1051
transport.ensure_base()
906
1052
self.reset_smart_call_log()
915
1061
self.assertEqual(2, rpc_count)
918
class TestFormat5(TestCaseWithTransport):
919
"""Tests specific to the version 5 bzrdir format."""
921
def test_same_lockfiles_between_tree_repo_branch(self):
922
# this checks that only a single lockfiles instance is created
923
# for format 5 objects
924
dir = bzrdir.BzrDirFormat5().initialize(self.get_url())
925
def check_dir_components_use_same_lock(dir):
926
ctrl_1 = dir.open_repository().control_files
927
ctrl_2 = dir.open_branch().control_files
928
ctrl_3 = dir.open_workingtree()._control_files
929
self.assertTrue(ctrl_1 is ctrl_2)
930
self.assertTrue(ctrl_2 is ctrl_3)
931
check_dir_components_use_same_lock(dir)
932
# and if we open it normally.
933
dir = bzrdir.BzrDir.open(self.get_url())
934
check_dir_components_use_same_lock(dir)
936
def test_can_convert(self):
937
# format 5 dirs are convertable
938
dir = bzrdir.BzrDirFormat5().initialize(self.get_url())
939
self.assertTrue(dir.can_convert_format())
941
def test_needs_conversion(self):
942
# format 5 dirs need a conversion if they are not the default,
944
dir = bzrdir.BzrDirFormat5().initialize(self.get_url())
945
# don't need to convert it to itself
946
self.assertFalse(dir.needs_format_conversion(bzrdir.BzrDirFormat5()))
947
# do need to convert it to the current default
948
self.assertTrue(dir.needs_format_conversion(
949
bzrdir.BzrDirFormat.get_default_format()))
952
class TestFormat6(TestCaseWithTransport):
953
"""Tests specific to the version 6 bzrdir format."""
955
def test_same_lockfiles_between_tree_repo_branch(self):
956
# this checks that only a single lockfiles instance is created
957
# for format 6 objects
958
dir = bzrdir.BzrDirFormat6().initialize(self.get_url())
959
def check_dir_components_use_same_lock(dir):
960
ctrl_1 = dir.open_repository().control_files
961
ctrl_2 = dir.open_branch().control_files
962
ctrl_3 = dir.open_workingtree()._control_files
963
self.assertTrue(ctrl_1 is ctrl_2)
964
self.assertTrue(ctrl_2 is ctrl_3)
965
check_dir_components_use_same_lock(dir)
966
# and if we open it normally.
967
dir = bzrdir.BzrDir.open(self.get_url())
968
check_dir_components_use_same_lock(dir)
970
def test_can_convert(self):
971
# format 6 dirs are convertable
972
dir = bzrdir.BzrDirFormat6().initialize(self.get_url())
973
self.assertTrue(dir.can_convert_format())
975
def test_needs_conversion(self):
976
# format 6 dirs need an conversion if they are not the default.
977
dir = bzrdir.BzrDirFormat6().initialize(self.get_url())
978
self.assertTrue(dir.needs_format_conversion(
979
bzrdir.BzrDirFormat.get_default_format()))
982
class NotBzrDir(bzrlib.bzrdir.BzrDir):
983
"""A non .bzr based control directory."""
985
def __init__(self, transport, format):
986
self._format = format
987
self.root_transport = transport
988
self.transport = transport.clone('.not')
991
class NotBzrDirFormat(bzrlib.bzrdir.BzrDirFormat):
992
"""A test class representing any non-.bzr based disk format."""
994
def initialize_on_transport(self, transport):
995
"""Initialize a new .not dir in the base directory of a Transport."""
996
transport.mkdir('.not')
997
return self.open(transport)
999
def open(self, transport):
1000
"""Open this directory."""
1001
return NotBzrDir(transport, self)
1004
def _known_formats(self):
1005
return set([NotBzrDirFormat()])
1008
def probe_transport(self, transport):
1009
"""Our format is present if the transport ends in '.not/'."""
1010
if transport.has('.not'):
1011
return NotBzrDirFormat()
1014
class TestNotBzrDir(TestCaseWithTransport):
1015
"""Tests for using the bzrdir api with a non .bzr based disk format.
1017
If/when one of these is in the core, we can let the implementation tests
1021
def test_create_and_find_format(self):
1022
# create a .notbzr dir
1023
format = NotBzrDirFormat()
1024
dir = format.initialize(self.get_url())
1025
self.assertIsInstance(dir, NotBzrDir)
1027
bzrlib.bzrdir.BzrDirFormat.register_control_format(format)
1029
found = bzrlib.bzrdir.BzrDirFormat.find_format(
1030
get_transport(self.get_url()))
1031
self.assertIsInstance(found, NotBzrDirFormat)
1033
bzrlib.bzrdir.BzrDirFormat.unregister_control_format(format)
1035
def test_included_in_known_formats(self):
1036
bzrlib.bzrdir.BzrDirFormat.register_control_format(NotBzrDirFormat)
1038
formats = bzrlib.bzrdir.BzrDirFormat.known_formats()
1039
for format in formats:
1040
if isinstance(format, NotBzrDirFormat):
1042
self.fail("No NotBzrDirFormat in %s" % formats)
1044
bzrlib.bzrdir.BzrDirFormat.unregister_control_format(NotBzrDirFormat)
1047
1064
class NonLocalTests(TestCaseWithTransport):
1048
1065
"""Tests for bzrdir static behaviour on non local paths."""
1050
1067
def setUp(self):
1051
1068
super(NonLocalTests, self).setUp()
1052
self.vfs_transport_factory = MemoryServer
1069
self.vfs_transport_factory = memory.MemoryServer
1054
1071
def test_create_branch_convenience(self):
1055
1072
# outside a repo the default convenience output is a repo+branch_tree
1056
format = bzrdir.format_registry.make_bzrdir('knit')
1073
format = controldir.format_registry.make_bzrdir('knit')
1057
1074
branch = bzrdir.BzrDir.create_branch_convenience(
1058
1075
self.get_url('foo'), format=format)
1059
1076
self.assertRaises(errors.NoWorkingTree,
1325
1369
url = transport.base
1326
1370
err = self.assertRaises(errors.BzrError, bzrdir.BzrDir.open, url)
1327
1371
self.assertEqual('fail', err._preformatted_string)
1373
def test_post_repo_init(self):
1374
from bzrlib.controldir import RepoInitHookParams
1376
bzrdir.BzrDir.hooks.install_named_hook('post_repo_init',
1378
self.make_repository('foo')
1379
self.assertLength(1, calls)
1381
self.assertIsInstance(params, RepoInitHookParams)
1382
self.assertTrue(hasattr(params, 'bzrdir'))
1383
self.assertTrue(hasattr(params, 'repository'))
1385
def test_post_repo_init_hook_repr(self):
1387
bzrdir.BzrDir.hooks.install_named_hook('post_repo_init',
1388
lambda params: param_reprs.append(repr(params)), None)
1389
self.make_repository('foo')
1390
self.assertLength(1, param_reprs)
1391
param_repr = param_reprs[0]
1392
self.assertStartsWith(param_repr, '<RepoInitHookParams for ')
1395
class TestGenerateBackupName(TestCaseWithMemoryTransport):
1396
# FIXME: This may need to be unified with test_osutils.TestBackupNames or
1397
# moved to per_bzrdir or per_transport for better coverage ?
1401
super(TestGenerateBackupName, self).setUp()
1402
self._transport = self.get_transport()
1403
bzrdir.BzrDir.create(self.get_url(),
1404
possible_transports=[self._transport])
1405
self._bzrdir = bzrdir.BzrDir.open_from_transport(self._transport)
1408
self.assertEqual("a.~1~", self._bzrdir._available_backup_name("a"))
1410
def test_exiting(self):
1411
self._transport.put_bytes("a.~1~", "some content")
1412
self.assertEqual("a.~2~", self._bzrdir._available_backup_name("a"))
1415
class TestMeta1DirColoFormat(TestCaseWithTransport):
1416
"""Tests specific to the meta1 dir with colocated branches format."""
1418
def test_supports_colo(self):
1419
format = bzrdir.BzrDirMetaFormat1Colo()
1420
self.assertTrue(format.colocated_branches)
1422
def test_upgrade_from_2a(self):
1423
tree = self.make_branch_and_tree('.', format='2a')
1424
format = bzrdir.BzrDirMetaFormat1Colo()
1425
self.assertTrue(tree.bzrdir.needs_format_conversion(format))
1426
converter = tree.bzrdir._format.get_converter(format)
1427
result = converter.convert(tree.bzrdir, None)
1428
self.assertIsInstance(result._format, bzrdir.BzrDirMetaFormat1Colo)
1429
self.assertFalse(result.needs_format_conversion(format))
1431
def test_downgrade_to_2a(self):
1432
tree = self.make_branch_and_tree('.', format='development-colo')
1433
format = bzrdir.BzrDirMetaFormat1()
1434
self.assertTrue(tree.bzrdir.needs_format_conversion(format))
1435
converter = tree.bzrdir._format.get_converter(format)
1436
result = converter.convert(tree.bzrdir, None)
1437
self.assertIsInstance(result._format, bzrdir.BzrDirMetaFormat1)
1438
self.assertFalse(result.needs_format_conversion(format))
1440
def test_downgrade_to_2a_too_many_branches(self):
1441
tree = self.make_branch_and_tree('.', format='development-colo')
1442
tree.bzrdir.create_branch(name="another-colocated-branch")
1443
converter = tree.bzrdir._format.get_converter(
1444
bzrdir.BzrDirMetaFormat1())
1445
result = converter.convert(tree.bzrdir, bzrdir.BzrDirMetaFormat1())
1446
self.assertIsInstance(result._format, bzrdir.BzrDirMetaFormat1)
1448
def test_nested(self):
1449
tree = self.make_branch_and_tree('.', format='development-colo')
1450
tree.bzrdir.create_branch(name='foo')
1451
tree.bzrdir.create_branch(name='fool/bla')
1453
errors.ParentBranchExists, tree.bzrdir.create_branch,
1456
def test_parent(self):
1457
tree = self.make_branch_and_tree('.', format='development-colo')
1458
tree.bzrdir.create_branch(name='fool/bla')
1459
tree.bzrdir.create_branch(name='foo/bar')
1461
errors.AlreadyBranchError, tree.bzrdir.create_branch,
1465
class SampleBzrFormat(bzrdir.BzrFormat):
1468
def get_format_string(cls):
1469
return "First line\n"
1472
class TestBzrFormat(TestCase):
1473
"""Tests for BzrFormat."""
1475
def test_as_string(self):
1476
format = SampleBzrFormat()
1477
format.features = {"foo": "required"}
1478
self.assertEquals(format.as_string(),
1481
format.features["another"] = "optional"
1482
self.assertEquals(format.as_string(),
1485
"optional another\n")
1487
def test_network_name(self):
1488
# The network string should include the feature info
1489
format = SampleBzrFormat()
1490
format.features = {"foo": "required"}
1492
"First line\nrequired foo\n",
1493
format.network_name())
1495
def test_from_string_no_features(self):
1497
format = SampleBzrFormat.from_string(
1499
self.assertEquals({}, format.features)
1501
def test_from_string_with_feature(self):
1503
format = SampleBzrFormat.from_string(
1504
"First line\nrequired foo\n")
1505
self.assertEquals("required", format.features.get("foo"))
1507
def test_from_string_format_string_mismatch(self):
1508
# The first line has to match the format string
1509
self.assertRaises(AssertionError, SampleBzrFormat.from_string,
1510
"Second line\nrequired foo\n")
1512
def test_from_string_missing_space(self):
1513
# At least one space is required in the feature lines
1514
self.assertRaises(errors.ParseFormatError, SampleBzrFormat.from_string,
1515
"First line\nfoo\n")
1517
def test_from_string_with_spaces(self):
1518
# Feature with spaces (in case we add stuff like this in the future)
1519
format = SampleBzrFormat.from_string(
1520
"First line\nrequired foo with spaces\n")
1521
self.assertEquals("required", format.features.get("foo with spaces"))
1524
format1 = SampleBzrFormat()
1525
format1.features = {"nested-trees": "optional"}
1526
format2 = SampleBzrFormat()
1527
format2.features = {"nested-trees": "optional"}
1528
self.assertEquals(format1, format1)
1529
self.assertEquals(format1, format2)
1530
format3 = SampleBzrFormat()
1531
self.assertNotEquals(format1, format3)
1533
def test_check_support_status_optional(self):
1534
# Optional, so silently ignore
1535
format = SampleBzrFormat()
1536
format.features = {"nested-trees": "optional"}
1537
format.check_support_status(True)
1538
self.addCleanup(SampleBzrFormat.unregister_feature, "nested-trees")
1539
SampleBzrFormat.register_feature("nested-trees")
1540
format.check_support_status(True)
1542
def test_check_support_status_required(self):
1543
# Optional, so trigger an exception
1544
format = SampleBzrFormat()
1545
format.features = {"nested-trees": "required"}
1546
self.assertRaises(errors.MissingFeature, format.check_support_status,
1548
self.addCleanup(SampleBzrFormat.unregister_feature, "nested-trees")
1549
SampleBzrFormat.register_feature("nested-trees")
1550
format.check_support_status(True)
1552
def test_check_support_status_unknown(self):
1553
# treat unknown necessity as required
1554
format = SampleBzrFormat()
1555
format.features = {"nested-trees": "unknown"}
1556
self.assertRaises(errors.MissingFeature, format.check_support_status,
1558
self.addCleanup(SampleBzrFormat.unregister_feature, "nested-trees")
1559
SampleBzrFormat.register_feature("nested-trees")
1560
format.check_support_status(True)
1562
def test_feature_already_registered(self):
1563
# a feature can only be registered once
1564
self.addCleanup(SampleBzrFormat.unregister_feature, "nested-trees")
1565
SampleBzrFormat.register_feature("nested-trees")
1566
self.assertRaises(errors.FeatureAlreadyRegistered,
1567
SampleBzrFormat.register_feature, "nested-trees")
1569
def test_feature_with_space(self):
1570
# spaces are not allowed in feature names
1571
self.assertRaises(ValueError, SampleBzrFormat.register_feature,
added
removed
bzrlib/tests/test_bzrdir.py
