1
# Copyright (C) 2006-2011 Canonical Ltd
1
# Copyright (C) 2006 by Canonical Ltd
2
# Authors: Robert Collins <robert.collins@canonical.com>
3
# -*- coding: utf-8 -*-
3
5
# This program is free software; you can redistribute it and/or modify
4
6
# it under the terms of the GNU General Public License as published by
5
7
# the Free Software Foundation; either version 2 of the License, or
6
8
# (at your option) any later version.
8
10
# This program is distributed in the hope that it will be useful,
9
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
13
# GNU General Public License for more details.
13
15
# You should have received a copy of the GNU General Public License
14
16
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
19
"""Black box tests for the upgrade ui."""
28
from breezy.bzr import (
31
from breezy.tests import (
33
TestCaseWithTransport,
35
from breezy.tests.test_sftp_transport import TestCaseWithSFTPServer
36
from breezy.bzr.knitpack_repo import RepositoryFormatKnitPack1
39
class OldBzrDir(bzrdir.BzrDirMeta1):
40
"""An test brz dir implementation"""
42
def needs_format_conversion(self, format):
43
return not isinstance(format, self.__class__)
46
class ConvertOldTestToMeta(controldir.Converter):
47
"""A trivial converter, used for testing."""
49
def convert(self, to_convert, pb):
50
ui.ui_factory.note('starting upgrade from old test format to 2a')
51
to_convert.control_transport.put_bytes(
53
bzrdir.BzrDirMetaFormat1().get_format_string(),
54
mode=to_convert._get_file_mode())
55
return controldir.ControlDir.open(to_convert.user_url)
58
class OldBzrDirFormat(bzrdir.BzrDirMetaFormat1):
60
_lock_class = lockable_files.TransportLock
62
def get_converter(self, format=None):
63
return ConvertOldTestToMeta()
66
def get_format_string(cls):
67
return b"Ancient Test Format"
69
def _open(self, transport):
70
return OldBzrDir(transport, self)
24
import bzrlib.bzrdir as bzrdir
25
import bzrlib.repository as repository
26
from bzrlib.tests import TestCaseWithTransport
27
from bzrlib.tests.blackbox import TestUIFactory
28
from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer
29
from bzrlib.transport import get_transport
30
import bzrlib.ui as ui
73
33
class TestWithUpgradableBranches(TestCaseWithTransport):
76
36
super(TestWithUpgradableBranches, self).setUp()
78
def make_current_format_branch_and_checkout(self):
79
current_tree = self.make_branch_and_tree('current_format_branch',
81
current_tree.branch.create_checkout(
82
self.get_url('current_format_checkout'), lightweight=True)
37
self.old_format = bzrdir.BzrDirFormat.get_default_format()
38
self.old_repo_format = \
39
bzrlib.repository.RepositoryFormat.get_default_format()
40
self.old_ui_factory = ui.ui_factory
41
self.addCleanup(self.restoreDefaults)
43
ui.ui_factory = TestUIFactory()
44
bzrdir.BzrDirFormat.set_default_format(bzrdir.BzrDirMetaFormat1())
45
bzrlib.repository.RepositoryFormat.set_default_format(
46
bzrlib.repository.RepositoryFormat7())
47
# FIXME RBC 20060120 we should be able to do this via ui calls only.
48
# setup a format 5 branch we can upgrade from.
49
t = get_transport(self.get_url())
50
t.mkdir('format_5_branch')
51
bzrdir.BzrDirFormat5().initialize(self.get_url('format_5_branch'))
52
bzrdir.BzrDir.create_standalone_workingtree('current_format_branch')
53
d = bzrdir.BzrDir.create('metadir_weave_branch')
56
d.create_workingtree()
57
self.run_bzr('checkout',
59
self.get_url('current_format_branch'),
60
'current_format_checkout')
62
def restoreDefaults(self):
63
bzrdir.BzrDirFormat.set_default_format(self.old_format)
64
bzrlib.repository.RepositoryFormat.set_default_format(
66
ui.ui_factory = self.old_ui_factory
84
68
def test_readonly_url_error(self):
85
self.make_branch_and_tree("old_format_branch", format="knit")
86
(out, err) = self.run_bzr(
87
['upgrade', self.get_readonly_url("old_format_branch")], retcode=3)
88
err_msg = 'Upgrade URL cannot work with readonly URLs.'
89
self.assertEqualDiff('conversion error: %s\nbrz: ERROR: %s\n'
69
(out, err) = self.run_bzr_captured(
70
['upgrade', self.get_readonly_url('format_5_branch')], 3)
71
self.assertEqual(out, "")
72
self.assertEqual(err, "bzr: ERROR: Upgrade URL cannot work with readonly URL's.\n")
93
74
def test_upgrade_up_to_date(self):
94
self.make_current_format_branch_and_checkout()
95
75
# when up to date we should get a message to that effect
96
burl = self.get_transport('current_format_branch').local_abspath(".")
97
(out, err) = self.run_bzr('upgrade current_format_branch', retcode=0)
99
'Upgrading branch %s/ ...\n'
100
'The branch format %s is already at the most recent format.\n'
101
% (burl, 'Meta directory format 1'),
76
(out, err) = self.run_bzr_captured(
77
['upgrade', 'current_format_branch'], 3)
78
self.assertEqual("", out)
79
self.assertEqualDiff("bzr: ERROR: The branch format Bazaar-NG meta "
80
"directory, format 1 is already at the most "
81
"recent format.\n", err)
104
83
def test_upgrade_up_to_date_checkout_warns_branch_left_alone(self):
105
self.make_current_format_branch_and_checkout()
106
84
# when upgrading a checkout, the branch location and a suggestion
107
# to upgrade it should be emitted even if the checkout is up to
85
# to upgrade it should be emitted even if the checkout is up to
109
burl = self.get_transport('current_format_branch').local_abspath(".")
110
curl = self.get_transport('current_format_checkout').local_abspath(".")
111
(out, err) = self.run_bzr('upgrade current_format_checkout', retcode=0)
113
'Upgrading branch %s/ ...\nThis is a checkout.'
114
' The branch (%s/) needs to be upgraded separately.\n'
115
'The branch format %s is already at the most recent format.\n'
116
% (curl, burl, 'Meta directory format 1'),
87
(out, err) = self.run_bzr_captured(
88
['upgrade', 'current_format_checkout'], 3)
89
self.assertEqual("This is a checkout. The branch (%s) needs to be "
90
"upgraded separately.\n"
91
% get_transport(self.get_url('current_format_branch')).base,
93
self.assertEqualDiff("bzr: ERROR: The branch format Bazaar-NG meta "
94
"directory, format 1 is already at the most "
95
"recent format.\n", err)
119
97
def test_upgrade_checkout(self):
120
98
# upgrading a checkout should work
123
101
def test_upgrade_repository_scans_branches(self):
124
# we should get individual upgrade notes for each branch even the
102
# we should get individual upgrade notes for each branch even the
125
103
# anonymous branch
128
def test_upgrade_branch_in_repo(self):
106
def test_ugrade_branch_in_repo(self):
129
107
# upgrading a branch in a repo should warn about not upgrading the repo
132
def test_upgrade_control_dir(self):
133
old_format = OldBzrDirFormat()
134
self.addCleanup(bzr.BzrProber.formats.remove,
135
old_format.get_format_string())
136
bzr.BzrProber.formats.register(old_format.get_format_string(),
138
self.addCleanup(controldir.ControlDirFormat._set_default_format,
139
controldir.ControlDirFormat.get_default_format())
141
# setup an old format branch we can upgrade from.
142
path = 'old_format_branch'
143
self.make_branch_and_tree(path, format=old_format)
144
transport = self.get_transport(path)
146
display_url = transport.local_abspath('.')
110
def test_upgrade_explicit_metaformat(self):
111
# users can force an upgrade to metadir format.
112
url = get_transport(self.get_url('format_5_branch')).base
147
113
# check --format takes effect
148
controldir.ControlDirFormat._set_default_format(old_format)
149
backup_dir = 'backup.bzr.~1~'
150
(out, err) = self.run_bzr(
151
['upgrade', '--format=2a', url])
152
self.assertEqualDiff("""Upgrading branch %s/ ...
153
starting upgrade of %s/
154
making backup of %s/.bzr
156
starting upgrade from old test format to 2a
114
bzrdir.BzrDirFormat.set_default_format(bzrdir.BzrDirFormat5())
115
(out, err) = self.run_bzr_captured(
116
['upgrade', '--format=metadir', url])
117
self.assertEqualDiff("""starting upgrade of %s
118
making backup of tree history
119
%s.bzr has been backed up to %s.bzr.backup
120
if conversion fails, you can move this directory back to .bzr
121
if it succeeds, you can remove this directory if you wish
122
starting upgrade from format 5 to 6
123
adding prefixes to weaves
124
adding prefixes to revision-store
125
starting upgrade from format 6 to metadir
158
""" % (display_url, display_url, display_url, display_url, backup_dir), out)
127
""" % (url, url, url), out)
159
128
self.assertEqualDiff("", err)
160
129
self.assertTrue(isinstance(
161
controldir.ControlDir.open(self.get_url(path))._format,
130
bzrdir.BzrDir.open(self.get_url('format_5_branch'))._format,
162
131
bzrdir.BzrDirMetaFormat1))
164
133
def test_upgrade_explicit_knit(self):
165
# users can force an upgrade to knit format from a metadir pack 0.92
166
# branch to a 2a branch.
167
self.make_branch_and_tree('branch', format='knit')
168
transport = self.get_transport('branch')
170
display_url = transport.local_abspath('.')
134
# users can force an upgrade to knit format from a metadir weave
136
url = get_transport(self.get_url('metadir_weave_branch')).base
171
137
# check --format takes effect
172
backup_dir = 'backup.bzr.~1~'
173
(out, err) = self.run_bzr(
174
['upgrade', '--format=pack-0.92', url])
175
self.assertEqualDiff("""Upgrading branch %s/ ...
176
starting upgrade of %s/
177
making backup of %s/.bzr
138
bzrdir.BzrDirFormat.set_default_format(bzrdir.BzrDirFormat5())
139
(out, err) = self.run_bzr_captured(
140
['upgrade', '--format=knit', url])
141
self.assertEqualDiff("""starting upgrade of %s
142
making backup of tree history
143
%s.bzr has been backed up to %s.bzr.backup
144
if conversion fails, you can move this directory back to .bzr
145
if it succeeds, you can remove this directory if you wish
179
146
starting repository conversion
180
147
repository converted
182
""" % (display_url, display_url, display_url, display_url, backup_dir),
149
""" % (url, url, url), out)
184
150
self.assertEqualDiff("", err)
185
converted_dir = controldir.ControlDir.open(self.get_url('branch'))
151
converted_dir = bzrdir.BzrDir.open(self.get_url('metadir_weave_branch'))
186
152
self.assertTrue(isinstance(converted_dir._format,
187
153
bzrdir.BzrDirMetaFormat1))
188
154
self.assertTrue(isinstance(converted_dir.open_repository()._format,
189
RepositoryFormatKnitPack1))
155
repository.RepositoryFormatKnit1))
191
157
def test_upgrade_repo(self):
192
self.run_bzr('init-shared-repository --format=pack-0.92 repo')
193
self.run_bzr('upgrade --format=2a repo')
195
def assertLegalOption(self, option_str):
196
# Confirm that an option is legal. (Lower level tests are
197
# expected to validate the actual functionality.)
198
self.run_bzr('init --format=pack-0.92 branch-foo')
199
self.run_bzr('upgrade --format=2a branch-foo %s' % (option_str,))
201
def assertBranchFormat(self, dir, format):
202
branch = controldir.ControlDir.open_tree_or_branch(self.get_url(dir))[
204
branch_format = branch._format
205
meta_format = controldir.format_registry.make_controldir(format)
206
expected_format = meta_format.get_branch_format()
207
self.assertEqual(expected_format, branch_format)
209
def test_upgrade_clean_supported(self):
210
self.assertLegalOption('--clean')
211
self.assertBranchFormat('branch-foo', '2a')
212
backup_bzr_dir = os.path.join("branch-foo", "backup.bzr.~1~")
213
self.assertFalse(os.path.exists(backup_bzr_dir))
215
def test_upgrade_dry_run_supported(self):
216
self.assertLegalOption('--dry-run')
217
self.assertBranchFormat('branch-foo', 'pack-0.92')
219
def test_upgrade_permission_check(self):
220
"""'backup.bzr' should retain permissions of .bzr. Bug #262450"""
221
self.requireFeature(features.posix_permissions_feature)
222
old_perms = stat.S_IRUSR | stat.S_IWUSR | stat.S_IXUSR
223
backup_dir = 'backup.bzr.~1~'
224
self.run_bzr('init --format=1.6')
225
os.chmod('.bzr', old_perms)
226
self.run_bzr('upgrade')
227
new_perms = os.stat(backup_dir).st_mode & 0o777
228
self.assertTrue(new_perms == old_perms)
230
def test_upgrade_with_existing_backup_dir(self):
231
self.make_branch_and_tree("old_format_branch", format="knit")
232
t = self.get_transport("old_format_branch")
234
display_url = t.local_abspath('.')
235
backup_dir1 = 'backup.bzr.~1~'
236
backup_dir2 = 'backup.bzr.~2~'
237
# explicitly create backup_dir1. brz should create the .~2~ directory
240
(out, err) = self.run_bzr(
241
['upgrade', '--format=2a', url])
242
self.assertEqualDiff("""Upgrading branch %s/ ...
243
starting upgrade of %s/
244
making backup of %s/.bzr
246
starting repository conversion
249
""" % (display_url, display_url, display_url, display_url, backup_dir2), out)
250
self.assertEqualDiff("", err)
251
self.assertTrue(isinstance(
252
controldir.ControlDir.open(
253
self.get_url("old_format_branch"))._format,
254
bzrdir.BzrDirMetaFormat1))
255
self.assertTrue(t.has(backup_dir2))
158
self.run_bzr('init-repository', '--format=metadir', 'repo')
159
self.run_bzr('upgrade', '--format=knit', 'repo')
258
162
class SFTPTests(TestCaseWithSFTPServer):
259
163
"""Tests for upgrade over sftp."""
166
super(SFTPTests, self).setUp()
167
self.old_ui_factory = ui.ui_factory
168
self.addCleanup(self.restoreDefaults)
170
ui.ui_factory = TestUIFactory()
172
def restoreDefaults(self):
173
ui.ui_factory = self.old_ui_factory
261
175
def test_upgrade_url(self):
262
self.run_bzr('init --format=pack-0.92')
263
t = self.get_transport()
176
self.run_bzr('init', '--format=weave')
177
t = get_transport(self.get_url())
265
display_url = urlutils.unescape_for_display(url,
267
out, err = self.run_bzr(['upgrade', '--format=2a', url])
268
backup_dir = 'backup.bzr.~1~'
269
self.assertEqualDiff("""Upgrading branch %s ...
270
starting upgrade of %s
271
making backup of %s.bzr
179
out, err = self.run_bzr('upgrade', '--format=knit', url)
180
self.assertEqualDiff("""starting upgrade of %s
181
making backup of tree history
182
%s.bzr has been backed up to %s.bzr.backup
183
if conversion fails, you can move this directory back to .bzr
184
if it succeeds, you can remove this directory if you wish
185
starting upgrade from format 6 to metadir
273
186
starting repository conversion
274
187
repository converted
276
""" % (display_url, display_url, display_url, display_url, backup_dir), out)
189
""" % (url, url, url), out)
277
190
self.assertEqual('', err)
280
class UpgradeRecommendedTests(TestCaseWithTransport):
282
def test_recommend_upgrade_wt4(self):
283
# using a deprecated format gives a warning
284
self.run_bzr('init --format=knit a')
285
out, err = self.run_bzr('status a')
286
self.assertContainsRe(err, 'brz upgrade .*[/\\\\]a')
288
def test_no_upgrade_recommendation_from_bzrdir(self):
289
# we should only get a recommendation to upgrade when we're accessing
290
# the actual workingtree, not when we only open a bzrdir that contains
292
self.run_bzr('init --format=knit a')
293
out, err = self.run_bzr('revno a')
294
if err.find('upgrade') > -1:
295
self.fail("message shouldn't suggest upgrade:\n%s" % err)
297
def test_upgrade_shared_repo(self):
298
repo = self.make_repository('repo', format='2a', shared=True)
299
branch = self.make_branch_and_tree('repo/branch', format="pack-0.92")
300
self.get_transport('repo/branch/.bzr/repository').delete_tree('.')
301
out, err = self.run_bzr(['upgrade'], working_dir='repo/branch')
added
removed
bzrlib/tests/blackbox/test_upgrade.py
