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.branch import Branch
33
from bzrlib.builtins import merge
34
from bzrlib.revisionspec import RevisionSpec
35
from bzrlib.status import show_status
36
from bzrlib.tests import TestCaseWithTransport
37
from bzrlib.workingtree import WorkingTree
40
class BranchStatus(TestCaseWithTransport):
42
def test_branch_status(self):
43
"""Test basic branch status"""
44
wt = self.make_branch_and_tree('.')
49
show_status(b, to_file=tof)
50
self.assertEquals(tof.getvalue(), "")
53
self.build_tree(['hello.c', 'bye.c'])
54
wt.add_pending_merge('pending@pending-0-0')
55
show_status(b, to_file=tof)
57
self.assertEquals(tof.readlines(),
62
' pending@pending-0-0\n'
65
def test_branch_status_revisions(self):
66
"""Tests branch status with revisions"""
67
wt = self.make_branch_and_tree('.')
71
self.build_tree(['hello.c', 'bye.c'])
74
wt.commit('Test message')
78
revs.append(RevisionSpec(0))
80
show_status(b, to_file=tof, revision=revs)
83
self.assertEquals(tof.readlines(),
88
self.build_tree(['more.c'])
90
wt.commit('Another test message')
93
revs.append(RevisionSpec(1))
95
show_status(b, to_file=tof, revision=revs)
98
self.assertEquals(tof.readlines(),
103
def status_string(self, branch):
104
# use a real file rather than StringIO because it doesn't handle
106
tof = codecs.getwriter('utf-8')(TemporaryFile())
107
show_status(branch, to_file=tof)
109
return tof.read().decode('utf-8')
111
def test_pending(self):
112
"""Pending merges display works, including Unicode"""
114
wt = self.make_branch_and_tree('branch')
116
wt.commit("Empty commit 1")
117
b_2_dir = b.bzrdir.sprout('./copy')
118
b_2 = b_2_dir.open_branch()
119
wt2 = b_2_dir.open_workingtree()
120
wt.commit(u"\N{TIBETAN DIGIT TWO} Empty commit 2")
121
merge(["./branch", -1], [None, None], this_dir = './copy')
122
message = self.status_string(b_2)
123
self.assert_(message.startswith("pending merges:\n"))
124
self.assert_(message.endswith("Empty commit 2\n"))
126
# must be long to make sure we see elipsis at the end
127
wt.commit("Empty commit 3 " +
128
"blah blah blah blah " * 10)
129
merge(["./branch", -1], [None, None], this_dir = './copy')
130
message = self.status_string(b_2)
131
self.assert_(message.startswith("pending merges:\n"))
132
self.assert_("Empty commit 3" in message)
133
self.assert_(message.endswith("...\n"))
135
def test_branch_status_specific_files(self):
136
"""Tests branch status with given specific files"""
137
wt = self.make_branch_and_tree('.')
140
self.build_tree(['directory/','directory/hello.c', 'bye.c','test.c','dir2/'])
146
show_status(b, to_file=tof)
148
self.assertEquals(tof.readlines(),
152
' directory/hello.c\n'
156
show_status(b, specific_files=['bye.c','test.c','absent.c'], to_file=tof)
158
self.assertEquals(tof.readlines(),
164
show_status(b, specific_files=['directory'], to_file=tof)
166
self.assertEquals(tof.readlines(),
168
' directory/hello.c\n'
171
show_status(b, specific_files=['dir2'], to_file=tof)
173
self.assertEquals(tof.readlines(),