← 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: 2007-05-04 18:59:36 UTC
  • mto: This revision was merged to the branch mainline in revision 2643.
  • Revision ID: john@arbash-meinel.com-20070504185936-1mjdoqmtz74xe5mg
A C implementation of _fields_to_entry_0_parents drops the time from 400ms to 330ms for a 21k-entry tree

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/blackbox/test_revert.py
1
 
# Copyright (C) 2005, 2007 Canonical Ltd
 
1
# Copyright (C) 2005 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
27
27
class TestRevert(ExternalBase):
28
28
 
29
29
    def _prepare_tree(self):
30
 
        self.run_bzr('init')
31
 
        self.run_bzr('mkdir dir')
 
30
        self.runbzr('init')
 
31
        self.runbzr('mkdir dir')
32
32
 
33
33
        f = file('dir/file', 'wb')
34
34
        f.write('spam')
35
35
        f.close()
36
 
        self.run_bzr('add dir/file')
 
36
        self.runbzr('add dir/file')
37
37
 
38
 
        self.run_bzr('commit -m1')
 
38
        self.runbzr('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.run_bzr('status')[0])
 
46
        self.assertEquals('modified:\n  dir/file\n', self.capture('status'))
47
47
 
48
48
    def _prepare_rename_mod_tree(self):
49
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.run_bzr('revno')[0])
70
 
        self.run_bzr('revert %s file' % param)
 
69
        self.assertEquals('1\n', self.capture('revno'))
 
70
        self.runbzr('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.run_bzr('checkout --lightweight . ../sprach')
86
 
        self.run_bzr('commit -m more')
 
85
        self.runbzr('checkout --lightweight . ../sprach')
 
86
        self.runbzr('commit -m more')
87
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])
 
88
        self.assertEqual('', self.capture('status'))
 
89
        self.runbzr('revert')
 
90
        self.assertEqual('', self.capture('status'))
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
99
        self.failUnlessExists('j')
100
100
        self.failUnlessExists('h')
101
 
        self.run_bzr('revert f')
 
101
        self.run_bzr('revert', 'f')
102
102
        self.failIfExists('j')
103
103
        self.failIfExists('h')
104
104
        self.failUnlessExists('a/d/e')
119
119
        self.run_bzr('init')
120
120
 
121
121
        file('hello', 'wt').write('foo')
122
 
        self.run_bzr('add hello')
123
 
        self.run_bzr('commit -m setup hello')
 
122
        self.run_bzr('add', 'hello')
 
123
        self.run_bzr('commit', '-m', 'setup', 'hello')
124
124
 
125
125
        file('goodbye', 'wt').write('baz')
126
 
        self.run_bzr('add goodbye')
127
 
        self.run_bzr('commit -m setup goodbye')
 
126
        self.run_bzr('add', 'goodbye')
 
127
        self.run_bzr('commit', '-m', 'setup', 'goodbye')
128
128
 
129
129
        file('hello', 'wt').write('bar')
130
130
        file('goodbye', 'wt').write('qux')
131
 
        self.run_bzr('revert hello')
 
131
        self.run_bzr('revert', 'hello')
132
132
        self.check_file_contents('hello', 'foo')
133
133
        self.check_file_contents('goodbye', 'qux')
134
134
        self.run_bzr('revert')
135
135
        self.check_file_contents('goodbye', 'baz')
136
136
 
137
137
        os.mkdir('revertdir')
138
 
        self.run_bzr('add revertdir')
139
 
        self.run_bzr('commit -m f')
 
138
        self.run_bzr('add', 'revertdir')
 
139
        self.run_bzr('commit', '-m', 'f')
140
140
        os.rmdir('revertdir')
141
141
        self.run_bzr('revert')
142
142
 
143
143
        if bzrlib.osutils.has_symlinks():
144
144
            os.symlink('/unlikely/to/exist', 'symlink')
145
 
            self.run_bzr('add symlink')
146
 
            self.run_bzr('commit -m f')
 
145
            self.run_bzr('add', 'symlink')
 
146
            self.run_bzr('commit', '-m', 'f')
147
147
            os.unlink('symlink')
148
148
            self.run_bzr('revert')
149
149
            self.failUnlessExists('symlink')
156
156
            self.log("skipping revert symlink tests")
157
157
        
158
158
        file('hello', 'wt').write('xyz')
159
 
        self.run_bzr('commit -m xyz hello')
160
 
        self.run_bzr('revert -r 1 hello')
 
159
        self.run_bzr('commit', '-m', 'xyz', 'hello')
 
160
        self.run_bzr('revert', '-r', '1', 'hello')
161
161
        self.check_file_contents('hello', 'foo')
162
 
        self.run_bzr('revert hello')
 
162
        self.run_bzr('revert', 'hello')
163
163
        self.check_file_contents('hello', 'xyz')
164
164
        os.chdir('revertdir')
165
165
        self.run_bzr('revert')
183
183
        self.build_tree(['file'])
184
184
        tree.add(['file'])
185
185
        tree.commit('add file')
186
 
        out, err = self.run_bzr('revert -r -2')
 
186
        out, err = self.run_bzr('revert', '-r', '-2')
187
187
        self.assertEqual('', out)
188
188
        self.assertEqual('-D  file\n', err)
189
 
 
190
 
    def test_revert_forget_merges(self):
191
 
        # revert --forget-merges removes any pending merges into the tree, but
192
 
        # leaves the files unchanged
193
 
        tree = self.make_branch_and_tree('.')
194
 
        # forget-merges before first commit, though pointless, does not fail
195
 
        self.run_bzr(['revert', '--forget-merges'])
196
 
        self.build_tree(['file'])
197
 
        first_rev_id = tree.commit('initial commit')
198
 
        self.build_tree_contents([('file', 'new content')])
199
 
        existing_parents = tree.get_parent_ids()
200
 
        self.assertEquals([first_rev_id], existing_parents)
201
 
        merged_parents = existing_parents + ['merged-in-rev']
202
 
        tree.set_parent_ids(merged_parents)
203
 
        self.assertEquals(merged_parents, tree.get_parent_ids())
204
 
        self.run_bzr(['revert', '--forget-merges'])
205
 
        self.assertEquals([first_rev_id], tree.get_parent_ids())
206
 
        # changed files are not reverted
207
 
        self.assertFileEqual('new content', 'file')
208
 
        # you can give it the path of a tree
209
 
        self.run_bzr(['revert', '--forget-merges', tree.abspath('.')])