1
# Copyright (C) 2005, 2006 Canonical Ltd
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.
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.
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
17
"""Tests of status command.
19
Most of these depend on the particular formatting used.
20
As such they really are blackbox tests even though some of the
21
tests are not using self.capture. If we add tests for the programmatic
22
interface later, they will be non blackbox tests.
25
from cStringIO import StringIO
27
from os import mkdir, chdir
29
from tempfile import TemporaryFile
31
from bzrlib import bzrdir, errors
33
from bzrlib.builtins import merge
34
from bzrlib.osutils import pathjoin
35
from bzrlib.revisionspec import RevisionSpec
36
from bzrlib.status import show_tree_status
37
from bzrlib.tests import TestCaseWithTransport, TestSkipped
38
from bzrlib.workingtree import WorkingTree
41
class BranchStatus(TestCaseWithTransport):
43
def assertStatus(self, output_lines, working_tree,
44
revision=None, short=False):
45
"""Run status in working_tree and look for output.
47
:param output_lines: The lines to look for.
48
:param working_tree: The tree to run status in.
50
output_string = self.status_string(working_tree, revision, short)
51
self.assertEqual(output_lines, output_string.splitlines(True))
53
def status_string(self, wt, revision=None, short=False):
54
# use a real file rather than StringIO because it doesn't handle
56
tof = codecs.getwriter('utf-8')(TemporaryFile())
57
show_tree_status(wt, to_file=tof, revision=revision, short=short)
59
return tof.read().decode('utf-8')
61
def test_branch_status(self):
62
"""Test basic branch status"""
63
wt = self.make_branch_and_tree('.')
65
# status with no commits or files - it must
66
# work and show no output. We do this with no
67
# commits to be sure that it's not going to fail
69
self.assertStatus([], wt)
71
self.build_tree(['hello.c', 'bye.c'])
84
# add a commit to allow showing pending merges.
85
wt.commit('create a parent to allow testing merge output')
87
wt.add_parent_tree_id('pending@pending-0-0')
93
' pending@pending-0-0\n',
99
'P pending@pending-0-0\n',
103
def test_branch_status_revisions(self):
104
"""Tests branch status with revisions"""
105
wt = self.make_branch_and_tree('.')
107
self.build_tree(['hello.c', 'bye.c'])
110
wt.commit('Test message')
112
revs = [RevisionSpec.from_string('0')]
121
self.build_tree(['more.c'])
123
wt.commit('Another test message')
125
revs.append(RevisionSpec.from_string('1'))
134
def test_pending(self):
135
"""Pending merges display works, including Unicode"""
137
wt = self.make_branch_and_tree('branch')
139
wt.commit("Empty commit 1")
140
b_2_dir = b.bzrdir.sprout('./copy')
141
b_2 = b_2_dir.open_branch()
142
wt2 = b_2_dir.open_workingtree()
143
wt.commit(u"\N{TIBETAN DIGIT TWO} Empty commit 2")
144
merge(["./branch", -1], [None, None], this_dir = './copy')
145
message = self.status_string(wt2)
146
self.assert_(message.startswith("pending merges:\n"))
147
self.assert_(message.endswith("Empty commit 2\n"))
149
# must be long to make sure we see elipsis at the end
150
wt.commit("Empty commit 3 " +
151
"blah blah blah blah " * 10)
152
merge(["./branch", -1], [None, None], this_dir = './copy')
153
message = self.status_string(wt2)
154
self.assert_(message.startswith("pending merges:\n"))
155
self.assert_("Empty commit 3" in message)
156
self.assert_(message.endswith("...\n"))
158
def test_branch_status_specific_files(self):
159
"""Tests branch status with given specific files"""
160
wt = self.make_branch_and_tree('.')
163
self.build_tree(['directory/','directory/hello.c', 'bye.c','test.c','dir2/'])
172
' directory/hello.c\n'
179
'? directory/hello.c\n'
184
self.assertRaises(errors.PathsDoNotExist,
186
wt, specific_files=['bye.c','test.c','absent.c'],
190
show_tree_status(wt, specific_files=['directory'], to_file=tof)
192
self.assertEquals(tof.readlines(),
194
' directory/hello.c\n'
197
show_tree_status(wt, specific_files=['directory'], to_file=tof,
200
self.assertEquals(tof.readlines(), ['? directory/hello.c\n'])
203
show_tree_status(wt, specific_files=['dir2'], to_file=tof)
205
self.assertEquals(tof.readlines(),
210
show_tree_status(wt, specific_files=['dir2'], to_file=tof, short=True)
212
self.assertEquals(tof.readlines(), ['? dir2\n'])
214
def test_status_nonexistent_file(self):
215
# files that don't exist in either the basis tree or working tree
216
# should give an error
217
wt = self.make_branch_and_tree('.')
218
out, err = self.run_bzr('status', 'does-not-exist', retcode=3)
219
self.assertContainsRe(err, r'do not exist.*does-not-exist')
221
def test_status_out_of_date(self):
222
"""Simulate status of out-of-date tree after remote push"""
223
tree = self.make_branch_and_tree('.')
224
self.build_tree_contents([('a', 'foo\n')])
228
tree.commit('add test file')
229
# simulate what happens after a remote push
230
tree.set_last_revision("0")
231
out, err = self.run_bzr('status')
232
self.assertEqual("working tree is out of date, run 'bzr update'\n",
238
class CheckoutStatus(BranchStatus):
241
super(CheckoutStatus, self).setUp()
245
def make_branch_and_tree(self, relpath):
246
source = self.make_branch(pathjoin('..', relpath))
247
checkout = bzrdir.BzrDirMetaFormat1().initialize(relpath)
248
bzrlib.branch.BranchReferenceFormat().initialize(checkout, source)
249
return checkout.create_workingtree()
252
class TestStatus(TestCaseWithTransport):
254
def test_status(self):
257
self.build_tree(['hello.txt'])
258
result = self.run_bzr("status")[0]
259
self.assert_("unknown:\n hello.txt\n" in result, result)
260
result = self.run_bzr("status","--short")[0]
261
self.assert_("? hello.txt\n" in result, result)
263
self.run_bzr("add", "hello.txt")
264
result = self.run_bzr("status")[0]
265
self.assert_("added:\n hello.txt\n" in result, result)
266
result = self.run_bzr("status","--short")[0]
267
self.assert_("A hello.txt\n" in result, result)
269
self.run_bzr("commit", "-m", "added")
270
result = self.run_bzr("status", "-r", "0..1")[0]
271
self.assert_("added:\n hello.txt\n" in result, result)
272
result = self.run_bzr("status", "--short", "-r", "0..1")[0]
273
self.assert_("A hello.txt\n" in result, result)
275
self.build_tree(['world.txt'])
276
result = self.run_bzr("status", "-r", "0")[0]
277
self.assert_("added:\n hello.txt\n" \
278
"unknown:\n world.txt\n" in result, result)
279
result2 = self.run_bzr("status", "-r", "0..")[0]
280
self.assertEquals(result2, result)
281
result = self.run_bzr("status", "--short", "-r", "0")[0]
282
self.assert_("A hello.txt\n" \
283
"? world.txt\n" in result, result)
284
result2 = self.run_bzr("status", "--short", "-r", "0..")[0]
285
self.assertEquals(result2, result)
288
class TestStatusEncodings(TestCaseWithTransport):
291
TestCaseWithTransport.setUp(self)
292
self.user_encoding = bzrlib.user_encoding
293
self.stdout = sys.stdout
296
bzrlib.user_encoding = self.user_encoding
297
sys.stdout = self.stdout
298
TestCaseWithTransport.tearDown(self)
300
def make_uncommitted_tree(self):
301
"""Build a branch with uncommitted unicode named changes in the cwd."""
302
working_tree = self.make_branch_and_tree(u'.')
303
filename = u'hell\u00d8'
305
self.build_tree_contents([(filename, 'contents of hello')])
306
except UnicodeEncodeError:
307
raise TestSkipped("can't build unicode working tree in "
308
"filesystem encoding %s" % sys.getfilesystemencoding())
309
working_tree.add(filename)
312
def test_stdout_ascii(self):
313
sys.stdout = StringIO()
314
bzrlib.user_encoding = 'ascii'
315
working_tree = self.make_uncommitted_tree()
316
stdout, stderr = self.run_bzr("status")
318
self.assertEquals(stdout, """\
323
def test_stdout_latin1(self):
324
sys.stdout = StringIO()
325
bzrlib.user_encoding = 'latin-1'
326
working_tree = self.make_uncommitted_tree()
327
stdout, stderr = self.run_bzr('status')
329
self.assertEquals(stdout, u"""\
332
""".encode('latin-1'))
added
removed
bzrlib/tests/blackbox/test_status.py
