/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar
1850.3.4 by John Arbash Meinel
Add copyright to 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."""
1558.1.12 by Aaron Bentley
Got uncommit working properly with checkouts
18
19
import os
20
1850.3.1 by John Arbash Meinel
cleanup the uncommit tests
21
from bzrlib import uncommit, workingtree
1558.9.1 by Aaron Bentley
Fix uncommit to handle bound branches, and to do locking
22
from bzrlib.bzrdir import BzrDirMetaFormat1
23
from bzrlib.errors import BzrError, BoundBranchOutOfDate
1850.3.1 by John Arbash Meinel
cleanup the uncommit tests
24
from bzrlib.tests import TestCaseWithTransport
25
26
27
class TestUncommit(TestCaseWithTransport):
1956.1.1 by John Arbash Meinel
Fix bug #57660: 'bzr uncommit' should preserve pending merges
28
1850.3.1 by John Arbash Meinel
cleanup the uncommit tests
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'])
1850.3.2 by John Arbash Meinel
Change uncommit -r 10 so that it uncommits *to* 10, rather than removing 10
33
        wt.commit('initial commit', rev_id='a1')
1850.3.1 by John Arbash Meinel
cleanup the uncommit tests
34
35
        open('tree/a', 'wb').write('new contents of a\n')
1850.3.2 by John Arbash Meinel
Change uncommit -r 10 so that it uncommits *to* 10, rather than removing 10
36
        wt.commit('second commit', rev_id='a2')
1850.3.1 by John Arbash Meinel
cleanup the uncommit tests
37
38
        return wt
39
0.3.11 by John Arbash Meinel
Updated to latest bzr.dev code, and added tests.
40
    def test_uncommit(self):
41
        """Test uncommit functionality."""
1850.3.1 by John Arbash Meinel
cleanup the uncommit tests
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')
1850.3.2 by John Arbash Meinel
Change uncommit -r 10 so that it uncommits *to* 10, rather than removing 10
47
        self.assertNotContainsRe(out, 'initial commit')
48
        self.assertContainsRe(out, 'second commit')
1850.3.1 by John Arbash Meinel
cleanup the uncommit tests
49
50
        # Nothing has changed
51
        self.assertEqual('a2', wt.last_revision())
52
53
        # Uncommit, don't prompt
1850.3.2 by John Arbash Meinel
Change uncommit -r 10 so that it uncommits *to* 10, rather than removing 10
54
        out, err = self.run_bzr('uncommit', '--force')
55
        self.assertNotContainsRe(out, 'initial commit')
56
        self.assertContainsRe(out, 'second commit')
1850.3.1 by John Arbash Meinel
cleanup the uncommit tests
57
58
        # This should look like we are back in revno 1
59
        self.assertEqual('a1', wt.last_revision())
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.last_revision())
70
71
        os.chdir('checkout')
72
        out, err = self.run_bzr('uncommit', '--dry-run', '--force')
73
        self.assertContainsRe(out, 'Dry-run')
1850.3.2 by John Arbash Meinel
Change uncommit -r 10 so that it uncommits *to* 10, rather than removing 10
74
        self.assertNotContainsRe(out, 'initial commit')
75
        self.assertContainsRe(out, 'second commit')
1850.3.1 by John Arbash Meinel
cleanup the uncommit tests
76
77
        self.assertEqual('a2', checkout_tree.last_revision())
78
79
        out, err = self.run_bzr('uncommit', '--force')
1850.3.2 by John Arbash Meinel
Change uncommit -r 10 so that it uncommits *to* 10, rather than removing 10
80
        self.assertNotContainsRe(out, 'initial commit')
81
        self.assertContainsRe(out, 'second commit')
1850.3.1 by John Arbash Meinel
cleanup the uncommit tests
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.last_revision())
86
        self.assertEquals('a1', wt.branch.last_revision())
87
        self.assertEquals('a2', wt.last_revision())
1558.9.1 by Aaron Bentley
Fix uncommit to handle bound branches, and to do locking
88
89
    def test_uncommit_bound(self):
90
        os.mkdir('a')
91
        a = BzrDirMetaFormat1().initialize('a')
92
        a.create_repository()
93
        a.create_branch()
94
        t = a.create_workingtree()
95
        t.commit('commit 1')
96
        t.commit('commit 2')
97
        t.commit('commit 3')
98
        b = t.bzrdir.sprout('b').open_branch()
99
        b.bind(t.branch)
1850.3.1 by John Arbash Meinel
cleanup the uncommit tests
100
        uncommit.uncommit(b)
1558.9.1 by Aaron Bentley
Fix uncommit to handle bound branches, and to do locking
101
        t.set_last_revision(t.branch.last_revision())
102
        self.assertEqual(len(b.revision_history()), 2)
103
        self.assertEqual(len(t.branch.revision_history()), 2)
104
        t.commit('commit 3b')
1850.3.1 by John Arbash Meinel
cleanup the uncommit tests
105
        self.assertRaises(BoundBranchOutOfDate, uncommit.uncommit, b)
1558.9.1 by Aaron Bentley
Fix uncommit to handle bound branches, and to do locking
106
        b.pull(t.branch)
1850.3.1 by John Arbash Meinel
cleanup the uncommit tests
107
        uncommit.uncommit(b)
1558.9.1 by Aaron Bentley
Fix uncommit to handle bound branches, and to do locking
108
1850.3.2 by John Arbash Meinel
Change uncommit -r 10 so that it uncommits *to* 10, rather than removing 10
109
    def test_uncommit_revision(self):
110
        wt = self.create_simple_tree()
111
112
        os.chdir('tree')
113
        out, err = self.run_bzr('uncommit', '-r1', '--force')
114
115
        self.assertNotContainsRe(out, 'initial commit')
116
        self.assertContainsRe(out, 'second commit')
117
        self.assertEqual('a1', wt.last_revision())
118
        self.assertEqual('a1', wt.branch.last_revision())
119
120
    def test_uncommit_neg_1(self):
121
        wt = self.create_simple_tree()
122
        os.chdir('tree')
123
        out, err = self.run_bzr('uncommit', '-r', '-1', retcode=1)
124
        self.assertEqual('No revisions to uncommit.\n', out)
1850.3.5 by John Arbash Meinel
Fix bug 31426, have uncommit keep track of pending merges.
125
126
    def test_uncommit_merges(self):
127
        wt = self.create_simple_tree()
128
129
        tree2 = wt.bzrdir.sprout('tree2').open_workingtree()
130
131
        tree2.commit('unchanged', rev_id='b3')
132
        tree2.commit('unchanged', rev_id='b4')
133
134
        wt.branch.fetch(tree2.branch)
135
        wt.set_pending_merges(['b4'])
136
        wt.commit('merge b4', rev_id='a3')
137
138
        self.assertEqual('a3', wt.last_revision())
139
        self.assertEqual([], wt.pending_merges())
140
141
        os.chdir('tree')
142
        out, err = self.run_bzr('uncommit', '--force')
143
144
        self.assertEqual('a2', wt.last_revision())
145
        self.assertEqual(['b4'], wt.pending_merges())
146
1956.1.1 by John Arbash Meinel
Fix bug #57660: 'bzr uncommit' should preserve pending merges
147
    def test_uncommit_pending_merge(self):
148
        wt = self.create_simple_tree()
149
        tree2 = wt.bzrdir.sprout('tree2').open_workingtree()
150
        tree2.commit('unchanged', rev_id='b3')
151
152
        wt.branch.fetch(tree2.branch)
153
        wt.set_pending_merges(['b3'])
154
155
        os.chdir('tree')
156
        out, err = self.run_bzr('uncommit', '--force')
157
        self.assertEqual('a1', wt.last_revision())
158
        self.assertEqual(['b3'], wt.pending_merges())
159
1850.3.5 by John Arbash Meinel
Fix bug 31426, have uncommit keep track of pending merges.
160
    def test_uncommit_multiple_merge(self):
161
        wt = self.create_simple_tree()
162
163
        tree2 = wt.bzrdir.sprout('tree2').open_workingtree()
164
165
        tree2.commit('unchanged', rev_id='b3')
166
        wt.branch.fetch(tree2.branch)
167
        wt.set_pending_merges(['b3'])
168
        wt.commit('merge b3', rev_id='a3')
169
170
        tree2.commit('unchanged', rev_id='b4')
171
        wt.branch.fetch(tree2.branch)
172
        wt.set_pending_merges(['b4'])
173
        wt.commit('merge b4', rev_id='a4')
174
175
        self.assertEqual('a4', wt.last_revision())
176
        self.assertEqual([], wt.pending_merges())
177
178
        os.chdir('tree')
179
        out, err = self.run_bzr('uncommit', '--force', '-r', '2')
180
181
        self.assertEqual('a2', wt.last_revision())
182
        self.assertEqual(['b3', 'b4'], wt.pending_merges())
183
1956.1.1 by John Arbash Meinel
Fix bug #57660: 'bzr uncommit' should preserve pending merges
184
    def test_uncommit_merge_plus_pending(self):
185
        wt = self.create_simple_tree()
186
187
        tree2 = wt.bzrdir.sprout('tree2').open_workingtree()
188
189
        tree2.commit('unchanged', rev_id='b3')
190
        wt.branch.fetch(tree2.branch)
191
        wt.set_pending_merges(['b3'])
192
        wt.commit('merge b3', rev_id='a3')
193
194
        tree2.commit('unchanged', rev_id='b4')
195
        wt.branch.fetch(tree2.branch)
196
        wt.set_pending_merges(['b4'])
197
198
        self.assertEqual('a3', wt.last_revision())
199
        self.assertEqual(['b4'], wt.pending_merges())
200
201
        os.chdir('tree')
202
        out, err = self.run_bzr('uncommit', '--force', '-r', '2')
203
204
        self.assertEqual('a2', wt.last_revision())
205
        self.assertEqual(['b3', 'b4'], wt.pending_merges())
206
1850.3.5 by John Arbash Meinel
Fix bug 31426, have uncommit keep track of pending merges.
207
    def test_uncommit_octopus_merge(self):
208
        # Check that uncommit keeps the pending merges in the same order
209
        wt = self.create_simple_tree()
210
211
        tree2 = wt.bzrdir.sprout('tree2').open_workingtree()
212
        tree3 = wt.bzrdir.sprout('tree3').open_workingtree()
213
214
        tree2.commit('unchanged', rev_id='b3')
215
        tree3.commit('unchanged', rev_id='c3')
216
        wt.branch.fetch(tree2.branch)
217
        wt.branch.fetch(tree3.branch)
218
        wt.set_pending_merges(['b3', 'c3'])
219
        wt.commit('merge b3, c3', rev_id='a3')
220
221
        tree2.commit('unchanged', rev_id='b4')
222
        tree3.commit('unchanged', rev_id='c4')
223
        wt.branch.fetch(tree2.branch)
224
        wt.branch.fetch(tree3.branch)
225
        wt.set_pending_merges(['c4', 'b4'])
226
        wt.commit('merge b4, c4', rev_id='a4')
227
228
        self.assertEqual('a4', wt.last_revision())
229
        self.assertEqual([], wt.pending_merges())
230
231
        os.chdir('tree')
232
        out, err = self.run_bzr('uncommit', '--force', '-r', '2')
233
234
        self.assertEqual('a2', wt.last_revision())
235
        self.assertEqual(['b3', 'c3', 'c4', 'b4'], wt.pending_merges())