~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

  • Committer: John Arbash Meinel
  • Date: 2007-03-14 20:15:52 UTC
  • mto: (2353.4.2 locking)
  • mto: This revision was merged to the branch mainline in revision 2360.
  • Revision ID: john@arbash-meinel.com-20070314201552-bjtfua57456dviep
Update the lock code and test code so that if more than one
lock implementation is available, they will both be tested.

It is quite a bit of overhead, for a case where we are likely to only have 1
real lock implementation per platform, but hey, for now we have 2.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006-2010 Canonical Ltd
 
1
# Copyright (C) 2006 Canonical Ltd
 
2
# Authors: Robert Collins <robert.collins@canonical.com>
 
3
# -*- coding: utf-8 -*-
2
4
#
3
5
# This program is free software; you can redistribute it and/or modify
4
6
# it under the terms of the GNU General Public License as published by
12
14
#
13
15
# You should have received a copy of the GNU General Public License
14
16
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
18
 
17
19
"""Black box tests for the upgrade ui."""
 
20
 
18
21
import os
19
 
import stat
20
22
 
21
 
from bzrlib import (
22
 
    bzrdir,
23
 
    repository,
24
 
    )
25
 
from bzrlib.tests import (
26
 
    features,
27
 
    TestCaseInTempDir,
28
 
    TestCaseWithTransport,
29
 
    )
 
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 import TestUIFactory
30
28
from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer
31
29
from bzrlib.transport import get_transport
 
30
import bzrlib.ui as ui
32
31
from bzrlib.repofmt.knitrepo import (
33
32
    RepositoryFormatKnit1,
34
33
    )
38
37
 
39
38
    def setUp(self):
40
39
        super(TestWithUpgradableBranches, self).setUp()
41
 
        self.addCleanup(bzrdir.BzrDirFormat._set_default_format,
42
 
                        bzrdir.BzrDirFormat.get_default_format())
43
 
 
44
 
    def make_current_format_branch_and_checkout(self):
 
40
        self.old_format = bzrdir.BzrDirFormat.get_default_format()
 
41
        self.old_ui_factory = ui.ui_factory
 
42
        self.addCleanup(self.restoreDefaults)
 
43
 
 
44
        ui.ui_factory = TestUIFactory()
 
45
        # setup a format 5 branch we can upgrade from.
 
46
        self.make_branch_and_tree('format_5_branch',
 
47
                                  format=bzrdir.BzrDirFormat5())
 
48
 
45
49
        current_tree = self.make_branch_and_tree('current_format_branch',
46
50
                                                 format='default')
 
51
        self.make_branch_and_tree('metadir_weave_branch', format='metaweave')
47
52
        current_tree.branch.create_checkout(
48
53
            self.get_url('current_format_checkout'), lightweight=True)
49
54
 
50
 
    def make_format_5_branch(self):
51
 
        # setup a format 5 branch we can upgrade from.
52
 
        self.make_branch_and_tree('format_5_branch',
53
 
                                  format=bzrdir.BzrDirFormat5())
54
 
 
55
 
    def make_metadir_weave_branch(self):
56
 
        self.make_branch_and_tree('metadir_weave_branch', format='metaweave')
 
55
    def restoreDefaults(self):
 
56
        ui.ui_factory = self.old_ui_factory
 
57
        bzrdir.BzrDirFormat._set_default_format(self.old_format)
57
58
 
58
59
    def test_readonly_url_error(self):
59
 
        self.make_format_5_branch()
60
 
        (out, err) = self.run_bzr(
61
 
            ['upgrade', self.get_readonly_url('format_5_branch')], retcode=3)
 
60
        (out, err) = self.run_bzr_captured(
 
61
            ['upgrade', self.get_readonly_url('format_5_branch')], 3)
62
62
        self.assertEqual(out, "")
63
63
        self.assertEqual(err, "bzr: ERROR: Upgrade URL cannot work with readonly URLs.\n")
64
64
 
65
65
    def test_upgrade_up_to_date(self):
66
 
        self.make_current_format_branch_and_checkout()
67
66
        # when up to date we should get a message to that effect
68
 
        (out, err) = self.run_bzr('upgrade current_format_branch', retcode=3)
 
67
        (out, err) = self.run_bzr_captured(
 
68
            ['upgrade', 'current_format_branch'], 3)
69
69
        self.assertEqual("", out)
70
 
        self.assertEqualDiff("bzr: ERROR: The branch format Meta "
71
 
                             "directory format 1 is already at the most "
 
70
        self.assertEqualDiff("bzr: ERROR: The branch format Bazaar-NG meta "
 
71
                             "directory, format 1 is already at the most "
72
72
                             "recent format.\n", err)
73
73
 
74
74
    def test_upgrade_up_to_date_checkout_warns_branch_left_alone(self):
75
 
        self.make_current_format_branch_and_checkout()
76
75
        # when upgrading a checkout, the branch location and a suggestion
77
 
        # to upgrade it should be emitted even if the checkout is up to
 
76
        # to upgrade it should be emitted even if the checkout is up to 
78
77
        # date
79
 
        (out, err) = self.run_bzr('upgrade current_format_checkout', retcode=3)
 
78
        (out, err) = self.run_bzr_captured(
 
79
            ['upgrade', 'current_format_checkout'], 3)
80
80
        self.assertEqual("This is a checkout. The branch (%s) needs to be "
81
 
                         "upgraded separately.\n"
 
81
                         "upgraded separately.\n" 
82
82
                         % get_transport(self.get_url('current_format_branch')).base,
83
83
                         out)
84
 
        self.assertEqualDiff("bzr: ERROR: The branch format Meta "
85
 
                             "directory format 1 is already at the most "
 
84
        self.assertEqualDiff("bzr: ERROR: The branch format Bazaar-NG meta "
 
85
                             "directory, format 1 is already at the most "
86
86
                             "recent format.\n", err)
87
87
 
88
88
    def test_upgrade_checkout(self):
90
90
        pass
91
91
 
92
92
    def test_upgrade_repository_scans_branches(self):
93
 
        # we should get individual upgrade notes for each branch even the
 
93
        # we should get individual upgrade notes for each branch even the 
94
94
        # anonymous branch
95
95
        pass
96
96
 
100
100
 
101
101
    def test_upgrade_explicit_metaformat(self):
102
102
        # users can force an upgrade to metadir format.
103
 
        self.make_format_5_branch()
104
103
        url = get_transport(self.get_url('format_5_branch')).base
105
104
        # check --format takes effect
106
105
        bzrdir.BzrDirFormat._set_default_format(bzrdir.BzrDirFormat5())
107
 
        backup_dir = 'backup.bzr.~1~'
108
 
        (out, err) = self.run_bzr(
 
106
        (out, err) = self.run_bzr_captured(
109
107
            ['upgrade', '--format=metaweave', url])
110
108
        self.assertEqualDiff("""starting upgrade of %s
111
 
making backup of %s.bzr
112
 
  to %s%s
 
109
making backup of tree history
 
110
%s.bzr has been backed up to %s.bzr.backup
 
111
if conversion fails, you can move this directory back to .bzr
 
112
if it succeeds, you can remove this directory if you wish
113
113
starting upgrade from format 5 to 6
114
114
adding prefixes to weaves
115
115
adding prefixes to revision-store
116
116
starting upgrade from format 6 to metadir
117
117
finished
118
 
""" % (url, url, url, backup_dir), out)
 
118
""" % (url, url, url), out)
119
119
        self.assertEqualDiff("", err)
120
120
        self.assertTrue(isinstance(
121
121
            bzrdir.BzrDir.open(self.get_url('format_5_branch'))._format,
122
122
            bzrdir.BzrDirMetaFormat1))
123
123
 
124
124
    def test_upgrade_explicit_knit(self):
125
 
        # users can force an upgrade to knit format from a metadir weave
 
125
        # users can force an upgrade to knit format from a metadir weave 
126
126
        # branch
127
 
        self.make_metadir_weave_branch()
128
127
        url = get_transport(self.get_url('metadir_weave_branch')).base
129
128
        # check --format takes effect
130
129
        bzrdir.BzrDirFormat._set_default_format(bzrdir.BzrDirFormat5())
131
 
        backup_dir = 'backup.bzr.~1~'
132
 
        (out, err) = self.run_bzr(
 
130
        (out, err) = self.run_bzr_captured(
133
131
            ['upgrade', '--format=knit', url])
134
132
        self.assertEqualDiff("""starting upgrade of %s
135
 
making backup of %s.bzr
136
 
  to %s%s
 
133
making backup of tree history
 
134
%s.bzr has been backed up to %s.bzr.backup
 
135
if conversion fails, you can move this directory back to .bzr
 
136
if it succeeds, you can remove this directory if you wish
137
137
starting repository conversion
138
138
repository converted
139
139
finished
140
 
""" % (url, url, url, backup_dir), out)
 
140
""" % (url, url, url), out)
141
141
        self.assertEqualDiff("", err)
142
142
        converted_dir = bzrdir.BzrDir.open(self.get_url('metadir_weave_branch'))
143
143
        self.assertTrue(isinstance(converted_dir._format,
146
146
                                   RepositoryFormatKnit1))
147
147
 
148
148
    def test_upgrade_repo(self):
149
 
        self.run_bzr('init-repository --format=metaweave repo')
150
 
        self.run_bzr('upgrade --format=knit repo')
151
 
 
152
 
    def test_upgrade_permission_check(self):
153
 
        """'backup.bzr' should retain permissions of .bzr. Bug #262450"""
154
 
        self.requireFeature(features.posix_permissions_feature)
155
 
        old_perms = stat.S_IRUSR | stat.S_IWUSR | stat.S_IXUSR
156
 
        backup_dir = 'backup.bzr.~1~'
157
 
        self.run_bzr('init --format=1.6')
158
 
        os.chmod('.bzr', old_perms)
159
 
        self.run_bzr('upgrade')
160
 
        new_perms = os.stat(backup_dir).st_mode & 0777
161
 
        self.assertTrue(new_perms == old_perms)
162
 
 
163
 
 
164
 
    def test_upgrade_with_existing_backup_dir(self):
165
 
        self.make_format_5_branch()
166
 
        transport = get_transport(self.get_url('format_5_branch'))
167
 
        url = transport.base
168
 
        bzrdir.BzrDirFormat._set_default_format(bzrdir.BzrDirFormat5())
169
 
        backup_dir1 = 'backup.bzr.~1~'
170
 
        backup_dir2 = 'backup.bzr.~2~'
171
 
        # explicitly create backup_dir1. bzr should create the .~2~ directory
172
 
        # as backup
173
 
        transport.mkdir(backup_dir1)
174
 
        (out, err) = self.run_bzr(
175
 
            ['upgrade', '--format=metaweave', url])
176
 
        self.assertEqualDiff("""starting upgrade of %s
177
 
making backup of %s.bzr
178
 
  to %s%s
179
 
starting upgrade from format 5 to 6
180
 
adding prefixes to weaves
181
 
adding prefixes to revision-store
182
 
starting upgrade from format 6 to metadir
183
 
finished
184
 
""" % (url, url, url, backup_dir2), out)
185
 
        self.assertEqualDiff("", err)
186
 
        self.assertTrue(isinstance(
187
 
            bzrdir.BzrDir.open(self.get_url('format_5_branch'))._format,
188
 
            bzrdir.BzrDirMetaFormat1))
189
 
        self.assertTrue(transport.has(backup_dir2))
 
149
        self.run_bzr('init-repository', '--format=metaweave', 'repo')
 
150
        self.run_bzr('upgrade', '--format=knit', 'repo')
 
151
 
190
152
 
191
153
class SFTPTests(TestCaseWithSFTPServer):
192
154
    """Tests for upgrade over sftp."""
193
155
 
 
156
    def setUp(self):
 
157
        super(SFTPTests, self).setUp()
 
158
        self.old_ui_factory = ui.ui_factory
 
159
        self.addCleanup(self.restoreDefaults)
 
160
 
 
161
        ui.ui_factory = TestUIFactory()
 
162
 
 
163
    def restoreDefaults(self):
 
164
        ui.ui_factory = self.old_ui_factory
 
165
 
194
166
    def test_upgrade_url(self):
195
 
        self.run_bzr('init --format=weave')
 
167
        self.run_bzr('init', '--format=weave')
196
168
        t = get_transport(self.get_url())
197
169
        url = t.base
198
 
        out, err = self.run_bzr(['upgrade', '--format=knit', url])
199
 
        backup_dir = 'backup.bzr.~1~'
 
170
        out, err = self.run_bzr('upgrade', '--format=knit', url)
200
171
        self.assertEqualDiff("""starting upgrade of %s
201
 
making backup of %s.bzr
202
 
  to %s%s
 
172
making backup of tree history
 
173
%s.bzr has been backed up to %s.bzr.backup
 
174
if conversion fails, you can move this directory back to .bzr
 
175
if it succeeds, you can remove this directory if you wish
203
176
starting upgrade from format 6 to metadir
204
177
starting repository conversion
205
178
repository converted
206
179
finished
207
 
""" % (url, url, url,backup_dir), out)
 
180
""" % (url, url, url), out)
208
181
        self.assertEqual('', err)
209
 
 
210
 
 
211
 
class UpgradeRecommendedTests(TestCaseWithTransport):
212
 
 
213
 
    def test_recommend_upgrade_wt4(self):
214
 
        # using a deprecated format gives a warning
215
 
        self.run_bzr('init --knit a')
216
 
        out, err = self.run_bzr('status a')
217
 
        self.assertContainsRe(err, 'bzr upgrade .*[/\\\\]a')
218
 
 
219
 
    def test_no_upgrade_recommendation_from_bzrdir(self):
220
 
        # we should only get a recommendation to upgrade when we're accessing
221
 
        # the actual workingtree, not when we only open a bzrdir that contains
222
 
        # an old workngtree
223
 
        self.run_bzr('init --knit a')
224
 
        out, err = self.run_bzr('revno a')
225
 
        if err.find('upgrade') > -1:
226
 
            self.fail("message shouldn't suggest upgrade:\n%s" % err)
227
 
 
228
 
    def test_upgrade_shared_repo(self):
229
 
        repo = self.make_repository('repo', format='2a', shared=True)
230
 
        branch = self.make_branch_and_tree('repo/branch', format="pack-0.92")
231
 
        self.get_transport('repo/branch/.bzr/repository').delete_tree('.')
232
 
        out, err = self.run_bzr(['upgrade'], working_dir='repo/branch')