~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005 by Canonical Ltd
 
1
# Copyright (C) 2005, 2007 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
 
27
27
class TestRevert(ExternalBase):
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')
 
95
        self.run_bzr('revert a')
96
96
        self.failUnlessExists('a/b')
97
97
        self.failUnlessExists('a/d')
98
98
        self.failIfExists('a/g')
99
 
        self.failUnlessExists('j')
 
99
        self.expectFailure(
 
100
            "j is in the delta revert applies because j was renamed too",
 
101
            self.failUnlessExists, 'j')
100
102
        self.failUnlessExists('h')
101
 
        self.run_bzr('revert', 'f')
 
103
        self.run_bzr('revert f')
102
104
        self.failIfExists('j')
103
105
        self.failIfExists('h')
104
106
        self.failUnlessExists('a/d/e')
105
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
 
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
151
            self.failUnlessExists('symlink')
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('.')])