~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-15 22:35:35 UTC
  • mto: This revision was merged to the branch mainline in revision 2363.
  • Revision ID: john@arbash-meinel.com-20070315223535-d3d4964oe1hc8zhg
Add an overzealous test, for Unicode support of _iter_changes.
For both knowns and unknowns.
And include a basic, if suboptimal, fix.
I would rather defer the decoding until we've determined that we are going to return the tuple.
There is still something broken with added files, but I'll get to that.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2006 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 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
from bzrlib.repofmt.knitrepo import (
 
32
    RepositoryFormatKnit1,
 
33
    )
 
34
 
 
35
 
 
36
class TestWithUpgradableBranches(TestCaseWithTransport):
 
37
 
 
38
    def setUp(self):
 
39
        super(TestWithUpgradableBranches, self).setUp()
 
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
 
 
49
        current_tree = self.make_branch_and_tree('current_format_branch',
 
50
                                                 format='default')
 
51
        self.make_branch_and_tree('metadir_weave_branch', format='metaweave')
 
52
        current_tree.branch.create_checkout(
 
53
            self.get_url('current_format_checkout'), lightweight=True)
 
54
 
 
55
    def restoreDefaults(self):
 
56
        ui.ui_factory = self.old_ui_factory
 
57
        bzrdir.BzrDirFormat._set_default_format(self.old_format)
 
58
 
 
59
    def test_readonly_url_error(self):
 
60
        (out, err) = self.run_bzr_captured(
 
61
            ['upgrade', self.get_readonly_url('format_5_branch')], 3)
 
62
        self.assertEqual(out, "")
 
63
        self.assertEqual(err, "bzr: ERROR: Upgrade URL cannot work with readonly URLs.\n")
 
64
 
 
65
    def test_upgrade_up_to_date(self):
 
66
        # when up to date we should get a message to that effect
 
67
        (out, err) = self.run_bzr_captured(
 
68
            ['upgrade', 'current_format_branch'], 3)
 
69
        self.assertEqual("", out)
 
70
        self.assertEqualDiff("bzr: ERROR: The branch format Bazaar-NG meta "
 
71
                             "directory, format 1 is already at the most "
 
72
                             "recent format.\n", err)
 
73
 
 
74
    def test_upgrade_up_to_date_checkout_warns_branch_left_alone(self):
 
75
        # when upgrading a checkout, the branch location and a suggestion
 
76
        # to upgrade it should be emitted even if the checkout is up to 
 
77
        # date
 
78
        (out, err) = self.run_bzr_captured(
 
79
            ['upgrade', 'current_format_checkout'], 3)
 
80
        self.assertEqual("This is a checkout. The branch (%s) needs to be "
 
81
                         "upgraded separately.\n" 
 
82
                         % get_transport(self.get_url('current_format_branch')).base,
 
83
                         out)
 
84
        self.assertEqualDiff("bzr: ERROR: The branch format Bazaar-NG meta "
 
85
                             "directory, format 1 is already at the most "
 
86
                             "recent format.\n", err)
 
87
 
 
88
    def test_upgrade_checkout(self):
 
89
        # upgrading a checkout should work
 
90
        pass
 
91
 
 
92
    def test_upgrade_repository_scans_branches(self):
 
93
        # we should get individual upgrade notes for each branch even the 
 
94
        # anonymous branch
 
95
        pass
 
96
 
 
97
    def test_ugrade_branch_in_repo(self):
 
98
        # upgrading a branch in a repo should warn about not upgrading the repo
 
99
        pass
 
100
 
 
101
    def test_upgrade_explicit_metaformat(self):
 
102
        # users can force an upgrade to metadir format.
 
103
        url = get_transport(self.get_url('format_5_branch')).base
 
104
        # check --format takes effect
 
105
        bzrdir.BzrDirFormat._set_default_format(bzrdir.BzrDirFormat5())
 
106
        (out, err) = self.run_bzr_captured(
 
107
            ['upgrade', '--format=metaweave', url])
 
108
        self.assertEqualDiff("""starting upgrade of %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
starting upgrade from format 5 to 6
 
114
adding prefixes to weaves
 
115
adding prefixes to revision-store
 
116
starting upgrade from format 6 to metadir
 
117
finished
 
118
""" % (url, url, url), out)
 
119
        self.assertEqualDiff("", err)
 
120
        self.assertTrue(isinstance(
 
121
            bzrdir.BzrDir.open(self.get_url('format_5_branch'))._format,
 
122
            bzrdir.BzrDirMetaFormat1))
 
123
 
 
124
    def test_upgrade_explicit_knit(self):
 
125
        # users can force an upgrade to knit format from a metadir weave 
 
126
        # branch
 
127
        url = get_transport(self.get_url('metadir_weave_branch')).base
 
128
        # check --format takes effect
 
129
        bzrdir.BzrDirFormat._set_default_format(bzrdir.BzrDirFormat5())
 
130
        (out, err) = self.run_bzr_captured(
 
131
            ['upgrade', '--format=knit', url])
 
132
        self.assertEqualDiff("""starting upgrade of %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
starting repository conversion
 
138
repository converted
 
139
finished
 
140
""" % (url, url, url), out)
 
141
        self.assertEqualDiff("", err)
 
142
        converted_dir = bzrdir.BzrDir.open(self.get_url('metadir_weave_branch'))
 
143
        self.assertTrue(isinstance(converted_dir._format,
 
144
                                   bzrdir.BzrDirMetaFormat1))
 
145
        self.assertTrue(isinstance(converted_dir.open_repository()._format,
 
146
                                   RepositoryFormatKnit1))
 
147
 
 
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
 
 
153
class SFTPTests(TestCaseWithSFTPServer):
 
154
    """Tests for upgrade over sftp."""
 
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
 
 
166
    def test_upgrade_url(self):
 
167
        self.run_bzr('init', '--format=weave')
 
168
        t = get_transport(self.get_url())
 
169
        url = t.base
 
170
        out, err = self.run_bzr('upgrade', '--format=knit', url)
 
171
        self.assertEqualDiff("""starting upgrade of %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
 
176
starting upgrade from format 6 to metadir
 
177
starting repository conversion
 
178
repository converted
 
179
finished
 
180
""" % (url, url, url), out)
 
181
        self.assertEqual('', err)