← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2006-06-29 18:21:57 UTC
  • mfrom: (1711.2.72 jam-integration)
  • Revision ID: pqm@pqm.ubuntu.com-20060629182157-cdc39b5f9e4fac85
(jam) add run_bzr_error, clean up remerge tests

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/blackbox/test_uncommit.py
1
 
# Copyright (C) 2005, 2006 by 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
 
 
17
 
"""Test the uncommit command."""
 
1
"""\
 
2
Test the uncommit command.
 
3
"""
18
4
 
19
5
import os
20
6
 
21
 
from bzrlib import uncommit, workingtree
22
7
from bzrlib.bzrdir import BzrDirMetaFormat1
23
8
from bzrlib.errors import BzrError, BoundBranchOutOfDate
24
 
from bzrlib.tests import TestCaseWithTransport
25
 
 
26
 
 
27
 
class TestUncommit(TestCaseWithTransport):
28
 
 
29
 
    def create_simple_tree(self):
30
 
        wt = self.make_branch_and_tree('tree')
31
 
        self.build_tree(['tree/a', 'tree/b', 'tree/c'])
32
 
        wt.add(['a', 'b', 'c'])
33
 
        wt.commit('initial commit', rev_id='a1')
34
 
 
35
 
        open('tree/a', 'wb').write('new contents of a\n')
36
 
        wt.commit('second commit', rev_id='a2')
37
 
 
38
 
        return wt
39
 
 
 
9
from bzrlib.uncommit import uncommit
 
10
from bzrlib.tests import TestCaseInTempDir
 
11
 
 
12
class TestUncommit(TestCaseInTempDir):
40
13
    def test_uncommit(self):
41
14
        """Test uncommit functionality."""
42
 
        wt = self.create_simple_tree()
43
 
 
44
 
        os.chdir('tree')
45
 
        out, err = self.run_bzr('uncommit', '--dry-run', '--force')
46
 
        self.assertContainsRe(out, 'Dry-run')
47
 
        self.assertNotContainsRe(out, 'initial commit')
48
 
        self.assertContainsRe(out, 'second commit')
49
 
 
50
 
        # Nothing has changed
51
 
        self.assertEqual(['a2'], wt.get_parent_ids())
52
 
 
53
 
        # Uncommit, don't prompt
54
 
        out, err = self.run_bzr('uncommit', '--force')
55
 
        self.assertNotContainsRe(out, 'initial commit')
56
 
        self.assertContainsRe(out, 'second commit')
57
 
 
58
 
        # This should look like we are back in revno 1
59
 
        self.assertEqual(['a1'], wt.get_parent_ids())
60
 
        out, err = self.run_bzr('status')
61
 
        self.assertEquals(out, 'modified:\n  a\n')
62
 
 
63
 
    def test_uncommit_checkout(self):
64
 
        wt = self.create_simple_tree()
65
 
 
66
 
        checkout_tree = wt.bzrdir.sprout('checkout').open_workingtree()
67
 
        checkout_tree.branch.bind(wt.branch)
68
 
 
69
 
        self.assertEqual(['a2'], checkout_tree.get_parent_ids())
70
 
 
71
 
        os.chdir('checkout')
72
 
        out, err = self.run_bzr('uncommit', '--dry-run', '--force')
73
 
        self.assertContainsRe(out, 'Dry-run')
74
 
        self.assertNotContainsRe(out, 'initial commit')
75
 
        self.assertContainsRe(out, 'second commit')
76
 
 
77
 
        self.assertEqual(['a2'], checkout_tree.get_parent_ids())
78
 
 
79
 
        out, err = self.run_bzr('uncommit', '--force')
80
 
        self.assertNotContainsRe(out, 'initial commit')
81
 
        self.assertContainsRe(out, 'second commit')
82
 
 
83
 
        # uncommit in a checkout should uncommit the parent branch
84
 
        # (but doesn't effect the other working tree)
85
 
        self.assertEquals(['a1'], checkout_tree.get_parent_ids())
86
 
        self.assertEquals('a1', wt.branch.last_revision())
87
 
        self.assertEquals(['a2'], wt.get_parent_ids())
 
15
        bzr = self.capture 
 
16
        os.mkdir('branch')
 
17
        os.chdir('branch')
 
18
        bzr('init')
 
19
        self.build_tree(['a', 'b', 'c'])
 
20
 
 
21
        bzr('add')
 
22
        bzr('commit -m initial')
 
23
 
 
24
        self.assertEquals(bzr('revno'), '1\n')
 
25
 
 
26
        open('a', 'wb').write('new contents of a\n')
 
27
        self.assertEquals(bzr('status'), 'modified:\n  a\n')
 
28
        bzr('commit -m second')
 
29
 
 
30
        self.assertEquals(bzr('status'), '')
 
31
        self.assertEquals(bzr('revno'), '2\n')
 
32
 
 
33
        txt = bzr('uncommit --dry-run --force')
 
34
        self.failIfEqual(txt.find('Dry-run'), -1)
 
35
 
 
36
        self.assertEquals(bzr('status'), '')
 
37
        self.assertEquals(bzr('revno'), '2\n')
 
38
 
 
39
        txt = bzr('uncommit --force')
 
40
 
 
41
        self.assertEquals(bzr('revno'), '1\n')
 
42
        self.assertEquals(bzr('status'), 'modified:\n  a\n')
 
43
        
 
44
        bzr('checkout . ../checkout')
 
45
        os.chdir('../checkout')
 
46
        self.assertEquals("", bzr('status'))
 
47
        self.assertEquals(bzr('revno'), '1\n')
 
48
 
 
49
        open('a', 'wb').write('new contents of a\n')
 
50
        self.assertEquals(bzr('status'), 'modified:\n  a\n')
 
51
        bzr('commit -m second')
 
52
 
 
53
        self.assertEquals(bzr('status'), '')
 
54
        self.assertEquals(bzr('revno'), '2\n')
 
55
 
 
56
        txt = bzr('uncommit --dry-run --force')
 
57
        self.failIfEqual(txt.find('Dry-run'), -1)
 
58
 
 
59
        self.assertEquals(bzr('status'), '')
 
60
        self.assertEquals(bzr('revno'), '2\n')
 
61
 
 
62
        txt = bzr('uncommit --force')
 
63
 
 
64
        self.assertEquals(bzr('revno'), '1\n')
 
65
        self.assertEquals(bzr('status'), 'modified:\n  a\n')
88
66
 
89
67
    def test_uncommit_bound(self):
90
68
        os.mkdir('a')
91
69
        a = BzrDirMetaFormat1().initialize('a')
92
70
        a.create_repository()
93
71
        a.create_branch()
94
 
        t_a = a.create_workingtree()
95
 
        t_a.commit('commit 1')
96
 
        t_a.commit('commit 2')
97
 
        t_a.commit('commit 3')
98
 
        b = t_a.bzrdir.sprout('b').open_branch()
99
 
        b.bind(t_a.branch)
100
 
        uncommit.uncommit(b)
 
72
        t = a.create_workingtree()
 
73
        t.commit('commit 1')
 
74
        t.commit('commit 2')
 
75
        t.commit('commit 3')
 
76
        b = t.bzrdir.sprout('b').open_branch()
 
77
        b.bind(t.branch)
 
78
        uncommit(b)
 
79
        t.set_last_revision(t.branch.last_revision())
101
80
        self.assertEqual(len(b.revision_history()), 2)
102
 
        self.assertEqual(len(t_a.branch.revision_history()), 2)
103
 
        # update A's tree to not have the uncomitted revision referenced.
104
 
        t_a.update()
105
 
        t_a.commit('commit 3b')
106
 
        self.assertRaises(BoundBranchOutOfDate, uncommit.uncommit, b)
107
 
        b.pull(t_a.branch)
108
 
        uncommit.uncommit(b)
109
 
 
110
 
    def test_uncommit_revision(self):
111
 
        wt = self.create_simple_tree()
112
 
 
113
 
        os.chdir('tree')
114
 
        out, err = self.run_bzr('uncommit', '-r1', '--force')
115
 
 
116
 
        self.assertNotContainsRe(out, 'initial commit')
117
 
        self.assertContainsRe(out, 'second commit')
118
 
        self.assertEqual(['a1'], wt.get_parent_ids())
119
 
        self.assertEqual('a1', wt.branch.last_revision())
120
 
 
121
 
    def test_uncommit_neg_1(self):
122
 
        wt = self.create_simple_tree()
123
 
        os.chdir('tree')
124
 
        out, err = self.run_bzr('uncommit', '-r', '-1', retcode=1)
125
 
        self.assertEqual('No revisions to uncommit.\n', out)
126
 
 
127
 
    def test_uncommit_merges(self):
128
 
        wt = self.create_simple_tree()
129
 
 
130
 
        tree2 = wt.bzrdir.sprout('tree2').open_workingtree()
131
 
 
132
 
        tree2.commit('unchanged', rev_id='b3')
133
 
        tree2.commit('unchanged', rev_id='b4')
134
 
 
135
 
        wt.merge_from_branch(tree2.branch)
136
 
        wt.commit('merge b4', rev_id='a3')
137
 
 
138
 
        self.assertEqual(['a3'], wt.get_parent_ids())
139
 
 
140
 
        os.chdir('tree')
141
 
        out, err = self.run_bzr('uncommit', '--force')
142
 
 
143
 
        self.assertEqual(['a2', 'b4'], wt.get_parent_ids())
144
 
 
145
 
    def test_uncommit_pending_merge(self):
146
 
        wt = self.create_simple_tree()
147
 
        tree2 = wt.bzrdir.sprout('tree2').open_workingtree()
148
 
        tree2.commit('unchanged', rev_id='b3')
149
 
 
150
 
        wt.branch.fetch(tree2.branch)
151
 
        wt.set_pending_merges(['b3'])
152
 
 
153
 
        os.chdir('tree')
154
 
        out, err = self.run_bzr('uncommit', '--force')
155
 
        self.assertEqual(['a1', 'b3'], wt.get_parent_ids())
156
 
 
157
 
    def test_uncommit_multiple_merge(self):
158
 
        wt = self.create_simple_tree()
159
 
 
160
 
        tree2 = wt.bzrdir.sprout('tree2').open_workingtree()
161
 
 
162
 
        tree2.commit('unchanged', rev_id='b3')
163
 
 
164
 
        wt.merge_from_branch(tree2.branch)
165
 
        wt.commit('merge b3', rev_id='a3')
166
 
 
167
 
        tree2.commit('unchanged', rev_id='b4')
168
 
 
169
 
        wt.merge_from_branch(tree2.branch)
170
 
        wt.commit('merge b4', rev_id='a4')
171
 
 
172
 
        self.assertEqual(['a4'], wt.get_parent_ids())
173
 
 
174
 
        os.chdir('tree')
175
 
        out, err = self.run_bzr('uncommit', '--force', '-r', '2')
176
 
 
177
 
        self.assertEqual(['a2', 'b3', 'b4'], wt.get_parent_ids())
178
 
 
179
 
    def test_uncommit_merge_plus_pending(self):
180
 
        wt = self.create_simple_tree()
181
 
 
182
 
        tree2 = wt.bzrdir.sprout('tree2').open_workingtree()
183
 
 
184
 
        tree2.commit('unchanged', rev_id='b3')
185
 
        wt.branch.fetch(tree2.branch)
186
 
        wt.set_pending_merges(['b3'])
187
 
        wt.commit('merge b3', rev_id='a3')
188
 
 
189
 
        tree2.commit('unchanged', rev_id='b4')
190
 
        wt.branch.fetch(tree2.branch)
191
 
        wt.set_pending_merges(['b4'])
192
 
 
193
 
        self.assertEqual(['a3', 'b4'], wt.get_parent_ids())
194
 
 
195
 
        os.chdir('tree')
196
 
        out, err = self.run_bzr('uncommit', '--force', '-r', '2')
197
 
 
198
 
        self.assertEqual(['a2', 'b3', 'b4'], wt.get_parent_ids())
199
 
 
200
 
    def test_uncommit_octopus_merge(self):
201
 
        # Check that uncommit keeps the pending merges in the same order
202
 
        wt = self.create_simple_tree()
203
 
 
204
 
        tree2 = wt.bzrdir.sprout('tree2').open_workingtree()
205
 
        tree3 = wt.bzrdir.sprout('tree3').open_workingtree()
206
 
 
207
 
        tree2.commit('unchanged', rev_id='b3')
208
 
        tree3.commit('unchanged', rev_id='c3')
209
 
        
210
 
        wt.merge_from_branch(tree2.branch)
211
 
        wt.merge_from_branch(tree3.branch)
212
 
        wt.commit('merge b3, c3', rev_id='a3')
213
 
 
214
 
        tree2.commit('unchanged', rev_id='b4')
215
 
        tree3.commit('unchanged', rev_id='c4')
216
 
 
217
 
        wt.merge_from_branch(tree3.branch)
218
 
        wt.merge_from_branch(tree2.branch)
219
 
        wt.commit('merge b4, c4', rev_id='a4')
220
 
 
221
 
        self.assertEqual(['a4'], wt.get_parent_ids())
222
 
 
223
 
        os.chdir('tree')
224
 
        out, err = self.run_bzr('uncommit', '--force', '-r', '2')
225
 
 
226
 
        self.assertEqual(['a2', 'b3', 'c3', 'c4', 'b4'], wt.get_parent_ids())
 
81
        self.assertEqual(len(t.branch.revision_history()), 2)
 
82
        t.commit('commit 3b')
 
83
        self.assertRaises(BoundBranchOutOfDate, uncommit, b)
 
84
        b.pull(t.branch)
 
85
        uncommit(b)
 
86