1
# Copyright (C) 2005 by Canonical Ltd
1
# Copyright (C) 2005, 2006 Canonical Ltd
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
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
11
# GNU General Public License for more details.
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18
17
"""Tests of status command.
20
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
24
from bzrlib.selftest import TestCaseInTempDir
31
from bzrlib import bzrdir, errors
33
from bzrlib.builtins import merge
34
from bzrlib.osutils import pathjoin
25
35
from bzrlib.revisionspec import RevisionSpec
26
from bzrlib.merge import merge
27
from cStringIO import StringIO
28
from bzrlib.status import show_status
29
from bzrlib.branch import Branch
31
from bzrlib.clone import copy_branch
33
class BranchStatus(TestCaseInTempDir):
35
def test_branch_status(self):
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,
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)
51
self.assertEqual(output_lines, output_string.splitlines(True))
53
def status_string(self, wt, revision=None):
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)
59
return tof.read().decode('utf-8')
61
def test_branch_status(self):
36
62
"""Test basic branch status"""
37
from cStringIO import StringIO
38
from bzrlib.status import show_status
39
from bzrlib.branch import Branch
41
b = Branch.initialize('.')
45
show_status(b, to_file=tof)
46
self.assertEquals(tof.getvalue(), "")
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)
49
71
self.build_tree(['hello.c', 'bye.c'])
50
b.add_pending_merge('pending@pending-0-0')
51
show_status(b, to_file=tof)
53
self.assertEquals(tof.readlines(),
58
' pending@pending-0-0\n'
79
# add a commit to allow showing pending merges.
80
wt.commit('create a parent to allow testing merge output')
82
wt.add_parent_tree_id('pending@pending-0-0')
88
' pending@pending-0-0\n',
61
92
def test_branch_status_revisions(self):
62
93
"""Tests branch status with revisions"""
64
b = Branch.initialize('.')
94
wt = self.make_branch_and_tree('.')
67
96
self.build_tree(['hello.c', 'bye.c'])
70
b.commit('Test message')
99
wt.commit('Test message')
74
revs.append(RevisionSpec(0))
76
show_status(b, to_file=tof, revision=revs)
79
self.assertEquals(tof.readlines(),
101
revs = [RevisionSpec.from_string('0')]
84
110
self.build_tree(['more.c'])
86
b.commit('Another test message')
89
revs.append(RevisionSpec(1))
91
show_status(b, to_file=tof, revision=revs)
94
self.assertEquals(tof.readlines(),
99
def status_string(self, branch):
101
show_status(branch, to_file=tof)
103
return tof.getvalue()
112
wt.commit('Another test message')
114
revs.append(RevisionSpec.from_string('1'))
105
123
def test_pending(self):
106
"""Pending merges display works"""
124
"""Pending merges display works, including Unicode"""
107
125
mkdir("./branch")
108
b = Branch.initialize('./branch')
109
b.commit("Empty commit 1")
110
b_2 = copy_branch(b, './copy')
111
b.commit("Empty commit 2")
112
merge(["./branch", -1], [None, None], this_dir = './copy')
113
message = self.status_string(b_2)
114
assert (message.startswith("pending merges:\n")), message
115
assert (message.endswith("Empty commit 2\n")), message
117
b.commit("Empty commit 3 blah blah blah blah blah blah blah blah blah")
118
merge(["./branch", -1], [None, None], this_dir = './copy')
119
message = self.status_string(b_2)
120
assert (message.startswith("pending merges:\n")), message
121
assert ("Empty commit 3" in message), message
122
assert (message.endswith("...\n")), message
126
wt = self.make_branch_and_tree('branch')
128
wt.commit("Empty commit 1")
129
b_2_dir = b.bzrdir.sprout('./copy')
130
b_2 = b_2_dir.open_branch()
131
wt2 = b_2_dir.open_workingtree()
132
wt.commit(u"\N{TIBETAN DIGIT TWO} Empty commit 2")
133
merge(["./branch", -1], [None, None], this_dir = './copy')
134
message = self.status_string(wt2)
135
self.assert_(message.startswith("pending merges:\n"))
136
self.assert_(message.endswith("Empty commit 2\n"))
138
# must be long to make sure we see elipsis at the end
139
wt.commit("Empty commit 3 " +
140
"blah blah blah blah " * 10)
141
merge(["./branch", -1], [None, None], this_dir = './copy')
142
message = self.status_string(wt2)
143
self.assert_(message.startswith("pending merges:\n"))
144
self.assert_("Empty commit 3" in message)
145
self.assert_(message.endswith("...\n"))
124
147
def test_branch_status_specific_files(self):
125
148
"""Tests branch status with given specific files"""
126
from cStringIO import StringIO
127
from bzrlib.status import show_status
128
from bzrlib.branch import Branch
130
b = Branch.initialize('.')
149
wt = self.make_branch_and_tree('.')
132
152
self.build_tree(['directory/','directory/hello.c', 'bye.c','test.c','dir2/'])
138
show_status(b, to_file=tof)
158
show_tree_status(wt, to_file=tof)
140
160
self.assertEquals(tof.readlines(),
144
164
' directory/hello.c\n'
148
show_status(b, specific_files=['bye.c','test.c','absent.c'], to_file=tof)
150
self.assertEquals(tof.readlines(),
167
self.assertRaises(errors.PathsDoNotExist,
169
wt, specific_files=['bye.c','test.c','absent.c'],
156
show_status(b, specific_files=['directory'], to_file=tof)
173
show_tree_status(wt, specific_files=['directory'], to_file=tof)
158
175
self.assertEquals(tof.readlines(),
160
177
' directory/hello.c\n'
163
show_status(b, specific_files=['dir2'], to_file=tof)
180
show_tree_status(wt, specific_files=['dir2'], to_file=tof)
165
182
self.assertEquals(tof.readlines(),
187
def test_status_nonexistent_file(self):
188
# files that don't exist in either the basis tree or working tree
189
# should give an error
190
wt = self.make_branch_and_tree('.')
191
out, err = self.run_bzr('status', 'does-not-exist', retcode=3)
192
self.assertContainsRe(err, r'do not exist.*does-not-exist')
194
def test_status_out_of_date(self):
195
"""Simulate status of out-of-date tree after remote push"""
196
tree = self.make_branch_and_tree('.')
197
self.build_tree_contents([('a', 'foo\n')])
201
tree.commit('add test file')
202
# simulate what happens after a remote push
203
tree.set_last_revision("0")
204
out, err = self.run_bzr('status')
205
self.assertEqual("working tree is out of date, run 'bzr update'\n",
211
class CheckoutStatus(BranchStatus):
214
super(CheckoutStatus, self).setUp()
218
def make_branch_and_tree(self, relpath):
219
source = self.make_branch(pathjoin('..', relpath))
220
checkout = bzrdir.BzrDirMetaFormat1().initialize(relpath)
221
bzrlib.branch.BranchReferenceFormat().initialize(checkout, source)
222
return checkout.create_workingtree()
225
class TestStatus(TestCaseWithTransport):
227
def test_status(self):
229
self.build_tree(['hello.txt'])
230
result = self.run_bzr("status")[0]
231
self.assert_("unknown:\n hello.txt\n" in result, result)
232
self.run_bzr("add", "hello.txt")
233
result = self.run_bzr("status")[0]
234
self.assert_("added:\n hello.txt\n" in result, result)
235
self.run_bzr("commit", "-m", "added")
236
result = self.run_bzr("status", "-r", "0..1")[0]
237
self.assert_("added:\n hello.txt\n" in result, result)
238
self.build_tree(['world.txt'])
239
result = self.run_bzr("status", "-r", "0")[0]
240
self.assert_("added:\n hello.txt\n" \
241
"unknown:\n world.txt\n" in result, result)
243
result2 = self.run_bzr("status", "-r", "0..")[0]
244
self.assertEquals(result2, result)
247
class TestStatusEncodings(TestCaseWithTransport):
250
TestCaseWithTransport.setUp(self)
251
self.user_encoding = bzrlib.user_encoding
252
self.stdout = sys.stdout
255
bzrlib.user_encoding = self.user_encoding
256
sys.stdout = self.stdout
257
TestCaseWithTransport.tearDown(self)
259
def make_uncommitted_tree(self):
260
"""Build a branch with uncommitted unicode named changes in the cwd."""
261
working_tree = self.make_branch_and_tree(u'.')
262
filename = u'hell\u00d8'
264
self.build_tree_contents([(filename, 'contents of hello')])
265
except UnicodeEncodeError:
266
raise TestSkipped("can't build unicode working tree in "
267
"filesystem encoding %s" % sys.getfilesystemencoding())
268
working_tree.add(filename)
271
def test_stdout_ascii(self):
272
sys.stdout = StringIO()
273
bzrlib.user_encoding = 'ascii'
274
working_tree = self.make_uncommitted_tree()
275
stdout, stderr = self.run_bzr("status")
277
self.assertEquals(stdout, """\
282
def test_stdout_latin1(self):
283
sys.stdout = StringIO()
284
bzrlib.user_encoding = 'latin-1'
285
working_tree = self.make_uncommitted_tree()
286
stdout, stderr = self.run_bzr('status')
288
self.assertEquals(stdout, u"""\
291
""".encode('latin-1'))
added
removed
bzrlib/tests/blackbox/test_status.py
