~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

  • Committer: Patch Queue Manager
  • Date: 2012-10-25 11:13:27 UTC
  • mfrom: (6570.1.6 rubberstamp)
  • Revision ID: pqm@pqm.ubuntu.com-20121025111327-p0ylql0nh9fla0rs
(gz) Set approved revision and vote "Approve" when using lp-propose
 --approve (Jonathan Lange)

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005 by Canonical Ltd
 
1
# Copyright (C) 2006, 2007, 2009, 2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
"""Black-box tests for bzr revert."""
18
18
 
19
19
import os
20
20
 
21
 
from bzrlib.tests.blackbox import ExternalBase
 
21
import bzrlib.osutils
 
22
from bzrlib.tests import TestCaseWithTransport
22
23
from bzrlib.trace import mutter
23
 
 
24
 
 
25
 
class TestRevert(ExternalBase):
 
24
from bzrlib.workingtree import WorkingTree
 
25
 
 
26
 
 
27
class TestRevert(TestCaseWithTransport):
26
28
 
27
29
    def _prepare_tree(self):
28
 
        self.runbzr('init')
29
 
        self.runbzr('mkdir dir')
 
30
        self.run_bzr('init')
 
31
        self.run_bzr('mkdir dir')
30
32
 
31
33
        f = file('dir/file', 'wb')
32
34
        f.write('spam')
33
35
        f.close()
34
 
        self.runbzr('add dir/file')
 
36
        self.run_bzr('add dir/file')
35
37
 
36
 
        self.runbzr('commit -m1')
 
38
        self.run_bzr('commit -m1')
37
39
 
38
40
        # modify file
39
41
        f = file('dir/file', 'wb')
41
43
        f.close()
42
44
 
43
45
        # check status
44
 
        self.assertEquals('modified:\n  dir/file\n', self.capture('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')
45
61
 
46
62
    def helper(self, param=''):
47
63
        self._prepare_tree()
50
66
        os.chdir('dir')
51
67
        mutter('cd dir\n')
52
68
 
53
 
        self.assertEquals('1\n', self.capture('revno'))
54
 
        self.runbzr('revert %s file' % param)
 
69
        self.assertEquals('1\n', self.run_bzr('revno')[0])
 
70
        self.run_bzr('revert %s file' % param)
55
71
        self.assertEquals('spam', open('file', 'rb').read())
56
72
 
57
73
    def test_revert_in_subdir(self):
66
82
        os.mkdir('brach')
67
83
        os.chdir('brach')
68
84
        self._prepare_tree()
69
 
        self.runbzr('checkout --lightweight . ../sprach')
70
 
        self.runbzr('commit -m more')
 
85
        self.run_bzr('checkout --lightweight . ../sprach')
 
86
        self.run_bzr('commit -m more')
71
87
        os.chdir('../sprach')
72
 
        self.assertEqual('', self.capture('status'))
73
 
        self.runbzr('revert')
74
 
        self.assertEqual('', self.capture('status'))
 
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.assertPathExists('a/b')
 
97
        self.assertPathExists('a/d')
 
98
        self.assertPathDoesNotExist('a/g')
 
99
        self.expectFailure(
 
100
            "j is in the delta revert applies because j was renamed too",
 
101
            self.assertPathExists, 'j')
 
102
        self.assertPathExists('h')
 
103
        self.run_bzr('revert f')
 
104
        self.assertPathDoesNotExist('j')
 
105
        self.assertPathDoesNotExist('h')
 
106
        self.assertPathExists('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
        with file('hello', 'wt') as f: f.write('foo')
 
124
        self.run_bzr('add hello')
 
125
        self.run_bzr('commit -m setup hello')
 
126
 
 
127
        with file('goodbye', 'wt') as f: f.write('baz')
 
128
        self.run_bzr('add goodbye')
 
129
        self.run_bzr('commit -m setup goodbye')
 
130
 
 
131
        with file('hello', 'wt') as f: f.write('bar')
 
132
        with file('goodbye', 'wt') as f: f.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.assertPathExists('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
        with file('hello', 'wt') as f: f.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('.')])