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.
26
from bzrlib.selftest import TestCase, TestCaseInTempDir
27
30
from bzrlib.branch import Branch
31
import bzrlib.bzrdir as bzrdir
32
import bzrlib.repository as repository
28
33
from bzrlib.revision import is_ancestor
34
from bzrlib.tests import TestCase, TestCaseInTempDir
35
from bzrlib.transport import get_transport
29
36
from bzrlib.upgrade import upgrade
30
from bzrlib.selftest.treeshape import build_tree_contents
37
import bzrlib.workingtree as workingtree
33
40
class TestUpgrade(TestCaseInTempDir):
35
42
def test_build_tree(self):
36
43
"""Test tree-building test helper"""
37
build_tree_contents(_upgrade1_template)
44
self.build_tree_contents(_upgrade1_template)
38
45
self.failUnlessExists('foo')
39
46
self.failUnlessExists('.bzr/README')
41
48
def test_upgrade_simple(self):
42
"""Upgrade simple v0.0.4 format to v5"""
49
"""Upgrade simple v0.0.4 format to latest format"""
43
50
eq = self.assertEquals
44
build_tree_contents(_upgrade1_template)
47
eq(b._branch_format, 5)
51
self.build_tree_contents(_upgrade1_template)
53
control = bzrdir.BzrDir.open('.')
54
b = control.open_branch()
55
# tsk, peeking under the covers.
59
bzrdir.BzrDirFormat.get_default_format().__class__))
48
60
rh = b.revision_history()
50
62
['mbp@sourcefrog.net-20051004035611-176b16534b086b3c',
51
63
'mbp@sourcefrog.net-20051004035756-235f2b7dcdddd8dd'])
52
t = b.revision_tree(rh[0])
64
rt = b.repository.revision_tree(rh[0])
53
65
foo_id = 'foo-20051004035605-91e788d1875603ae'
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')
66
eq(rt.get_file_text(foo_id), 'initial contents\n')
67
rt = b.repository.revision_tree(rh[1])
68
eq(rt.get_file_text(foo_id), 'new contents\n')
69
# check a backup was made:
70
transport = get_transport(b.base)
71
transport.stat('.bzr.backup')
72
transport.stat('.bzr.backup/README')
73
transport.stat('.bzr.backup/branch-format')
74
transport.stat('.bzr.backup/revision-history')
75
transport.stat('.bzr.backup/merged-patches')
76
transport.stat('.bzr.backup/pending-merged-patches')
77
transport.stat('.bzr.backup/pending-merges')
78
transport.stat('.bzr.backup/branch-name')
79
transport.stat('.bzr.backup/branch-lock')
80
transport.stat('.bzr.backup/inventory')
81
transport.stat('.bzr.backup/stat-cache')
82
transport.stat('.bzr.backup/text-store')
83
transport.stat('.bzr.backup/text-store/foo-20051004035611-1591048e9dc7c2d4.gz')
84
transport.stat('.bzr.backup/text-store/foo-20051004035756-4081373d897c3453.gz')
85
transport.stat('.bzr.backup/inventory-store/')
86
transport.stat('.bzr.backup/inventory-store/mbp@sourcefrog.net-20051004035611-176b16534b086b3c.gz')
87
transport.stat('.bzr.backup/inventory-store/mbp@sourcefrog.net-20051004035756-235f2b7dcdddd8dd.gz')
88
transport.stat('.bzr.backup/revision-store/')
89
transport.stat('.bzr.backup/revision-store/mbp@sourcefrog.net-20051004035611-176b16534b086b3c.gz')
90
transport.stat('.bzr.backup/revision-store/mbp@sourcefrog.net-20051004035756-235f2b7dcdddd8dd.gz')
58
92
def test_upgrade_with_ghosts(self):
59
93
"""Upgrade v0.0.4 tree containing ghost references.
61
95
That is, some of the parents of revisions mentioned in the branch
62
aren't present in the branches storage.
96
aren't present in the branch's storage.
64
98
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
99
bzr, but can happen in branches imported from baz and arch, or from
100
other systems, where the importer knows about a revision but not
68
102
eq = self.assertEquals
69
build_tree_contents(_ghost_template)
103
self.build_tree_contents(_ghost_template)
105
b = Branch.open(u'.')
72
106
revision_id = b.revision_history()[1]
73
rev = b.get_revision(revision_id)
107
rev = b.repository.get_revision(revision_id)
74
108
eq(len(rev.parent_ids), 2)
75
109
eq(rev.parent_ids[1], 'wibble@wobble-2')
77
111
def test_upgrade_makes_dir_weaves(self):
78
build_tree_contents(_upgrade_dir_template)
112
self.build_tree_contents(_upgrade_dir_template)
113
old_repodir = bzrlib.bzrdir.BzrDir.open_unsupported('.')
114
old_repo_format = old_repodir.open_repository()._format
80
self.failUnlessExists(
81
'.bzr/weaves/dir-20051005095101-da1441ea3fa6917a.weave')
116
# this is the path to the literal file. As format changes
117
# occur it needs to be updated. FIXME: ask the store for the
119
repo = bzrlib.repository.Repository.open('.')
120
# it should have changed the format
121
self.assertNotEqual(old_repo_format.__class__, repo._format.__class__)
122
# and we should be able to read the names for the file id
123
# 'dir-20051005095101-da1441ea3fa6917a'
126
repo.text_store.get_weave(
127
'dir-20051005095101-da1441ea3fa6917a',
128
repo.get_transaction()))
130
def test_upgrade_to_meta_sets_workingtree_last_revision(self):
131
self.build_tree_contents(_upgrade_dir_template)
132
upgrade('.', bzrdir.BzrDirMetaFormat1())
133
tree = workingtree.WorkingTree.open('.')
134
self.assertEqual([tree.branch.revision_history()[-1]],
135
tree.get_parent_ids())
137
def test_upgrade_v6_to_meta_no_workingtree(self):
138
# Some format6 branches do not have checkout files. Upgrading
139
# such a branch to metadir must not setup a working tree.
140
self.build_tree_contents(_upgrade1_template)
141
upgrade('.', bzrdir.BzrDirFormat6())
142
transport = get_transport('.')
143
transport.delete_multi(['.bzr/pending-merges', '.bzr/inventory'])
144
assert not transport.has('.bzr/stat-cache')
145
# XXX: upgrade fails if a .bzr.backup is already present
146
# -- David Allouche 2006-08-11
147
transport.delete_tree('.bzr.backup')
148
# At this point, we have a format6 branch without checkout files.
149
upgrade('.', bzrdir.BzrDirMetaFormat1())
150
# The upgrade should not have set up a working tree.
151
control = bzrdir.BzrDir.open('.')
152
self.assertFalse(control.has_workingtree())
153
# We have covered the scope of this test, we may as well check that
154
# upgrade has not eaten our data, even if it's a bit redundant with
156
self.failUnless(isinstance(control._format, bzrdir.BzrDirMetaFormat1))
157
branch = control.open_branch()
158
self.assertEquals(branch.revision_history(),
159
['mbp@sourcefrog.net-20051004035611-176b16534b086b3c',
160
'mbp@sourcefrog.net-20051004035756-235f2b7dcdddd8dd'])
84
163
_upgrade1_template = \
added
removed
bzrlib/tests/test_upgrade.py
