1
# Copyright (C) 2005 by 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
18
"""Tests of status command.
20
Most of these depend on the particular formatting used.
21
As such they really are blackbox tests even though some of the
22
tests are not using self.capture. If we add tests for the programmatic
23
interface later, they will be non blackbox tests.
27
from cStringIO import StringIO
29
from tempfile import TemporaryFile
32
from bzrlib.builtins import merge
33
from bzrlib.revisionspec import RevisionSpec
34
from bzrlib.status import show_status
35
from bzrlib.tests import TestCaseWithTransport
36
from bzrlib.workingtree import WorkingTree
39
class BranchStatus(TestCaseWithTransport):
41
def test_branch_status(self):
42
"""Test basic branch status"""
43
wt = self.make_branch_and_tree('.')
48
show_status(b, to_file=tof)
49
self.assertEquals(tof.getvalue(), "")
52
self.build_tree(['hello.c', 'bye.c'])
53
wt.add_pending_merge('pending@pending-0-0')
54
show_status(b, to_file=tof)
56
self.assertEquals(tof.readlines(),
61
' pending@pending-0-0\n'
64
def test_branch_status_revisions(self):
65
"""Tests branch status with revisions"""
66
wt = self.make_branch_and_tree('.')
70
self.build_tree(['hello.c', 'bye.c'])
73
wt.commit('Test message')
77
revs.append(RevisionSpec(0))
79
show_status(b, to_file=tof, revision=revs)
82
self.assertEquals(tof.readlines(),
87
self.build_tree(['more.c'])
89
wt.commit('Another test message')
92
revs.append(RevisionSpec(1))
94
show_status(b, to_file=tof, revision=revs)
97
self.assertEquals(tof.readlines(),
102
def status_string(self, branch):
103
# use a real file rather than StringIO because it doesn't handle
105
tof = codecs.getwriter('utf-8')(TemporaryFile())
106
show_status(branch, to_file=tof)
108
return tof.read().decode('utf-8')
110
def test_pending(self):
111
"""Pending merges display works, including Unicode"""
113
wt = self.make_branch_and_tree('branch')
115
wt.commit("Empty commit 1")
116
b_2_dir = b.bzrdir.sprout('./copy')
117
b_2 = b_2_dir.open_branch()
118
wt2 = b_2_dir.open_workingtree()
119
wt.commit(u"\N{TIBETAN DIGIT TWO} Empty commit 2")
120
merge(["./branch", -1], [None, None], this_dir = './copy')
121
message = self.status_string(b_2)
122
self.assert_(message.startswith("pending merges:\n"))
123
self.assert_(message.endswith("Empty commit 2\n"))
125
# must be long to make sure we see elipsis at the end
126
wt.commit("Empty commit 3 " +
127
"blah blah blah blah " * 10)
128
merge(["./branch", -1], [None, None], this_dir = './copy')
129
message = self.status_string(b_2)
130
self.assert_(message.startswith("pending merges:\n"))
131
self.assert_("Empty commit 3" in message)
132
self.assert_(message.endswith("...\n"))
134
def test_branch_status_specific_files(self):
135
"""Tests branch status with given specific files"""
136
wt = self.make_branch_and_tree('.')
139
self.build_tree(['directory/','directory/hello.c', 'bye.c','test.c','dir2/'])
145
show_status(b, to_file=tof)
147
self.assertEquals(tof.readlines(),
151
' directory/hello.c\n'
155
show_status(b, specific_files=['bye.c','test.c','absent.c'], to_file=tof)
157
self.assertEquals(tof.readlines(),
163
show_status(b, specific_files=['directory'], to_file=tof)
165
self.assertEquals(tof.readlines(),
167
' directory/hello.c\n'
170
show_status(b, specific_files=['dir2'], to_file=tof)
172
self.assertEquals(tof.readlines(),
178
class TestStatus(TestCaseWithTransport):
180
def test_status(self):
182
self.build_tree(['hello.txt'])
183
result = self.run_bzr("status")[0]
184
self.assert_("unknown:\n hello.txt\n" in result, result)
185
self.run_bzr("add", "hello.txt")
186
result = self.run_bzr("status")[0]
187
self.assert_("added:\n hello.txt\n" in result, result)
188
self.run_bzr("commit", "-m", "added")
189
result = self.run_bzr("status", "-r", "0..1")[0]
190
self.assert_("added:\n hello.txt\n" in result, result)
191
self.build_tree(['world.txt'])
192
result = self.run_bzr("status", "-r", "0")[0]
193
self.assert_("added:\n hello.txt\n" \
194
"unknown:\n world.txt\n" in result, result)
196
result2 = self.run_bzr("status", "-r", "0..")[0]
197
self.assertEquals(result2, result)