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
17
17
"""Tests for upgrade of old trees.
19
This file contains canned versions of some old trees, which are instantiated
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
39
27
from bzrlib.branch import Branch
40
from bzrlib.tests import TestCaseWithTransport
28
from bzrlib.revision import is_ancestor
41
29
from bzrlib.upgrade import upgrade
44
class TestUpgrade(TestCaseWithTransport):
30
from bzrlib.selftest.treeshape import build_tree_contents
33
class TestUpgrade(TestCaseInTempDir):
35
def test_build_tree(self):
36
"""Test tree-building test helper"""
37
build_tree_contents(_upgrade1_template)
38
self.failUnlessExists('foo')
39
self.failUnlessExists('.bzr/README')
46
41
def test_upgrade_simple(self):
47
"""Upgrade simple v0.0.4 format to latest format"""
42
"""Upgrade simple v0.0.4 format to v6"""
48
43
eq = self.assertEquals
49
self.build_tree_contents(_upgrade1_template)
51
control = bzrdir.BzrDir.open('.')
52
b = control.open_branch()
53
# tsk, peeking under the covers.
57
bzrdir.BzrDirFormat.get_default_format().__class__))
44
build_tree_contents(_upgrade1_template)
47
eq(b._branch_format, 6)
58
48
rh = b.revision_history()
60
50
['mbp@sourcefrog.net-20051004035611-176b16534b086b3c',
61
51
'mbp@sourcefrog.net-20051004035756-235f2b7dcdddd8dd'])
62
rt = b.repository.revision_tree(rh[0])
52
t = b.revision_tree(rh[0])
63
53
foo_id = 'foo-20051004035605-91e788d1875603ae'
66
eq(rt.get_file_text(foo_id), 'initial contents\n')
69
rt = b.repository.revision_tree(rh[1])
72
eq(rt.get_file_text(foo_id), 'new contents\n')
75
# check a backup was made:
76
backup_dir = 'backup.bzr.~1~'
77
t = transport.get_transport(b.base)
79
t.stat(backup_dir + '/README')
80
t.stat(backup_dir + '/branch-format')
81
t.stat(backup_dir + '/revision-history')
82
t.stat(backup_dir + '/merged-patches')
83
t.stat(backup_dir + '/pending-merged-patches')
84
t.stat(backup_dir + '/pending-merges')
85
t.stat(backup_dir + '/branch-name')
86
t.stat(backup_dir + '/branch-lock')
87
t.stat(backup_dir + '/inventory')
88
t.stat(backup_dir + '/stat-cache')
89
t.stat(backup_dir + '/text-store')
90
t.stat(backup_dir + '/text-store/foo-20051004035611-1591048e9dc7c2d4.gz')
91
t.stat(backup_dir + '/text-store/foo-20051004035756-4081373d897c3453.gz')
92
t.stat(backup_dir + '/inventory-store/')
93
t.stat(backup_dir + '/inventory-store/mbp@sourcefrog.net-20051004035611-176b16534b086b3c.gz')
94
t.stat(backup_dir + '/inventory-store/mbp@sourcefrog.net-20051004035756-235f2b7dcdddd8dd.gz')
95
t.stat(backup_dir + '/revision-store/')
96
t.stat(backup_dir + '/revision-store/mbp@sourcefrog.net-20051004035611-176b16534b086b3c.gz')
97
t.stat(backup_dir + '/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')
99
58
def test_upgrade_with_ghosts(self):
100
59
"""Upgrade v0.0.4 tree containing ghost references.
102
61
That is, some of the parents of revisions mentioned in the branch
103
aren't present in the branch's storage.
62
aren't present in the branches storage.
105
This shouldn't normally happen in branches created entirely in
106
bzr, but can happen in branches imported from baz and arch, or from
107
other systems, where the importer knows about a revision but not
64
This shouldn't normally happen in branches created entirely in
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
109
68
eq = self.assertEquals
110
self.build_tree_contents(_ghost_template)
112
b = Branch.open(u'.')
69
build_tree_contents(_ghost_template)
113
72
revision_id = b.revision_history()[1]
114
rev = b.repository.get_revision(revision_id)
73
rev = b.get_revision(revision_id)
115
74
eq(len(rev.parent_ids), 2)
116
75
eq(rev.parent_ids[1], 'wibble@wobble-2')
118
77
def test_upgrade_makes_dir_weaves(self):
119
self.build_tree_contents(_upgrade_dir_template)
120
old_repodir = bzrlib.bzrdir.BzrDir.open_unsupported('.')
121
old_repo_format = old_repodir.open_repository()._format
78
build_tree_contents(_upgrade_dir_template)
123
# this is the path to the literal file. As format changes
80
# this is the path to the literal file. As format changes
124
81
# occur it needs to be updated. FIXME: ask the store for the
126
repo = bzrlib.repository.Repository.open('.')
127
# it should have changed the format
128
self.assertNotEqual(old_repo_format.__class__, repo._format.__class__)
129
# and we should be able to read the names for the file id
130
# 'dir-20051005095101-da1441ea3fa6917a'
132
self.addCleanup(repo.unlock)
133
text_keys = repo.texts.keys()
134
dir_keys = [key for key in text_keys if key[0] ==
135
'dir-20051005095101-da1441ea3fa6917a']
136
self.assertNotEqual([], dir_keys)
138
def test_upgrade_to_meta_sets_workingtree_last_revision(self):
139
self.build_tree_contents(_upgrade_dir_template)
140
upgrade('.', bzrdir.BzrDirMetaFormat1())
141
tree = workingtree.WorkingTree.open('.')
142
self.assertEqual([tree.branch.revision_history()[-1]],
143
tree.get_parent_ids())
145
def test_upgrade_v6_to_meta_no_workingtree(self):
146
# Some format6 branches do not have checkout files. Upgrading
147
# such a branch to metadir must not setup a working tree.
148
self.build_tree_contents(_upgrade1_template)
149
upgrade('.', bzrdir.BzrDirFormat6())
150
t = transport.get_transport('.')
151
t.delete_multi(['.bzr/pending-merges', '.bzr/inventory'])
152
self.assertFalse(t.has('.bzr/stat-cache'))
153
# XXX: upgrade fails if a backup.bzr is already present
154
# -- David Allouche 2006-08-11
155
t.delete_tree('backup.bzr.~1~')
156
# At this point, we have a format6 branch without checkout files.
157
upgrade('.', bzrdir.BzrDirMetaFormat1())
158
# The upgrade should not have set up a working tree.
159
control = bzrdir.BzrDir.open('.')
160
self.assertFalse(control.has_workingtree())
161
# We have covered the scope of this test, we may as well check that
162
# upgrade has not eaten our data, even if it's a bit redundant with
164
self.failUnless(isinstance(control._format, bzrdir.BzrDirMetaFormat1))
165
branch = control.open_branch()
166
self.assertEquals(branch.revision_history(),
167
['mbp@sourcefrog.net-20051004035611-176b16534b086b3c',
168
'mbp@sourcefrog.net-20051004035756-235f2b7dcdddd8dd'])
170
def test_upgrade_rich_root(self):
171
tree = self.make_branch_and_tree('tree', format='rich-root')
172
rev_id = tree.commit('first post')
175
def test_convert_branch5_branch6(self):
176
branch = self.make_branch('branch', format='knit')
177
branch.set_revision_history(['AB', 'CD'])
178
branch.set_parent('file:///EF')
179
branch.set_bound_location('file:///GH')
180
branch.set_push_location('file:///IJ')
181
target = bzrdir.format_registry.make_bzrdir('dirstate-with-subtree')
182
converter = branch.bzrdir._format.get_converter(target)
183
converter.convert(branch.bzrdir, None)
184
new_branch = _mod_branch.Branch.open(self.get_url('branch'))
185
self.assertIs(new_branch.__class__, _mod_branch.BzrBranch6)
186
self.assertEqual('CD', new_branch.last_revision())
187
self.assertEqual('file:///EF', new_branch.get_parent())
188
self.assertEqual('file:///GH', new_branch.get_bound_location())
189
branch_config = new_branch.get_config()._get_branch_data_config()
190
self.assertEqual('file:///IJ',
191
branch_config.get_user_option('push_location'))
193
branch2 = self.make_branch('branch2', format='knit')
194
converter = branch2.bzrdir._format.get_converter(target)
195
converter.convert(branch2.bzrdir, None)
196
branch2 = _mod_branch.Branch.open(self.get_url('branch'))
197
self.assertIs(branch2.__class__, _mod_branch.BzrBranch6)
199
def test_convert_branch7_branch8(self):
200
branch = self.make_branch('branch', format='1.9')
201
target = bzrdir.format_registry.make_bzrdir('1.9')
202
target.set_branch_format(_mod_branch.BzrBranchFormat8())
203
converter = branch.bzrdir._format.get_converter(target)
204
converter.convert(branch.bzrdir, None)
205
branch = _mod_branch.Branch.open(self.get_url('branch'))
206
self.assertIs(branch.__class__, _mod_branch.BzrBranch8)
207
self.assertEqual({}, branch._get_all_reference_info())
209
def test_convert_knit_dirstate_empty(self):
210
# test that asking for an upgrade from knit to dirstate works.
211
tree = self.make_branch_and_tree('tree', format='knit')
212
target = bzrdir.format_registry.make_bzrdir('dirstate')
213
converter = tree.bzrdir._format.get_converter(target)
214
converter.convert(tree.bzrdir, None)
215
new_tree = workingtree.WorkingTree.open('tree')
216
self.assertIs(new_tree.__class__, workingtree_4.WorkingTree4)
217
self.assertEqual('null:', new_tree.last_revision())
219
def test_convert_knit_dirstate_content(self):
220
# smoke test for dirstate conversion: we call dirstate primitives,
221
# and its there that the core logic is tested.
222
tree = self.make_branch_and_tree('tree', format='knit')
223
self.build_tree(['tree/file'])
224
tree.add(['file'], ['file-id'])
225
target = bzrdir.format_registry.make_bzrdir('dirstate')
226
converter = tree.bzrdir._format.get_converter(target)
227
converter.convert(tree.bzrdir, None)
228
new_tree = workingtree.WorkingTree.open('tree')
229
self.assertIs(new_tree.__class__, workingtree_4.WorkingTree4)
230
self.assertEqual('null:', new_tree.last_revision())
232
def test_convert_knit_one_parent_dirstate(self):
233
# test that asking for an upgrade from knit to dirstate works.
234
tree = self.make_branch_and_tree('tree', format='knit')
235
rev_id = tree.commit('first post')
236
target = bzrdir.format_registry.make_bzrdir('dirstate')
237
converter = tree.bzrdir._format.get_converter(target)
238
converter.convert(tree.bzrdir, None)
239
new_tree = workingtree.WorkingTree.open('tree')
240
self.assertIs(new_tree.__class__, workingtree_4.WorkingTree4)
241
self.assertEqual(rev_id, new_tree.last_revision())
242
for path in ['basis-inventory-cache', 'inventory', 'last-revision',
243
'pending-merges', 'stat-cache']:
244
self.failIfExists('tree/.bzr/checkout/' + path)
246
def test_convert_knit_merges_dirstate(self):
247
tree = self.make_branch_and_tree('tree', format='knit')
248
rev_id = tree.commit('first post')
249
merge_tree = tree.bzrdir.sprout('tree2').open_workingtree()
250
rev_id2 = tree.commit('second post')
251
rev_id3 = merge_tree.commit('second merge post')
252
tree.merge_from_branch(merge_tree.branch)
253
target = bzrdir.format_registry.make_bzrdir('dirstate')
254
converter = tree.bzrdir._format.get_converter(target)
255
converter.convert(tree.bzrdir, None)
256
new_tree = workingtree.WorkingTree.open('tree')
257
self.assertIs(new_tree.__class__, workingtree_4.WorkingTree4)
258
self.assertEqual(rev_id2, new_tree.last_revision())
259
self.assertEqual([rev_id2, rev_id3], new_tree.get_parent_ids())
260
for path in ['basis-inventory-cache', 'inventory', 'last-revision',
261
'pending-merges', 'stat-cache']:
262
self.failIfExists('tree/.bzr/checkout/' + path)
83
self.failUnlessExists(
84
'.bzr/weaves/de/dir-20051005095101-da1441ea3fa6917a.weave')
265
87
_upgrade1_template = \
added
removed
bzrlib/selftest/test_upgrade.py
