← 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

Make a new BzrDir format that uses a versioned branch format in a branch/ subdirectory.

Show diffs side-by-side

added added

removed removed

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