/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

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

  • Committer: John Arbash Meinel
  • Date: 2009-12-22 16:28:47 UTC
  • mto: This revision was merged to the branch mainline in revision 4922.
  • Revision ID: john@arbash-meinel.com-20091222162847-tvnsc69to4l4uf5r
Implement a permute_for_extension helper.

Use it for all of the 'simple' extension permutations.
It basically permutes all tests in the current module, by setting TestCase.module.
Which works well for most of our extension tests. Some had more advanced
handling of permutations (extra permutations, custom vars, etc.)

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