← 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: Sidnei da Silva
  • Date: 2009-05-29 14:19:29 UTC
  • mto: (4531.1.1 integration)
  • mto: This revision was merged to the branch mainline in revision 4532.
  • Revision ID: sidnei.da.silva@canonical.com-20090529141929-3heywbvj36po72a5
- Add initial config

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