19
19
This file contains canned versions of some old trees, which are instantiated
20
20
and then upgraded to the new format."""
22
# TODO queue for upgrade:
23
# test the error message when upgrading an unknown BzrDir format.
30
branch as _mod_branch,
26
from bzrlib.selftest import TestCase, TestCaseInTempDir
36
27
from bzrlib.branch import Branch
37
28
from bzrlib.revision import is_ancestor
38
from bzrlib.tests import TestCase, TestCaseInTempDir
39
from bzrlib.transport import get_transport
40
29
from bzrlib.upgrade import upgrade
41
import bzrlib.workingtree as workingtree
30
from bzrlib.selftest.treeshape import build_tree_contents
44
33
class TestUpgrade(TestCaseInTempDir):
46
35
def test_build_tree(self):
47
36
"""Test tree-building test helper"""
48
self.build_tree_contents(_upgrade1_template)
37
build_tree_contents(_upgrade1_template)
49
38
self.failUnlessExists('foo')
50
39
self.failUnlessExists('.bzr/README')
52
41
def test_upgrade_simple(self):
53
"""Upgrade simple v0.0.4 format to latest format"""
42
"""Upgrade simple v0.0.4 format to v5"""
54
43
eq = self.assertEquals
55
self.build_tree_contents(_upgrade1_template)
57
control = bzrdir.BzrDir.open('.')
58
b = control.open_branch()
59
# tsk, peeking under the covers.
63
bzrdir.BzrDirFormat.get_default_format().__class__))
44
build_tree_contents(_upgrade1_template)
47
eq(b._branch_format, 5)
64
48
rh = b.revision_history()
66
50
['mbp@sourcefrog.net-20051004035611-176b16534b086b3c',
67
51
'mbp@sourcefrog.net-20051004035756-235f2b7dcdddd8dd'])
68
rt = b.repository.revision_tree(rh[0])
52
t = b.revision_tree(rh[0])
69
53
foo_id = 'foo-20051004035605-91e788d1875603ae'
70
eq(rt.get_file_text(foo_id), 'initial contents\n')
71
rt = b.repository.revision_tree(rh[1])
72
eq(rt.get_file_text(foo_id), 'new contents\n')
73
# check a backup was made:
74
transport = get_transport(b.base)
75
transport.stat('.bzr.backup')
76
transport.stat('.bzr.backup/README')
77
transport.stat('.bzr.backup/branch-format')
78
transport.stat('.bzr.backup/revision-history')
79
transport.stat('.bzr.backup/merged-patches')
80
transport.stat('.bzr.backup/pending-merged-patches')
81
transport.stat('.bzr.backup/pending-merges')
82
transport.stat('.bzr.backup/branch-name')
83
transport.stat('.bzr.backup/branch-lock')
84
transport.stat('.bzr.backup/inventory')
85
transport.stat('.bzr.backup/stat-cache')
86
transport.stat('.bzr.backup/text-store')
87
transport.stat('.bzr.backup/text-store/foo-20051004035611-1591048e9dc7c2d4.gz')
88
transport.stat('.bzr.backup/text-store/foo-20051004035756-4081373d897c3453.gz')
89
transport.stat('.bzr.backup/inventory-store/')
90
transport.stat('.bzr.backup/inventory-store/mbp@sourcefrog.net-20051004035611-176b16534b086b3c.gz')
91
transport.stat('.bzr.backup/inventory-store/mbp@sourcefrog.net-20051004035756-235f2b7dcdddd8dd.gz')
92
transport.stat('.bzr.backup/revision-store/')
93
transport.stat('.bzr.backup/revision-store/mbp@sourcefrog.net-20051004035611-176b16534b086b3c.gz')
94
transport.stat('.bzr.backup/revision-store/mbp@sourcefrog.net-20051004035756-235f2b7dcdddd8dd.gz')
54
eq(t.get_file_text(foo_id), 'initial contents\n')
55
t = b.revision_tree(rh[1])
56
eq(t.get_file_text(foo_id), 'new contents\n')
96
58
def test_upgrade_with_ghosts(self):
97
59
"""Upgrade v0.0.4 tree containing ghost references.
99
61
That is, some of the parents of revisions mentioned in the branch
100
aren't present in the branch's storage.
62
aren't present in the branches storage.
102
64
This shouldn't normally happen in branches created entirely in
103
bzr, but can happen in branches imported from baz and arch, or from
104
other systems, where the importer knows about a revision but not
65
bzr but can happen in imports from baz and arch, or from other
66
systems, where the importer knows about a revision but not
106
68
eq = self.assertEquals
107
self.build_tree_contents(_ghost_template)
109
b = Branch.open(u'.')
69
build_tree_contents(_ghost_template)
110
72
revision_id = b.revision_history()[1]
111
rev = b.repository.get_revision(revision_id)
73
rev = b.get_revision(revision_id)
112
74
eq(len(rev.parent_ids), 2)
113
75
eq(rev.parent_ids[1], 'wibble@wobble-2')
115
77
def test_upgrade_makes_dir_weaves(self):
116
self.build_tree_contents(_upgrade_dir_template)
117
old_repodir = bzrlib.bzrdir.BzrDir.open_unsupported('.')
118
old_repo_format = old_repodir.open_repository()._format
78
build_tree_contents(_upgrade_dir_template)
120
# this is the path to the literal file. As format changes
121
# occur it needs to be updated. FIXME: ask the store for the
123
repo = bzrlib.repository.Repository.open('.')
124
# it should have changed the format
125
self.assertNotEqual(old_repo_format.__class__, repo._format.__class__)
126
# and we should be able to read the names for the file id
127
# 'dir-20051005095101-da1441ea3fa6917a'
130
repo.text_store.get_weave(
131
'dir-20051005095101-da1441ea3fa6917a',
132
repo.get_transaction()))
134
def test_upgrade_to_meta_sets_workingtree_last_revision(self):
135
self.build_tree_contents(_upgrade_dir_template)
136
upgrade('.', bzrdir.BzrDirMetaFormat1())
137
tree = workingtree.WorkingTree.open('.')
138
self.assertEqual([tree.branch.revision_history()[-1]],
139
tree.get_parent_ids())
141
def test_upgrade_v6_to_meta_no_workingtree(self):
142
# Some format6 branches do not have checkout files. Upgrading
143
# such a branch to metadir must not setup a working tree.
144
self.build_tree_contents(_upgrade1_template)
145
upgrade('.', bzrdir.BzrDirFormat6())
146
transport = get_transport('.')
147
transport.delete_multi(['.bzr/pending-merges', '.bzr/inventory'])
148
assert not transport.has('.bzr/stat-cache')
149
# XXX: upgrade fails if a .bzr.backup is already present
150
# -- David Allouche 2006-08-11
151
transport.delete_tree('.bzr.backup')
152
# At this point, we have a format6 branch without checkout files.
153
upgrade('.', bzrdir.BzrDirMetaFormat1())
154
# The upgrade should not have set up a working tree.
155
control = bzrdir.BzrDir.open('.')
156
self.assertFalse(control.has_workingtree())
157
# We have covered the scope of this test, we may as well check that
158
# upgrade has not eaten our data, even if it's a bit redundant with
160
self.failUnless(isinstance(control._format, bzrdir.BzrDirMetaFormat1))
161
branch = control.open_branch()
162
self.assertEquals(branch.revision_history(),
163
['mbp@sourcefrog.net-20051004035611-176b16534b086b3c',
164
'mbp@sourcefrog.net-20051004035756-235f2b7dcdddd8dd'])
166
def test_convert_branch5_branch6(self):
167
branch = self.make_branch('branch', format='knit')
168
branch.set_revision_history(['AB', 'CD'])
169
branch.set_parent('file:///EF')
170
branch.set_bound_location('file:///GH')
171
branch.set_push_location('file:///IJ')
172
target = bzrdir.format_registry.make_bzrdir('experimental-branch6')
173
converter = branch.bzrdir._format.get_converter(target)
174
converter.convert(branch.bzrdir, progress.DummyProgress())
175
new_branch = _mod_branch.Branch.open(self.get_url('branch'))
176
self.assertIs(new_branch.__class__, _mod_branch.BzrBranch6)
177
self.assertEqual('CD', new_branch.last_revision())
178
self.assertEqual('file:///EF', new_branch.get_parent())
179
self.assertEqual('file:///GH', new_branch.get_bound_location())
180
branch_config = new_branch.get_config()._get_branch_data_config()
181
self.assertEqual('file:///IJ',
182
branch_config.get_user_option('push_location'))
184
branch2 = self.make_branch('branch2', format='knit')
185
converter = branch2.bzrdir._format.get_converter(target)
186
converter.convert(branch2.bzrdir, progress.DummyProgress())
80
self.failUnlessExists(
81
'.bzr/weaves/dir-20051005095101-da1441ea3fa6917a.weave')
189
84
_upgrade1_template = \
added
removed
bzrlib/selftest/test_upgrade.py
