← 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: 2010-02-11 04:02:41 UTC
  • mfrom: (5017.2.2 tariff)
  • Revision ID: pqm@pqm.ubuntu.com-20100211040241-w6n021dz0uus341n
(mbp) add import-tariff tests

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/blackbox/test_upgrade.py
16
16
 
17
17
"""Black box tests for the upgrade ui."""
18
18
 
19
 
import os
20
 
 
21
19
from bzrlib import (
22
20
    bzrdir,
23
21
    repository,
24
 
    ui,
25
22
    )
26
23
from bzrlib.tests import (
27
24
    TestCaseInTempDir,
28
25
    TestCaseWithTransport,
29
 
    TestUIFactory,
30
26
    )
31
27
from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer
32
28
from bzrlib.transport import get_transport
39
35
 
40
36
    def setUp(self):
41
37
        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
 
 
46
 
        ui.ui_factory = TestUIFactory()
 
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):
47
48
        # setup a format 5 branch we can upgrade from.
48
49
        self.make_branch_and_tree('format_5_branch',
49
50
                                  format=bzrdir.BzrDirFormat5())
50
51
 
51
 
        current_tree = self.make_branch_and_tree('current_format_branch',
52
 
                                                 format='default')
 
52
    def make_metadir_weave_branch(self):
53
53
        self.make_branch_and_tree('metadir_weave_branch', format='metaweave')
54
 
        current_tree.branch.create_checkout(
55
 
            self.get_url('current_format_checkout'), lightweight=True)
56
 
 
57
 
    def restoreDefaults(self):
58
 
        ui.ui_factory = self.old_ui_factory
59
 
        bzrdir.BzrDirFormat._set_default_format(self.old_format)
60
54
 
61
55
    def test_readonly_url_error(self):
 
56
        self.make_format_5_branch()
62
57
        (out, err) = self.run_bzr(
63
58
            ['upgrade', self.get_readonly_url('format_5_branch')], retcode=3)
64
59
        self.assertEqual(out, "")
65
60
        self.assertEqual(err, "bzr: ERROR: Upgrade URL cannot work with readonly URLs.\n")
66
61
 
67
62
    def test_upgrade_up_to_date(self):
 
63
        self.make_current_format_branch_and_checkout()
68
64
        # when up to date we should get a message to that effect
69
65
        (out, err) = self.run_bzr('upgrade current_format_branch', retcode=3)
70
66
        self.assertEqual("", out)
73
69
                             "recent format.\n", err)
74
70
 
75
71
    def test_upgrade_up_to_date_checkout_warns_branch_left_alone(self):
 
72
        self.make_current_format_branch_and_checkout()
76
73
        # when upgrading a checkout, the branch location and a suggestion
77
74
        # to upgrade it should be emitted even if the checkout is up to
78
75
        # date
100
97
 
101
98
    def test_upgrade_explicit_metaformat(self):
102
99
        # users can force an upgrade to metadir format.
 
100
        self.make_format_5_branch()
103
101
        url = get_transport(self.get_url('format_5_branch')).base
104
102
        # check --format takes effect
105
103
        bzrdir.BzrDirFormat._set_default_format(bzrdir.BzrDirFormat5())
122
120
    def test_upgrade_explicit_knit(self):
123
121
        # users can force an upgrade to knit format from a metadir weave
124
122
        # branch
 
123
        self.make_metadir_weave_branch()
125
124
        url = get_transport(self.get_url('metadir_weave_branch')).base
126
125
        # check --format takes effect
127
126
        bzrdir.BzrDirFormat._set_default_format(bzrdir.BzrDirFormat5())
149
148
class SFTPTests(TestCaseWithSFTPServer):
150
149
    """Tests for upgrade over sftp."""
151
150
 
152
 
    def setUp(self):
153
 
        super(SFTPTests, self).setUp()
154
 
        self.old_ui_factory = ui.ui_factory
155
 
        self.addCleanup(self.restoreDefaults)
156
 
 
157
 
        ui.ui_factory = TestUIFactory()
158
 
 
159
 
    def restoreDefaults(self):
160
 
        ui.ui_factory = self.old_ui_factory
161
 
 
162
151
    def test_upgrade_url(self):
163
152
        self.run_bzr('init --format=weave')
164
153
        t = get_transport(self.get_url())
175
164
        self.assertEqual('', err)
176
165
 
177
166
 
178
 
class UpgradeRecommendedTests(TestCaseInTempDir):
 
167
class UpgradeRecommendedTests(TestCaseWithTransport):
179
168
 
180
169
    def test_recommend_upgrade_wt4(self):
181
170
        # using a deprecated format gives a warning
191
180
        out, err = self.run_bzr('revno a')
192
181
        if err.find('upgrade') > -1:
193
182
            self.fail("message shouldn't suggest upgrade:\n%s" % err)
 
183
 
 
184
    def test_upgrade_shared_repo(self):
 
185
        repo = self.make_repository('repo', format='2a', shared=True)
 
186
        branch = self.make_branch_and_tree('repo/branch', format="pack-0.92")
 
187
        self.get_transport('repo/branch/.bzr/repository').delete_tree('.')
 
188
        out, err = self.run_bzr(['upgrade'], working_dir='repo/branch')