← 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: Arnaud Jeansen
  • Date: 2010-03-19 23:58:06 UTC
  • mto: This revision was merged to the branch mainline in revision 5126.
  • Revision ID: arnaud.jeansen@gmail.com-20100319235806-n0owdq874qsrb12u
Go back to unified report_delta method (i.e. former TreeDelta.show())

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/blackbox/test_upgrade.py
1
 
# Copyright (C) 2006 by Canonical Ltd
2
 
# Authors: Robert Collins <robert.collins@canonical.com>
3
 
# -*- coding: utf-8 -*-
4
 
 
 
1
# Copyright (C) 2006, 2007, 2009 Canonical Ltd
 
2
#
5
3
# This program is free software; you can redistribute it and/or modify
6
4
# it under the terms of the GNU General Public License as published by
7
5
# the Free Software Foundation; either version 2 of the License, or
8
6
# (at your option) any later version.
9
 
 
 
7
#
10
8
# This program is distributed in the hope that it will be useful,
11
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
11
# GNU General Public License for more details.
14
 
 
 
12
#
15
13
# You should have received a copy of the GNU General Public License
16
14
# along with this program; if not, write to the Free Software
17
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18
16
 
19
17
"""Black box tests for the upgrade ui."""
20
18
 
21
 
import os
22
 
 
23
 
import bzrlib
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
 
19
from bzrlib import (
 
20
    bzrdir,
 
21
    repository,
 
22
    )
 
23
from bzrlib.tests import (
 
24
    TestCaseInTempDir,
 
25
    TestCaseWithTransport,
 
26
    )
28
27
from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer
29
28
from bzrlib.transport import get_transport
30
 
import bzrlib.ui as ui
 
29
from bzrlib.repofmt.knitrepo import (
 
30
    RepositoryFormatKnit1,
 
31
    )
31
32
 
32
33
 
33
34
class TestWithUpgradableBranches(TestCaseWithTransport):
34
35
 
35
36
    def setUp(self):
36
37
        super(TestWithUpgradableBranches, self).setUp()
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)
42
 
 
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.
 
38
        self.addCleanup(bzrdir.BzrDirFormat._set_default_format,
 
39
                        bzrdir.BzrDirFormat.get_default_format())
 
40
 
 
41
    def make_current_format_branch_and_checkout(self):
 
42
        current_tree = self.make_branch_and_tree('current_format_branch',
 
43
                                                 format='default')
 
44
        current_tree.branch.create_checkout(
 
45
            self.get_url('current_format_checkout'), lightweight=True)
 
46
 
 
47
    def make_format_5_branch(self):
48
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')
54
 
        d.create_repository()
55
 
        d.create_branch()
56
 
        d.create_workingtree()
57
 
        self.run_bzr('checkout',
58
 
                     '--lightweight',
59
 
                     self.get_url('current_format_branch'),
60
 
                     'current_format_checkout')
 
49
        self.make_branch_and_tree('format_5_branch',
 
50
                                  format=bzrdir.BzrDirFormat5())
61
51
 
62
 
    def restoreDefaults(self):
63
 
        bzrdir.BzrDirFormat.set_default_format(self.old_format)
64
 
        bzrlib.repository.RepositoryFormat.set_default_format(
65
 
            self.old_repo_format)
66
 
        ui.ui_factory = self.old_ui_factory
 
52
    def make_metadir_weave_branch(self):
 
53
        self.make_branch_and_tree('metadir_weave_branch', format='metaweave')
67
54
 
68
55
    def test_readonly_url_error(self):
69
 
        (out, err) = self.run_bzr_captured(
70
 
            ['upgrade', self.get_readonly_url('format_5_branch')], 3)
 
56
        self.make_format_5_branch()
 
57
        (out, err) = self.run_bzr(
 
58
            ['upgrade', self.get_readonly_url('format_5_branch')], retcode=3)
71
59
        self.assertEqual(out, "")
72
 
        self.assertEqual(err, "bzr: ERROR: Upgrade URL cannot work with readonly URL's.\n")
 
60
        self.assertEqual(err, "bzr: ERROR: Upgrade URL cannot work with readonly URLs.\n")
73
61
 
74
62
    def test_upgrade_up_to_date(self):
 
63
        self.make_current_format_branch_and_checkout()
75
64
        # when up to date we should get a message to that effect
76
 
        (out, err) = self.run_bzr_captured(
77
 
            ['upgrade', 'current_format_branch'], 3)
 
65
        (out, err) = self.run_bzr('upgrade current_format_branch', retcode=3)
78
66
        self.assertEqual("", out)
79
 
        self.assertEqualDiff("bzr: ERROR: The branch format Bazaar-NG meta "
80
 
                             "directory, format 1 is already at the most "
 
67
        self.assertEqualDiff("bzr: ERROR: The branch format Meta "
 
68
                             "directory format 1 is already at the most "
81
69
                             "recent format.\n", err)
82
70
 
83
71
    def test_upgrade_up_to_date_checkout_warns_branch_left_alone(self):
 
72
        self.make_current_format_branch_and_checkout()
84
73
        # when upgrading a checkout, the branch location and a suggestion
85
 
        # to upgrade it should be emitted even if the checkout is up to 
 
74
        # to upgrade it should be emitted even if the checkout is up to
86
75
        # date
87
 
        (out, err) = self.run_bzr_captured(
88
 
            ['upgrade', 'current_format_checkout'], 3)
 
76
        (out, err) = self.run_bzr('upgrade current_format_checkout', retcode=3)
89
77
        self.assertEqual("This is a checkout. The branch (%s) needs to be "
90
 
                         "upgraded separately.\n" 
 
78
                         "upgraded separately.\n"
91
79
                         % get_transport(self.get_url('current_format_branch')).base,
92
80
                         out)
93
 
        self.assertEqualDiff("bzr: ERROR: The branch format Bazaar-NG meta "
94
 
                             "directory, format 1 is already at the most "
 
81
        self.assertEqualDiff("bzr: ERROR: The branch format Meta "
 
82
                             "directory format 1 is already at the most "
95
83
                             "recent format.\n", err)
96
84
 
97
85
    def test_upgrade_checkout(self):
99
87
        pass
100
88
 
101
89
    def test_upgrade_repository_scans_branches(self):
102
 
        # we should get individual upgrade notes for each branch even the 
 
90
        # we should get individual upgrade notes for each branch even the
103
91
        # anonymous branch
104
92
        pass
105
93
 
109
97
 
110
98
    def test_upgrade_explicit_metaformat(self):
111
99
        # users can force an upgrade to metadir format.
 
100
        self.make_format_5_branch()
112
101
        url = get_transport(self.get_url('format_5_branch')).base
113
102
        # check --format takes effect
114
 
        bzrdir.BzrDirFormat.set_default_format(bzrdir.BzrDirFormat5())
115
 
        (out, err) = self.run_bzr_captured(
 
103
        bzrdir.BzrDirFormat._set_default_format(bzrdir.BzrDirFormat5())
 
104
        backup_dir = 'backup.bzr.~1~'
 
105
        (out, err) = self.run_bzr(
116
106
            ['upgrade', '--format=metaweave', url])
117
107
        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
 
108
making backup of %s.bzr
 
109
  to %s%s
122
110
starting upgrade from format 5 to 6
123
111
adding prefixes to weaves
124
112
adding prefixes to revision-store
125
113
starting upgrade from format 6 to metadir
126
114
finished
127
 
""" % (url, url, url), out)
 
115
""" % (url, url, url, backup_dir), out)
128
116
        self.assertEqualDiff("", err)
129
117
        self.assertTrue(isinstance(
130
118
            bzrdir.BzrDir.open(self.get_url('format_5_branch'))._format,
131
119
            bzrdir.BzrDirMetaFormat1))
132
120
 
133
121
    def test_upgrade_explicit_knit(self):
134
 
        # users can force an upgrade to knit format from a metadir weave 
 
122
        # users can force an upgrade to knit format from a metadir weave
135
123
        # branch
 
124
        self.make_metadir_weave_branch()
136
125
        url = get_transport(self.get_url('metadir_weave_branch')).base
137
126
        # check --format takes effect
138
 
        bzrdir.BzrDirFormat.set_default_format(bzrdir.BzrDirFormat5())
139
 
        (out, err) = self.run_bzr_captured(
 
127
        bzrdir.BzrDirFormat._set_default_format(bzrdir.BzrDirFormat5())
 
128
        backup_dir = 'backup.bzr.~1~'
 
129
        (out, err) = self.run_bzr(
140
130
            ['upgrade', '--format=knit', url])
141
131
        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
 
132
making backup of %s.bzr
 
133
  to %s%s
146
134
starting repository conversion
147
135
repository converted
148
136
finished
149
 
""" % (url, url, url), out)
 
137
""" % (url, url, url, backup_dir), out)
150
138
        self.assertEqualDiff("", err)
151
139
        converted_dir = bzrdir.BzrDir.open(self.get_url('metadir_weave_branch'))
152
140
        self.assertTrue(isinstance(converted_dir._format,
153
141
                                   bzrdir.BzrDirMetaFormat1))
154
142
        self.assertTrue(isinstance(converted_dir.open_repository()._format,
155
 
                                   repository.RepositoryFormatKnit1))
 
143
                                   RepositoryFormatKnit1))
156
144
 
157
145
    def test_upgrade_repo(self):
158
 
        self.run_bzr('init-repository', '--format=metaweave', 'repo')
159
 
        self.run_bzr('upgrade', '--format=knit', 'repo')
160
 
 
 
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))
161
176
 
162
177
class SFTPTests(TestCaseWithSFTPServer):
163
178
    """Tests for upgrade over sftp."""
164
179
 
165
 
    def setUp(self):
166
 
        super(SFTPTests, self).setUp()
167
 
        self.old_ui_factory = ui.ui_factory
168
 
        self.addCleanup(self.restoreDefaults)
169
 
 
170
 
        ui.ui_factory = TestUIFactory()
171
 
 
172
 
    def restoreDefaults(self):
173
 
        ui.ui_factory = self.old_ui_factory
174
 
 
175
180
    def test_upgrade_url(self):
176
 
        self.run_bzr('init', '--format=weave')
 
181
        self.run_bzr('init --format=weave')
177
182
        t = get_transport(self.get_url())
178
183
        url = t.base
179
 
        out, err = self.run_bzr('upgrade', '--format=knit', url)
 
184
        out, err = self.run_bzr(['upgrade', '--format=knit', url])
 
185
        backup_dir = 'backup.bzr.~1~'
180
186
        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
 
187
making backup of %s.bzr
 
188
  to %s%s
185
189
starting upgrade from format 6 to metadir
186
190
starting repository conversion
187
191
repository converted
188
192
finished
189
 
""" % (url, url, url), out)
 
193
""" % (url, url, url,backup_dir), out)
190
194
        self.assertEqual('', err)
 
195
 
 
196
 
 
197
class UpgradeRecommendedTests(TestCaseWithTransport):
 
198
 
 
199
    def test_recommend_upgrade_wt4(self):
 
200
        # using a deprecated format gives a warning
 
201
        self.run_bzr('init --knit a')
 
202
        out, err = self.run_bzr('status a')
 
203
        self.assertContainsRe(err, 'bzr upgrade .*[/\\\\]a')
 
204
 
 
205
    def test_no_upgrade_recommendation_from_bzrdir(self):
 
206
        # we should only get a recommendation to upgrade when we're accessing
 
207
        # the actual workingtree, not when we only open a bzrdir that contains
 
208
        # an old workngtree
 
209
        self.run_bzr('init --knit a')
 
210
        out, err = self.run_bzr('revno a')
 
211
        if err.find('upgrade') > -1:
 
212
            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')