1
# Copyright (C) 2006, 2007, 2009 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
"""Black box tests for the upgrade ui."""
26
from bzrlib.tests import (
28
TestCaseWithTransport,
31
from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer
32
from bzrlib.transport import get_transport
33
from bzrlib.repofmt.knitrepo import (
34
RepositoryFormatKnit1,
38
class TestWithUpgradableBranches(TestCaseWithTransport):
41
super(TestWithUpgradableBranches, self).setUp()
42
self.old_format = bzrdir.BzrDirFormat.get_default_format()
43
self.old_ui_factory = ui.ui_factory
44
self.addCleanup(self.restoreDefaults)
46
ui.ui_factory = TestUIFactory()
47
# setup a format 5 branch we can upgrade from.
48
self.make_branch_and_tree('format_5_branch',
49
format=bzrdir.BzrDirFormat5())
51
current_tree = self.make_branch_and_tree('current_format_branch',
53
self.make_branch_and_tree('metadir_weave_branch', format='metaweave')
54
current_tree.branch.create_checkout(
55
self.get_url('current_format_checkout'), lightweight=True)
57
def restoreDefaults(self):
58
ui.ui_factory = self.old_ui_factory
59
bzrdir.BzrDirFormat._set_default_format(self.old_format)
61
def test_readonly_url_error(self):
62
(out, err) = self.run_bzr(
63
['upgrade', self.get_readonly_url('format_5_branch')], retcode=3)
64
self.assertEqual(out, "")
65
self.assertEqual(err, "bzr: ERROR: Upgrade URL cannot work with readonly URLs.\n")
67
def test_upgrade_up_to_date(self):
68
# when up to date we should get a message to that effect
69
(out, err) = self.run_bzr('upgrade current_format_branch', retcode=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('upgrade current_format_checkout', retcode=3)
80
self.assertEqual("This is a checkout. The branch (%s) needs to be "
81
"upgraded separately.\n"
82
% get_transport(self.get_url('current_format_branch')).base,
84
self.assertEqualDiff("bzr: ERROR: The branch format Bazaar-NG meta "
85
"directory, format 1 is already at the most "
86
"recent format.\n", err)
88
def test_upgrade_checkout(self):
89
# upgrading a checkout should work
92
def test_upgrade_repository_scans_branches(self):
93
# we should get individual upgrade notes for each branch even the
97
def test_ugrade_branch_in_repo(self):
98
# upgrading a branch in a repo should warn about not upgrading the repo
101
def test_upgrade_explicit_metaformat(self):
102
# users can force an upgrade to metadir format.
103
url = get_transport(self.get_url('format_5_branch')).base
104
# check --format takes effect
105
bzrdir.BzrDirFormat._set_default_format(bzrdir.BzrDirFormat5())
106
(out, err) = self.run_bzr(
107
['upgrade', '--format=metaweave', url])
108
self.assertEqualDiff("""starting upgrade of %s
109
making backup of tree history
110
%s.bzr has been backed up to %sbackup.bzr
111
if conversion fails, you can move this directory back to .bzr
112
if it succeeds, you can remove this directory if you wish
113
starting upgrade from format 5 to 6
114
adding prefixes to weaves
115
adding prefixes to revision-store
116
starting upgrade from format 6 to metadir
118
""" % (url, url, url), out)
119
self.assertEqualDiff("", err)
120
self.assertTrue(isinstance(
121
bzrdir.BzrDir.open(self.get_url('format_5_branch'))._format,
122
bzrdir.BzrDirMetaFormat1))
124
def test_upgrade_explicit_knit(self):
125
# users can force an upgrade to knit format from a metadir weave
127
url = get_transport(self.get_url('metadir_weave_branch')).base
128
# check --format takes effect
129
bzrdir.BzrDirFormat._set_default_format(bzrdir.BzrDirFormat5())
130
(out, err) = self.run_bzr(
131
['upgrade', '--format=knit', url])
132
self.assertEqualDiff("""starting upgrade of %s
133
making backup of tree history
134
%s.bzr has been backed up to %sbackup.bzr
135
if conversion fails, you can move this directory back to .bzr
136
if it succeeds, you can remove this directory if you wish
137
starting repository conversion
140
""" % (url, url, url), out)
141
self.assertEqualDiff("", err)
142
converted_dir = bzrdir.BzrDir.open(self.get_url('metadir_weave_branch'))
143
self.assertTrue(isinstance(converted_dir._format,
144
bzrdir.BzrDirMetaFormat1))
145
self.assertTrue(isinstance(converted_dir.open_repository()._format,
146
RepositoryFormatKnit1))
148
def test_upgrade_repo(self):
149
self.run_bzr('init-repository --format=metaweave repo')
150
self.run_bzr('upgrade --format=knit repo')
153
class SFTPTests(TestCaseWithSFTPServer):
154
"""Tests for upgrade over sftp."""
157
super(SFTPTests, self).setUp()
158
self.old_ui_factory = ui.ui_factory
159
self.addCleanup(self.restoreDefaults)
161
ui.ui_factory = TestUIFactory()
163
def restoreDefaults(self):
164
ui.ui_factory = self.old_ui_factory
166
def test_upgrade_url(self):
167
self.run_bzr('init --format=weave')
168
t = get_transport(self.get_url())
170
out, err = self.run_bzr(['upgrade', '--format=knit', url])
171
self.assertEqualDiff("""starting upgrade of %s
172
making backup of tree history
173
%s.bzr has been backed up to %sbackup.bzr
174
if conversion fails, you can move this directory back to .bzr
175
if it succeeds, you can remove this directory if you wish
176
starting upgrade from format 6 to metadir
177
starting repository conversion
180
""" % (url, url, url), out)
181
self.assertEqual('', err)
184
class UpgradeRecommendedTests(TestCaseInTempDir):
186
def test_recommend_upgrade_wt4(self):
187
# using a deprecated format gives a warning
188
self.run_bzr('init --knit a')
189
out, err = self.run_bzr('status a')
190
self.assertContainsRe(err, 'bzr upgrade .*[/\\\\]a')
192
def test_no_upgrade_recommendation_from_bzrdir(self):
193
# we should only get a recommendation to upgrade when we're accessing
194
# the actual workingtree, not when we only open a bzrdir that contains
196
self.run_bzr('init --knit a')
197
out, err = self.run_bzr('revno a')
198
if err.find('upgrade') > -1:
199
self.fail("message shouldn't suggest upgrade:\n%s" % err)
added
removed
bzrlib/tests/blackbox/test_upgrade.py
