~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

  • Committer: Martin Pool
  • Date: 2006-03-23 19:00:53 UTC
  • mto: This revision was merged to the branch mainline in revision 1626.
  • Revision ID: mbp@sourcefrog.net-20060323190053-ac6b735ff74f56d7
Handle 'bzr ?', etc.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006, 2007 Canonical Ltd
 
1
# Copyright (C) 2006 by Canonical Ltd
2
2
# Authors: Robert Collins <robert.collins@canonical.com>
3
 
#          and others
4
 
#
 
3
# -*- coding: utf-8 -*-
 
4
 
5
5
# This program is free software; you can redistribute it and/or modify
6
6
# it under the terms of the GNU General Public License as published by
7
7
# the Free Software Foundation; either version 2 of the License, or
8
8
# (at your option) any later version.
9
 
#
 
9
 
10
10
# This program is distributed in the hope that it will be useful,
11
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
13
# GNU General Public License for more details.
14
 
#
 
14
 
15
15
# You should have received a copy of the GNU General Public License
16
16
# along with this program; if not, write to the Free Software
17
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
20
20
 
21
21
import os
22
22
 
23
 
from bzrlib import (
24
 
    bzrdir,
25
 
    repository,
26
 
    ui,
27
 
    )
28
 
from bzrlib.tests import (
29
 
    TestCaseInTempDir,
30
 
    TestCaseWithTransport,
31
 
    TestUIFactory,
32
 
    )
33
 
from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer
 
23
import bzrlib.bzrdir as bzrdir
 
24
import bzrlib.repository as repository
 
25
from bzrlib.tests import TestCaseWithTransport
 
26
from bzrlib.tests.blackbox import TestUIFactory
34
27
from bzrlib.transport import get_transport
35
 
from bzrlib.repofmt.knitrepo import (
36
 
    RepositoryFormatKnit1,
37
 
    )
 
28
import bzrlib.ui as ui
38
29
 
39
30
 
40
31
class TestWithUpgradableBranches(TestCaseWithTransport):
46
37
        self.addCleanup(self.restoreDefaults)
47
38
 
48
39
        ui.ui_factory = TestUIFactory()
 
40
        bzrdir.BzrDirFormat.set_default_format(bzrdir.BzrDirMetaFormat1())
 
41
        # FIXME RBC 20060120 we should be able to do this via ui calls only.
49
42
        # setup a format 5 branch we can upgrade from.
50
 
        self.make_branch_and_tree('format_5_branch',
51
 
                                  format=bzrdir.BzrDirFormat5())
52
 
 
53
 
        current_tree = self.make_branch_and_tree('current_format_branch',
54
 
                                                 format='default')
55
 
        self.make_branch_and_tree('metadir_weave_branch', format='metaweave')
56
 
        current_tree.branch.create_checkout(
57
 
            self.get_url('current_format_checkout'), lightweight=True)
 
43
        t = get_transport(self.get_url())
 
44
        t.mkdir('format_5_branch')
 
45
        bzrdir.BzrDirFormat5().initialize(self.get_url('format_5_branch'))
 
46
        bzrdir.BzrDir.create_standalone_workingtree('current_format_branch')
 
47
        d = bzrdir.BzrDir.create('metadir_weave_branch')
 
48
        d.create_repository()
 
49
        d.create_branch()
 
50
        d.create_workingtree()
 
51
        self.run_bzr('checkout',
 
52
                     '--lightweight',
 
53
                     self.get_url('current_format_branch'),
 
54
                     'current_format_checkout')
58
55
 
59
56
    def restoreDefaults(self):
 
57
        bzrdir.BzrDirFormat.set_default_format(self.old_format)
60
58
        ui.ui_factory = self.old_ui_factory
61
 
        bzrdir.BzrDirFormat._set_default_format(self.old_format)
62
59
 
63
60
    def test_readonly_url_error(self):
64
 
        (out, err) = self.run_bzr(
65
 
            ['upgrade', self.get_readonly_url('format_5_branch')], retcode=3)
 
61
        (out, err) = self.run_bzr_captured(
 
62
            ['upgrade', self.get_readonly_url('format_5_branch')], 3)
66
63
        self.assertEqual(out, "")
67
 
        self.assertEqual(err, "bzr: ERROR: Upgrade URL cannot work with readonly URLs.\n")
 
64
        self.assertEqual(err, "bzr: ERROR: Upgrade URL cannot work with readonly URL's.\n")
68
65
 
69
66
    def test_upgrade_up_to_date(self):
70
67
        # when up to date we should get a message to that effect
71
 
        (out, err) = self.run_bzr(
72
 
            ['upgrade', 'current_format_branch'], retcode=3)
 
68
        (out, err) = self.run_bzr_captured(
 
69
            ['upgrade', 'current_format_branch'], 3)
73
70
        self.assertEqual("", out)
74
71
        self.assertEqualDiff("bzr: ERROR: The branch format Bazaar-NG meta "
75
72
                             "directory, format 1 is already at the most "
79
76
        # when upgrading a checkout, the branch location and a suggestion
80
77
        # to upgrade it should be emitted even if the checkout is up to 
81
78
        # date
82
 
        (out, err) = self.run_bzr(
83
 
            ['upgrade', 'current_format_checkout'], retcode=3)
 
79
        (out, err) = self.run_bzr_captured(
 
80
            ['upgrade', 'current_format_checkout'], 3)
84
81
        self.assertEqual("This is a checkout. The branch (%s) needs to be "
85
82
                         "upgraded separately.\n" 
86
83
                         % get_transport(self.get_url('current_format_branch')).base,
106
103
        # users can force an upgrade to metadir format.
107
104
        url = get_transport(self.get_url('format_5_branch')).base
108
105
        # check --format takes effect
109
 
        bzrdir.BzrDirFormat._set_default_format(bzrdir.BzrDirFormat5())
110
 
        (out, err) = self.run_bzr(
111
 
            ['upgrade', '--format=metaweave', url])
 
106
        bzrdir.BzrDirFormat.set_default_format(bzrdir.BzrDirFormat5())
 
107
        (out, err) = self.run_bzr_captured(
 
108
            ['upgrade', '--format=metadir', url])
112
109
        self.assertEqualDiff("""starting upgrade of %s
113
110
making backup of tree history
114
111
%s.bzr has been backed up to %s.bzr.backup
130
127
        # branch
131
128
        url = get_transport(self.get_url('metadir_weave_branch')).base
132
129
        # check --format takes effect
133
 
        bzrdir.BzrDirFormat._set_default_format(bzrdir.BzrDirFormat5())
134
 
        (out, err) = self.run_bzr(
 
130
        bzrdir.BzrDirFormat.set_default_format(bzrdir.BzrDirFormat5())
 
131
        (out, err) = self.run_bzr_captured(
135
132
            ['upgrade', '--format=knit', url])
136
133
        self.assertEqualDiff("""starting upgrade of %s
137
134
making backup of tree history
147
144
        self.assertTrue(isinstance(converted_dir._format,
148
145
                                   bzrdir.BzrDirMetaFormat1))
149
146
        self.assertTrue(isinstance(converted_dir.open_repository()._format,
150
 
                                   RepositoryFormatKnit1))
151
 
 
152
 
    def test_upgrade_repo(self):
153
 
        self.run_bzr('init-repository', '--format=metaweave', 'repo')
154
 
        self.run_bzr('upgrade', '--format=knit', 'repo')
155
 
 
156
 
 
157
 
class SFTPTests(TestCaseWithSFTPServer):
158
 
    """Tests for upgrade over sftp."""
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
 
 
170
 
    def test_upgrade_url(self):
171
 
        self.run_bzr('init', '--format=weave')
172
 
        t = get_transport(self.get_url())
173
 
        url = t.base
174
 
        out, err = self.run_bzr('upgrade', '--format=knit', url)
175
 
        self.assertEqualDiff("""starting upgrade of %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
180
 
starting upgrade from format 6 to metadir
181
 
starting repository conversion
182
 
repository converted
183
 
finished
184
 
""" % (url, url, url), out)
185
 
        self.assertEqual('', err)
186
 
 
187
 
 
188
 
class UpgradeRecommendedTests(TestCaseInTempDir):
189
 
 
190
 
    def test_recommend_upgrade_wt4(self):
191
 
        # using a deprecated format gives a warning
192
 
        self.run_bzr('init', '--knit', 'a')
193
 
        out, err = self.run_bzr('status', 'a')
194
 
        self.assertContainsRe(err, 'bzr upgrade .*[/\\\\]a')
195
 
 
196
 
    def test_no_upgrade_recommendation_from_bzrdir(self):
197
 
        # we should only get a recommendation to upgrade when we're accessing
198
 
        # the actual workingtree, not when we only open a bzrdir that contains
199
 
        # an old workngtree
200
 
        self.run_bzr('init', '--knit', 'a')
201
 
        out, err = self.run_bzr('checkout', 'a', 'b')
202
 
        if err.find('upgrade') > -1:
203
 
            self.fail("message shouldn't suggest upgrade:\n%s" % err)
 
147
                                   repository.RepositoryFormatKnit1))