23
from bzrlib.branch import Branch, BzrBranchFormat5
23
import bzrlib.bzrdir as bzrdir
24
import bzrlib.repository as repository
24
25
from bzrlib.tests import TestCaseWithTransport
26
from bzrlib.tests.blackbox import TestUIFactory
25
27
from bzrlib.transport import get_transport
28
class TestUpgrade(TestCaseWithTransport):
28
import bzrlib.ui as ui
31
class TestWithUpgradableBranches(TestCaseWithTransport):
31
super(TestUpgrade, self).setUp()
34
super(TestWithUpgradableBranches, self).setUp()
35
self.old_format = bzrdir.BzrDirFormat.get_default_format()
36
self.old_ui_factory = ui.ui_factory
37
self.addCleanup(self.restoreDefaults)
39
ui.ui_factory = TestUIFactory()
40
bzrdir.BzrDirFormat.set_default_format(bzrdir.BzrDirMetaFormat1())
32
41
# FIXME RBC 20060120 we should be able to do this via ui calls only.
33
42
# setup a format 5 branch we can upgrade from.
34
43
t = get_transport(self.get_url())
36
BzrBranchFormat5().initialize(self.get_url('old_branch'))
44
t.mkdir('format_5_branch')
45
bzrdir.BzrDirFormat5().initialize(self.get_url('format_5_branch'))
46
bzrdir.BzrDir.create_standalone_workingtree('current_format_branch')
47
d = bzrdir.BzrDir.create('metadir_weave_branch')
50
d.create_workingtree()
51
self.run_bzr('checkout',
53
self.get_url('current_format_branch'),
54
'current_format_checkout')
56
def restoreDefaults(self):
57
bzrdir.BzrDirFormat.set_default_format(self.old_format)
58
ui.ui_factory = self.old_ui_factory
38
60
def test_readonly_url_error(self):
39
61
(out, err) = self.run_bzr_captured(
40
['upgrade', self.get_readonly_url('old_branch')], 3)
62
['upgrade', self.get_readonly_url('format_5_branch')], 3)
41
63
self.assertEqual(out, "")
42
64
self.assertEqual(err, "bzr: ERROR: Upgrade URL cannot work with readonly URL's.\n")
66
def test_upgrade_up_to_date(self):
67
# when up to date we should get a message to that effect
68
(out, err) = self.run_bzr_captured(
69
['upgrade', 'current_format_branch'], 3)
70
self.assertEqual("", out)
71
self.assertEqualDiff("bzr: ERROR: The branch format Bazaar-NG meta "
72
"directory, format 1 is already at the most "
73
"recent format.\n", err)
75
def test_upgrade_up_to_date_checkout_warns_branch_left_alone(self):
76
# when upgrading a checkout, the branch location and a suggestion
77
# to upgrade it should be emitted even if the checkout is up to
79
(out, err) = self.run_bzr_captured(
80
['upgrade', 'current_format_checkout'], 3)
81
self.assertEqual("This is a checkout. The branch (%s) needs to be "
82
"upgraded separately.\n"
83
% get_transport(self.get_url('current_format_branch')).base,
85
self.assertEqualDiff("bzr: ERROR: The branch format Bazaar-NG meta "
86
"directory, format 1 is already at the most "
87
"recent format.\n", err)
89
def test_upgrade_checkout(self):
90
# upgrading a checkout should work
93
def test_upgrade_repository_scans_branches(self):
94
# we should get individual upgrade notes for each branch even the
98
def test_ugrade_branch_in_repo(self):
99
# upgrading a branch in a repo should warn about not upgrading the repo
102
def test_upgrade_explicit_metaformat(self):
103
# users can force an upgrade to metadir format.
104
url = get_transport(self.get_url('format_5_branch')).base
105
# check --format takes effect
106
bzrdir.BzrDirFormat.set_default_format(bzrdir.BzrDirFormat5())
107
(out, err) = self.run_bzr_captured(
108
['upgrade', '--format=metadir', url])
109
self.assertEqualDiff("""starting upgrade of %s
110
making backup of tree history
111
%s.bzr has been backed up to %s.bzr.backup
112
if conversion fails, you can move this directory back to .bzr
113
if it succeeds, you can remove this directory if you wish
114
starting upgrade from format 5 to 6
115
adding prefixes to weaves
116
adding prefixes to revision-store
117
starting upgrade from format 6 to metadir
119
""" % (url, url, url), out)
120
self.assertEqualDiff("", err)
121
self.assertTrue(isinstance(
122
bzrdir.BzrDir.open(self.get_url('format_5_branch'))._format,
123
bzrdir.BzrDirMetaFormat1))
125
def test_upgrade_explicit_knit(self):
126
# users can force an upgrade to knit format from a metadir weave
128
url = get_transport(self.get_url('metadir_weave_branch')).base
129
# check --format takes effect
130
bzrdir.BzrDirFormat.set_default_format(bzrdir.BzrDirFormat5())
131
(out, err) = self.run_bzr_captured(
132
['upgrade', '--format=knit', url])
133
self.assertEqualDiff("""starting upgrade of %s
134
making backup of tree history
135
%s.bzr has been backed up to %s.bzr.backup
136
if conversion fails, you can move this directory back to .bzr
137
if it succeeds, you can remove this directory if you wish
138
starting repository conversion
141
""" % (url, url, url), out)
142
self.assertEqualDiff("", err)
143
converted_dir = bzrdir.BzrDir.open(self.get_url('metadir_weave_branch'))
144
self.assertTrue(isinstance(converted_dir._format,
145
bzrdir.BzrDirMetaFormat1))
146
self.assertTrue(isinstance(converted_dir.open_repository()._format,
147
repository.RepositoryFormatKnit1))
added
removed
bzrlib/tests/blackbox/test_upgrade.py
