← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

Add a group cache to decompression, 5 times faster than knit at decompression when accessing everything in a group.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/blackbox/test_revert.py
1
 
# Copyright (C) 2006, 2007, 2009, 2010 Canonical Ltd
2
 
#
3
 
# This program is free software; you can redistribute it and/or modify
4
 
# it under the terms of the GNU General Public License as published by
5
 
# the Free Software Foundation; either version 2 of the License, or
6
 
# (at your option) any later version.
7
 
#
8
 
# This program is distributed in the hope that it will be useful,
9
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
 
# GNU General Public License for more details.
12
 
#
13
 
# You should have received a copy of the GNU General Public License
14
 
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
 
 
17
 
"""Black-box tests for bzr revert."""
18
 
 
19
 
import os
20
 
 
21
 
import bzrlib.osutils
22
 
from bzrlib.tests import TestCaseWithTransport
23
 
from bzrlib.trace import mutter
24
 
from bzrlib.workingtree import WorkingTree
25
 
 
26
 
 
27
 
class TestRevert(TestCaseWithTransport):
28
 
 
29
 
    def _prepare_tree(self):
30
 
        self.run_bzr('init')
31
 
        self.run_bzr('mkdir dir')
32
 
 
33
 
        f = file('dir/file', 'wb')
34
 
        f.write('spam')
35
 
        f.close()
36
 
        self.run_bzr('add dir/file')
37
 
 
38
 
        self.run_bzr('commit -m1')
39
 
 
40
 
        # modify file
41
 
        f = file('dir/file', 'wb')
42
 
        f.write('eggs')
43
 
        f.close()
44
 
 
45
 
        # check status
46
 
        self.assertEquals('modified:\n  dir/file\n', self.run_bzr('status')[0])
47
 
 
48
 
    def _prepare_rename_mod_tree(self):
49
 
        self.build_tree(['a/', 'a/b', 'a/c', 'a/d/', 'a/d/e', 'f/', 'f/g',
50
 
                         'f/h', 'f/i'])
51
 
        self.run_bzr('init')
52
 
        self.run_bzr('add')
53
 
        self.run_bzr('commit -m 1')
54
 
        wt = WorkingTree.open('.')
55
 
        wt.rename_one('a/b', 'f/b')
56
 
        wt.rename_one('a/d/e', 'f/e')
57
 
        wt.rename_one('a/d', 'f/d')
58
 
        wt.rename_one('f/g', 'a/g')
59
 
        wt.rename_one('f/h', 'h')
60
 
        wt.rename_one('f', 'j')
61
 
 
62
 
    def helper(self, param=''):
63
 
        self._prepare_tree()
64
 
        # change dir
65
 
        # revert to default revision for file in subdir does work
66
 
        os.chdir('dir')
67
 
        mutter('cd dir\n')
68
 
 
69
 
        self.assertEquals('1\n', self.run_bzr('revno')[0])
70
 
        self.run_bzr('revert %s file' % param)
71
 
        self.assertEquals('spam', open('file', 'rb').read())
72
 
 
73
 
    def test_revert_in_subdir(self):
74
 
        self.helper()
75
 
 
76
 
    def test_revert_to_revision_in_subdir(self):
77
 
        # test case for bug #29424:
78
 
        # revert to specific revision for file in subdir does not work
79
 
        self.helper('-r 1')
80
 
 
81
 
    def test_revert_in_checkout(self):
82
 
        os.mkdir('brach')
83
 
        os.chdir('brach')
84
 
        self._prepare_tree()
85
 
        self.run_bzr('checkout --lightweight . ../sprach')
86
 
        self.run_bzr('commit -m more')
87
 
        os.chdir('../sprach')
88
 
        self.assertEqual('', self.run_bzr('status')[0])
89
 
        self.run_bzr('revert')
90
 
        self.assertEqual('', self.run_bzr('status')[0])
91
 
 
92
 
    def test_revert_dirname(self):
93
 
        """Test that revert DIRECTORY does what's expected"""
94
 
        self._prepare_rename_mod_tree()
95
 
        self.run_bzr('revert a')
96
 
        self.failUnlessExists('a/b')
97
 
        self.failUnlessExists('a/d')
98
 
        self.failIfExists('a/g')
99
 
        self.expectFailure(
100
 
            "j is in the delta revert applies because j was renamed too",
101
 
            self.failUnlessExists, 'j')
102
 
        self.failUnlessExists('h')
103
 
        self.run_bzr('revert f')
104
 
        self.failIfExists('j')
105
 
        self.failIfExists('h')
106
 
        self.failUnlessExists('a/d/e')
107
 
 
108
 
    def test_revert_chatter(self):
109
 
        self._prepare_rename_mod_tree()
110
 
        chatter = self.run_bzr('revert')[1]
111
 
        self.assertEqualDiff(
112
 
            'R   a/g => f/g\n'
113
 
            'R   h => f/h\n'
114
 
            'R   j/ => f/\n'
115
 
            'R   j/b => a/b\n'
116
 
            'R   j/d/ => a/d/\n'
117
 
            'R   j/e => a/d/e\n',
118
 
            chatter)
119
 
 
120
 
    def test_revert(self):
121
 
        self.run_bzr('init')
122
 
 
123
 
        file('hello', 'wt').write('foo')
124
 
        self.run_bzr('add hello')
125
 
        self.run_bzr('commit -m setup hello')
126
 
 
127
 
        file('goodbye', 'wt').write('baz')
128
 
        self.run_bzr('add goodbye')
129
 
        self.run_bzr('commit -m setup goodbye')
130
 
 
131
 
        file('hello', 'wt').write('bar')
132
 
        file('goodbye', 'wt').write('qux')
133
 
        self.run_bzr('revert hello')
134
 
        self.check_file_contents('hello', 'foo')
135
 
        self.check_file_contents('goodbye', 'qux')
136
 
        self.run_bzr('revert')
137
 
        self.check_file_contents('goodbye', 'baz')
138
 
 
139
 
        os.mkdir('revertdir')
140
 
        self.run_bzr('add revertdir')
141
 
        self.run_bzr('commit -m f')
142
 
        os.rmdir('revertdir')
143
 
        self.run_bzr('revert')
144
 
 
145
 
        if bzrlib.osutils.has_symlinks():
146
 
            os.symlink('/unlikely/to/exist', 'symlink')
147
 
            self.run_bzr('add symlink')
148
 
            self.run_bzr('commit -m f')
149
 
            os.unlink('symlink')
150
 
            self.run_bzr('revert')
151
 
            self.failUnlessExists('symlink')
152
 
            os.unlink('symlink')
153
 
            os.symlink('a-different-path', 'symlink')
154
 
            self.run_bzr('revert')
155
 
            self.assertEqual('/unlikely/to/exist',
156
 
                             os.readlink('symlink'))
157
 
        else:
158
 
            self.log("skipping revert symlink tests")
159
 
 
160
 
        file('hello', 'wt').write('xyz')
161
 
        self.run_bzr('commit -m xyz hello')
162
 
        self.run_bzr('revert -r 1 hello')
163
 
        self.check_file_contents('hello', 'foo')
164
 
        self.run_bzr('revert hello')
165
 
        self.check_file_contents('hello', 'xyz')
166
 
        os.chdir('revertdir')
167
 
        self.run_bzr('revert')
168
 
        os.chdir('..')
169
 
 
170
 
    def test_revert_newly_added(self):
171
 
        # this tests the UI reports reverting a newly added file
172
 
        # correct (such files are not deleted)
173
 
        tree = self.make_branch_and_tree('.')
174
 
        self.build_tree(['file'])
175
 
        tree.add(['file'])
176
 
        out, err = self.run_bzr('revert')
177
 
        self.assertEqual('', out)
178
 
        self.assertEqual('-   file\n', err)
179
 
 
180
 
    def test_revert_removing_file(self):
181
 
        # this tests the UI reports reverting a file which has been committed
182
 
        # to a revision that did not have it, reports it as being deleted.
183
 
        tree = self.make_branch_and_tree('.')
184
 
        tree.commit('empty commit')
185
 
        self.build_tree(['file'])
186
 
        tree.add(['file'])
187
 
        tree.commit('add file')
188
 
        out, err = self.run_bzr('revert -r -2')
189
 
        self.assertEqual('', out)
190
 
        self.assertEqual('-D  file\n', err)
191
 
 
192
 
    def test_revert_forget_merges(self):
193
 
        # revert --forget-merges removes any pending merges into the tree, but
194
 
        # leaves the files unchanged
195
 
        tree = self.make_branch_and_tree('.')
196
 
        # forget-merges before first commit, though pointless, does not fail
197
 
        self.run_bzr(['revert', '--forget-merges'])
198
 
        self.build_tree(['file'])
199
 
        first_rev_id = tree.commit('initial commit')
200
 
        self.build_tree_contents([('file', 'new content')])
201
 
        existing_parents = tree.get_parent_ids()
202
 
        self.assertEquals([first_rev_id], existing_parents)
203
 
        merged_parents = existing_parents + ['merged-in-rev']
204
 
        tree.set_parent_ids(merged_parents)
205
 
        self.assertEquals(merged_parents, tree.get_parent_ids())
206
 
        self.run_bzr(['revert', '--forget-merges'])
207
 
        self.assertEquals([first_rev_id], tree.get_parent_ids())
208
 
        # changed files are not reverted
209
 
        self.assertFileEqual('new content', 'file')
210
 
        # you can give it the path of a tree
211
 
        self.run_bzr(['revert', '--forget-merges', tree.abspath('.')])