← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/blackbox/test_upgrade.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2008-03-16 14:01:20 UTC
  • mfrom: (3280.2.5 integration)
  • Revision ID: pqm@pqm.ubuntu.com-20080316140120-i3yq8yr1l66m11h7
Start 1.4 development

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/blackbox/test_upgrade.py
1
 
# Copyright (C) 2006-2010 Canonical Ltd
 
1
# Copyright (C) 2006, 2007 Canonical Ltd
 
2
# Authors: Robert Collins <robert.collins@canonical.com>
 
3
#          and others
2
4
#
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
12
14
#
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
16
18
 
17
19
"""Black box tests for the upgrade ui."""
 
20
 
18
21
import os
19
 
import stat
20
22
 
21
23
from bzrlib import (
22
24
    bzrdir,
23
 
    controldir,
24
 
    transport,
 
25
    repository,
 
26
    ui,
25
27
    )
26
28
from bzrlib.tests import (
27
 
    features,
 
29
    TestCaseInTempDir,
28
30
    TestCaseWithTransport,
 
31
    TestUIFactory,
29
32
    )
30
33
from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer
 
34
from bzrlib.transport import get_transport
31
35
from bzrlib.repofmt.knitrepo import (
32
36
    RepositoryFormatKnit1,
33
37
    )
37
41
 
38
42
    def setUp(self):
39
43
        super(TestWithUpgradableBranches, self).setUp()
40
 
        self.addCleanup(controldir.ControlDirFormat._set_default_format,
41
 
                        controldir.ControlDirFormat.get_default_format())
42
 
 
43
 
    def make_current_format_branch_and_checkout(self):
 
44
        self.old_format = bzrdir.BzrDirFormat.get_default_format()
 
45
        self.old_ui_factory = ui.ui_factory
 
46
        self.addCleanup(self.restoreDefaults)
 
47
 
 
48
        ui.ui_factory = TestUIFactory()
 
49
        # setup a format 5 branch we can upgrade from.
 
50
        self.make_branch_and_tree('format_5_branch',
 
51
                                  format=bzrdir.BzrDirFormat5())
 
52
 
44
53
        current_tree = self.make_branch_and_tree('current_format_branch',
45
54
                                                 format='default')
 
55
        self.make_branch_and_tree('metadir_weave_branch', format='metaweave')
46
56
        current_tree.branch.create_checkout(
47
57
            self.get_url('current_format_checkout'), lightweight=True)
48
58
 
49
 
    def make_format_5_branch(self):
50
 
        # setup a format 5 branch we can upgrade from.
51
 
        self.make_branch_and_tree('format_5_branch',
52
 
                                  format=bzrdir.BzrDirFormat5())
53
 
 
54
 
    def make_metadir_weave_branch(self):
55
 
        self.make_branch_and_tree('metadir_weave_branch', format='metaweave')
 
59
    def restoreDefaults(self):
 
60
        ui.ui_factory = self.old_ui_factory
 
61
        bzrdir.BzrDirFormat._set_default_format(self.old_format)
56
62
 
57
63
    def test_readonly_url_error(self):
58
 
        self.make_format_5_branch()
59
64
        (out, err) = self.run_bzr(
60
65
            ['upgrade', self.get_readonly_url('format_5_branch')], retcode=3)
61
66
        self.assertEqual(out, "")
62
67
        self.assertEqual(err, "bzr: ERROR: Upgrade URL cannot work with readonly URLs.\n")
63
68
 
64
69
    def test_upgrade_up_to_date(self):
65
 
        self.make_current_format_branch_and_checkout()
66
70
        # when up to date we should get a message to that effect
67
71
        (out, err) = self.run_bzr('upgrade current_format_branch', retcode=3)
68
72
        self.assertEqual("", out)
69
 
        self.assertEqualDiff("bzr: ERROR: The branch format Meta "
70
 
                             "directory format 1 is already at the most "
 
73
        self.assertEqualDiff("bzr: ERROR: The branch format Bazaar-NG meta "
 
74
                             "directory, format 1 is already at the most "
71
75
                             "recent format.\n", err)
72
76
 
73
77
    def test_upgrade_up_to_date_checkout_warns_branch_left_alone(self):
74
 
        self.make_current_format_branch_and_checkout()
75
78
        # when upgrading a checkout, the branch location and a suggestion
76
 
        # to upgrade it should be emitted even if the checkout is up to
 
79
        # to upgrade it should be emitted even if the checkout is up to 
77
80
        # date
78
81
        (out, err) = self.run_bzr('upgrade current_format_checkout', retcode=3)
79
82
        self.assertEqual("This is a checkout. The branch (%s) needs to be "
80
83
                         "upgraded separately.\n"
81
 
                         % transport.get_transport(
82
 
                self.get_url('current_format_branch')).base,
 
84
                         % get_transport(self.get_url('current_format_branch')).base,
83
85
                         out)
84
 
        self.assertEqualDiff("bzr: ERROR: The branch format Meta "
85
 
                             "directory format 1 is already at the most "
 
86
        self.assertEqualDiff("bzr: ERROR: The branch format Bazaar-NG meta "
 
87
                             "directory, format 1 is already at the most "
86
88
                             "recent format.\n", err)
87
89
 
88
90
    def test_upgrade_checkout(self):
90
92
        pass
91
93
 
92
94
    def test_upgrade_repository_scans_branches(self):
93
 
        # we should get individual upgrade notes for each branch even the
 
95
        # we should get individual upgrade notes for each branch even the 
94
96
        # anonymous branch
95
97
        pass
96
98
 
100
102
 
101
103
    def test_upgrade_explicit_metaformat(self):
102
104
        # users can force an upgrade to metadir format.
103
 
        self.make_format_5_branch()
104
 
        url = transport.get_transport(self.get_url('format_5_branch')).base
 
105
        url = get_transport(self.get_url('format_5_branch')).base
105
106
        # check --format takes effect
106
 
        controldir.ControlDirFormat._set_default_format(bzrdir.BzrDirFormat5())
107
 
        backup_dir = 'backup.bzr.~1~'
 
107
        bzrdir.BzrDirFormat._set_default_format(bzrdir.BzrDirFormat5())
108
108
        (out, err) = self.run_bzr(
109
109
            ['upgrade', '--format=metaweave', url])
110
110
        self.assertEqualDiff("""starting upgrade of %s
111
 
making backup of %s.bzr
112
 
  to %s%s
 
111
making backup of tree history
 
112
%s.bzr has been backed up to %sbackup.bzr
 
113
if conversion fails, you can move this directory back to .bzr
 
114
if it succeeds, you can remove this directory if you wish
113
115
starting upgrade from format 5 to 6
114
116
adding prefixes to weaves
115
117
adding prefixes to revision-store
116
118
starting upgrade from format 6 to metadir
117
119
finished
118
 
""" % (url, url, url, backup_dir), out)
 
120
""" % (url, url, url), out)
119
121
        self.assertEqualDiff("", err)
120
122
        self.assertTrue(isinstance(
121
123
            bzrdir.BzrDir.open(self.get_url('format_5_branch'))._format,
122
124
            bzrdir.BzrDirMetaFormat1))
123
125
 
124
126
    def test_upgrade_explicit_knit(self):
125
 
        # users can force an upgrade to knit format from a metadir weave
 
127
        # users can force an upgrade to knit format from a metadir weave 
126
128
        # branch
127
 
        self.make_metadir_weave_branch()
128
 
        url = transport.get_transport(self.get_url('metadir_weave_branch')).base
 
129
        url = get_transport(self.get_url('metadir_weave_branch')).base
129
130
        # check --format takes effect
130
 
        controldir.ControlDirFormat._set_default_format(bzrdir.BzrDirFormat5())
131
 
        backup_dir = 'backup.bzr.~1~'
 
131
        bzrdir.BzrDirFormat._set_default_format(bzrdir.BzrDirFormat5())
132
132
        (out, err) = self.run_bzr(
133
133
            ['upgrade', '--format=knit', url])
134
134
        self.assertEqualDiff("""starting upgrade of %s
135
 
making backup of %s.bzr
136
 
  to %s%s
 
135
making backup of tree history
 
136
%s.bzr has been backed up to %sbackup.bzr
 
137
if conversion fails, you can move this directory back to .bzr
 
138
if it succeeds, you can remove this directory if you wish
137
139
starting repository conversion
138
140
repository converted
139
141
finished
140
 
""" % (url, url, url, backup_dir), out)
 
142
""" % (url, url, url), out)
141
143
        self.assertEqualDiff("", err)
142
144
        converted_dir = bzrdir.BzrDir.open(self.get_url('metadir_weave_branch'))
143
145
        self.assertTrue(isinstance(converted_dir._format,
149
151
        self.run_bzr('init-repository --format=metaweave repo')
150
152
        self.run_bzr('upgrade --format=knit repo')
151
153
 
152
 
    def test_upgrade_permission_check(self):
153
 
        """'backup.bzr' should retain permissions of .bzr. Bug #262450"""
154
 
        self.requireFeature(features.posix_permissions_feature)
155
 
        old_perms = stat.S_IRUSR | stat.S_IWUSR | stat.S_IXUSR
156
 
        backup_dir = 'backup.bzr.~1~'
157
 
        self.run_bzr('init --format=1.6')
158
 
        os.chmod('.bzr', old_perms)
159
 
        self.run_bzr('upgrade')
160
 
        new_perms = os.stat(backup_dir).st_mode & 0777
161
 
        self.assertTrue(new_perms == old_perms)
162
 
 
163
 
 
164
 
    def test_upgrade_with_existing_backup_dir(self):
165
 
        self.make_format_5_branch()
166
 
        t = transport.get_transport(self.get_url('format_5_branch'))
167
 
        url = t.base
168
 
        controldir.ControlDirFormat._set_default_format(bzrdir.BzrDirFormat5())
169
 
        backup_dir1 = 'backup.bzr.~1~'
170
 
        backup_dir2 = 'backup.bzr.~2~'
171
 
        # explicitly create backup_dir1. bzr should create the .~2~ directory
172
 
        # as backup
173
 
        t.mkdir(backup_dir1)
174
 
        (out, err) = self.run_bzr(
175
 
            ['upgrade', '--format=metaweave', url])
176
 
        self.assertEqualDiff("""starting upgrade of %s
177
 
making backup of %s.bzr
178
 
  to %s%s
179
 
starting upgrade from format 5 to 6
180
 
adding prefixes to weaves
181
 
adding prefixes to revision-store
182
 
starting upgrade from format 6 to metadir
183
 
finished
184
 
""" % (url, url, url, backup_dir2), out)
185
 
        self.assertEqualDiff("", err)
186
 
        self.assertTrue(isinstance(
187
 
            bzrdir.BzrDir.open(self.get_url('format_5_branch'))._format,
188
 
            bzrdir.BzrDirMetaFormat1))
189
 
        self.assertTrue(t.has(backup_dir2))
190
154
 
191
155
class SFTPTests(TestCaseWithSFTPServer):
192
156
    """Tests for upgrade over sftp."""
193
157
 
 
158
    def setUp(self):
 
159
        super(SFTPTests, self).setUp()
 
160
        self.old_ui_factory = ui.ui_factory
 
161
        self.addCleanup(self.restoreDefaults)
 
162
 
 
163
        ui.ui_factory = TestUIFactory()
 
164
 
 
165
    def restoreDefaults(self):
 
166
        ui.ui_factory = self.old_ui_factory
 
167
 
194
168
    def test_upgrade_url(self):
195
169
        self.run_bzr('init --format=weave')
196
 
        t = transport.get_transport(self.get_url())
 
170
        t = get_transport(self.get_url())
197
171
        url = t.base
198
172
        out, err = self.run_bzr(['upgrade', '--format=knit', url])
199
 
        backup_dir = 'backup.bzr.~1~'
200
173
        self.assertEqualDiff("""starting upgrade of %s
201
 
making backup of %s.bzr
202
 
  to %s%s
 
174
making backup of tree history
 
175
%s.bzr has been backed up to %sbackup.bzr
 
176
if conversion fails, you can move this directory back to .bzr
 
177
if it succeeds, you can remove this directory if you wish
203
178
starting upgrade from format 6 to metadir
204
179
starting repository conversion
205
180
repository converted
206
181
finished
207
 
""" % (url, url, url,backup_dir), out)
 
182
""" % (url, url, url), out)
208
183
        self.assertEqual('', err)
209
184
 
210
185
 
211
 
class UpgradeRecommendedTests(TestCaseWithTransport):
 
186
class UpgradeRecommendedTests(TestCaseInTempDir):
212
187
 
213
188
    def test_recommend_upgrade_wt4(self):
214
189
        # using a deprecated format gives a warning
224
199
        out, err = self.run_bzr('revno a')
225
200
        if err.find('upgrade') > -1:
226
201
            self.fail("message shouldn't suggest upgrade:\n%s" % err)
227
 
 
228
 
    def test_upgrade_shared_repo(self):
229
 
        repo = self.make_repository('repo', format='2a', shared=True)
230
 
        branch = self.make_branch_and_tree('repo/branch', format="pack-0.92")
231
 
        self.get_transport('repo/branch/.bzr/repository').delete_tree('.')
232
 
        out, err = self.run_bzr(['upgrade'], working_dir='repo/branch')