← 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: Aaron Bentley
  • Date: 2007-06-21 23:43:17 UTC
  • mto: (2520.5.2 bzr.mpbundle)
  • mto: This revision was merged to the branch mainline in revision 2631.
  • Revision ID: abentley@panoramicfeedback.com-20070621234317-5w3h8h36oe90sups
Implement new merge directive format

Show diffs side-by-side

added added

removed removed

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