1
# Copyright (C) 2006 Canonical Ltd
2
# Authors: Robert Collins <robert.collins@canonical.com>
3
# -*- coding: utf-8 -*-
5
# This program is free software; you can redistribute it and/or modify
6
# it under the terms of the GNU General Public License as published by
7
# the Free Software Foundation; either version 2 of the License, or
8
# (at your option) any later version.
10
# This program is distributed in the hope that it will be useful,
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13
# GNU General Public License for more details.
15
# You should have received a copy of the GNU General Public License
16
# along with this program; if not, write to the Free Software
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19
"""Black box tests for the upgrade ui."""
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
31
from bzrlib.repofmt.knitrepo import (
32
RepositoryFormatKnit1,
33
RepositoryFormatKnit2,
37
class TestWithUpgradableBranches(TestCaseWithTransport):
40
super(TestWithUpgradableBranches, self).setUp()
41
self.old_format = bzrdir.BzrDirFormat.get_default_format()
42
self.old_ui_factory = ui.ui_factory
43
self.addCleanup(self.restoreDefaults)
45
ui.ui_factory = TestUIFactory()
46
# setup a format 5 branch we can upgrade from.
47
self.make_branch_and_tree('format_5_branch',
48
format=bzrdir.BzrDirFormat5())
50
current_tree = self.make_branch_and_tree('current_format_branch',
52
self.make_branch_and_tree('metadir_weave_branch', format='metaweave')
53
current_tree.branch.create_checkout(
54
self.get_url('current_format_checkout'), lightweight=True)
56
def restoreDefaults(self):
57
ui.ui_factory = self.old_ui_factory
58
bzrdir.BzrDirFormat._set_default_format(self.old_format)
60
def test_readonly_url_error(self):
61
(out, err) = self.run_bzr_captured(
62
['upgrade', self.get_readonly_url('format_5_branch')], 3)
63
self.assertEqual(out, "")
64
self.assertEqual(err, "bzr: ERROR: Upgrade URL cannot work with readonly URLs.\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=metaweave', 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
RepositoryFormatKnit1))
149
def test_upgrade_repo(self):
150
self.run_bzr('init-repository', '--format=metaweave', 'repo')
151
self.run_bzr('upgrade', '--format=knit', 'repo')
154
class SFTPTests(TestCaseWithSFTPServer):
155
"""Tests for upgrade over sftp."""
158
super(SFTPTests, self).setUp()
159
self.old_ui_factory = ui.ui_factory
160
self.addCleanup(self.restoreDefaults)
162
ui.ui_factory = TestUIFactory()
164
def restoreDefaults(self):
165
ui.ui_factory = self.old_ui_factory
167
def test_upgrade_url(self):
168
self.run_bzr('init', '--format=weave')
169
t = get_transport(self.get_url())
171
out, err = self.run_bzr('upgrade', '--format=knit', url)
172
self.assertEqualDiff("""starting upgrade of %s
173
making backup of tree history
174
%s.bzr has been backed up to %s.bzr.backup
175
if conversion fails, you can move this directory back to .bzr
176
if it succeeds, you can remove this directory if you wish
177
starting upgrade from format 6 to metadir
178
starting repository conversion
181
""" % (url, url, url), out)
182
self.assertEqual('', err)
added
removed
bzrlib/tests/blackbox/test_upgrade.py
