← 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: Martin Pool
  • Date: 2006-06-15 05:36:34 UTC
  • mto: This revision was merged to the branch mainline in revision 1797.
  • Revision ID: mbp@sourcefrog.net-20060615053634-4fd52ba691855659
Clean up many exception classes.

Errors indicating a user error are now shown with is_user_error on the
exception; use this rather than hardcoding a list of exceptions that should be
handled this way.

Exceptions now inherit from BzrNewException where possible to use consistent
formatting method.

Remove rather obsolete docstring test on Branch.missing_revisions.

Remove dead code from find_merge_base.


Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/blackbox/test_upgrade.py
 
1
# Copyright (C) 2006 by Canonical Ltd
 
2
# Authors: Robert Collins <robert.collins@canonical.com>
 
3
# -*- coding: utf-8 -*-
 
4
 
 
5
# This program is free software; you can redistribute it and/or modify
 
6
# it under the terms of the GNU General Public License as published by
 
7
# the Free Software Foundation; either version 2 of the License, or
 
8
# (at your option) any later version.
 
9
 
 
10
# This program is distributed in the hope that it will be useful,
 
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
13
# GNU General Public License for more details.
 
14
 
 
15
# You should have received a copy of the GNU General Public License
 
16
# along with this program; if not, write to the Free Software
 
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
18
 
 
19
"""Black box tests for the upgrade ui."""
 
20
 
 
21
import os
 
22
 
 
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.blackbox import TestUIFactory
 
28
from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer
 
29
from bzrlib.transport import get_transport
 
30
import bzrlib.ui as ui
 
31
 
 
32
 
 
33
class TestWithUpgradableBranches(TestCaseWithTransport):
 
34
 
 
35
    def setUp(self):
 
36
        super(TestWithUpgradableBranches, self).setUp()
 
37
        self.old_format = bzrdir.BzrDirFormat.get_default_format()
 
38
        self.old_repo_format = \
 
39
            bzrlib.repository.RepositoryFormat.get_default_format()
 
40
        self.old_ui_factory = ui.ui_factory
 
41
        self.addCleanup(self.restoreDefaults)
 
42
 
 
43
        ui.ui_factory = TestUIFactory()
 
44
        bzrdir.BzrDirFormat.set_default_format(bzrdir.BzrDirMetaFormat1())
 
45
        bzrlib.repository.RepositoryFormat.set_default_format(
 
46
            bzrlib.repository.RepositoryFormat7())
 
47
        # FIXME RBC 20060120 we should be able to do this via ui calls only.
 
48
        # setup a format 5 branch we can upgrade from.
 
49
        t = get_transport(self.get_url())
 
50
        t.mkdir('format_5_branch')
 
51
        bzrdir.BzrDirFormat5().initialize(self.get_url('format_5_branch'))
 
52
        bzrdir.BzrDir.create_standalone_workingtree('current_format_branch')
 
53
        d = bzrdir.BzrDir.create('metadir_weave_branch')
 
54
        d.create_repository()
 
55
        d.create_branch()
 
56
        d.create_workingtree()
 
57
        self.run_bzr('checkout',
 
58
                     '--lightweight',
 
59
                     self.get_url('current_format_branch'),
 
60
                     'current_format_checkout')
 
61
 
 
62
    def restoreDefaults(self):
 
63
        bzrdir.BzrDirFormat.set_default_format(self.old_format)
 
64
        bzrlib.repository.RepositoryFormat.set_default_format(
 
65
            self.old_repo_format)
 
66
        ui.ui_factory = self.old_ui_factory
 
67
 
 
68
    def test_readonly_url_error(self):
 
69
        (out, err) = self.run_bzr_captured(
 
70
            ['upgrade', self.get_readonly_url('format_5_branch')], 3)
 
71
        self.assertEqual(out, "")
 
72
        self.assertEqual(err, "bzr: ERROR: Upgrade URL cannot work with readonly URL's.\n")
 
73
 
 
74
    def test_upgrade_up_to_date(self):
 
75
        # when up to date we should get a message to that effect
 
76
        (out, err) = self.run_bzr_captured(
 
77
            ['upgrade', 'current_format_branch'], 3)
 
78
        self.assertEqual("", out)
 
79
        self.assertEqualDiff("bzr: ERROR: The branch format Bazaar-NG meta "
 
80
                             "directory, format 1 is already at the most "
 
81
                             "recent format.\n", err)
 
82
 
 
83
    def test_upgrade_up_to_date_checkout_warns_branch_left_alone(self):
 
84
        # when upgrading a checkout, the branch location and a suggestion
 
85
        # to upgrade it should be emitted even if the checkout is up to 
 
86
        # date
 
87
        (out, err) = self.run_bzr_captured(
 
88
            ['upgrade', 'current_format_checkout'], 3)
 
89
        self.assertEqual("This is a checkout. The branch (%s) needs to be "
 
90
                         "upgraded separately.\n" 
 
91
                         % get_transport(self.get_url('current_format_branch')).base,
 
92
                         out)
 
93
        self.assertEqualDiff("bzr: ERROR: The branch format Bazaar-NG meta "
 
94
                             "directory, format 1 is already at the most "
 
95
                             "recent format.\n", err)
 
96
 
 
97
    def test_upgrade_checkout(self):
 
98
        # upgrading a checkout should work
 
99
        pass
 
100
 
 
101
    def test_upgrade_repository_scans_branches(self):
 
102
        # we should get individual upgrade notes for each branch even the 
 
103
        # anonymous branch
 
104
        pass
 
105
 
 
106
    def test_ugrade_branch_in_repo(self):
 
107
        # upgrading a branch in a repo should warn about not upgrading the repo
 
108
        pass
 
109
 
 
110
    def test_upgrade_explicit_metaformat(self):
 
111
        # users can force an upgrade to metadir format.
 
112
        url = get_transport(self.get_url('format_5_branch')).base
 
113
        # check --format takes effect
 
114
        bzrdir.BzrDirFormat.set_default_format(bzrdir.BzrDirFormat5())
 
115
        (out, err) = self.run_bzr_captured(
 
116
            ['upgrade', '--format=metaweave', url])
 
117
        self.assertEqualDiff("""starting upgrade of %s
 
118
making backup of tree history
 
119
%s.bzr has been backed up to %s.bzr.backup
 
120
if conversion fails, you can move this directory back to .bzr
 
121
if it succeeds, you can remove this directory if you wish
 
122
starting upgrade from format 5 to 6
 
123
adding prefixes to weaves
 
124
adding prefixes to revision-store
 
125
starting upgrade from format 6 to metadir
 
126
finished
 
127
""" % (url, url, url), out)
 
128
        self.assertEqualDiff("", err)
 
129
        self.assertTrue(isinstance(
 
130
            bzrdir.BzrDir.open(self.get_url('format_5_branch'))._format,
 
131
            bzrdir.BzrDirMetaFormat1))
 
132
 
 
133
    def test_upgrade_explicit_knit(self):
 
134
        # users can force an upgrade to knit format from a metadir weave 
 
135
        # branch
 
136
        url = get_transport(self.get_url('metadir_weave_branch')).base
 
137
        # check --format takes effect
 
138
        bzrdir.BzrDirFormat.set_default_format(bzrdir.BzrDirFormat5())
 
139
        (out, err) = self.run_bzr_captured(
 
140
            ['upgrade', '--format=knit', url])
 
141
        self.assertEqualDiff("""starting upgrade of %s
 
142
making backup of tree history
 
143
%s.bzr has been backed up to %s.bzr.backup
 
144
if conversion fails, you can move this directory back to .bzr
 
145
if it succeeds, you can remove this directory if you wish
 
146
starting repository conversion
 
147
repository converted
 
148
finished
 
149
""" % (url, url, url), out)
 
150
        self.assertEqualDiff("", err)
 
151
        converted_dir = bzrdir.BzrDir.open(self.get_url('metadir_weave_branch'))
 
152
        self.assertTrue(isinstance(converted_dir._format,
 
153
                                   bzrdir.BzrDirMetaFormat1))
 
154
        self.assertTrue(isinstance(converted_dir.open_repository()._format,
 
155
                                   repository.RepositoryFormatKnit1))
 
156
 
 
157
    def test_upgrade_repo(self):
 
158
        self.run_bzr('init-repository', '--format=metaweave', 'repo')
 
159
        self.run_bzr('upgrade', '--format=knit', 'repo')
 
160
 
 
161
 
 
162
class SFTPTests(TestCaseWithSFTPServer):
 
163
    """Tests for upgrade over sftp."""
 
164
 
 
165
    def setUp(self):
 
166
        super(SFTPTests, self).setUp()
 
167
        self.old_ui_factory = ui.ui_factory
 
168
        self.addCleanup(self.restoreDefaults)
 
169
 
 
170
        ui.ui_factory = TestUIFactory()
 
171
 
 
172
    def restoreDefaults(self):
 
173
        ui.ui_factory = self.old_ui_factory
 
174
 
 
175
    def test_upgrade_url(self):
 
176
        self.run_bzr('init', '--format=weave')
 
177
        t = get_transport(self.get_url())
 
178
        url = t.base
 
179
        out, err = self.run_bzr('upgrade', '--format=knit', url)
 
180
        self.assertEqualDiff("""starting upgrade of %s
 
181
making backup of tree history
 
182
%s.bzr has been backed up to %s.bzr.backup
 
183
if conversion fails, you can move this directory back to .bzr
 
184
if it succeeds, you can remove this directory if you wish
 
185
starting upgrade from format 6 to metadir
 
186
starting repository conversion
 
187
repository converted
 
188
finished
 
189
""" % (url, url, url), out)
 
190
        self.assertEqual('', err)