← 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: John Arbash Meinel
  • Date: 2011-05-11 11:35:28 UTC
  • mto: This revision was merged to the branch mainline in revision 5851.
  • Revision ID: john@arbash-meinel.com-20110511113528-qepibuwxicjrbb2h
Break compatibility with python <2.6.

This includes auditing the code for places where we were doing
explicit 'sys.version' checks and removing them as appropriate.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/blackbox/test_revert.py
1
 
# Copyright (C) 2005 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
21
import bzrlib.osutils
22
 
from bzrlib.tests.blackbox import ExternalBase
 
22
from bzrlib.tests import TestCaseWithTransport
23
23
from bzrlib.trace import mutter
24
24
from bzrlib.workingtree import WorkingTree
25
25
 
26
26
 
27
 
class TestRevert(ExternalBase):
 
27
class TestRevert(TestCaseWithTransport):
28
28
 
29
29
    def _prepare_tree(self):
30
 
        self.runbzr('init')
31
 
        self.runbzr('mkdir dir')
 
30
        self.run_bzr('init')
 
31
        self.run_bzr('mkdir dir')
32
32
 
33
33
        f = file('dir/file', 'wb')
34
34
        f.write('spam')
35
35
        f.close()
36
 
        self.runbzr('add dir/file')
 
36
        self.run_bzr('add dir/file')
37
37
 
38
 
        self.runbzr('commit -m1')
 
38
        self.run_bzr('commit -m1')
39
39
 
40
40
        # modify file
41
41
        f = file('dir/file', 'wb')
43
43
        f.close()
44
44
 
45
45
        # check status
46
 
        self.assertEquals('modified:\n  dir/file\n', self.capture('status'))
 
46
        self.assertEquals('modified:\n  dir/file\n', self.run_bzr('status')[0])
47
47
 
48
48
    def _prepare_rename_mod_tree(self):
49
 
        self.build_tree(['a/', 'a/b', 'a/c', 'a/d/', 'a/d/e', 'f/', 'f/g', 
 
49
        self.build_tree(['a/', 'a/b', 'a/c', 'a/d/', 'a/d/e', 'f/', 'f/g',
50
50
                         'f/h', 'f/i'])
51
51
        self.run_bzr('init')
52
52
        self.run_bzr('add')
53
 
        self.run_bzr('commit', '-m', '1')
 
53
        self.run_bzr('commit -m 1')
54
54
        wt = WorkingTree.open('.')
55
55
        wt.rename_one('a/b', 'f/b')
56
56
        wt.rename_one('a/d/e', 'f/e')
66
66
        os.chdir('dir')
67
67
        mutter('cd dir\n')
68
68
 
69
 
        self.assertEquals('1\n', self.capture('revno'))
70
 
        self.runbzr('revert %s file' % param)
 
69
        self.assertEquals('1\n', self.run_bzr('revno')[0])
 
70
        self.run_bzr('revert %s file' % param)
71
71
        self.assertEquals('spam', open('file', 'rb').read())
72
72
 
73
73
    def test_revert_in_subdir(self):
82
82
        os.mkdir('brach')
83
83
        os.chdir('brach')
84
84
        self._prepare_tree()
85
 
        self.runbzr('checkout --lightweight . ../sprach')
86
 
        self.runbzr('commit -m more')
 
85
        self.run_bzr('checkout --lightweight . ../sprach')
 
86
        self.run_bzr('commit -m more')
87
87
        os.chdir('../sprach')
88
 
        self.assertEqual('', self.capture('status'))
89
 
        self.runbzr('revert')
90
 
        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
91
 
92
92
    def test_revert_dirname(self):
93
93
        """Test that revert DIRECTORY does what's expected"""
94
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.failUnlessExists('j')
100
 
        self.failUnlessExists('h')
101
 
        self.run_bzr('revert', 'f')
102
 
        self.failIfExists('j')
103
 
        self.failIfExists('h')
104
 
        self.failUnlessExists('a/d/e')
 
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)
105
119
 
106
120
    def test_revert(self):
107
121
        self.run_bzr('init')
108
122
 
109
123
        file('hello', 'wt').write('foo')
110
 
        self.run_bzr('add', 'hello')
111
 
        self.run_bzr('commit', '-m', 'setup', 'hello')
 
124
        self.run_bzr('add hello')
 
125
        self.run_bzr('commit -m setup hello')
112
126
 
113
127
        file('goodbye', 'wt').write('baz')
114
 
        self.run_bzr('add', 'goodbye')
115
 
        self.run_bzr('commit', '-m', 'setup', 'goodbye')
 
128
        self.run_bzr('add goodbye')
 
129
        self.run_bzr('commit -m setup goodbye')
116
130
 
117
131
        file('hello', 'wt').write('bar')
118
132
        file('goodbye', 'wt').write('qux')
119
 
        self.run_bzr('revert', 'hello')
 
133
        self.run_bzr('revert hello')
120
134
        self.check_file_contents('hello', 'foo')
121
135
        self.check_file_contents('goodbye', 'qux')
122
136
        self.run_bzr('revert')
123
137
        self.check_file_contents('goodbye', 'baz')
124
138
 
125
139
        os.mkdir('revertdir')
126
 
        self.run_bzr('add', 'revertdir')
127
 
        self.run_bzr('commit', '-m', 'f')
 
140
        self.run_bzr('add revertdir')
 
141
        self.run_bzr('commit -m f')
128
142
        os.rmdir('revertdir')
129
143
        self.run_bzr('revert')
130
144
 
131
145
        if bzrlib.osutils.has_symlinks():
132
146
            os.symlink('/unlikely/to/exist', 'symlink')
133
 
            self.run_bzr('add', 'symlink')
134
 
            self.run_bzr('commit', '-m', 'f')
 
147
            self.run_bzr('add symlink')
 
148
            self.run_bzr('commit -m f')
135
149
            os.unlink('symlink')
136
150
            self.run_bzr('revert')
137
 
            self.failUnlessExists('symlink')
 
151
            self.assertPathExists('symlink')
138
152
            os.unlink('symlink')
139
153
            os.symlink('a-different-path', 'symlink')
140
154
            self.run_bzr('revert')
142
156
                             os.readlink('symlink'))
143
157
        else:
144
158
            self.log("skipping revert symlink tests")
145
 
        
 
159
 
146
160
        file('hello', 'wt').write('xyz')
147
 
        self.run_bzr('commit', '-m', 'xyz', 'hello')
148
 
        self.run_bzr('revert', '-r', '1', 'hello')
 
161
        self.run_bzr('commit -m xyz hello')
 
162
        self.run_bzr('revert -r 1 hello')
149
163
        self.check_file_contents('hello', 'foo')
150
 
        self.run_bzr('revert', 'hello')
 
164
        self.run_bzr('revert hello')
151
165
        self.check_file_contents('hello', 'xyz')
152
166
        os.chdir('revertdir')
153
167
        self.run_bzr('revert')
154
168
        os.chdir('..')
155
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('.')])