← 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: Alexander Belchenko
  • Date: 2007-01-24 13:03:32 UTC
  • mto: This revision was merged to the branch mainline in revision 2242.
  • Revision ID: bialix@ukr.net-20070124130332-ane2eqz3eqrtm9u1
Use new API for testing

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/blackbox/test_uncommit.py
 
1
# Copyright (C) 2005, 2006 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."""
 
18
 
 
19
import os
 
20
 
 
21
from bzrlib import uncommit, workingtree
 
22
from bzrlib.bzrdir import BzrDirMetaFormat1
 
23
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
 
 
40
    def test_uncommit(self):
 
41
        """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
        checkout_tree = wt.branch.create_checkout('checkout')
 
66
 
 
67
        self.assertEqual(['a2'], checkout_tree.get_parent_ids())
 
68
 
 
69
        os.chdir('checkout')
 
70
        out, err = self.run_bzr('uncommit', '--dry-run', '--force')
 
71
        self.assertContainsRe(out, 'Dry-run')
 
72
        self.assertNotContainsRe(out, 'initial commit')
 
73
        self.assertContainsRe(out, 'second commit')
 
74
 
 
75
        self.assertEqual(['a2'], checkout_tree.get_parent_ids())
 
76
 
 
77
        out, err = self.run_bzr('uncommit', '--force')
 
78
        self.assertNotContainsRe(out, 'initial commit')
 
79
        self.assertContainsRe(out, 'second commit')
 
80
 
 
81
        # uncommit in a checkout should uncommit the parent branch
 
82
        # (but doesn't effect the other working tree)
 
83
        self.assertEquals(['a1'], checkout_tree.get_parent_ids())
 
84
        self.assertEquals('a1', wt.branch.last_revision())
 
85
        self.assertEquals(['a2'], wt.get_parent_ids())
 
86
 
 
87
    def test_uncommit_bound(self):
 
88
        os.mkdir('a')
 
89
        a = BzrDirMetaFormat1().initialize('a')
 
90
        a.create_repository()
 
91
        a.create_branch()
 
92
        t_a = a.create_workingtree()
 
93
        t_a.commit('commit 1')
 
94
        t_a.commit('commit 2')
 
95
        t_a.commit('commit 3')
 
96
        b = t_a.branch.create_checkout('b').branch
 
97
        uncommit.uncommit(b)
 
98
        self.assertEqual(len(b.revision_history()), 2)
 
99
        self.assertEqual(len(t_a.branch.revision_history()), 2)
 
100
        # update A's tree to not have the uncomitted revision referenced.
 
101
        t_a.update()
 
102
        t_a.commit('commit 3b')
 
103
        self.assertRaises(BoundBranchOutOfDate, uncommit.uncommit, b)
 
104
        b.pull(t_a.branch)
 
105
        uncommit.uncommit(b)
 
106
 
 
107
    def test_uncommit_revision(self):
 
108
        wt = self.create_simple_tree()
 
109
 
 
110
        os.chdir('tree')
 
111
        out, err = self.run_bzr('uncommit', '-r1', '--force')
 
112
 
 
113
        self.assertNotContainsRe(out, 'initial commit')
 
114
        self.assertContainsRe(out, 'second commit')
 
115
        self.assertEqual(['a1'], wt.get_parent_ids())
 
116
        self.assertEqual('a1', wt.branch.last_revision())
 
117
 
 
118
    def test_uncommit_neg_1(self):
 
119
        wt = self.create_simple_tree()
 
120
        os.chdir('tree')
 
121
        out, err = self.run_bzr('uncommit', '-r', '-1', retcode=1)
 
122
        self.assertEqual('No revisions to uncommit.\n', out)
 
123
 
 
124
    def test_uncommit_merges(self):
 
125
        wt = self.create_simple_tree()
 
126
 
 
127
        tree2 = wt.bzrdir.sprout('tree2').open_workingtree()
 
128
 
 
129
        tree2.commit('unchanged', rev_id='b3')
 
130
        tree2.commit('unchanged', rev_id='b4')
 
131
 
 
132
        wt.merge_from_branch(tree2.branch)
 
133
        wt.commit('merge b4', rev_id='a3')
 
134
 
 
135
        self.assertEqual(['a3'], wt.get_parent_ids())
 
136
 
 
137
        os.chdir('tree')
 
138
        out, err = self.run_bzr('uncommit', '--force')
 
139
 
 
140
        self.assertEqual(['a2', 'b4'], wt.get_parent_ids())
 
141
 
 
142
    def test_uncommit_pending_merge(self):
 
143
        wt = self.create_simple_tree()
 
144
        tree2 = wt.bzrdir.sprout('tree2').open_workingtree()
 
145
        tree2.commit('unchanged', rev_id='b3')
 
146
 
 
147
        wt.branch.fetch(tree2.branch)
 
148
        wt.set_pending_merges(['b3'])
 
149
 
 
150
        os.chdir('tree')
 
151
        out, err = self.run_bzr('uncommit', '--force')
 
152
        self.assertEqual(['a1', 'b3'], wt.get_parent_ids())
 
153
 
 
154
    def test_uncommit_multiple_merge(self):
 
155
        wt = self.create_simple_tree()
 
156
 
 
157
        tree2 = wt.bzrdir.sprout('tree2').open_workingtree()
 
158
 
 
159
        tree2.commit('unchanged', rev_id='b3')
 
160
 
 
161
        wt.merge_from_branch(tree2.branch)
 
162
        wt.commit('merge b3', rev_id='a3')
 
163
 
 
164
        tree2.commit('unchanged', rev_id='b4')
 
165
 
 
166
        wt.merge_from_branch(tree2.branch)
 
167
        wt.commit('merge b4', rev_id='a4')
 
168
 
 
169
        self.assertEqual(['a4'], wt.get_parent_ids())
 
170
 
 
171
        os.chdir('tree')
 
172
        out, err = self.run_bzr('uncommit', '--force', '-r', '2')
 
173
 
 
174
        self.assertEqual(['a2', 'b3', 'b4'], wt.get_parent_ids())
 
175
 
 
176
    def test_uncommit_merge_plus_pending(self):
 
177
        wt = self.create_simple_tree()
 
178
 
 
179
        tree2 = wt.bzrdir.sprout('tree2').open_workingtree()
 
180
 
 
181
        tree2.commit('unchanged', rev_id='b3')
 
182
        wt.branch.fetch(tree2.branch)
 
183
        wt.set_pending_merges(['b3'])
 
184
        wt.commit('merge b3', rev_id='a3')
 
185
 
 
186
        tree2.commit('unchanged', rev_id='b4')
 
187
        wt.branch.fetch(tree2.branch)
 
188
        wt.set_pending_merges(['b4'])
 
189
 
 
190
        self.assertEqual(['a3', 'b4'], wt.get_parent_ids())
 
191
 
 
192
        os.chdir('tree')
 
193
        out, err = self.run_bzr('uncommit', '--force', '-r', '2')
 
194
 
 
195
        self.assertEqual(['a2', 'b3', 'b4'], wt.get_parent_ids())
 
196
 
 
197
    def test_uncommit_octopus_merge(self):
 
198
        # Check that uncommit keeps the pending merges in the same order
 
199
        wt = self.create_simple_tree()
 
200
 
 
201
        tree2 = wt.bzrdir.sprout('tree2').open_workingtree()
 
202
        tree3 = wt.bzrdir.sprout('tree3').open_workingtree()
 
203
 
 
204
        tree2.commit('unchanged', rev_id='b3')
 
205
        tree3.commit('unchanged', rev_id='c3')
 
206
        
 
207
        wt.merge_from_branch(tree2.branch)
 
208
        wt.merge_from_branch(tree3.branch)
 
209
        wt.commit('merge b3, c3', rev_id='a3')
 
210
 
 
211
        tree2.commit('unchanged', rev_id='b4')
 
212
        tree3.commit('unchanged', rev_id='c4')
 
213
 
 
214
        wt.merge_from_branch(tree3.branch)
 
215
        wt.merge_from_branch(tree2.branch)
 
216
        wt.commit('merge b4, c4', rev_id='a4')
 
217
 
 
218
        self.assertEqual(['a4'], wt.get_parent_ids())
 
219
 
 
220
        os.chdir('tree')
 
221
        out, err = self.run_bzr('uncommit', '--force', '-r', '2')
 
222
 
 
223
        self.assertEqual(['a2', 'b3', 'c3', 'c4', 'b4'], wt.get_parent_ids())