← 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

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-08-30 22:49:20 UTC
  • mfrom: (5397.1.6 jam-integration)
  • Revision ID: pqm@pqm.ubuntu.com-20100830224920-w9zw1vhsd5oiyljv
(vila, jam) Get PQM running correctly again (bug #626667),
        skip test_bzr_connect_to_bzr_ssh (bug #626876)

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('.')])